package darkknight.jewelrycraft.util;

import java.util.ArrayList;

import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;

public class JewelryNBT {
	// TODO NBT Tag Adding
	public static void addItem(ItemStack item, ItemStack target) {
		if (target != null) {
			NBTTagCompound itemStackData;

			if (item.hasTagCompound()) {
				itemStackData = item.getTagCompound();
			} else {
				itemStackData = new NBTTagCompound();
				item.setTagCompound(itemStackData);
			}

			NBTTagCompound targetNBT = new NBTTagCompound();
			target.writeToNBT(targetNBT);
			itemStackData.setTag("target", targetNBT);
		}
	}

	/**
	 * @param item
	 *            The item you want to add the NBT data on
	 * @param metal
	 *            The metal you want to add on the item
	 */
	public static void addMetal(ItemStack item, ItemStack metal) {
		NBTTagCompound itemStackData;

		if (item.hasTagCompound()) {
			itemStackData = item.getTagCompound();
		} else {
			itemStackData = new NBTTagCompound();
			item.setTagCompound(itemStackData);
		}

		if (metal != null) {
			NBTTagCompound ingotNBT = new NBTTagCompound();
			metal.writeToNBT(ingotNBT);
			itemStackData.setTag("ingot", ingotNBT);
		}
	}

	/**
	 * @param item
	 *            The item you want to add the NBT data on
	 * @param gem
	 *            The gem you want to add on the item
	 */
	public static void addGem(ItemStack item, ItemStack gem) {
		if (gem != null) {
			NBTTagCompound itemStackData;

			if (item.hasTagCompound()) {
				itemStackData = item.getTagCompound();
			} else {
				itemStackData = new NBTTagCompound();
				item.setTagCompound(itemStackData);
			}

			NBTTagCompound gemNBT = new NBTTagCompound();
			gem.writeToNBT(gemNBT);
			itemStackData.setTag("gem", gemNBT);
		}
	}
	
	/**
	 * @param item
	 *            The item you want to add the NBT data on
	 * @param prefix
	 *            The prefix you want to add on the item
	 */
	public static void addPrefix(ItemStack item, String prefix) {
		if (prefix != null) {
			NBTTagCompound itemStackData;

			if (item.hasTagCompound()) {
				itemStackData = item.getTagCompound();
			} else {
				itemStackData = new NBTTagCompound();
				item.setTagCompound(itemStackData);
			}

			itemStackData.setString("prefix", prefix);
		}
	}
	
	/**
	 * @param item
	 *            The item you want to add the NBT data on
	 * @param suffix
	 *            The suffix you want to add on the item
	 */
	public static void addSuffix(ItemStack item, String suffix) {
		if (suffix != null) {
			NBTTagCompound itemStackData;

			if (item.hasTagCompound()) {
				itemStackData = item.getTagCompound();
			} else {
				itemStackData = new NBTTagCompound();
				item.setTagCompound(itemStackData);
			}

			itemStackData.setString("suffix", suffix);
		}
	}

	/**
	 * @param item
	 *            The item you want to add the NBT data on
	 * @param modifier
	 *            The modifier you want to add on the item
	 */
	public static void addModifiers(ItemStack item, ArrayList<ItemStack> modifier) {
		if (modifier != null) {
			NBTTagCompound itemStackData;

			if (item.hasTagCompound()) {
				itemStackData = item.getTagCompound();
			} else {
				itemStackData = new NBTTagCompound();
				item.setTagCompound(itemStackData);
			}

			for (int i = 0; i < modifier.size(); i++) {
				NBTTagCompound modifierNBT = new NBTTagCompound();
				modifier.get(i).writeToNBT(modifierNBT);
				itemStackData.setTag("modifier" + i, modifierNBT);
			}

			itemStackData.setInteger("modifierSize", modifier.size());
		}
	}

	/**
	 * @param item
	 * @param color
	 */
	public static void addIngotColor(ItemStack item, int color) {
		NBTTagCompound itemStackData;

		if (item.hasTagCompound()) {
			itemStackData = item.getTagCompound();
		} else {
			itemStackData = new NBTTagCompound();
			item.setTagCompound(itemStackData);
		}

		itemStackData.setInteger("ingotColor", color);
	}

	// TODO
	/**
	 * @param item
	 * @param color
	 */
	public static void addGemColor(ItemStack item, int color) {
		NBTTagCompound itemStackData;

		if (item.hasTagCompound()) {
			itemStackData = item.getTagCompound();
		} else {
			itemStackData = new NBTTagCompound();
			item.setTagCompound(itemStackData);
		}

		itemStackData.setInteger("gemColor", color);
	}

	// TODO NTB Tag Checking
	/**
	 * @param item
	 * @param tag
	 * @return
	 */
	public static boolean hasTag(ItemStack item, String tag) {
		NBTTagCompound itemStackData;

		if (item.hasTagCompound()) {
			itemStackData = item.getTagCompound();
		} else {
			itemStackData = new NBTTagCompound();
			item.setTagCompound(itemStackData);
		}

		if (itemStackData.hasKey(tag)) {
			return true;
		}

		return false;
	}

	/**
	 * @param stack
	 * @param gem
	 * @return
	 */
	public static boolean isGemX(ItemStack stack, ItemStack gem) {
		if (gem(stack) != null) {
			if (gem(stack).getItem() == gem.getItem()) {
				if (gem(stack).getItemDamage() == gem.getItemDamage()) {
					return true;
				}
			}
		}

		return false;
	}

	/**
	 * @param stack
	 * @param modifier
	 * @return
	 */
	public static boolean doesModifierExist(ItemStack stack, ItemStack modifier) {
		if (modifier(stack) != null) {
			ArrayList<ItemStack> list = modifier(stack);

			for (int i = 0; i < list.size(); i++) {
				if (list.get(i).getItem() == modifier.getItem()) {
					if (list.get(i).getItemDamage() == modifier.getItemDamage()) {
						return true;
					}
				}
			}
		}

		return false;
	}

	public static int modifierSize(ItemStack stack, ItemStack modifier) {
		if (modifier(stack) != null) {
			ArrayList<ItemStack> list = modifier(stack);

			for (int i = 0; i < list.size(); i++) {
				if (list.get(i).getItem() == modifier.getItem()) {
					if (list.get(i).getItemDamage() == modifier.getItemDamage()) {
						return list.get(i).stackSize;
					}
				}
			}
		}

		return -1;
	}

	public static int numberOfModifiers(ItemStack stack) {
		if (modifier(stack) != null) {
			return modifier(stack).size();
		}

		return -1;
	}

	/**
	 * @param stack
	 * @param ingot
	 * @return
	 */
	public static boolean isIngotX(ItemStack stack, ItemStack ingot) {
		if (ingot(stack) != null) {
			if (ingot(stack).getItem() == ingot.getItem()) {
				if (ingot(stack).getItemDamage() == ingot.getItemDamage())
					return true;
			}
		}

		return false;
	}

	// TODO Return components based on NBT
	public static ItemStack item(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound()) {
					if (stack.getTagCompound().hasKey("target")) {
						NBTTagCompound itemNBT = (NBTTagCompound) stack.getTagCompound().getTag("target");

						ItemStack target = new ItemStack(Item.getItemById(0), 0, 0);

						target.readFromNBT(itemNBT);

						return target;
					}
				}
			}
		}

		return null;
	}

	/**
	 * @param stack
	 * @return
	 */
	public static ItemStack gem(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound() && stack.getTagCompound().hasKey("gem")) {
					NBTTagCompound jewelNBT = (NBTTagCompound) stack.getTagCompound().getTag("gem");

					ItemStack gem = new ItemStack(Item.getItemById(0), 0, 0);

					gem.readFromNBT(jewelNBT);

					return gem;
				}
			}
		}

		return null;
	}
	
	public static String prefix(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound() && stack.getTagCompound().hasKey("prefix")) {
					return stack.getTagCompound().getString("prefix");
				}
			}
		}

		return null;
	}
	
	public static String suffix(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound() && stack.getTagCompound().hasKey("suffix")) {
					return stack.getTagCompound().getString("suffix");
				}
			}
		}

		return null;
	}

	/**
	 * @param stack
	 * @return
	 */
	public static ArrayList<ItemStack> modifier(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound()) {
					int size = stack.getTagCompound().getInteger("modifierSize");

					ArrayList<ItemStack> list = new ArrayList<>();

					for (int i = 0; i < size; i++) {
						ItemStack modifier = new ItemStack(Item.getItemById(0), 0, 0);

						NBTTagCompound modifierNBT = (NBTTagCompound) stack.getTagCompound().getTag("modifier" + i);

						modifier.readFromNBT(modifierNBT);

						list.add(modifier);
					}

					return list;
				}
			}
		}

		return null;
	}

	/**
	 * @param stack
	 * @return
	 */
	public static ItemStack ingot(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound()) {
					if (stack.getTagCompound().hasKey("ingot")) {
						NBTTagCompound ingotNBT = (NBTTagCompound) stack.getTagCompound().getTag("ingot");

						ItemStack ingot = new ItemStack(Item.getItemById(0), 0, 0);

						ingot.readFromNBT(ingotNBT);

						return ingot;
					}
				}
			}
		}

		return null;
	}

	/**
	 * @param stack
	 * @return
	 */
	public static int ingotColor(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound()) {
					if (stack.getTagCompound().hasKey("ingotColor")) {
						return stack.getTagCompound().getInteger("ingotColor");
					}
				}
			}
		}

		return 16777215;
	}

	// TODO
	/**
	 * @param stack
	 * @return
	 */
	public static int gemColor(ItemStack stack) {
		if (stack != null) {
			if (stack != new ItemStack(Item.getItemById(0), 0, 0)) {
				if (stack.hasTagCompound()) {
					if (stack.getTagCompound().hasKey("gemColor"))
						return stack.getTagCompound().getInteger("gemColor");
				}
			}
		}

		return 16777215;
	}
}