Move source into src

src/Patches/ContextMenuShortcutPatches.cs

@@ -0,0 +1,263 @@
+using Comfort.Common;
+using EFT.InventoryLogic;
+using EFT.UI;
+using EFT.UI.DragAndDrop;
+using HarmonyLib;
+using SPT.Reflection.Patching;
+using System.Reflection;
+using TMPro;
+using UnityEngine.EventSystems;
+
+namespace UIFixes;
+
+public static class ContextMenuShortcutPatches
+{
+    private static TMP_InputField LastSelectedInput = null;
+
+    public static void Enable()
+    {
+        new ItemUiContextPatch().Enable();
+
+        new HideoutItemViewRegisterContextPatch().Enable();
+
+        new TradingPanelRegisterContextPatch().Enable();
+        new TradingPanelUnregisterContextPatch().Enable();
+
+        new SelectCurrentContextPatch().Enable();
+        new DeselectCurrentContextPatch().Enable();
+    }
+
+    public class ItemUiContextPatch : ModulePatch
+    {
+        private static ItemInfoInteractionsAbstractClass<EItemInfoButton> Interactions;
+
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.Update));
+        }
+
+        [PatchPostfix]
+        public static void Postfix(ItemUiContext __instance)
+        {
+            // Need an item context to operate on
+            ItemContextAbstractClass itemContext = __instance.R().ItemContext;
+            if (itemContext == null)
+            {
+                return;
+            }
+
+            if (!Settings.ItemContextBlocksTextInputs.Value && Plugin.TextboxActive())
+            {
+                return;
+            }
+
+            if (Settings.InspectKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Inspect);
+            }
+
+            if (Settings.OpenKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Open);
+            }
+
+            if (Settings.TopUpKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.TopUp);
+            }
+
+            if (Settings.UseKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Use);
+            }
+
+            if (Settings.UseAllKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.UseAll, [EItemInfoButton.Use]);
+            }
+
+            if (Settings.ReloadKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Reload);
+            }
+
+            if (Settings.UnloadKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Unload, [EItemInfoButton.UnloadAmmo]);
+            }
+
+            if (Settings.UnpackKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Unpack);
+            }
+
+            if (Settings.FilterByKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.FilterSearch);
+            }
+
+            if (Settings.LinkedSearchKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.LinkedSearch);
+            }
+
+            if (Settings.SortingTableKeyBind.Value.IsDown())
+            {
+                MoveToFromSortingTable(itemContext, __instance);
+            }
+
+            if (Settings.ExamineKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButton.Examine,
+                    [EItemInfoButton.Fold, EItemInfoButton.Unfold, EItemInfoButton.TurnOn, EItemInfoButton.TurnOff, EItemInfoButton.CheckMagazine]);
+            }
+
+            if (Settings.AddOfferKeyBind.Value.IsDown())
+            {
+                TryInteraction(__instance, itemContext, EItemInfoButtonExt.AddOffer);
+            }
+
+            Interactions = null;
+        }
+
+        private static void TryInteraction(ItemUiContext itemUiContext, ItemContextAbstractClass itemContext, EItemInfoButton interaction, EItemInfoButton[] fallbackInteractions = null)
+        {
+            Interactions ??= itemUiContext.GetItemContextInteractions(itemContext, null);
+            if (!Interactions.ExecuteInteraction(interaction) && fallbackInteractions != null)
+            {
+                foreach (var fallbackInteraction in fallbackInteractions)
+                {
+                    if (Interactions.ExecuteInteraction(fallbackInteraction))
+                    {
+                        return;
+                    }
+                }
+            }
+        }
+
+        private static void MoveToFromSortingTable(ItemContextAbstractClass itemContext, ItemUiContext itemUiContext)
+        {
+            Item item = itemContext.Item;
+            if (item.Owner is not InventoryControllerClass controller)
+            {
+                return;
+            }
+
+            SortingTableClass sortingTable = controller.Inventory.SortingTable;
+            bool isInSortingTable = sortingTable != null && item.Parent.Container.ParentItem == sortingTable;
+
+            var operation = isInSortingTable ? itemUiContext.QuickFindAppropriatePlace(itemContext, controller, false, true, true) : itemUiContext.QuickMoveToSortingTable(item, true);
+            if (operation.Succeeded && controller.CanExecute(operation.Value))
+            {
+                if (operation.Value is IDestroyResult destroyResult && destroyResult.ItemsDestroyRequired)
+                {
+                    NotificationManagerClass.DisplayWarningNotification(new DestroyError(item, destroyResult.ItemsToDestroy).GetLocalizedDescription());
+                    return;
+                }
+
+                controller.RunNetworkTransaction(operation.Value, null);
+                if (itemUiContext.Tooltip != null)
+                {
+                    itemUiContext.Tooltip.Close();
+                }
+
+                Singleton<GUISounds>.Instance.PlayItemSound(item.ItemSound, EInventorySoundType.pickup, false);
+            }
+        }
+    }
+
+    // HideoutItemViews don't register themselves with ItemUiContext for some reason
+    public class HideoutItemViewRegisterContextPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.DeclaredMethod(typeof(HideoutItemView), nameof(HideoutItemView.OnPointerEnter));
+        }
+
+        [PatchPostfix]
+        public static void Postfix(HideoutItemView __instance, ItemUiContext ___ItemUiContext)
+        {
+            ___ItemUiContext.RegisterCurrentItemContext(__instance.ItemContext);
+        }
+    }
+
+    public class TradingPanelRegisterContextPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(TradingRequisitePanel), nameof(TradingRequisitePanel.method_1)); // OnHoverStart
+        }
+
+        [PatchPostfix]
+        public static void Postfix(ItemUiContext ___itemUiContext_0, ItemContextAbstractClass ___itemContextAbstractClass)
+        {
+            ___itemUiContext_0.RegisterCurrentItemContext(___itemContextAbstractClass);
+        }
+    }
+
+    public class TradingPanelUnregisterContextPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(TradingRequisitePanel), nameof(TradingRequisitePanel.method_2)); // OnHoverEnd
+        }
+
+        [PatchPostfix]
+        public static void Postfix(ItemUiContext ___itemUiContext_0, ItemContextAbstractClass ___itemContextAbstractClass)
+        {
+            ___itemUiContext_0.UnregisterCurrentItemContext(___itemContextAbstractClass);
+        }
+    }
+
+    // Keybinds don't work if a textbox has focus - setting the textbox to readonly seems the best way to fix this
+    // without changing selection and causing weird side effects
+    public class SelectCurrentContextPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.RegisterCurrentItemContext));
+        }
+
+        [PatchPostfix]
+        public static void Postfix()
+        {
+            if (!Settings.ItemContextBlocksTextInputs.Value)
+            {
+                return;
+            }
+
+            if (EventSystem.current?.currentSelectedGameObject != null)
+            {
+                LastSelectedInput = EventSystem.current.currentSelectedGameObject.GetComponent<TMP_InputField>();
+                if (LastSelectedInput != null)
+                {
+                    LastSelectedInput.readOnly = true;
+                }
+            }
+        }
+    }
+
+    public class DeselectCurrentContextPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.UnregisterCurrentItemContext));
+        }
+
+        [PatchPostfix]
+        public static void Postfix()
+        {
+            if (!Settings.ItemContextBlocksTextInputs.Value)
+            {
+                return;
+            }
+
+            if (LastSelectedInput != null)
+            {
+                LastSelectedInput.readOnly = false;
+            }
+
+            LastSelectedInput = null;
+        }
+    }
+}