From 2e0dce868b79998fd1c1eec312f8645993208058 Mon Sep 17 00:00:00 2001
From: zontreck <tarapiccari@gmail.com>
Date: Sun, 14 Jan 2024 19:20:53 -0700
Subject: [PATCH] Fix compile error

---
 .../java/dev/zontreck/libzontreck/chestgui/ChestGUI.java | 4 ++++
 .../zontreck/libzontreck/chestgui/ChestGUIButton.java    | 9 ++++++++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUI.java b/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUI.java
index 055adaa..ae829ec 100644
--- a/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUI.java
+++ b/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUI.java
@@ -83,6 +83,7 @@ public class ChestGUI
             return this;
         }
         buttons.add(button);
+        button.withContainer(container);
         container.setStackInSlot(button.getSlotNum(), button.buildIcon());
 
         return this;
@@ -113,6 +114,7 @@ public class ChestGUI
             ItemStack remStack = new ItemStack(ModItems.CHESTGUI_REM.get(), 1);
 
             ChestGUIButton rem = new ChestGUIButton(remStack, onRemove, new Vector2i(2, 3));
+            rem.withContainer(container);
 
             removeBtn = rem;
 
@@ -124,6 +126,7 @@ public class ChestGUI
             ItemStack resStack = new ItemStack(ModItems.CHESTGUI_RESET.get(), 1);
 
             ChestGUIButton rem = new ChestGUIButton(resStack, onReset, new Vector2i(2, 4));
+            rem.withContainer(container);
 
             resetBtn = rem;
 
@@ -137,6 +140,7 @@ public class ChestGUI
             ItemStack remStack = new ItemStack(ModItems.CHESTGUI_ADD.get(), 1);
 
             ChestGUIButton rem = new ChestGUIButton(remStack, onAdd, new Vector2i(2, 5));
+            rem.withContainer(container);
 
             addBtn = rem;
 
diff --git a/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUIButton.java b/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUIButton.java
index 42ae498..613a666 100644
--- a/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUIButton.java
+++ b/src/main/java/dev/zontreck/libzontreck/chestgui/ChestGUIButton.java
@@ -28,6 +28,7 @@ public class ChestGUIButton
      */
     private Vector2i position;
     private ItemStack built;
+    private ItemStackHandler container;
 
     /**
      * Sets the name of the ChestGUI Button (Item Name)
@@ -102,6 +103,12 @@ public class ChestGUIButton
         return st;
     }
 
+    protected ChestGUIButton withContainer(ItemStackHandler handler)
+    {
+        this.container=handler;
+        return this;
+    }
+
     /**
      * Adds a line to the Lore (Tooltip) of the button
      * @param line The line to add
@@ -147,6 +154,6 @@ public class ChestGUIButton
 
     public void clicked()
     {
-        callback.run(built);
+        callback.run(built, container);
     }
 }