NaturesAura/src/main/java/de/ellpeck/naturesaura/api/aura/container/BasicAuraContainer.java

package de.ellpeck.naturesaura.api.aura.container;

import de.ellpeck.naturesaura.api.aura.type.IAuraType;
import net.minecraft.nbt.CompoundTag;

public class BasicAuraContainer implements IAuraContainer {

    protected final IAuraType type;
    protected final int maxAura;
    protected int aura;

    public BasicAuraContainer(IAuraType type, int maxAura) {
        this.type = type;
        this.maxAura = maxAura;
    }

    @Override
    public int storeAura(int amountToStore, boolean simulate) {
        var actual = Math.min(amountToStore, this.maxAura - this.aura);
        if (!simulate) {
            this.aura += actual;
        }
        return actual;
    }

    @Override
    public int drainAura(int amountToDrain, boolean simulate) {
        var actual = Math.min(amountToDrain, this.aura);
        if (!simulate) {
            this.aura -= actual;
        }
        return actual;
    }

    @Override
    public int getStoredAura() {
        return this.aura;
    }

    @Override
    public int getMaxAura() {
        return this.maxAura;
    }

    @Override
    public int getAuraColor() {
        return 0x1E891E;
    }

    @Override
    public boolean isAcceptableType(IAuraType type) {
        return this.type == null || type.isSimilar(this.type);
    }

    public void writeNBT(CompoundTag compound) {
        compound.putInt("aura", this.aura);
    }

    public void readNBT(CompoundTag compound) {
        this.aura = compound.getInt("aura");
    }
}