ActuallyAdditions/src/main/java/de/ellpeck/actuallyadditions/mod/inventory/SackContainer.java

/*
 * This file ("ContainerBag.java") is part of the Actually Additions mod for Minecraft.
 * It is created and owned by Ellpeck and distributed
 * under the Actually Additions License to be found at
 * http://ellpeck.de/actaddlicense
 * View the source code at https://github.com/Ellpeck/ActuallyAdditions
 *
 * © 2015-2017 Ellpeck
 */

package de.ellpeck.actuallyadditions.mod.inventory;

import de.ellpeck.actuallyadditions.api.ActuallyTags;
import de.ellpeck.actuallyadditions.mod.inventory.slot.SlotDeletion;
import de.ellpeck.actuallyadditions.mod.inventory.slot.SlotFilter;
import de.ellpeck.actuallyadditions.mod.inventory.slot.SlotImmovable;
import de.ellpeck.actuallyadditions.mod.inventory.slot.SlotItemHandlerUnconditioned;
import de.ellpeck.actuallyadditions.mod.items.Sack;
import de.ellpeck.actuallyadditions.mod.network.gui.IButtonReactor;
import de.ellpeck.actuallyadditions.mod.tile.FilterSettings;
import de.ellpeck.actuallyadditions.mod.util.ItemStackHandlerAA;
import net.minecraft.entity.player.PlayerEntity;
import net.minecraft.entity.player.PlayerInventory;
import net.minecraft.inventory.container.ClickType;
import net.minecraft.inventory.container.Container;
import net.minecraft.inventory.container.Slot;
import net.minecraft.item.ItemStack;
import net.minecraft.nbt.CompoundNBT;
import net.minecraft.network.PacketBuffer;
import net.minecraftforge.api.distmarker.Dist;
import net.minecraftforge.api.distmarker.OnlyIn;

import javax.annotation.Nonnull;
import java.util.UUID;

public class SackContainer extends Container implements IButtonReactor {

    public final FilterSettings filter = new FilterSettings(4, false, false, false);
    private final ItemStackHandlerAA bagInventory;
    private final PlayerInventory inventory;
    public boolean autoInsert;
    private boolean oldAutoInsert;

    public static final int SIZE = 28;

    public static SackContainer fromNetwork(int windowId, PlayerInventory inv, PacketBuffer data) {
        return new SackContainer(windowId, inv, data.readUUID(), new ItemStackHandlerAA(28));
    }

    public SackContainer(int windowId, PlayerInventory playerInventory, UUID uuid, ItemStackHandlerAA handler) {
        super(ActuallyContainers.BAG_CONTAINER.get(), windowId);

        this.inventory = playerInventory;
        this.bagInventory = handler;

        for (int row = 0; row < 4; row++) {
            this.addSlot(new SlotFilter(this.filter, row, 155, 10 + row * 18));
        }

        if (false) { // TODO isvoid, move to its own container
            this.addSlot(new SlotDeletion(this.bagInventory, 0, 64, 65) {
                @Override
                public boolean mayPlace(ItemStack stack) {
                    return SackContainer.this.filter.check(stack);
                }
            });
        }

        // Sack inventory
        for (int row = 0; row < 4; row++) {
            for (int col = 0; col < 7; col++) {
                this.addSlot(new SlotItemHandlerUnconditioned(this.bagInventory, col + row * 7, 10 + col * 18, 10 + row * 18) {
                    @Override
                    public boolean mayPlace(ItemStack stack) {
                        return !stack.getItem().is(ActuallyTags.Items.HOLDS_ITEMS) && SackContainer.this.filter.check(stack);
                    }
                });
            }
        }

        // Player Inventory
        for (int row = 0; row < 3; row++) {
            for (int col = 0; col < 9; col++) {
                this.addSlot(new Slot(playerInventory, col + row * 9 + 9, 8 + col * 18, 94 + row * 18));
            }
        }

        // Player Hotbar
        for (int i = 0; i < 9; i++) {
            if (i == playerInventory.selected) {
                this.addSlot(new SlotImmovable(playerInventory, i, 8 + i * 18, 152));
            } else {
                this.addSlot(new Slot(playerInventory, i, 8 + i * 18, 152));
            }
        }

        ItemStack stack = playerInventory.getSelected();
        if (!stack.isEmpty() && stack.getItem() instanceof Sack) {
            //DrillItem.loadSlotsFromNBT(this.bagInventory, playerInventory.getSelected());
            if (stack.hasTag()) {
                CompoundNBT compound = stack.getOrCreateTag();
                this.filter.readFromNBT(compound, "Filter");
                this.autoInsert = compound.getBoolean("AutoInsert");
            }
        }
    }

    @Override
    public void broadcastChanges() { // TODO is this needed anymore?
        super.broadcastChanges();

        if (this.filter.needsUpdateSend() || this.autoInsert != this.oldAutoInsert) {
            /*
            for (IContainerListener listener : this.containerListeners) {
                listener.setContainerData(this, 0, this.filter.isWhitelist
                    ? 1
                    : 0);
                listener.setContainerData(this, 1, this.filter.respectMeta
                    ? 1
                    : 0);
                listener.setContainerData(this, 2, this.filter.respectNBT
                    ? 1
                    : 0);
                listener.setContainerData(this, 3, this.filter.respectOredict);
                listener.setContainerData(this, 4, this.autoInsert
                    ? 1
                    : 0);
                listener.setContainerData(this, 5, this.filter.respectMod
                    ? 1
                    : 0);
            }
             */
            this.filter.updateLasts();
            this.oldAutoInsert = this.autoInsert;
        }
    }

    @Override
    @OnlyIn(Dist.CLIENT)
    public void setData(int id, int data) {
        if (id == 0) {
            this.filter.isWhitelist = data == 1;
        } else if (id == 1) {
            this.filter.respectNBT = data == 1;
        } else if (id == 2) {
            this.autoInsert = data == 1;
        } else if (id == 3) {
            this.filter.respectMod = data == 1;
        }
    }

    @Override
    public ItemStack quickMoveStack(@Nonnull PlayerEntity player, int slot) {
        int inventoryStart = this.bagInventory.getSlots() + 4;
        int inventoryEnd = inventoryStart + 26;
        int hotbarStart = inventoryEnd + 1;
        int hotbarEnd = hotbarStart + 8;

        Slot theSlot = this.slots.get(slot);

        if (theSlot != null && theSlot.hasItem()) {
            ItemStack newStack = theSlot.getItem();
            ItemStack currentStack = newStack.copy();

            //Other Slots in Inventory excluded
            if (slot >= inventoryStart) {
                //Shift from Inventory
                if (!this.filter.check(newStack) || !this.moveItemStackTo(newStack, 4, 32, false)) {
                    if (slot <= inventoryEnd) {
                        if (!this.moveItemStackTo(newStack, hotbarStart, hotbarEnd + 1, false)) {
                            return ItemStack.EMPTY;
                        }
                    } else if (slot >= inventoryEnd + 1 && slot < hotbarEnd + 1 && !this.moveItemStackTo(newStack, inventoryStart, inventoryEnd + 1, false)) {
                        return ItemStack.EMPTY;
                    }
                }
                //

            } else if (!this.moveItemStackTo(newStack, inventoryStart, hotbarEnd + 1, false)) {
                return ItemStack.EMPTY;
            }

            if (newStack.isEmpty()) {
                theSlot.set(ItemStack.EMPTY);
            } else {
                theSlot.setChanged();
            }

            if (newStack.getCount() == currentStack.getCount()) {
                return ItemStack.EMPTY;
            }
            theSlot.onTake(player, newStack);

            return currentStack;
        }
        return ItemStack.EMPTY;
    }

    @Override
    public ItemStack clicked(int slotId, int dragType, ClickType clickTypeIn, PlayerEntity player) {
        if (SlotFilter.checkFilter(this, slotId, player)) {
            return ItemStack.EMPTY;
        } else if (clickTypeIn == ClickType.SWAP && dragType == this.inventory.selected) {
            return ItemStack.EMPTY;
        } else {
            return super.clicked(slotId, dragType, clickTypeIn, player);
        }
    }

    @Override
    public void removed(PlayerEntity player) {
        ItemStack stack = this.inventory.getSelected();
        if (!stack.isEmpty() && stack.getItem() instanceof Sack) {
            //DrillItem.writeSlotsToNBT(this.bagInventory, this.inventory.getSelected());
            CompoundNBT compound = stack.getOrCreateTag();
            this.filter.writeToNBT(compound, "Filter");
            compound.putBoolean("AutoInsert", this.autoInsert);
        }
        super.removed(player);
    }

    @Override
    public boolean stillValid(PlayerEntity player) {
        return true;
    }

    @Override
    public void onButtonPressed(int buttonID, PlayerEntity player) {
        if (buttonID == 0) {
            this.autoInsert = !this.autoInsert;
        } else {
            //this.filter.onButtonPressed(buttonID); //TODO
        }
    }
}