package com.example.classicchestmenus.screen; import com.example.classicchestmenus.ClassicChestMenus; import com.example.classicchestmenus.data.MenuDefinition; import com.example.classicchestmenus.data.MenuItem; import net.minecraft.entity.player.PlayerEntity; import net.minecraft.entity.player.PlayerInventory; import net.minecraft.inventory.Inventory; import net.minecraft.inventory.SimpleInventory; import net.minecraft.item.ItemStack; import net.minecraft.registry.Registries; import net.minecraft.registry.RegistryKey; import net.minecraft.registry.RegistryKeys; import net.minecraft.screen.ScreenHandler; import net.minecraft.screen.ScreenHandlerType; import net.minecraft.screen.slot.Slot; import net.minecraft.util.Identifier; public class CustomChestScreenHandler extends ScreenHandler { public static ScreenHandlerType TYPE; private final Inventory inventory; private final int rows; public CustomChestScreenHandler(int syncId, PlayerInventory playerInventory) { this(syncId, playerInventory, new SimpleInventory(54), 6); } public CustomChestScreenHandler(int syncId, PlayerInventory playerInventory, Inventory inventory, int rows) { super(TYPE, syncId); this.inventory = inventory; this.rows = rows; inventory.onOpen(playerInventory.player); int i = (rows - 4) * 18; for (int j = 0; j < rows; ++j) { for (int k = 0; k < 9; ++k) { this.addSlot(new Slot(inventory, k + j * 9, 8 + k * 18, 18 + j * 18)); } } for (int j = 0; j < 3; ++j) { for (int k = 0; k < 9; ++k) { this.addSlot(new Slot(playerInventory, k + j * 9 + 9, 8 + k * 18, 103 + j * 18 + i)); } } for (int j = 0; j < 9; ++j) { this.addSlot(new Slot(playerInventory, j, 8 + j * 18, 161 + i)); } } @Override public boolean canUse(PlayerEntity player) { return this.inventory.canPlayerUse(player); } @Override public ItemStack quickMove(PlayerEntity player, int slot) { ItemStack itemStack = ItemStack.EMPTY; Slot slot2 = this.slots.get(slot); if (slot2 != null && slot2.hasStack()) { ItemStack itemStack2 = slot2.getStack(); itemStack = itemStack2.copy(); if (slot < this.rows * 9) { if (!this.insertItem(itemStack2, this.rows * 9, this.slots.size(), true)) { return ItemStack.EMPTY; } } else if (!this.insertItem(itemStack2, 0, this.rows * 9, false)) { return ItemStack.EMPTY; } if (itemStack2.isEmpty()) { slot2.setStack(ItemStack.EMPTY); } else { slot2.markDirty(); } } return itemStack; } @Override public void onClosed(PlayerEntity player) { super.onClosed(player); this.inventory.onClose(player); } public int getRows() { return rows; } public static void register() { RegistryKey> key = RegistryKey.of( RegistryKeys.SCREEN_HANDLER, Identifier.of(ClassicChestMenus.MOD_ID, "custom_chest") ); TYPE = new ScreenHandlerType<>(CustomChestScreenHandler::new, net.minecraft.resource.featuretoggle.FeatureSet.empty()); Registries.SCREEN_HANDLER.add(key, TYPE); } }