From ed2dff20f6ddc7b504f106ef9410a616c15a4cbc Mon Sep 17 00:00:00 2001
From: Tyfon <29051038+tyfon7@users.noreply.github.com>
Date: Mon, 3 Jun 2024 02:49:16 -0700
Subject: [PATCH] Add snap buttons and keybinds to container grids

---
 Patches/GridWindowButtonsPatch.cs     | 100 ++++++++++++++++++++++++++
 Patches/InspectWindowResizePatches.cs |  16 +++--
 Plugin.cs                             |   1 +
 R.cs                                  |   2 +-
 Settings.cs                           |  10 +++
 5 files changed, 123 insertions(+), 6 deletions(-)
 create mode 100644 Patches/GridWindowButtonsPatch.cs

diff --git a/Patches/GridWindowButtonsPatch.cs b/Patches/GridWindowButtonsPatch.cs
new file mode 100644
index 0000000..c34173a
--- /dev/null
+++ b/Patches/GridWindowButtonsPatch.cs
@@ -0,0 +1,100 @@
+using Aki.Reflection.Patching;
+using EFT.InventoryLogic;
+using EFT.UI;
+using HarmonyLib;
+using System.Reflection;
+using UnityEngine;
+using UnityEngine.UI;
+
+namespace UIFixes
+{
+    public class GridWindowButtonsPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.DeclaredMethod(typeof(GridWindow), nameof(GridWindow.Show));
+        }
+
+        [PatchPostfix]
+        public static void Postfix(GridWindow __instance)
+        {
+            if (Settings.AddContainerButtons.Value)
+            {
+                Transform closeButton = __instance.transform.Find("Caption Panel/Close Button");
+                Image sortBackground = __instance.transform.Find("Caption Panel/Sort Button")?.GetComponent<Image>();
+
+                // Left button
+                Button leftButton = CreateButton(closeButton, sortBackground.sprite, EItemAttributeId.RecoilBack);
+                leftButton.onClick.AddListener(() => SnapLeft(__instance));
+                __instance.R().UI.AddDisposable(() => leftButton.onClick.RemoveAllListeners());
+
+                // Right button
+                Button rightButton = CreateButton(closeButton, sortBackground.sprite, EItemAttributeId.RecoilBack);
+                rightButton.transform.Find("X").Rotate(0f, 180f, 0f);
+                rightButton.onClick.AddListener(() => SnapRight(__instance));
+                __instance.R().UI.AddDisposable(() => rightButton.onClick.RemoveAllListeners());
+
+                // Put close back on the end
+                closeButton.SetAsLastSibling();
+            }
+
+            // Keybinds
+            LeftRightKeybind leftRightKeybind = __instance.GetOrAddComponent<LeftRightKeybind>();
+            leftRightKeybind.Init(__instance);
+        }
+
+        private static Button CreateButton(Transform template, Sprite backgroundSprite, EItemAttributeId attributeIcon)
+        {
+            Transform transform = UnityEngine.Object.Instantiate(template, template.parent, false);
+
+            Image background = transform.GetComponent<Image>();
+            background.sprite = backgroundSprite;
+
+            Image icon = transform.Find("X").GetComponent<Image>();
+            icon.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(attributeIcon);
+            icon.overrideSprite = null;
+            icon.SetNativeSize();
+
+            Button button = transform.GetComponent<Button>();
+            button.navigation = new Navigation() { mode = Navigation.Mode.None };
+
+            return button;
+        }
+
+        public class LeftRightKeybind : MonoBehaviour
+        {
+            private GridWindow window;
+
+            public void Init(GridWindow window)
+            {
+                this.window = window;
+            }
+
+            public void Update()
+            {
+                bool isTopWindow = window.transform.GetSiblingIndex() == window.transform.parent.childCount - 1;
+                if (Settings.SnapLeftKeybind.Value.IsDown() && isTopWindow)
+                {
+                    SnapLeft(window);
+                }
+
+                if (Settings.SnapRightKeybind.Value.IsDown() && isTopWindow)
+                {
+                    SnapRight(window);
+                }
+            }
+        }
+
+        private static void SnapLeft(GridWindow window)
+        {
+            RectTransform inspectRect = (RectTransform)window.transform;
+            inspectRect.anchoredPosition = new Vector2((float)Screen.width / 4f / inspectRect.lossyScale.x, inspectRect.anchoredPosition.y);
+        }
+
+        private static void SnapRight(GridWindow window)
+        {
+            RectTransform inspectRect = (RectTransform)window.transform;
+            inspectRect.anchoredPosition = new Vector2((float)Screen.width * 3f / 4f / inspectRect.lossyScale.x, inspectRect.anchoredPosition.y);
+        }
+    }
+}
diff --git a/Patches/InspectWindowResizePatches.cs b/Patches/InspectWindowResizePatches.cs
index 6bfeb99..b31b98b 100644
--- a/Patches/InspectWindowResizePatches.cs
+++ b/Patches/InspectWindowResizePatches.cs
@@ -107,6 +107,8 @@ namespace UIFixes
 
                 Button restoreButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
                 restoreButton.name = "Restore";
+                restoreButton.navigation = new Navigation() { mode = Navigation.Mode.None };
+
                 RectTransform restoreRect = (RectTransform)restoreButton.transform;
                 restoreRect.localPosition = new Vector3(templateRect.localPosition.x - 3 * (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);
 
@@ -152,6 +154,8 @@ namespace UIFixes
                 RectTransform templateRect = (RectTransform)template.transform;
 
                 Button leftButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
+                leftButton.navigation = new Navigation() { mode = Navigation.Mode.None };
+
                 RectTransform leftRect = (RectTransform)leftButton.transform;
                 leftRect.localPosition = new Vector3(templateRect.localPosition.x - 2 * (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);
 
@@ -172,17 +176,19 @@ namespace UIFixes
                 RectTransform templateRect = (RectTransform)template.transform;
 
                 Button rightButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
+                rightButton.navigation = new Navigation() { mode = Navigation.Mode.None };
+
                 RectTransform rightRect = (RectTransform)rightButton.transform;
                 rightRect.localPosition = new Vector3(templateRect.localPosition.x - (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);
 
                 Image background = rightButton.GetComponent<Image>();
                 background.sprite = ButtonBackground.sprite;
 
-                Image leftImage = rightButton.transform.Find("X").GetComponent<Image>();
-                leftImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.RecoilBack);
-                leftImage.transform.Rotate(0f, 180f, 0f);
-                leftImage.overrideSprite = null;
-                leftImage.SetNativeSize();
+                Image rightImage = rightButton.transform.Find("X").GetComponent<Image>();
+                rightImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.RecoilBack);
+                rightImage.transform.Rotate(0f, 180f, 0f);
+                rightImage.overrideSprite = null;
+                rightImage.SetNativeSize();
 
                 rightButton.onClick.AddListener(() => SnapRight(inspectPanel));
                 inspectPanel.AddDisposable(() => rightButton.onClick.RemoveAllListeners());
diff --git a/Plugin.cs b/Plugin.cs
index 18475a2..ec38a0b 100644
--- a/Plugin.cs
+++ b/Plugin.cs
@@ -43,6 +43,7 @@ namespace UIFixes
             KeepMessagesOpenPatches.Enable();
             new FocusTradeQuantityPatch().Enable();
             new RememberRepairerPatch().Enable();
+            new GridWindowButtonsPatch().Enable();
         }
 
         public static bool InRaid()
diff --git a/R.cs b/R.cs
index 9815c16..b9a75fb 100644
--- a/R.cs
+++ b/R.cs
@@ -554,7 +554,7 @@ namespace UIFixes
             public TraderAssortmentControllerClass TraderAssortmentControler { get { return (TraderAssortmentControllerClass)TraderAssortmentControllerField.GetValue(Value); } }
         }
 
-        public class GridWindow(object value) : Wrapper(value)
+        public class GridWindow(object value) : UIInputNode(value)
         {
             public static Type Type { get; private set; }
             private static FieldInfo GridSortPanelField;
diff --git a/Settings.cs b/Settings.cs
index 5c4e564..accd101 100644
--- a/Settings.cs
+++ b/Settings.cs
@@ -60,6 +60,7 @@ namespace UIFixes
         public static ConfigEntry<KeyboardShortcut> SnapLeftKeybind { get; set; }
         public static ConfigEntry<KeyboardShortcut> SnapRightKeybind { get; set; }
         public static ConfigEntry<bool> StyleItemPanel { get; set; } // Advanced
+        public static ConfigEntry<bool> AddContainerButtons { get; set; } // Advanced
 
         // In Raid
         public static ConfigEntry<bool> RemoveDisabledActions { get; set; }
@@ -297,6 +298,15 @@ namespace UIFixes
                     null,
                     new ConfigurationManagerAttributes { IsAdvanced = true })));
 
+            configEntries.Add(AddContainerButtons = config.Bind(
+                InspectSection,
+                "Add Left/Right Buttons on Containers",
+                true,
+                new ConfigDescription(
+                    "Adds snap left and snap right buttons to container windows too",
+                    null,
+                    new ConfigurationManagerAttributes { IsAdvanced = true })));
+
             // In Raid
             configEntries.Add(RemoveDisabledActions = config.Bind(
                 InRaidSection,