Tarkov-UIFixes/Patches/ConfirmationDialogKeysPatches.cs

using EFT.UI;
using EFT.UI.Ragfair;
using HarmonyLib;
using SPT.Reflection.Patching;
using System.Reflection;
using UnityEngine;
using UnityEngine.UI;

namespace UIFixes
{
    public static class ConfirmDialogKeysPatches
    {
        public static void Enable()
        {
            new DialogWindowPatch().Enable();
            new ItemUiContextWindowPatch().Enable();
            new ErrorScreenPatch().Enable();
            new AddOfferPatch().Enable();

            new ClickOutPatch().Enable();
            new ClickOutSplitDialogPatch().Enable();
            new ClickOutItemsListPatch().Enable();
            new ClickOutErrorScreenPatch().Enable();
        }

        public class DialogWindowPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(R.DialogWindow.Type, "Update");
            }

            [PatchPostfix]
            public static void Postfix(object __instance, bool ___bool_0)
            {
                var instance = new R.DialogWindow(__instance);

                if (!___bool_0)
                {
                    return;
                }

                if (Input.GetKeyDown(KeyCode.Return) || Input.GetKeyDown(KeyCode.KeypadEnter) || Input.GetKeyDown(KeyCode.Space))
                {
                    instance.Accept();
                    return;
                }
            }
        }

        public class ItemUiContextWindowPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.Update));
            }

            [PatchPostfix]
            public static void Postfix(SplitDialog ___splitDialog_0, ItemsListWindow ____itemsListWindow)
            {
                if (Input.GetKeyDown(KeyCode.Return) || Input.GetKeyDown(KeyCode.KeypadEnter) || Input.GetKeyDown(KeyCode.Space))
                {
                    if (___splitDialog_0 != null && ___splitDialog_0.gameObject.activeSelf)
                    {
                        ___splitDialog_0.Accept();
                        return;
                    }

                    if (____itemsListWindow.isActiveAndEnabled)
                    {
                        ____itemsListWindow.Close();
                        return;
                    }
                }
            }
        }

        public class ErrorScreenPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ErrorScreen), nameof(ErrorScreen.Update));
            }

            [PatchPostfix]
            public static void Postfix(ErrorScreen __instance)
            {
                if (Input.GetKeyDown(KeyCode.Return) || Input.GetKeyDown(KeyCode.KeypadEnter) || Input.GetKeyDown(KeyCode.Space))
                {
                    __instance.method_4();
                }
            }
        }

        public class AddOfferPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(AddOfferWindow), nameof(AddOfferWindow.Update));
            }

            [PatchPostfix]
            public static void Postfix(AddOfferWindow __instance, InteractableElement ____addOfferButton)
            {
                if (!____addOfferButton.isActiveAndEnabled || !____addOfferButton.Interactable)
                {
                    return;
                }

                if (Input.GetKeyDown(KeyCode.Return) || Input.GetKeyDown(KeyCode.KeypadEnter) || Input.GetKeyDown(KeyCode.Space))
                {
                    __instance.method_1();
                }
            }
        }

        public class ClickOutPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.DeclaredMethod(typeof(MessageWindow), nameof(MessageWindow.Show));
            }

            [PatchPostfix]
            public static void Postfix(MessageWindow __instance)
            {
                if (!Settings.ClickOutOfDialogs.Value)
                {
                    return;
                }

                // Note the space after firewall, because unity doesn't trim names and BSG is incompetent.
                // Also for some reason some MessageWindows have a Window child and some don't.
                Transform firewall = __instance.transform.Find("Firewall ") ?? __instance.transform.Find("Window/Firewall ");
                Button button = firewall?.gameObject.GetOrAddComponent<Button>();
                if (button != null)
                {
                    button.transition = Selectable.Transition.None;
                    button.onClick.AddListener(__instance.Close);
                    __instance.R().UI.AddDisposable(button.onClick.RemoveAllListeners);
                }
            }
        }

        public class ClickOutSplitDialogPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                // Using method_0 because there's 2 Show(), and they have 10+ args and f that
                return AccessTools.Method(typeof(SplitDialog), nameof(SplitDialog.method_0));
            }

            [PatchPostfix]
            public static void Postfix(SplitDialog __instance)
            {
                if (!Settings.ClickOutOfDialogs.Value)
                {
                    return;
                }

                Button button = __instance.transform.Find("Background")?.gameObject.GetOrAddComponent<Button>();
                if (button != null)
                {
                    button.transition = Selectable.Transition.None;
                    button.onClick.RemoveAllListeners(); // There's no disposable here so keeping the listener count down
                    button.onClick.AddListener(__instance.method_2);
                }
            }
        }

        public class ClickOutItemsListPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ItemsListWindow), nameof(ItemsListWindow.Show));
            }

            [PatchPostfix]
            public static void Postfix(ItemsListWindow __instance)
            {
                if (!Settings.ClickOutOfDialogs.Value)
                {
                    return;
                }

                // Note the space after firewall, because unity doesn't trim names and BSG is incompetent
                Transform firewall = __instance.transform.Find("Firewall ");
                Button button = firewall?.gameObject.GetOrAddComponent<Button>();
                if (button != null)
                {
                    button.transition = Selectable.Transition.None;
                    button.onClick.AddListener(__instance.Close);
                    __instance.R().UI.AddDisposable(button.onClick.RemoveAllListeners);
                }
            }
        }

        public class ClickOutErrorScreenPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.DeclaredMethod(typeof(ErrorScreen), nameof(ErrorScreen.Show));
            }

            [PatchPostfix]
            public static void Postfix(ErrorScreen __instance)
            {
                if (!Settings.ClickOutOfDialogs.Value)
                {
                    return;
                }

                // Note the space after firewall, because unity doesn't trim names and BSG is incompetent
                Transform firewall = __instance.transform.Find("Firewall ");
                Button button = firewall?.gameObject.GetOrAddComponent<Button>();
                if (button != null)
                {
                    button.transition = Selectable.Transition.None;
                    button.onClick.AddListener(__instance.method_4);
                    __instance.R().UI.AddDisposable(button.onClick.RemoveAllListeners);
                }
            }
        }
    }
}