package codechicken.nei.recipe.stackinfo;

import net.minecraft.item.Item;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraftforge.fluids.FluidContainerRegistry;
import net.minecraftforge.fluids.FluidStack;
import net.minecraftforge.fluids.IFluidContainerItem;

import codechicken.nei.api.IStackStringifyHandler;
import cpw.mods.fml.common.registry.GameData;

public class DefaultStackStringifyHandler implements IStackStringifyHandler {

    public NBTTagCompound convertItemStackToNBT(ItemStack stack, boolean saveStackSize) {
        final String strId = Item.itemRegistry.getNameForObject(stack.getItem());

        if (strId == null) {
            return null;
        }

        final NBTTagCompound nbTag = new NBTTagCompound();
        nbTag.setString("strId", strId);
        nbTag.setInteger("Count", Math.max(saveStackSize ? stack.stackSize : 1, 1));
        nbTag.setShort("Damage", (short) stack.getItemDamage());

        if (stack.hasTagCompound()) {
            nbTag.setTag("tag", stack.getTagCompound());
        }

        return nbTag;
    }

    public ItemStack convertNBTToItemStack(NBTTagCompound nbtTag) {
        final String strId = nbtTag.getString("strId");

        nbtTag = (NBTTagCompound) nbtTag.copy();
        nbtTag.setInteger("id", GameData.getItemRegistry().getId(strId)); // getObject

        final ItemStack stack = ItemStack.loadItemStackFromNBT(nbtTag);

        if (stack != null) {
            stack.stackSize = nbtTag.getInteger("Count");
        }

        return stack;
    }

    public FluidStack getFluid(ItemStack stack) {
        FluidStack fluidStack = FluidContainerRegistry.getFluidForFilledItem(stack);

        if (fluidStack == null && stack.getItem() instanceof IFluidContainerItem) {
            fluidStack = ((IFluidContainerItem) stack.getItem()).getFluid(stack);
        }

        return fluidStack;
    }
}