Refresh creative search order on gamemode change

Author: NotRyken

common/src/main/java/dev/terminalmc/clientsort/client/order/CreativeSearchOrder.java

@@ -71,7 +71,7 @@ public static int getPosition(ItemStack stack) {
     public static void tryRefreshStackPositionMap() {
         if (Config.options().optimizeCreativeSorting) {
             if (ClientSort.emiReloading) {
-                dev.terminalmc.clientsort.client.ClientSort.updateBlockedByEmi = true;
+                ClientSort.updateBlockedByEmi = true;
                 ClientSort.LOG.info("Search order update blocked by EMI reload, waiting...");
             } else {
                 refreshStackPositionMap();

common/src/main/java/dev/terminalmc/clientsort/mixin/client/LocalPlayerMixin.java

@@ -21,6 +21,7 @@
 import dev.terminalmc.clientsort.client.interaction.InteractionManager;
 import dev.terminalmc.clientsort.client.order.CreativeSearchOrder;
 import net.minecraft.client.player.LocalPlayer;
+import net.minecraft.world.level.GameType;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
@@ -36,7 +37,7 @@ public abstract class LocalPlayerMixin {
             method = "clientSideCloseContainer",
             at = @At("HEAD")
     )
-    public void beforeContainerClose(CallbackInfo callbackInfo) {
+    private void beforeContainerClose(CallbackInfo callbackInfo) {
         InteractionManager.clear();
         ClientSort.operatingClient = false;
     }
@@ -45,10 +46,18 @@ public void beforeContainerClose(CallbackInfo callbackInfo) {
             method = "setPermissionLevel",
             at = @At("RETURN")
     )
-    public void afterPermissionLevelChange(int level, CallbackInfo ci) {
+    private void afterPermissionLevelChange(int level, CallbackInfo ci) {
         if (!ClientSort.searchOrderUpdated) {
             ClientSort.searchOrderUpdated = true;
             CreativeSearchOrder.tryRefreshStackPositionMap();
         }
     }
+
+    @Inject(
+            method = "onGameModeChanged",
+            at = @At("RETURN")
+    )
+    private void afterGameModeChange(GameType gameMode, CallbackInfo ci) {
+        CreativeSearchOrder.tryRefreshStackPositionMap();
+    }
 }