aether-legacy/src/main/java/com/legacy/aether/tileentity/TileEntityFreezer.java

package com.legacy.aether.tileentity;

import java.util.List;
import java.util.Map;

import net.minecraft.enchantment.Enchantment;
import net.minecraft.enchantment.EnchantmentHelper;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;

import com.legacy.aether.api.AetherAPI;
import com.legacy.aether.api.events.AetherHooks;
import com.legacy.aether.api.freezables.AetherFreezable;
import com.legacy.aether.blocks.BlocksAether;
import com.legacy.aether.blocks.container.BlockAetherContainer;
import com.legacy.aether.tileentity.util.AetherTileEntity;
import com.legacy.aether.util.FilledList;

import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;

public class TileEntityFreezer extends AetherTileEntity {

	public int progress, ticksRequired, powerRemaining;

	private final FilledList<ItemStack> frozenItemStacks = new FilledList<ItemStack>(3, null);

	private AetherFreezable currentFreezable;

	public TileEntityFreezer() {
		super("Freezer");
	}

	@Override
	public List<ItemStack> getTileInventory() {
		return this.frozenItemStacks;
	}

	@Override
	public void onSlotChanged(int index) {

	}

	@Override
	public void updateEntity() {
		boolean flag = this.isFreezing();

		if (this.powerRemaining > 0) {
			this.powerRemaining--;

			if (this.currentFreezable != null) {
				if (this.worldObj.getBlock(this.xCoord, this.yCoord, this.zCoord) == BlocksAether.icestone) {
					this.progress += 2;
				} else {
					this.progress++;
				}
			}
		}

		if (this.currentFreezable != null) {
			if (this.progress >= this.currentFreezable.getTimeRequired()) {
				if (!this.worldObj.isRemote) {
					ItemStack result = this.currentFreezable.getOutput().copy();

					EnchantmentHelper.setEnchantments(EnchantmentHelper.getEnchantments(this.getStackInSlot(0)), result);

					if (this.getStackInSlot(0).hasTagCompound())
					{
						result.setTagCompound(this.getStackInSlot(0).getTagCompound());
					}

					if (this.getStackInSlot(2) != null && this.getStackInSlot(2).isStackable()) {
						result.stackSize += (this.getStackInSlot(2).stackSize);

						this.setInventorySlotContents(2, result);
					} else {
						this.setInventorySlotContents(2, result);
					}

					if (this.getStackInSlot(0).getItem().hasContainerItem(this.getStackInSlot(0))) {
						this.setInventorySlotContents(0, this.getStackInSlot(0).getItem().getContainerItem(this.getStackInSlot(0)));
					} else {
						this.decrStackSize(0, 1);
					}
				}

				this.progress = 0;
				AetherHooks.onItemFreeze(this, this.currentFreezable);
			}

			if (this.getStackInSlot(0) == null || (this.getStackInSlot(0) != null && AetherAPI.instance().getFreezable(this.getStackInSlot(0)) != this.currentFreezable)) {
				this.currentFreezable = null;
				this.progress = 0;
			}

			if (this.powerRemaining <= 0) {
				if (this.getStackInSlot(1) != null && AetherAPI.instance().isFreezableFuel(this.getStackInSlot(1))) {
					this.powerRemaining += AetherAPI.instance().getFreezableFuel(this.getStackInSlot(1)).getTimeGiven();

					if (!this.worldObj.isRemote) {
						this.decrStackSize(1, 1);
					}
				} else {
					this.currentFreezable = null;
					this.progress = 0;
				}
			}
		} else if (this.getStackInSlot(0) != null) {
			ItemStack itemstack = this.getStackInSlot(0);
			AetherFreezable freezable = AetherAPI.instance().getFreezable(itemstack);

			if (freezable != null) {
				if (this.getStackInSlot(2) == null || (freezable.getOutput().getItem() == this.getStackInSlot(2).getItem() && freezable.getOutput().getItemDamage() == this.getStackInSlot(2).getItemDamage() && this.getStackInSlot(2).isStackable())) {
					this.currentFreezable = freezable;
					this.ticksRequired = this.currentFreezable.getTimeRequired();
					this.addEnchantmentWeight(itemstack);
					this.ticksRequired = AetherHooks.onSetFreezableTime(this, this.currentFreezable, this.ticksRequired);
				}
			}
		}

		if (flag != this.isFreezing()) {
			this.markDirty();
			BlockAetherContainer.setState(this.worldObj, this.xCoord, this.yCoord, this.zCoord, this.isFreezing());
		}
	}

	@SuppressWarnings("unchecked")
	public void addEnchantmentWeight(ItemStack stack) {
		Map<Enchantment, Integer> enchantments = EnchantmentHelper.getEnchantments(stack);

		if (!enchantments.isEmpty()) {
			for (int levels : enchantments.values()) {
				this.ticksRequired += (levels * 1250);
			}
		}
	}

	@SideOnly(Side.CLIENT)
	public int getFreezingProgressScaled(int i) {
		if (this.ticksRequired == 0) {
			return 0;
		}
		return (this.progress * i) / this.ticksRequired;
	}

	@SideOnly(Side.CLIENT)
	public int getFreezingTimeRemaining(int i) {
		return (this.powerRemaining * i) / 500;
	}

	public boolean isFreezing() {
		return this.powerRemaining > 0;
	}

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

		this.progress = compound.getInteger("progress");
		this.powerRemaining = compound.getInteger("powerRemaining");
		this.ticksRequired = compound.getInteger("ticksRequired");
	}

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

		compound.setInteger("progress", this.progress);
		compound.setInteger("powerRemaining", this.powerRemaining);
		compound.setInteger("ticksRequired", this.ticksRequired);
	}

	@Override
	public boolean isValidSlotItem(int slot, ItemStack stackInSlot) {
		if (stackInSlot != null) {
			if (AetherAPI.instance().hasFreezable(stackInSlot)) {
				return true;
			} else if (slot == 1 && AetherAPI.instance().isFreezableFuel(stackInSlot)) {
				return true;
			}
		}

		return false;
	}

	@Override
	public int[] getAccessibleSlotsFromSide(int side) {
		return side == 0 ? new int[]{2} : new int[]{0, 1};
	}

}