From c993dc6caab1624f77d4b38829de8d84506b50a0 Mon Sep 17 00:00:00 2001
From: AuriRex <xaurirexx@gmail.com>
Date: Thu, 19 Oct 2023 19:00:45 +0200
Subject: [PATCH] Added Feature: "LastUsedGearSwitcher"

closes #58
---
 .../Features/QoL/LastUsedGearSwitcher.cs      | 99 +++++++++++++++++++
 TheArchive.MONO/TheArchive.MONO.csproj        |  1 +
 2 files changed, 100 insertions(+)
 create mode 100644 TheArchive.IL2CPP/Features/QoL/LastUsedGearSwitcher.cs

diff --git a/TheArchive.IL2CPP/Features/QoL/LastUsedGearSwitcher.cs b/TheArchive.IL2CPP/Features/QoL/LastUsedGearSwitcher.cs
new file mode 100644
index 00000000..1e39510e
--- /dev/null
+++ 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
index 70223489..4072d1c8 100644
--- 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" />