✅ Add a Max Traits mutator

Author: Chasmical

RogueLibsCore.Test/RogueLibsCore.Test.csproj.DotSettings

@@ -4,6 +4,6 @@
 	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Cdisasters/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Ceffects/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Citems/@EntryIndexedValue">True</s:Boolean>
-	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Cmutators/@EntryIndexedValue">False</s:Boolean>
+	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Cmutators/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Ctraits/@EntryIndexedValue">True</s:Boolean>
 	<s:Boolean x:Key="/Default/CodeInspection/NamespaceProvider/NamespaceFoldersToSkip/=tests_005Cweapons/@EntryIndexedValue">True</s:Boolean></wpf:ResourceDictionary>

RogueLibsCore.Test/Tests/Mutators/MaxTraits.cs

@@ -0,0 +1,44 @@
+using System.Collections.Generic;
+
+namespace RogueLibsCore.Test
+{
+    public class MaxTraits : MutatorUnlock
+    {
+        public MaxTraits() : base(nameof(MaxTraits)) { UnlockCost = 3; }
+
+        [RLSetup]
+        public static void Setup()
+        {
+            RogueLibs.CreateCustomUnlock(new MaxTraits())
+                     .WithName(new CustomNameInfo("Max Traits"))
+                     .WithDescription(new CustomNameInfo("Everyone gets their traits upgraded!"));
+
+            RoguePatcher patcher = new RoguePatcher(TestPlugin.Instance) { TypeWithPatches = typeof(MaxTraits) };
+            patcher.Postfix(typeof(Agent), nameof(Agent.SetupAgentStats));
+        }
+
+        public static void Agent_SetupAgentStats(Agent __instance)
+        {
+            if (!gc.challenges.Contains(nameof(MaxTraits))) return;
+
+            List<Trait> upgradeable = __instance.statusEffects.TraitList.FindAll(static trait =>
+            {
+                UnlockWrapper? unlock = RogueFramework.Unlocks.Find(u => u.Type == UnlockTypes.Trait && u.Name == trait.traitName);
+                return !string.IsNullOrEmpty(unlock?.Unlock.upgrade);
+            });
+            if (upgradeable.Count > 0)
+            {
+                __instance.usingAugmentationBooth = true;
+                upgradeable.ForEach(trait =>
+                {
+                    UnlockWrapper unlock = RogueFramework.Unlocks.Find(u => u.Type == UnlockTypes.Trait && u.Name == trait.traitName);
+                    __instance.statusEffects.AddTrait(unlock.Unlock.upgrade);
+                });
+                __instance.usingAugmentationBooth = false;
+            }
+        }
+
+
+
+    }
+}