misc: 一些调整

NiFeather · Dec 16, 2023 · 4ce9900 · 4ce9900
1 parent b0eff61
commit 4ce9900
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/xiamomc/morph/backends/fallback/NilBackend.java b/src/main/java/xiamomc/morph/backends/fallback/NilBackend.java
@@ -76,6 +76,9 @@ public boolean disguise(Player player, DisguiseWrapper<?> rawWrapper)
         if (!(rawWrapper instanceof NilWrapper wrapper))
             return false;
 
+        if (playerFallbackWrapperMap.containsKey(player))
+            unDisguise(player);
+
         playerFallbackWrapperMap.put(player, wrapper);
         return true;
     }
@@ -84,6 +87,10 @@ public boolean disguise(Player player, DisguiseWrapper<?> rawWrapper)
     public boolean unDisguise(Player player)
     {
         var wrapper = playerFallbackWrapperMap.getOrDefault(player, null);
+
+        if (wrapper != null)
+            wrapper.dispose();
+
         playerFallbackWrapperMap.remove(player);
 
         return true;