package darkknight.jewelrycraft.worldGen;

import java.util.Random;
import cpw.mods.fml.common.event.FMLPreInitializationEvent;
import darkknight.jewelrycraft.block.BlockList;
import darkknight.jewelrycraft.item.ItemList;
import darkknight.jewelrycraft.util.JewelryNBT;
import darkknight.jewelrycraft.util.JewelrycraftUtil;
import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.util.WeightedRandomChestContent;
import net.minecraftforge.common.ChestGenHooks;

/**
 * @author Sorin
 *
 */
public class ChestGeneration
{
    static Item[] jewelry = new Item[]{ItemList.ring, ItemList.necklace, ItemList.bracelet, ItemList.earrings};
    static Random random = new Random();
    public static void preInit(FMLPreInitializationEvent e)
    {
        addItemToDifferentPlaces(new WeightedRandomChestContent(new ItemStack(ItemList.thiefGloves), 1, 1, 2), true, true, false, false, true);
        addItemToDifferentPlaces(new WeightedRandomChestContent(new ItemStack(ItemList.guide), 1, 1, 7), true, true, true, true, true, false, true, true);
        addVillageBlacksmithLoot(new WeightedRandomChestContent(new ItemStack(ItemList.shadowIngot), 1, 4, 5));
        for(int i = 0; i < 16 && i%3==0; i++) addItemToDifferentPlaces(new WeightedRandomChestContent(new ItemStack(BlockList.crystal, 1, i), 1, 3, 3), true, true, true, true);
        
        ItemStack special = new ItemStack(jewelry[random.nextInt(4)]);
        int randValue = random.nextInt(4);
        if(JewelrycraftUtil.metal.size() > 0) JewelryNBT.addMetal(special, JewelrycraftUtil.metal.get(random.nextInt(JewelrycraftUtil.metal.size())));
        if(JewelrycraftUtil.objects.size() > 0) JewelryNBT.addModifiers(special, JewelrycraftUtil.addRandomModifiers(randValue));
        if(JewelrycraftUtil.gem.size() > 0) JewelryNBT.addGem(special, JewelrycraftUtil.gem.get(random.nextInt(JewelrycraftUtil.gem.size())));
        addItemToDifferentPlaces(new WeightedRandomChestContent(special, 1, 1, 1), true, true, true, true);
    }
    
    /**
     * The booleans determine in which places should the items be added. The order is like so:
     * <p><ul>
     * <li> Dungeon
     * <li> Stronhold
     * <li> Pyramid
     * <li> Mineshaft
     * <li> Village Blacksmith
     * <li> Dispenser
     * <li> Bonus Chest
     * <li> Stronghold Library
     * </ul><p>
     */
    public static void addItemToDifferentPlaces(WeightedRandomChestContent item, Boolean ... options)
    {
        if(options.length > 0 && options[0]) addDungeonLoot(item);
        if(options.length > 1 && options[1]) addStrongholdLoot(item);
        if(options.length > 2 && options[2]) addPyramidLoot(item);
        if(options.length > 3 && options[3]) addMineshaftLoot(item);
        if(options.length > 4 && options[4]) addVillageBlacksmithLoot(item);
        if(options.length > 5 && options[5]) addDispenserLoot(item);
        if(options.length > 6 && options[6]) addBonusChestLoot(item);
        if(options.length > 7 && options[7]) addStrongholdLibraryLoot(item);
    }
    
    public static void addDungeonLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.DUNGEON_CHEST, item);
    }
    
    public static void addStrongholdLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.STRONGHOLD_CORRIDOR, item);
        ChestGenHooks.addItem(ChestGenHooks.STRONGHOLD_CROSSING, item);
    }
    
    public static void addPyramidLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.PYRAMID_DESERT_CHEST, item);
        ChestGenHooks.addItem(ChestGenHooks.PYRAMID_JUNGLE_CHEST, item);
    }
    
    public static void addMineshaftLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.MINESHAFT_CORRIDOR, item);
    }
        
    public static void addVillageBlacksmithLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.VILLAGE_BLACKSMITH, item);
    }
    
    public static void addDispenserLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.PYRAMID_JUNGLE_DISPENSER, item);
    }
    
    public static void addBonusChestLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.BONUS_CHEST, item);
    }
    
    public static void addStrongholdLibraryLoot(WeightedRandomChestContent item)
    {
        ChestGenHooks.addItem(ChestGenHooks.STRONGHOLD_LIBRARY, item);
    }
}