Added Feature: "LastUsedGearSwitcher"

closes #58
AuriRex · Oct 19, 2023 · c993dc6 · c993dc6
1 parent c04952b
commit c993dc6
Show file tree

Hide file tree

Showing 2 changed files with 100 additions and 0 deletions.
diff --git a/TheArchive.IL2CPP/Features/QoL/LastUsedGearSwitcher.cs b/TheArchive.IL2CPP/Features/QoL/LastUsedGearSwitcher.cs
@@ -0,0 +1,99 @@
+using Player;
+using TheArchive.Core.Attributes;
+using TheArchive.Core.Attributes.Feature.Settings;
+using TheArchive.Core.FeaturesAPI;
+using TheArchive.Interfaces;
+using TheArchive.Utilities;
+using UnityEngine;
+
+namespace TheArchive.Features.QoL
+{
+    public class LastUsedGearSwitcher : Feature
+    {
+        public override string Name => "Last Used Gear Switcher";
+
+        public override string Group => FeatureGroups.QualityOfLife;
+
+        public override string Description => "Allows you to swap between the last two used weapons via a keypress";
+
+        public new static IArchiveLogger FeatureLogger { get; set; }
+
+        [FeatureConfig]
+        public static LastUsedGearSwitcherSettings Settings { get; set; }
+
+        public class LastUsedGearSwitcherSettings
+        {
+            [FSDisplayName("Quick Swap Key")]
+            [FSDescription("Press this key to switch to the previously wielded gear.")]
+            public KeyCode QuickSwitchKey { get; set; } = KeyCode.X;
+
+            [FSHide]
+            [FSDescription("Prints debug info to the console")]
+            public bool DebugLog { get; set; } = false;
+        }
+
+        private static readonly eFocusState _eFocusState_FPS = Utils.GetEnumFromName<eFocusState>(nameof(eFocusState.FPS));
+        private static eFocusState _eFocusState_FPS_CommunicationDialog;
+
+        public override void Init()
+        {
+#if IL2CPP
+            if(Is.R6OrLater)
+            {
+                Utils.TryGetEnumFromName(nameof(eFocusState.FPS_CommunicationDialog), out _eFocusState_FPS_CommunicationDialog);
+            }
+#endif
+        }
+
+        public override void Update()
+        {
+            if (FocusStateManager.CurrentState != _eFocusState_FPS)
+            {
+                if (!Is.R6OrLater)
+                    return;
+
+                if(FocusStateManager.CurrentState != _eFocusState_FPS_CommunicationDialog)
+                    return;
+            }
+
+            if (!Input.GetKeyDown(Settings.QuickSwitchKey))
+                return;
+
+            SwitchToPreviousSlot();
+        }
+
+        private static InventorySlot _previousInventorySlot = InventorySlot.GearMelee;
+
+        public static void SwitchToPreviousSlot()
+        {
+            var player = PlayerManager.GetLocalPlayerAgent();
+
+            var previousSlot = _previousInventorySlot;
+            var currentSlot = player.Inventory.WieldedSlot;
+
+            if (Settings.DebugLog)
+                FeatureLogger.Debug($"Switching to previous slot: {currentSlot} -> {previousSlot}");
+
+            player.Sync.WantsToWieldSlot(previousSlot, false);
+        }
+
+        //WantsToWieldSlot(InventorySlot slot, bool broadcastOnly = false)
+        [ArchivePatch(typeof(PlayerSync), nameof(PlayerSync.WantsToWieldSlot))]
+        internal static class PlayerSync_WantsToWieldSlot_Patch
+        {
+            public static void Prefix(PlayerSync __instance, InventorySlot slot)
+            {
+                if (!(__instance.Replicator?.OwningPlayer?.IsLocal ?? false))
+                    return;
+
+                var currentSlot = __instance.GetWieldedSlot();
+
+                if (Settings.DebugLog)
+                    FeatureLogger.Debug($"PlayerSync.WantsToWieldSlot: currentSlot:{currentSlot} -> slot:{slot} | _previousInventorySlot:{_previousInventorySlot}");
+
+                if (currentSlot != slot)
+                    _previousInventorySlot = currentSlot;
+            }
+        }
+    }
+}
diff --git a/TheArchive.MONO/TheArchive.MONO.csproj b/TheArchive.MONO/TheArchive.MONO.csproj
@@ -261,6 +261,7 @@
 	  <Compile Include="..\TheArchive.IL2CPP\Features\Presence\SteamRichPresenceTweaks.cs" Link="Features\Presence\SteamRichPresenceTweaks.cs" />
 	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\IntroSkip.cs" Link="Features\QoL\IntroSkip.cs" />
 	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\L4DStylePacks.cs" Link="Features\QoL\L4DStylePacks.cs" />
+	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\LastUsedGearSwitcher.cs" Link="Features\QoL\LastUsedGearSwitcher.cs" />
 	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\LoadoutRandomizer.cs" Link="Features\QoL\LoadoutRandomizer.cs" />
 	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\NoDroppedMagazineSounds.cs" Link="Features\QoL\NoDroppedMagazineSounds.cs" />
 	  <Compile Include="..\TheArchive.IL2CPP\Features\QoL\NoNavMarkerHideInChat.cs" Link="Features\QoL\NoNavMarkerHideInChat.cs" />