package mods.tinker.tconstruct.modifiers;

import java.util.Iterator;
import java.util.Map;

import mods.tinker.common.ToolMod;
import mods.tinker.tconstruct.library.Weapon;
import net.minecraft.enchantment.Enchantment;
import net.minecraft.enchantment.EnchantmentHelper;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;

public class ModLapis extends ToolMod
{
    String tooltipName;
    int increase;
    int max;

    public ModLapis(ItemStack[] items, int effect, int inc)
    {
        super(items, effect, "Lapis");
        tooltipName = "\u00a79Luck";
        increase = inc;
        max = 100;
    }

    @Override
    protected boolean canModify (ItemStack tool, ItemStack[] input)
    {
        NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
        if (!tags.hasKey(key))
            return tags.getInteger("Modifiers") > 0;

        int keyPair[] = tags.getIntArray(key);
        if (keyPair[0] + increase <= 100)
            return true;
        else
            return false;
    }

    @Override
    public void modify (ItemStack[] input, ItemStack tool)
    {
        NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
        if (!tags.hasKey(key))
        {
            tags.setBoolean(key, true);
            
            String modName = "\u00a79Lapis (0/100)";
            int tooltipIndex = addToolTip(tool, "\u00a79Luck", modName);
            int[] keyPair = new int[] { 0, tooltipIndex };
            tags.setIntArray(key, keyPair);
            
            int modifiers = tags.getInteger("Modifiers");
            modifiers -= 1;
            tags.setInteger("Modifiers", modifiers);
        }
        
        int keyPair[] = tags.getIntArray(key);
        keyPair[0] += increase;
        tags.setIntArray(key, keyPair);
        if (tool.getItem() instanceof Weapon)
        {
            if (keyPair[0] >= 100)
                addEnchantment(tool, Enchantment.looting, 3);
            else if (keyPair[0] >= 50)
                addEnchantment(tool, Enchantment.looting, 2);
            else if (keyPair[0] >= 20)
                addEnchantment(tool, Enchantment.looting, 1);
        }
        else
        {
            if (keyPair[0] >= 100)
                addEnchantment(tool, Enchantment.fortune, 3);
            else if (keyPair[0] >= 50)
                addEnchantment(tool, Enchantment.fortune, 2);
            else if (keyPair[0] >= 20)
                addEnchantment(tool, Enchantment.fortune, 1);
        }
        
        updateModTag(tool, keyPair);
    }
    
    public void addEnchantment(ItemStack tool, Enchantment enchant, int level)
    {
        NBTTagList tags = new NBTTagList("ench");
        Map enchantMap = EnchantmentHelper.getEnchantments(tool);
        Iterator iterator = enchantMap.keySet().iterator();
        int index;
        int lvl;
        boolean hasEnchant = false;
        while (iterator.hasNext())
        {
            NBTTagCompound enchantTag = new NBTTagCompound();
            index = ((Integer)iterator.next()).intValue();
            lvl = (Integer) enchantMap.get(index);
            if (index == enchant.effectId)
            {
                hasEnchant = true;
                enchantTag.setShort("id", (short)index);
                enchantTag.setShort("lvl", (short)((byte)level));
                tags.appendTag(enchantTag);
            }
            else
            {
                enchantTag.setShort("id", (short)index);
                enchantTag.setShort("lvl", (short)((byte)lvl));
                tags.appendTag(enchantTag);
            }
        }
        if (!hasEnchant)
        {
            NBTTagCompound enchantTag = new NBTTagCompound();
            enchantTag.setShort("id", (short)enchant.effectId);
            enchantTag.setShort("lvl", (short)((byte)level));
            tags.appendTag(enchantTag);
        }
        tool.stackTagCompound.setTag("ench", tags);
    }
    
    void updateModTag (ItemStack tool, int[] keys)
    {
        NBTTagCompound tags = tool.getTagCompound().getCompoundTag("InfiTool");
        String tip = "ModifierTip"+keys[1];
        String modName = "\u00a79Lapis ("+keys[0]+"/100)";
        tags.setString(tip, modName);
    }
}