Fix Godmode showing the incorrect toggle state

Author: RhythmicSys

src/main/java/simplexity/adminhax/commands/hax/Fly.java

@@ -64,6 +64,5 @@ private void switchFlyState(Player player) {
         Util.flipPDCState(player, flyStatus, false);
         player.setAllowFlight(false);
         player.setFlying(false);
-        return;
     }
 }

src/main/java/simplexity/adminhax/commands/withargs/GodMode.java

@@ -11,7 +11,7 @@
 public class GodMode extends AbstractArgsCommands {
     public static final String GET_ARG = "get";
     public static final String TOGGLE_ARG = "toggle";
-    private NamespacedKey godModeStatus = Util.GODMODE_STATUS;
+    private final NamespacedKey godModeStatus = Util.GODMODE_STATUS;
 
     public GodMode(Permission DEFAULT_PERMISSION, String DEFAULT_ARG) {
         super(DEFAULT_PERMISSION, DEFAULT_ARG);
@@ -51,8 +51,14 @@ public void runLogic(CommandSender sender, Player player, String arg) {
     }
 
     public void toggleGodmode(Player player) {
-        player.setInvulnerable(!player.isInvulnerable());
-        Util.flipPDCState(player, godModeStatus, false);
+        boolean currentlyEnabled = Util.isPDCStateEnabled(player, godModeStatus, false);
+        if (currentlyEnabled) {
+            player.setInvulnerable(false);
+            Util.setPdcState(player, godModeStatus, false);
+            return;
+        }
+        player.setInvulnerable(true);
+        Util.setPdcState(player, godModeStatus, true);
     }
 
     public String getGodmodeStatus(Player player) {

src/main/java/simplexity/adminhax/util/Util.java

@@ -76,6 +76,11 @@ public static void flipPDCState(Player player, NamespacedKey namespacedKey, bool
         }
     }
 
+    public static void setPdcState(Player player, NamespacedKey namespacedKey, boolean state){
+        PersistentDataContainer playerPdc = player.getPersistentDataContainer();
+        playerPdc.set(namespacedKey, PersistentDataType.BOOLEAN, state);
+    }
+
     public static boolean isPDCStateEnabled(Player player, NamespacedKey namespacedKey, boolean defaultState){
         PersistentDataContainer playerPDC = player.getPersistentDataContainer();
         return playerPDC.getOrDefault(namespacedKey, PersistentDataType.BOOLEAN, defaultState);