package StevenDimDoors.mod_pocketDim.dungeon;

import java.util.Random;

import net.minecraft.block.Block;
import net.minecraft.block.BlockContainer;
import net.minecraft.inventory.IInventory;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.tileentity.TileEntityChest;
import net.minecraft.tileentity.TileEntityDispenser;
import net.minecraft.util.WeightedRandomChestContent;
import net.minecraft.world.World;
import net.minecraftforge.common.ChestGenHooks;
import StevenDimDoors.mod_pocketDim.DDLoot;
import StevenDimDoors.mod_pocketDim.schematic.WorldOperation;

public class FillContainersOperation extends WorldOperation
{
	private Random random;
	
	public FillContainersOperation(Random random)
	{
		super("FillContainersOperation");
		this.random = random;
	}

	@Override
	protected boolean applyToBlock(World world, int x, int y, int z)
	{
		int blockID = world.getBlockId(x, y, z);

		//Fill empty chests and dispensers
		if (Block.blocksList[blockID] instanceof BlockContainer)
		{
			TileEntity tileEntity = world.getBlockTileEntity(x, y, z);

			//Fill chests
			if (tileEntity instanceof TileEntityChest)
			{
				TileEntityChest chest = (TileEntityChest) tileEntity;
				if (isInventoryEmpty(chest))
				{
					ChestGenHooks info = DDLoot.DungeonChestInfo;
					WeightedRandomChestContent.generateChestContents(random, info.getItems(random), chest, info.getCount(random));
				}
			}
			
			//Fill dispensers
			if (tileEntity instanceof TileEntityDispenser)
			{
				TileEntityDispenser dispenser = (TileEntityDispenser) tileEntity;
				if (isInventoryEmpty(dispenser))
				{
					dispenser.addItem(new ItemStack(Item.arrow, 64));
				}
			}
		}
		return true;
	}
	
	private static boolean isInventoryEmpty(IInventory inventory)
	{
		int size = inventory.getSizeInventory();
		for (int index = 0; index < size; index++)
		{
			if (inventory.getStackInSlot(index) != null)
			{
				return false;
			}
		}
		return true;
	}
}