Tarkov-UIFixes/Patches/ProductionPatch.cs

using Aki.Reflection.Patching;
using EFT.Hideout;
using EFT.UI;
using HarmonyLib;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using UnityEngine.UI;

namespace UIFixes
{
    public class ProductionPanelPatches
    {
        private static FieldInfo ProductionPanelSearch;
        private static FieldInfo SubstrateContentLayoutField;

        private static readonly Dictionary<string, string> LastSearches = [];

        public static void Enable()
        {
            ProductionPanelSearch = AccessTools.Field(typeof(ProductionPanel), "_searchInputField");
            SubstrateContentLayoutField = AccessTools.Field(typeof(AreaScreenSubstrate), "_contentLayout");

            new LazyShowPatch().Enable();
            new ShowContentsPatch().Enable();
            new ClosePatch().Enable();
            new ReturnToPreviousStatePatch().Enable();
            new GetSortedProductsPatch().Enable();
            new OnSearchChangePatch().Enable();
        }

        // Deactivate ProduceViews as they lazy load if they don't match the search
        public class LazyShowPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                Type type = typeof(ProductionPanel).GetNestedTypes().First(t =>
                {
                    MethodInfo method = t.GetMethod("method_6");
                    return method != null && method.GetParameters().Length == 2 && method.GetParameters()[1].ParameterType == typeof(ProduceView);
                });

                return AccessTools.Method(type, "method_6");
            }

            [PatchPostfix]
            public static void Postfix(ProductionPanel.Class1631 __instance, GClass1923 scheme, ProduceView view)
            {
                var searchField = ProductionPanelSearch.GetValue(__instance.productionPanel_0) as ValidationInputField;
                if (searchField.text.Length > 0 && scheme.endProduct.LocalizedName().IndexOf(searchField.text, StringComparison.InvariantCultureIgnoreCase) < 0)
                {
                    view.GameObject.SetActive(false);
                }
            }
        }

        // Populate the search box, and force the window to render
        public class ShowContentsPatch : ModulePatch
        {

            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ProductionPanel), nameof(ProductionPanel.ShowContents));
            }

            [PatchPrefix]
            public static void Prefix(ProductionPanel __instance, ValidationInputField ____searchInputField)
            {
                if (LastSearches.TryGetValue(__instance.AreaData.ToString(), out string lastSearch))
                {
                    ____searchInputField.text = lastSearch;
                }
            }

            [PatchPostfix]
            public static void Postfix(ProductionPanel __instance, ValidationInputField ____searchInputField)
            {
                // Force it to render immediately, at full height, even if the search filtering would reduce the number of children
                if (__instance.method_4().Count() > 2)
                {
                    AreaScreenSubstrate areaScreenSubstrate = __instance.GetComponentInParent<AreaScreenSubstrate>();
                    LayoutElement layoutElement = SubstrateContentLayoutField.GetValue(areaScreenSubstrate) as LayoutElement;
                    layoutElement.minHeight = 750f; // aka areaScreenSubstrate._maxHeight
                    areaScreenSubstrate.method_8();
                }

                ____searchInputField.ActivateInputField();
                ____searchInputField.Select();
            }
        }

        // method_4 gets the sorted list of products. If there's a search term, prioritize the matching items so they load first
        public class GetSortedProductsPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ProductionPanel), nameof(ProductionPanel.method_4));
            }

            // Working with GClasses directly here, because this would be a nightmare with reflection
            // Copied directly from method_4
            [PatchPrefix]
            public static bool Prefix(ref IEnumerable<GClass1923> __result, ProductionPanel __instance, GClass1922[] ___gclass1922_0, ValidationInputField ____searchInputField)
            {
                __result = ___gclass1922_0.OfType<GClass1923>().Where(scheme => !scheme.locked)
                    .OrderBy(scheme => scheme.endProduct.LocalizedName().Contains(____searchInputField.text) ? 0 : 1) // search-matching items first
                    .ThenBy(__instance.method_10)
                    .ThenBy(scheme => scheme.FavoriteIndex)
                    .ThenBy(scheme => scheme.Level);

                return false;
            }
        }

        // method_9 activates/deactivates the product game objects based on the search. Need to resort the list due to above patch
        public class OnSearchChangePatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ProductionPanel), nameof(ProductionPanel.method_9));
            }

            [PatchPrefix]
            public static void Prefix(ProductionPanel __instance)
            {
                __instance.method_8(); // update sort order
            }
        }

        // Save the search as the window closes
        public class ClosePatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ProductionPanel), nameof(ProductionPanel.Close));
            }

            [PatchPrefix]
            public static void Prefix(ProductionPanel __instance, ValidationInputField ____searchInputField)
            {
                LastSearches[__instance.AreaData.ToString()] = ____searchInputField.text;

                // Reset the default behavior
                AreaScreenSubstrate areaScreenSubstrate = __instance.GetComponentInParent<AreaScreenSubstrate>();
                LayoutElement layoutElement = SubstrateContentLayoutField.GetValue(areaScreenSubstrate) as LayoutElement;
                layoutElement.minHeight = -1f;
            }
        }

        // Clear the search stuff when you exit out
        public class ReturnToPreviousStatePatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(HideoutScreenOverlay), nameof(HideoutScreenOverlay.ReturnToPreviousState));
            }

            [PatchPostfix]
            public static void Postfix()
            {
                LastSearches.Clear();
            }
        }
    }
}