PrettyPipes/src/main/java/de/ellpeck/prettypipes/pipe/modules/stacksize/StackSizeModuleItem.java

package de.ellpeck.prettypipes.pipe.modules.stacksize;

import de.ellpeck.prettypipes.Registry;
import de.ellpeck.prettypipes.items.IModule;
import de.ellpeck.prettypipes.items.ModuleItem;
import de.ellpeck.prettypipes.misc.ItemEquality;
import de.ellpeck.prettypipes.misc.ItemFilter;
import de.ellpeck.prettypipes.pipe.PipeBlockEntity;
import de.ellpeck.prettypipes.pipe.containers.AbstractPipeContainer;
import net.minecraft.world.entity.player.Inventory;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.neoforged.neoforge.items.IItemHandler;

public class StackSizeModuleItem extends ModuleItem {

    public StackSizeModuleItem() {
        super("stack_size_module");
    }

    public static int getMaxStackSizeForModule(ItemStack module) {
        if (module.hasTag()) {
            var amount = module.getTag().getInt("max_stack_size");
            if (amount > 0)
                return amount;
        }
        return 64;
    }

    public static void setMaxStackSize(ItemStack module, int amount) {
        module.getOrCreateTag().putInt("max_stack_size", amount);
    }

    public static boolean getLimitToMaxStackSize(ItemStack module) {
        if (module.hasTag())
            return module.getTag().getBoolean("limit_to_max_stack_size");
        return false;
    }

    public static void setLimitToMaxStackSize(ItemStack module, boolean yes) {
        module.getOrCreateTag().putBoolean("limit_to_max_stack_size", yes);
    }

    @Override
    public int getMaxInsertionAmount(ItemStack module, PipeBlockEntity tile, ItemStack stack, IItemHandler destination) {
        var types = ItemFilter.getEqualityTypes(tile);
        var max = StackSizeModuleItem.getMaxStackSizeForModule(module);
        if (StackSizeModuleItem.getLimitToMaxStackSize(module))
            max = Math.min(max, stack.getMaxStackSize());
        var amount = 0;
        for (var i = 0; i < destination.getSlots(); i++) {
            var stored = destination.getStackInSlot(i);
            if (stored.isEmpty())
                continue;
            if (!ItemEquality.compareItems(stored, stack, types))
                continue;
            amount += stored.getCount();
            if (amount >= max)
                return 0;
        }
        return max - amount;
    }

    @Override
    public boolean isCompatible(ItemStack module, PipeBlockEntity tile, IModule other) {
        return !(other instanceof StackSizeModuleItem);
    }

    @Override
    public boolean hasContainer(ItemStack module, PipeBlockEntity tile) {
        return true;
    }

    @Override
    public AbstractPipeContainer<?> getContainer(ItemStack module, PipeBlockEntity tile, int windowId, Inventory inv, Player player, int moduleIndex) {
        return new StackSizeModuleContainer(Registry.stackSizeModuleContainer, windowId, player, tile.getBlockPos(), moduleIndex);
    }

}