Add LoadoutSwitchWithoutSSC

sgkoishi · Jan 3, 2024 · 78b914c · 78b914c
1 parent 1dd1393
commit 78b914c
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/Core/Config.cs b/Core/Config.cs
@@ -682,6 +682,11 @@ public record class MitigationSettings
         /// </summary>
         public Optional<bool> AllowCrossJourney = Optional.Default(false);
 
+        /// <summary>
+        /// Loadout switch is kind of broken since it interacts with the SSC even when it's disabled.
+        /// </summary>
+        public Optional<bool> LoadoutSwitchWithoutSSC = Optional.Default(true, true);
+
         public enum DisabledDamageAction
         {
             AsIs,

diff --git a/Core/Mitigations.cs b/Core/Mitigations.cs
@@ -702,4 +702,10 @@ private void GDHook_Mitigation_PlayerBuffUpdate(object? sender, TShockAPI.GetDat
         }
         args.Data.Position = currentPosition;
     }
+
+    private bool Detour_Mitigation_HandleSyncLoadout(Func<TShockAPI.GetDataHandlerArgs, bool> orig, TShockAPI.GetDataHandlerArgs args)
+    {
+        var mitigation = this.config.Mitigation.Value;
+        return (mitigation.DisableAllMitigation || !mitigation.LoadoutSwitchWithoutSSC) && orig(args);
+    }
 }
diff --git a/Core/Plugin.cs b/Core/Plugin.cs
@@ -113,6 +113,12 @@ public Plugin(Main game) : base(game)
                 .GetMethod("AttemptConfigUpgrade", _bfany),
             this.Detour_Mitigation_ConfigUpdate
         );
+        this.Detour(
+          nameof(this.Detour_Mitigation_HandleSyncLoadout),
+           typeof(TShockAPI.GetDataHandlers)
+                .GetMethod("HandleSyncLoadout", _bfany),
+          this.Detour_Mitigation_HandleSyncLoadout
+        );
     }
 
     public event Action<Plugin>? OnConfigLoad;