Move source into src

2024-09-20 11:47:09 -07:00
parent 2ff3fba9bc
commit 698cd61a88
84 changed files with 0 additions and 0 deletions
--- a/src/Patches/LoadMultipleMagazinesPatches.cs
+++ b/src/Patches/LoadMultipleMagazinesPatches.cs
@@ -0,0 +1,149 @@
+using EFT.InventoryLogic;
+using EFT.UI;
+using HarmonyLib;
+using SPT.Reflection.Patching;
+using SPT.Reflection.Utils;
+using System;
+using System.Linq;
+using System.Reflection;
+using System.Threading.Tasks;
+using UnityEngine;
+
+namespace UIFixes;
+
+public static class LoadMultipleMagazinesPatches
+{
+    private static ItemFilter[] CombinedFilters;
+
+    public static void Enable()
+    {
+        new FindCompatibleAmmoPatch().Enable();
+        new CheckItemFilterPatch().Enable();
+        new LoadAmmoPatch().Enable();
+        new FilterMagPresetsPatch().Enable();
+        new LoadPresetPatch().Enable();
+    }
+
+    public class FindCompatibleAmmoPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.FindCompatibleAmmo));
+        }
+
+        [PatchPrefix]
+        public static void Prefix()
+        {
+            if (MultiSelect.Active)
+            {
+                CombinedFilters = MultiSelect.SortedItemContexts()
+                    .Select(itemContext => itemContext.Item)
+                    .OfType<MagazineClass>()
+                    .SelectMany(mag => mag.Cartridges.Filters)
+                    .ToArray();
+            }
+        }
+
+        [PatchPostfix]
+        public static void Postfix()
+        {
+            CombinedFilters = null;
+        }
+    }
+
+    public class CheckItemFilterPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(ItemFilterExtensions), nameof(ItemFilterExtensions.CheckItemFilter));
+        }
+
+        [PatchPrefix]
+        public static void Prefix(ref ItemFilter[] filters)
+        {
+            if (CombinedFilters == null)
+            {
+                return;
+            }
+
+            filters = CombinedFilters;
+        }
+    }
+
+    public class LoadAmmoPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            Type type = PatchConstants.EftTypes.Single(t => t.GetNestedType("EMagInteraction") != null);
+            return AccessTools.Method(type, "method_6");
+        }
+
+        [PatchPrefix]
+        public static bool Prefix(string ammoTemplateId, ref Task __result, ItemUiContext ___itemUiContext_0)
+        {
+            if (!MultiSelect.Active)
+            {
+                return true;
+            }
+
+            __result = MultiSelect.LoadAmmoAll(___itemUiContext_0, ammoTemplateId, false);
+            return false;
+        }
+    }
+
+    public class FilterMagPresetsPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            Type type = PatchConstants.EftTypes.Single(t => t.GetNestedType("EMagPresetInteraction") != null);
+            return AccessTools.Method(type, "method_7");
+        }
+
+        [PatchPrefix]
+        public static void Prefix()
+        {
+            if (MultiSelect.Active)
+            {
+                CombinedFilters = MultiSelect.SortedItemContexts()
+                    .Select(itemContext => itemContext.Item)
+                    .OfType<MagazineClass>()
+                    .SelectMany(mag => mag.Cartridges.Filters)
+                    .ToArray();
+            }
+        }
+
+        [PatchPostfix]
+        public static void Postfix()
+        {
+            CombinedFilters = null;
+        }
+    }
+
+    public class LoadPresetPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            Type type = PatchConstants.EftTypes.Single(t => t.GetNestedType("EMagPresetInteraction") != null);
+            return AccessTools.Method(type, "method_6");
+        }
+
+        [PatchPrefix]
+        public static bool Prefix(MagazineBuildPresetClass preset, ItemUiContext ___itemUiContext_1)
+        {
+            if (!MultiSelect.Active)
+            {
+                return true;
+            }
+
+            if (Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.LeftShift) || Input.GetKey(KeyCode.LeftControl) || Input.GetKey(KeyCode.RightControl))
+            {
+                return true;
+            }
+
+            var magazines = MultiSelect.SortedItemContexts().Select(itemContext => itemContext.Item).OfType<MagazineClass>();
+            ___itemUiContext_1.ApplyMagPreset(preset, magazines.ToList()).HandleExceptions();
+
+            return false;
+        }
+    }
+}