MagicBees/src/main/java/magicbees/tileentity/TileEntityPhialingCabinet.java

package magicbees.tileentity;

import java.util.Objects;

import forestry.api.apiculture.*;
import magicbees.bees.BeeManager;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.tileentity.TileEntity;
import net.minecraftforge.common.util.ForgeDirection;

import forestry.apiculture.genetics.BeeGenome;
import magicbees.bees.BeeSpecies;
import magicbees.main.Config;
import thaumcraft.api.aspects.Aspect;
import thaumcraft.api.aspects.AspectList;
import thaumcraft.api.aspects.IAspectContainer;
import thaumcraft.api.aspects.IEssentiaTransport;

public class TileEntityPhialingCabinet extends TileEntity implements IAspectContainer, IEssentiaTransport {

    public static final String tileEntityName = "Phialing Cabinet";

    public Aspect aspect;
    public AspectList essentia = new AspectList();

    public final int maxAmount = Config.thaumcraftEssentiaBeePhialingCabinetCapacity;

    private int increment = 0;

    public void setAspect(Aspect sAspect) {
        aspect = sAspect;
    }

    @Override
    public void updateEntity() {
        // If there's no stored aspect we shouldn't even check for the above block.
        // We also don't need to run all the logic if the cabinet is full.
        if (aspect == null || essentia.visSize() == maxAmount) return;

        if (increment % Config.thaumcraftEssentiaBeePhialingCabinetTimeBetween == 0) {
            try {
                TileEntity above = worldObj.getTileEntity(this.xCoord, this.yCoord + 1, this.zCoord);
                if (IBeeHousing.class.isAssignableFrom(above.getClass())) {
                    IBeeHousing beeHousing = (IBeeHousing) above;
                    IBeekeepingLogic beekeepingLogic = beeHousing.getBeekeepingLogic();

                    // This performs all the checks to see if the bee is a living queen and if the species conditions
                    // are met.
                    if (!beekeepingLogic.canWork()) return;

                    ItemStack queenStack = beeHousing.getBeeInventory().getQueen();
                    IAlleleBeeSpecies queenSpecies = BeeGenome.getSpecies(queenStack);
                    if (queenSpecies == null) return;

                    if (Objects.equals(queenSpecies.getUID(), BeeSpecies.TC_ESSENTIA.getSpecies().getUID())) {
                        IBeeModifier modifier = BeeManager.beeRoot.createBeeHousingModifier(beeHousing);
                        IBeeGenome queenGenome = BeeManager.beeRoot.getMember(queenStack).getGenome();
                        float productionMultiplier = Math.abs(modifier.getProductionModifier(queenGenome, 1.0F));

                        int amount = Math.round(Config.thaumcraftEssentiaBeePhialingCabinetAmount * productionMultiplier);

                        addToContainer(aspect, amount);
                    }
                }
            } catch (Exception ignored) {}
            increment = 0;
        }

        increment++;
    }

    @Override
    public AspectList getAspects() {
        return this.essentia;
    }

    @Override
    public void setAspects(AspectList aspects) {
        this.essentia = aspects;
    }

    @Override
    public boolean doesContainerAccept(Aspect tag) {
        return false;
    }

    @Override
    public int addToContainer(Aspect tag, int am) {
        int toAdd = Math.min(maxAmount - essentia.visSize(), am);

        if (aspect.equals(tag) && toAdd > 0) {
            essentia.add(aspect, toAdd);
            markDirty();
            return am - toAdd;
        }

        markDirty();
        return am;
    }

    @Override
    public boolean takeFromContainer(Aspect tag, int amount) {
        if (!this.worldObj.isRemote) {
            this.worldObj.markBlockForUpdate(this.xCoord, this.yCoord, this.zCoord);
        }
        if (this.essentia.getAmount(tag) >= amount) {
            this.essentia.reduce(tag, amount);
            return true;
        } else {
            return false;
        }
    }

    @Override
    public boolean takeFromContainer(AspectList ot) {
        // TODO Auto-generated method stub
        boolean hasIt = true;
        if (!this.worldObj.isRemote) {
            this.worldObj.markBlockForUpdate(this.xCoord, this.yCoord, this.zCoord);
        }
        for (Aspect next : ot.aspects.keySet()) {
            if (this.essentia.getAmount(next) < ot.getAmount(next)) hasIt = false;
        }
        if (hasIt) {
            for (Aspect next : ot.aspects.keySet()) {
                essentia.reduce(next, ot.getAmount(next));
            }
            return true;
        } else {
            return false;
        }
    }

    @Override
    public boolean doesContainerContainAmount(Aspect tag, int amount) {
        return this.essentia.getAmount(tag) >= amount;
    }

    @Override
    public boolean doesContainerContain(AspectList ot) {
        boolean hasIt = true;
        for (Aspect next : ot.aspects.keySet()) {
            if (this.essentia.getAmount(next) < ot.getAmount(next)) hasIt = false;
        }
        return hasIt;
    }

    @Override
    public int containerContains(Aspect tag) {
        return this.essentia.getAmount(tag);
    }

    @Override
    public boolean isConnectable(ForgeDirection face) {
        return (face != ForgeDirection.UP);
    }

    @Override
    public boolean canInputFrom(ForgeDirection face) {
        return false;
    }

    @Override
    public boolean canOutputTo(ForgeDirection face) {
        return (face != ForgeDirection.UP);
    }

    @Override
    public int takeEssentia(Aspect aspect, int amount, ForgeDirection face) {
        if (face != ForgeDirection.UP) {
            if (!this.worldObj.isRemote) {
                this.worldObj.markBlockForUpdate(this.xCoord, this.yCoord, this.zCoord);
            }
            if (amount > this.essentia.getAmount(aspect)) {
                int total = this.essentia.getAmount(aspect);
                this.essentia.reduce(aspect, total);
                return total;
            } else {
                this.essentia.reduce(aspect, amount);
                return amount;
            }

        } else {
            return 0;
        }
    }

    @Override
    public void setSuction(Aspect aspect, int amount) {}

    @Override
    public int getMinimumSuction() {
        return -1;
    }

    @Override
    public int getSuctionAmount(ForgeDirection face) {
        return -1;
    }

    @Override
    public boolean renderExtendedTube() {
        return false;
    }

    @Override
    public Aspect getSuctionType(ForgeDirection face) {
        return null;
    }

    @Override
    public Aspect getEssentiaType(ForgeDirection face) {
        return this.essentia.size() > 0
                ? this.essentia.getAspects()[this.worldObj.rand.nextInt(this.essentia.getAspects().length)]
                : null;
    }

    @Override
    public int getEssentiaAmount(ForgeDirection face) {
        return this.essentia.visSize();
    }

    @Override
    public int addEssentia(Aspect aspect, int amount, ForgeDirection face) {
        return this.canInputFrom(face) ? amount - this.addToContainer(aspect, amount) : 0;
    }

    /* Saving and loading */
    @Override
    public void writeToNBT(NBTTagCompound compound) {
        super.writeToNBT(compound);

        this.essentia.writeToNBT(compound);
        compound.setString("aspect", aspect.getTag());
        compound.setInteger("increment", increment);
    }

    @Override
    public void readFromNBT(NBTTagCompound compound) {
        super.readFromNBT(compound);

        this.essentia.readFromNBT(compound);
        if (this.essentia.visSize() > this.maxAmount) {
            this.essentia = new AspectList();
        }
        aspect = Aspect.getAspect(compound.getString("aspect"));
        increment = compound.getInteger("increment");
    }
}