added remove weapon trait debug action

LinnielDW · Apr 23, 2023 · f06d364 · f06d364
1 parent 494e4d0
commit f06d364
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/1.4/Assemblies/MorePersonaTraits.dll b/1.4/Assemblies/MorePersonaTraits.dll
diff --git a/Source/DebugActions/DebugActions.cs b/Source/DebugActions/DebugActions.cs
@@ -56,6 +56,31 @@ private static void AddTrait(
                 weapon.TryGetComp<CompBladelinkWeapon>().RegainBond();
             }
         }
+
+        [DebugAction("PersonaWeapons", "[MPT] Remove Weapon Trait", true, actionType = DebugActionType.ToolMap, allowedGameStates = AllowedGameStates.PlayingOnMap)]
+        private static void RemoveWeaponTrait()
+        {
+            List<DebugMenuOption> list = new List<DebugMenuOption>();
+            var weapon = (ThingWithComps)Find.CurrentMap.thingGrid.ThingsAt(UI.MouseCell()).First(t => t is ThingWithComps comps && comps.TryGetComp<CompBladelinkWeapon>() != null);
+
+            foreach (var trait in weapon.TryGetComp<CompBladelinkWeapon>().TraitsListForReading)
+            {
+                list.Add(new DebugMenuOption(trait.defName, DebugMenuOptionMode.Action,
+                    delegate { RemoveTrait(weapon, trait); }));
+            }
+
+            Find.WindowStack.Add(new Dialog_DebugOptionListLister(list));
+        }
+
+        private static void RemoveTrait(
+            ThingWithComps weapon,
+            WeaponTraitDef weaponTraitDef)
+        {
+            weapon.TryGetComp<CompBladelinkWeapon>().TempUnbond();
+            weapon.TryGetComp<CompBladelinkWeapon>().TraitsListForReading.Remove(weaponTraitDef);
+            if (weapon.TryGetComp<CompBladelinkWeapon>().TraitsListForReading.NullOrEmpty()) weapon.TryGetComp<CompBladelinkWeapon>().InitializeTraits();
+            weapon.TryGetComp<CompBladelinkWeapon>().RegainBond();
+        }
 
         [DebugAction("PersonaWeapons", "[MPT] Reroll Weapon Traits", true, actionType = DebugActionType.ToolMap, allowedGameStates = AllowedGameStates.PlayingOnMap)]
         private static void RerollTraits()