From 05d0339b626378adc718c5621ae845098addaff2 Mon Sep 17 00:00:00 2001
From: Tyfon <29051038+tyfon7@users.noreply.github.com>
Date: Sun, 2 Jun 2024 15:47:02 -0700
Subject: [PATCH] Block clicks around flea purchase button; use transform.find

---
 Patches/FixFleaPatches.cs             | 27 +++++++++++++++++++++------
 Patches/FleaPrevSearchPatches.cs      |  4 ++--
 Patches/InspectWindowResizePatches.cs | 18 ++++++++----------
 3 files changed, 31 insertions(+), 18 deletions(-)

diff --git a/Patches/FixFleaPatches.cs b/Patches/FixFleaPatches.cs
index d2304c5..dbd713e 100644
--- a/Patches/FixFleaPatches.cs
+++ b/Patches/FixFleaPatches.cs
@@ -1,12 +1,6 @@
 using Aki.Reflection.Patching;
-using EFT.InventoryLogic;
-using EFT.Quests;
-using EFT.UI;
 using EFT.UI.Ragfair;
 using HarmonyLib;
-using System;
-using System.Collections.Generic;
-using System.Linq;
 using System.Reflection;
 using TMPro;
 using UnityEngine;
@@ -23,6 +17,7 @@ namespace UIFixes
             new ToggleOnOpenPatch().Enable();
 
             new OfferItemFixMaskPatch().Enable();
+            new BlockActionsClickThroughPatch().Enable();
         }
 
         public class DoNotToggleOnMouseOverPatch : ModulePatch
@@ -76,5 +71,25 @@ namespace UIFixes
                 }
             }
         }
+
+        public class BlockActionsClickThroughPatch : ModulePatch
+        {
+            protected override MethodBase GetTargetMethod()
+            {
+                return AccessTools.Method(typeof(OfferView), nameof(OfferView.Awake));
+            }
+
+            [PatchPostfix]
+            public static void Postfix(OfferView __instance)
+            {
+                // Intercept clicks on the actions area
+                var blocker = __instance.transform.Find("Actions").gameObject.GetOrAddComponent<Button>();
+                blocker.transition = Selectable.Transition.None;
+
+                // But enable clicks specifically on the minimize button
+                var minimizeButton = __instance.transform.Find("Actions/MinimizeButton").gameObject.GetOrAddComponent<Button>();
+                minimizeButton.onClick.AddListener(() => __instance.OnPointerClick(null));
+            }
+        }
     }
 }
diff --git a/Patches/FleaPrevSearchPatches.cs b/Patches/FleaPrevSearchPatches.cs
index 4eed06b..931c68c 100644
--- a/Patches/FleaPrevSearchPatches.cs
+++ b/Patches/FleaPrevSearchPatches.cs
@@ -93,7 +93,7 @@ namespace UIFixes
             public static void Postfix(RagfairScreen __instance, ISession session, DefaultUIButton ____addOfferButton)
             {
                 // Delete the upper right display options, since they aren't even implemented
-                var tabs = __instance.GetComponentsInChildren<RectTransform>().FirstOrDefault(c => c.name == "Tabs");
+                var tabs = __instance.transform.Find("TopRightPanel/Tabs");
                 tabs?.gameObject.SetActive(false);
 
                 if (!Settings.EnableFleaHistory.Value)
@@ -107,7 +107,7 @@ namespace UIFixes
                 addOfferLayout.preferredWidth = -1;
 
                 // Recenter the add offer text
-                var addOfferLabel = ____addOfferButton.GetComponentsInChildren<RectTransform>().First(c => c.name == "SizeLabel");
+                var addOfferLabel = ____addOfferButton.transform.Find("SizeLabel");
                 addOfferLabel.localPosition = new Vector3(0f, 0f, 0f);
 
                 // For some reason the widths revert
diff --git a/Patches/InspectWindowResizePatches.cs b/Patches/InspectWindowResizePatches.cs
index e08e34f..6bfeb99 100644
--- a/Patches/InspectWindowResizePatches.cs
+++ b/Patches/InspectWindowResizePatches.cs
@@ -20,8 +20,6 @@ namespace UIFixes
         private const float DefaultPreferredWidth = 670f;
         private const float DefaultPreferredHeight = 500f;
 
-        private const string RestoreButtonName = "Restore";
-
         private const float ButtonPadding = 3f;
 
         public static Image ButtonBackground; // Nice gray background for the new buttons
@@ -52,7 +50,7 @@ namespace UIFixes
                 SavedPreferredWidth = ___layoutElement_0.preferredWidth;
                 SavedPreferredHeight = ___layoutElement_0.preferredHeight;
 
-                Button resizeButton = ___layoutElement_0.GetComponentsInChildren<Button>(true).FirstOrDefault(b => b.name == RestoreButtonName);
+                Button resizeButton = ___layoutElement_0.transform.Find("Inner/Caption Panel/Restore")?.GetComponent<Button>();
                 if (resizeButton != null && !resizeButton.IsActive())
                 {
                     resizeButton.gameObject.SetActive(true);
@@ -81,7 +79,7 @@ namespace UIFixes
             {
                 if (Settings.LockInspectPreviewSize.Value)
                 {
-                    LayoutElement previewPanel = __instance.GetComponentsInChildren<LayoutElement>().FirstOrDefault(e => e.name == "Preview Panel");
+                    LayoutElement previewPanel = __instance.transform.Find("Inner/Contents/Preview Panel")?.GetComponent<LayoutElement>();
                     if (previewPanel != null)
                     {
                         previewPanel.flexibleHeight = -1;
@@ -94,7 +92,7 @@ namespace UIFixes
                     ButtonBackground = ___itemUiContext_0.R().GridWindowTemplate.R().GridSortPanel.R().Button.image;
                 }
 
-                Button closeButton = __instance.GetComponentsInChildren<Button>().FirstOrDefault(b => b.name == "Close Button");
+                Button closeButton = __instance.transform.Find("Inner/Caption Panel/Close Button")?.GetComponent<Button>();
                 if (closeButton != null)
                 {
                     CreateRightButton(__instance, closeButton);
@@ -108,14 +106,14 @@ namespace UIFixes
                 RectTransform templateRect = (RectTransform)template.transform;
 
                 Button restoreButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
-                restoreButton.name = RestoreButtonName;
+                restoreButton.name = "Restore";
                 RectTransform restoreRect = (RectTransform)restoreButton.transform;
                 restoreRect.localPosition = new Vector3(templateRect.localPosition.x - 3 * (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);
 
                 Image background = restoreButton.GetComponent<Image>();
                 background.sprite = ButtonBackground.sprite;
 
-                Image restoreImage = restoreButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
+                Image restoreImage = restoreButton.transform.Find("X").GetComponent<Image>();
                 restoreImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.EffectiveDist);
                 restoreImage.overrideSprite = null;
                 restoreImage.SetNativeSize();
@@ -160,7 +158,7 @@ namespace UIFixes
                 Image background = leftButton.GetComponent<Image>();
                 background.sprite = ButtonBackground.sprite;
 
-                Image leftImage = leftButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
+                Image leftImage = leftButton.transform.Find("X").GetComponent<Image>();
                 leftImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.RecoilBack);
                 leftImage.overrideSprite = null;
                 leftImage.SetNativeSize();
@@ -180,7 +178,7 @@ namespace UIFixes
                 Image background = rightButton.GetComponent<Image>();
                 background.sprite = ButtonBackground.sprite;
 
-                Image leftImage = rightButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
+                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;
@@ -223,7 +221,7 @@ namespace UIFixes
                 return;
             }
 
-            LayoutElement scrollArea = inspectLayout.GetComponentsInChildren<LayoutElement>().FirstOrDefault(le => le.name == "Scroll Area");
+            LayoutElement scrollArea = inspectLayout.transform.Find("Inner/Contents/DescriptionPanel/DescriptionPanel/Scroll Area").GetComponent<LayoutElement>();
             if (inspectLayout != null && scrollArea != null && scrollArea.transform.childCount > 0)
             {
                 RectTransform description = (RectTransform)scrollArea.transform.GetChild(0);