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);
    }
}