shit works 2

Author: Midget807

src/main/java/net/midget807/jewellery_box/block/jewellery_box/JewelleryBoxBlock.java

@@ -71,11 +71,30 @@ public ActionResult onUse(BlockState state, World world, BlockPos pos, PlayerEnt
         if (world.isClient) {
             return ActionResult.SUCCESS;
         } else {
-            player.openHandledScreen(
-                    new SimpleNamedScreenHandlerFactory(
-                            (syncId, playerInventory, player1) -> JewelleryBoxScreenHandler.createFull(syncId, playerInventory), this.getName()
-                    )
-            );
+            switch (size) {
+                case 2:
+                    player.openHandledScreen(
+                            new SimpleNamedScreenHandlerFactory(
+                                    (syncId, playerInventory, player1) -> JewelleryBoxScreenHandler.createQuarter(syncId, playerInventory), this.getName()
+                            )
+                    );
+                    break;
+                case 4:
+                    player.openHandledScreen(
+                            new SimpleNamedScreenHandlerFactory(
+                                    (syncId, playerInventory, player1) -> JewelleryBoxScreenHandler.createHalf(syncId, playerInventory), this.getName()
+                            )
+                    );
+                    break;
+                default:
+                    player.openHandledScreen(
+                            new SimpleNamedScreenHandlerFactory(
+                                    (syncId, playerInventory, player1) -> JewelleryBoxScreenHandler.createFull(syncId, playerInventory), this.getName()
+                            )
+                    );
+                    break;
+            }
+
             player.incrementStat(this.getOpenStat());
             PiglinBrain.onGuardedBlockInteracted(player, true);
 

src/main/java/net/midget807/jewellery_box/client/screen/JewelleryBoxScreen.java

@@ -1,5 +1,6 @@
 package net.midget807.jewellery_box.client.screen;
 
+import net.midget807.jewellery_box.JewelleryBoxMain;
 import net.midget807.jewellery_box.screen.jewellery_box.JewelleryBoxScreenHandler;
 import net.minecraft.client.gui.DrawContext;
 import net.minecraft.client.gui.screen.ingame.HandledScreen;
@@ -9,12 +10,14 @@
 import net.minecraft.util.Identifier;
 
 public class JewelleryBoxScreen extends HandledScreen<JewelleryBoxScreenHandler> implements ScreenHandlerProvider<JewelleryBoxScreenHandler> {
-    private static final Identifier TEXTURE = new Identifier("textures/gui/container/generic_54.png");
+    private static final Identifier TEXTURE_8 = new Identifier(JewelleryBoxMain.MOD_ID, "textures/gui/container/jewellery_box_8.png");
+    private static final Identifier TEXTURE_4 = new Identifier(JewelleryBoxMain.MOD_ID, "textures/gui/container/jewellery_box_4.png");
+    private static final Identifier TEXTURE_2 = new Identifier(JewelleryBoxMain.MOD_ID, "textures/gui/container/jewellery_box_2.png");
     private final int columns;
     public JewelleryBoxScreen(JewelleryBoxScreenHandler handler, PlayerInventory inventory, Text title) {
         super(handler, inventory, title);
         this.columns = handler.getColumns();
-        this.backgroundHeight = 114;
+        this.backgroundHeight = 133;
         this.playerInventoryTitleY = this.backgroundHeight - 94;
     }
 
@@ -29,7 +32,15 @@ public void render(DrawContext context, int mouseX, int mouseY, float delta) {
     protected void drawBackground(DrawContext context, float delta, int mouseX, int mouseY) {
         int i = (this.width - this.backgroundWidth) / 2;
         int j = (this.height - this.backgroundHeight) / 2;
-        context.drawTexture(TEXTURE, i, j, 0, 0, this.columns * 18 + 17, this.backgroundHeight);
-        context.drawTexture(TEXTURE, i, j + 18 + 17, 0, 126, this.columns * 18 + 17, 96);
+        switch (columns) {
+            case 4:
+                context.drawTexture(TEXTURE_4, i, j, 0, 0, this.backgroundWidth, this.backgroundHeight);
+                break;
+            case 2:
+                context.drawTexture(TEXTURE_2, i, j, 0, 0, this.backgroundWidth, this.backgroundHeight);
+                break;
+            default:
+                context.drawTexture(TEXTURE_8, i, j, 0, 0, this.backgroundWidth, this.backgroundHeight);
+        }
     }
 }

src/main/java/net/midget807/jewellery_box/screen/jewellery_box/JewelleryBoxScreenHandler.java

@@ -29,22 +29,22 @@ public static JewelleryBoxScreenHandler createQuarter(int syncId, PlayerInventor
     }
     public JewelleryBoxScreenHandler(ScreenHandlerType<?> type, int syncId, PlayerInventory playerInventory, Inventory inventory, int columns) {
         super(type, syncId);
-        checkSize(inventory, columns);
         this.inventory = inventory;
         this.columns = columns;
+        checkSize(inventory, columns);
         inventory.onOpen(playerInventory.player);
-        int i = -3 * 18;
+        int i = 51;
         for (int j = 0; j < columns; j++) {
-            this.addSlot(new Slot(inventory, j, 8 + j * 18, 18));
+            this.addSlot(new Slot(inventory, j, 17 + j * 18, 20));
         }
         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));
+                this.addSlot(new Slot(playerInventory, k + j * 9 + 9, 8 + k * 18, j * 18 + i));
             }
         }
 
         for (int j = 0; j < 9; j++) {
-            this.addSlot(new Slot(playerInventory, j, 8 + j * 18, 161 + i));
+            this.addSlot(new Slot(playerInventory, j, 8 + j * 18, 109));
         }
     }
 
@@ -55,11 +55,11 @@ public ItemStack quickMove(PlayerEntity player, int slot) {
         if (slot2 != null && slot2.hasStack()) {
             ItemStack itemStack2 = slot2.getStack();
             itemStack = itemStack2.copy();
-            if (slot < columns) {
-                if (!this.insertItem(itemStack2, columns, this.slots.size(), true)) {
+            if (slot < this.inventory.size()) {
+                if (!this.insertItem(itemStack2, this.inventory.size(), this.slots.size(), true)) {
                     return ItemStack.EMPTY;
                 }
-            } else if (!this.insertItem(itemStack2, 0, columns, false)) {
+            } else if (!this.insertItem(itemStack2, 0, this.inventory.size(), false)) {
                 return ItemStack.EMPTY;
             }