From 4758eb342ad594db6c3b61b8e7dd316efce2a98b Mon Sep 17 00:00:00 2001
From: miozune <miozune@gmail.com>
Date: Sat, 7 Sep 2024 21:23:49 +0900
Subject: [PATCH] More custom wrapper compat (#78)

---
 .../java/com/cleanroommc/modularui/ClientEventHandler.java    | 4 ++--
 .../cleanroommc/modularui/core/mixin/GuiContainerMixin.java   | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/cleanroommc/modularui/ClientEventHandler.java b/src/main/java/com/cleanroommc/modularui/ClientEventHandler.java
index 01b27cb4..9f721268 100644
--- a/src/main/java/com/cleanroommc/modularui/ClientEventHandler.java
+++ b/src/main/java/com/cleanroommc/modularui/ClientEventHandler.java
@@ -1,7 +1,7 @@
 package com.cleanroommc.modularui;
 
+import com.cleanroommc.modularui.api.IMuiScreen;
 import com.cleanroommc.modularui.drawable.Stencil;
-import com.cleanroommc.modularui.screen.GuiContainerWrapper;
 
 import net.minecraftforge.client.event.GuiScreenEvent;
 import net.minecraftforge.fml.common.eventhandler.EventPriority;
@@ -65,6 +65,6 @@ public static void onGuiInput(GuiScreenEvent.KeyboardInputEvent.Pre event) {
     }
 
     private static boolean hasDraggable(GuiScreenEvent event) {
-        return event.getGui() instanceof GuiContainerWrapper screenWrapper && screenWrapper.getScreen().getContext().hasDraggable();
+        return event.getGui() instanceof IMuiScreen screenWrapper && screenWrapper.getScreen().getContext().hasDraggable();
     }
 }
diff --git a/src/main/java/com/cleanroommc/modularui/core/mixin/GuiContainerMixin.java b/src/main/java/com/cleanroommc/modularui/core/mixin/GuiContainerMixin.java
index 52930e96..51785d65 100644
--- a/src/main/java/com/cleanroommc/modularui/core/mixin/GuiContainerMixin.java
+++ b/src/main/java/com/cleanroommc/modularui/core/mixin/GuiContainerMixin.java
@@ -1,6 +1,6 @@
 package com.cleanroommc.modularui.core.mixin;
 
-import com.cleanroommc.modularui.screen.GuiContainerWrapper;
+import com.cleanroommc.modularui.api.IMuiScreen;
 
 import net.minecraft.client.gui.inventory.GuiContainer;
 import net.minecraft.inventory.Slot;
@@ -24,7 +24,7 @@ public class GuiContainerMixin {
      */
     @Inject(method = "getSlotAtPosition", at = @At("HEAD"), cancellable = true)
     public void getSlot(int x, int y, CallbackInfoReturnable<Slot> cir) {
-        if (((Object) this).getClass() == GuiContainerWrapper.class) {
+        if (IMuiScreen.class.isAssignableFrom(this.getClass())) {
             cir.setReturnValue(this.hoveredSlot);
         }
     }