add tool to reset traps

codengine · Feb 28, 2024 · b51329a · b51329a
1 parent ccbc11f
commit b51329a
Show file tree

Hide file tree

Showing 6 changed files with 91 additions and 4 deletions.
diff --git a/SOTFEdit/Model/Constants.cs b/SOTFEdit/Model/Constants.cs
@@ -83,6 +83,7 @@ public static class JsonKeys
         public const string StructureDestruction = "Data.StructureDestruction";
         public const string WorldItemManager = "Data.WorldItemManager";
         public const string ZipLineManager = "Data.ZipLineManager";
+        public const string ScrewTraps = "Data.ScrewTraps";
     }
 
 

diff --git a/SOTFEdit/Model/Savegame/SavegameStore.cs b/SOTFEdit/Model/Savegame/SavegameStore.cs
@@ -30,7 +30,8 @@ public enum FileType
         ScrewStructureNodeInstancesSaveData,
         StructureDestructionSaveData,
         WorldItemManagerSaveData,
-        ZipLineManagerSaveData
+        ZipLineManagerSaveData,
+        ScrewTrapsSaveData
     }
 
     private static readonly ILogger Logger = LogManager.GetCurrentClassLogger();

diff --git a/SOTFEdit/View/MainWindow.xaml b/SOTFEdit/View/MainWindow.xaml
@@ -166,12 +166,15 @@
                     <MenuItem Header="{infrastructure:Translate 'menu.tools.resetNumberOfCutTrees.text'}"
                               Command="{Binding ExperimentResetNumberCutTreesCommand}"
                               ToolTip="{infrastructure:Translate 'menu.tools.resetNumberOfCutTrees.tooltip'}" />
-                    <MenuItem Header="{infrastructure:Translate 'menu.tools.unlocks.text'}"
-                              Command="{Binding UnlocksCommand}"
-                              ToolTip="{infrastructure:Translate 'menu.tools.unlocks.tooltip'}" />
                     <MenuItem Header="{infrastructure:Translate 'menu.tools.resetContainers.text'}"
                               Command="{Binding ResetContainersCommand}"
                               ToolTip="{infrastructure:Translate 'menu.tools.resetContainers.tooltip'}" />
+                    <MenuItem Header="{infrastructure:Translate 'menu.tools.resetTraps.text'}"
+                              Command="{Binding ResetTrapsCommand}"
+                              ToolTip="{infrastructure:Translate 'menu.tools.resetTraps.tooltip'}" />
+                    <MenuItem Header="{infrastructure:Translate 'menu.tools.unlocks.text'}"
+                              Command="{Binding UnlocksCommand}"
+                              ToolTip="{infrastructure:Translate 'menu.tools.unlocks.tooltip'}" />
                 </MenuItem>
                 <MenuItem Header="{infrastructure:Translate 'menu.tools.map.text'}"
                           Command="{Binding OpenMapCommand}"

diff --git a/SOTFEdit/ViewModel/MainViewModel.cs b/SOTFEdit/ViewModel/MainViewModel.cs
@@ -176,6 +176,7 @@ private void RefreshCanExecuteChanged()
         ResetStructureDamageCommand.NotifyCanExecuteChanged();
         TeleportWorldItemCommand.NotifyCanExecuteChanged();
         ResetContainersCommand.NotifyCanExecuteChanged();
+        ResetTrapsCommand.NotifyCanExecuteChanged();
         OnPropertyChanged(nameof(CanEditTabs));
     }
 
@@ -465,6 +466,55 @@ private static void ResetContainers()
         WeakReferenceMessenger.Default.Send(new RequestResetContainersEvent());
     }
 
+    [RelayCommand(CanExecute = nameof(CanSaveAndEdit))]
+    private static void ResetTraps()
+    {
+        if (SavegameManager.SelectedSavegame is not { } selectedSavegame ||
+            selectedSavegame.SavegameStore.LoadJsonRaw(SavegameStore.FileType.ScrewTrapsSaveData) is not
+                { } saveDataWrapper ||
+            saveDataWrapper.GetJsonBasedToken(Constants.JsonKeys.ScrewTraps) is not { } traps ||
+            traps["_tarpStructureData"] is not JArray structureData || structureData.Count == 0)
+        {
+            ShowNoTrapsToResetMessage();
+            return;
+        }
+
+        var countReset = 0;
+
+        foreach (var trapData in structureData)
+        {
+            if (trapData["IsTriggered"] is { } isTriggeredToken && isTriggeredToken.Value<bool>())
+            {
+                trapData["IsTriggered"] = false;
+                countReset++;
+            }
+
+            if (trapData["Data1"] is { } data1Token && data1Token.Value<int>() < 5)
+            {
+                trapData["Data1"] = 5;
+                countReset++;
+            }
+        }
+
+        if (countReset == 0)
+        {
+            ShowNoTrapsToResetMessage();
+            return;
+        }
+
+        saveDataWrapper.MarkAsModified(Constants.JsonKeys.ScrewTraps);
+        WeakReferenceMessenger.Default.Send(new GenericMessageEvent(
+            TranslationManager.GetFormatted("windows.resetTraps.messages.trapsReset.text", countReset),
+            TranslationManager.Get("windows.resetTraps.messages.trapsReset.title")));
+    }
+
+    private static void ShowNoTrapsToResetMessage()
+    {
+        WeakReferenceMessenger.Default.Send(new GenericMessageEvent(
+            TranslationManager.Get("windows.resetTraps.messages.nothingToReset.text"),
+            TranslationManager.Get("windows.resetTraps.messages.nothingToReset.title")));
+    }
+
     public void OnPreviewKeyDown(KeyEventArgs e)
     {
         if (e.Key == Key.F5 && ReloadSavegameCommand.CanExecute(null))

diff --git a/SOTFEdit/data/lang/de.json b/SOTFEdit/data/lang/de.json
@@ -1148,6 +1148,10 @@
         "text": "Schadenszähler für Gebäude zurücksetzen",
         "tooltip": "Ermöglicht es, alle Gebäude zu reparieren."
       },
+      "resetTraps": {
+        "text": "Fallen zurücksetzen",
+        "tooltip": "Setzt alle Fallen zurück"
+      },
       "text": "Tools",
       "unlocks": {
         "text": "Freischaltung",
@@ -1439,6 +1443,18 @@
         }
       }
     },
+    "resetTraps": {
+      "messages": {
+        "nothingToReset": {
+          "text": "Nichts zum zurücksetzen gefunden",
+          "title": "Nichts gefunden"
+        },
+        "trapsReset": {
+          "text": "{0} Fallen wurden zurückgesetzt. Bitte speichere die Änderungen um sie anzuwenden",
+          "title": "Fallen zurückgesetzt"
+        }
+      }
+    },
     "selectSavegame": {
       "buttons": {
         "change": "Ändern...",

diff --git a/SOTFEdit/data/lang/en.json b/SOTFEdit/data/lang/en.json
@@ -1148,6 +1148,10 @@
         "text": "Reset Structure Damage",
         "tooltip": "Lets you repair all structures"
       },
+      "resetTraps": {
+        "text": "Reset Traps",
+        "tooltip": "Resets all traps"
+      },
       "text": "Tools",
       "unlocks": {
         "text": "Unlocks",
@@ -1439,6 +1443,18 @@
         }
       }
     },
+    "resetTraps": {
+      "messages": {
+        "nothingToReset": {
+          "text": "Nothing to reset found",
+          "title": "Nothing found"
+        },
+        "trapsReset": {
+          "text": "{0} traps have been reset. Please save the changes to apply them to the savegame",
+          "title": "Traps reset"
+        }
+      }
+    },
     "selectSavegame": {
       "buttons": {
         "change": "Change...",