InfiniteStackSizeHandler.java

package codechicken.nei;

import codechicken.lib.inventory.InventoryUtils;
import codechicken.nei.api.IInfiniteItemHandler;
import net.minecraft.entity.player.InventoryPlayer;
import net.minecraft.item.ItemStack;

public class InfiniteStackSizeHandler implements IInfiniteItemHandler
{
    @Override
    public void onPickup(ItemStack heldItem)
    {
        heldItem.stackSize = 1;
    }

    @Override
    public void onPlaceInfinite(ItemStack heldItem)
    {
        heldItem.stackSize = 111;
    }

    @Override
    public boolean canHandleItem(ItemStack stack)
    {
        return !stack.isItemStackDamageable();
    }

    @Override
    public boolean isItemInfinite(ItemStack stack)
    {
        return stack.stackSize == -1 || stack.stackSize > 100;
    }

    @Override
    public void replenishInfiniteStack(InventoryPlayer inv, int slotNo)
    {
        ItemStack stack = inv.getStackInSlot(slotNo);
        stack.stackSize = 111;
        
        for(int i = 0; i < inv.getSizeInventory(); i++)
        {
            if(i == slotNo)
                continue;
            
            if(NEIServerUtils.areStacksSameType(stack, inv.getStackInSlot(i)))
                inv.setInventorySlotContents(i, null);
        }
    }

    @Override
    public ItemStack getInfiniteItem(ItemStack typeStack)
    {
        return InventoryUtils.copyStack(typeStack, -1);
    }
}