Bindable tactical devices

2024-08-16 15:51:05 -07:00
parent 75c60fb09a
commit 852b9b0078
4 changed files with 186 additions and 0 deletions
--- a/Patches/TacticalBindsPatches.cs
+++ b/Patches/TacticalBindsPatches.cs
@@ -0,0 +1,138 @@
+using Comfort.Common;
+using EFT;
+using EFT.InventoryLogic;
+using HarmonyLib;
+using SPT.Reflection.Patching;
+using System.Reflection;
+using UnityEngine;
+
+namespace UIFixes;
+
+public static class TacticalBindsPatches
+{
+    public static void Enable()
+    {
+        new BindableTacticalPatch().Enable();
+        new ReachableTacticalPatch().Enable();
+        new UseTacticalPatch().Enable();
+    }
+
+    public class BindableTacticalPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(InventoryControllerClass), nameof(InventoryControllerClass.IsAtBindablePlace));
+        }
+
+        [PatchPostfix]
+        public static void Postfix(InventoryControllerClass __instance, Item item, ref bool __result)
+        {
+            if (__result)
+            {
+                return;
+            }
+
+            __result = IsEquippedTacticalDevice(__instance, item);
+        }
+    }
+
+    public class ReachableTacticalPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(InventoryControllerClass), nameof(InventoryControllerClass.IsAtReachablePlace));
+        }
+
+        [PatchPostfix]
+        public static void Postfix(InventoryControllerClass __instance, Item item, ref bool __result)
+        {
+            if (__result)
+            {
+                return;
+            }
+
+            __result = IsEquippedTacticalDevice(__instance, item);
+        }
+    }
+
+    public class UseTacticalPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(Player), nameof(Player.SetQuickSlotItem));
+        }
+
+        [PatchPrefix]
+        public static bool Prefix(Player __instance, EBoundItem quickSlot, Callback<IHandsController> callback)
+        {
+            Item boundItem = __instance.InventoryControllerClass.Inventory.FastAccess.GetBoundItem(quickSlot);
+            if (boundItem == null)
+            {
+                return true;
+            }
+
+            LightComponent lightComponent = boundItem.GetItemComponent<LightComponent>();
+            if (lightComponent == null)
+            {
+                return true;
+            }
+
+            // Don't return true past this point; if the default handler tries to use a tactical device, very bad things happen
+
+            if (__instance.HandsController is not Player.FirearmController firearmController ||
+                firearmController.Item != boundItem.GetRootItem())
+            {
+                callback(null);
+                return false;
+            }
+
+            FirearmLightStateStruct lightState = new()
+            {
+                Id = lightComponent.Item.Id,
+                IsActive = lightComponent.IsActive,
+                LightMode = lightComponent.SelectedMode
+            };
+
+            if (IsTacticalModeModifierPressed())
+            {
+                lightState.LightMode++;
+            }
+            else
+            {
+                lightState.IsActive = !lightState.IsActive;
+            }
+
+            firearmController.SetLightsState([lightState], false);
+
+            callback(null);
+            return false;
+        }
+    }
+
+    private static bool IsEquippedTacticalDevice(InventoryControllerClass inventoryController, Item item)
+    {
+        LightComponent lightComponent = item.GetItemComponent<LightComponent>();
+        if (lightComponent == null)
+        {
+            return false;
+        }
+
+        if (item.GetRootItem() is Weapon weapon)
+        {
+            return inventoryController.Inventory.Equipment.Contains(weapon);
+        }
+
+        return false;
+    }
+
+    private static bool IsTacticalModeModifierPressed()
+    {
+        return Settings.TacticalModeModifier.Value switch
+        {
+            TacticalBindModifier.Shift => Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.RightShift),
+            TacticalBindModifier.Control => Input.GetKey(KeyCode.LeftControl) || Input.GetKey(KeyCode.RightControl),
+            TacticalBindModifier.Alt => Input.GetKey(KeyCode.LeftAlt) || Input.GetKey(KeyCode.RightAlt),
+            _ => false,
+        };
+    }
+}