NotEnoughItems/src/main/java/codechicken/nei/InfiniteToolHandler.java

package codechicken.nei;

import net.minecraft.entity.player.InventoryPlayer;
import net.minecraft.item.ItemStack;

import codechicken.nei.api.IInfiniteItemHandler;

public class InfiniteToolHandler implements IInfiniteItemHandler {

    @Override
    public void onPickup(ItemStack heldItem) {
        heldItem.setItemDamage(0);
    }

    @Override
    public void onPlaceInfinite(ItemStack heldItem) {
        heldItem.setItemDamage(-32000);
    }

    @Override
    public void replenishInfiniteStack(InventoryPlayer inv, int slotNo) {
        inv.getStackInSlot(slotNo).setItemDamage(-32000);
    }

    @Override
    public boolean canHandleItem(ItemStack stack) {
        return stack.getItem().isDamageable() && stack.getMaxStackSize() == 1;
    }

    @Override
    public boolean isItemInfinite(ItemStack stack) {
        return stack.getItemDamage() < -30000;
    }

    @Override
    public ItemStack getInfiniteItem(ItemStack typeStack) {
        return new ItemStack(typeStack.getItem(), 1, -32000);
    }
}