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

namespace UIFixes
{
    internal class InspectWindowResizePatches
    {
        private static float SavedPreferredWidth = -1f;
        private static float SavedPreferredHeight = -1f;

        // Seems like this is the always the default for ItemSpecificationPanels
        private const float DefaultPreferredWidth = 670f;
        private const float DefaultPreferredHeight = 500f;

        private const string RestoreButtonName = "Restore";

        private const float ButtonPadding = 3f;

        public static void Enable()
        {
            new SaveInspectWindowSizePatch().Enable();
            new AddInspectWindowButtonsPatch().Enable();
            new GrowInspectWindowDescriptionPatch().Enable();
        }

        private class SaveInspectWindowSizePatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(StretchArea), nameof(StretchArea.OnDrag));
            }

            [PatchPostfix]
            public static void Postfix(LayoutElement ___layoutElement_0)
            {
                if (!Settings.RememberInspectSize.Value || ___layoutElement_0.GetComponent<ItemSpecificationPanel>() == null)
                {
                    return;
                }

                SavedPreferredWidth = ___layoutElement_0.preferredWidth;
                SavedPreferredHeight = ___layoutElement_0.preferredHeight;

                Button resizeButton = ___layoutElement_0.GetComponentsInChildren<Button>(true).FirstOrDefault(b => b.name == RestoreButtonName);
                if (resizeButton != null && !resizeButton.IsActive())
                {
                    resizeButton.gameObject.SetActive(true);
                }
            }
        }

        private class AddInspectWindowButtonsPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ItemSpecificationPanel), nameof(ItemSpecificationPanel.Show));
            }

            [PatchPrefix]
            public static void Prefix(LayoutElement ___layoutElement_0)
            {
                if (Settings.RememberInspectSize.Value)
                {
                    RestoreSavedSize(___layoutElement_0);
                }
            }

            [PatchPostfix]
            public static void Postfix(ItemSpecificationPanel __instance, LayoutElement ___layoutElement_0)
            {
                if (Settings.LockInspectPreviewSize.Value)
                {
                    LayoutElement previewPanel = __instance.GetComponentsInChildren<LayoutElement>().FirstOrDefault(e => e.name == "Preview Panel");
                    if (previewPanel != null)
                    {
                        previewPanel.flexibleHeight = -1;
                    }
                }

                Button closeButton = __instance.GetComponentsInChildren<Button>().FirstOrDefault(b => b.name == "Close Button");
                if (closeButton != null)
                {
                    CreateRightButton(__instance, closeButton);
                    CreateLeftButton(__instance, closeButton);
                    CreateRestoreButton(__instance, ___layoutElement_0, closeButton);
                }
            }

            private static void CreateRestoreButton(ItemSpecificationPanel inspectPanel, LayoutElement inspectLayout, Button template)
            {
                RectTransform templateRect = (RectTransform)template.transform;

                Button restoreButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
                restoreButton.name = RestoreButtonName;
                RectTransform restoreRect = (RectTransform)restoreButton.transform;
                restoreRect.localPosition = new Vector3(templateRect.localPosition.x - 3 * (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);

                Image restoreImage = restoreButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
                restoreImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.EffectiveDist);
                restoreImage.overrideSprite = null;
                restoreImage.SetNativeSize();
                restoreImage.transform.localScale = new Vector3(restoreImage.transform.localScale.x * 0.8f, restoreImage.transform.localScale.y * 0.8f, restoreImage.transform.localScale.z);

                Image restoreImage2 = UnityEngine.Object.Instantiate(restoreImage, restoreImage.transform.parent, false);
                restoreImage2.transform.Rotate(180f, 180f, 0f);

                Vector3 startPosition = restoreImage2.transform.localPosition;
                restoreImage.transform.localPosition = new Vector3(startPosition.x - 3f, startPosition.y - 3f, startPosition.z);
                restoreImage2.transform.localPosition = new Vector3(startPosition.x + 2.5f, startPosition.y + 2f, startPosition.z);

                if (SavedPreferredWidth < 0 && SavedPreferredHeight < 0)
                {
                    restoreButton.gameObject.SetActive(false);
                }

                restoreButton.onClick.AddListener(() =>
                {
                    SavedPreferredWidth = -1f;
                    SavedPreferredHeight = -1f;
                    RestoreSavedSize(inspectLayout);
                    restoreButton.gameObject.SetActive(false);

                    // I'm really not sure why this is necessary, but something in the layout gets borked trying to just restore the default size
                    // This recreates a lot of the children, but it works
                    inspectPanel.method_1();

                    StretchDescription(inspectLayout);
                });
                inspectPanel.AddDisposable(() => restoreButton.onClick.RemoveAllListeners());
            }

            private static void CreateLeftButton(ItemSpecificationPanel inspectPanel, Button template)
            {
                RectTransform templateRect = (RectTransform)template.transform;

                Button leftButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
                RectTransform leftRect = (RectTransform)leftButton.transform;
                leftRect.localPosition = new Vector3(templateRect.localPosition.x - 2 * (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);

                Image leftImage = leftButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
                leftImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.RecoilBack);
                leftImage.overrideSprite = null;
                leftImage.SetNativeSize();

                leftButton.onClick.AddListener(() =>
                {
                    RectTransform inspectRect = (RectTransform)inspectPanel.transform;
                    inspectRect.anchoredPosition = new Vector2((float)Screen.width / 4f / inspectRect.lossyScale.x, inspectRect.anchoredPosition.y);
                });
                inspectPanel.AddDisposable(() => leftButton.onClick.RemoveAllListeners());
            }

            private static void CreateRightButton(ItemSpecificationPanel inspectPanel, Button template)
            {
                RectTransform templateRect = (RectTransform)template.transform;

                Button rightButton = UnityEngine.Object.Instantiate(template, template.transform.parent, false);
                RectTransform rightRect = (RectTransform)rightButton.transform;
                rightRect.localPosition = new Vector3(templateRect.localPosition.x - (templateRect.rect.width + ButtonPadding), templateRect.localPosition.y, templateRect.localPosition.z);

                Image leftImage = rightButton.GetComponentsInChildren<Image>().First(i => i.name == "X");
                leftImage.sprite = EFTHardSettings.Instance.StaticIcons.GetAttributeIcon(EItemAttributeId.RecoilBack);
                leftImage.transform.Rotate(0f, 180f, 0f);
                leftImage.overrideSprite = null;
                leftImage.SetNativeSize();

                rightButton.onClick.AddListener(() =>
                {
                    RectTransform inspectRect = (RectTransform)inspectPanel.transform;
                    inspectRect.anchoredPosition = new Vector2((float)Screen.width * 3f/4f / inspectRect.lossyScale.x, inspectRect.anchoredPosition.y);
                });
                inspectPanel.AddDisposable(() => rightButton.onClick.RemoveAllListeners());
            }

            // Informed by StretchArea.OnDrag
            private static void RestoreSavedSize(LayoutElement layout)
            {
                RectTransform layoutRect = (RectTransform)layout.transform;

                layout.preferredWidth = SavedPreferredWidth > 0 ? SavedPreferredWidth : DefaultPreferredWidth;
                layout.preferredHeight = SavedPreferredHeight > 0 ? SavedPreferredHeight : DefaultPreferredHeight;

                LayoutRebuilder.ForceRebuildLayoutImmediate(layoutRect);

                layoutRect.CorrectPositionResolution(default);
            }
        }

        private static void StretchDescription(LayoutElement inspectLayout)
        {
            if (!Settings.ExpandDescriptionHeight.Value)
            {
                return;
            }

            LayoutElement scrollArea = inspectLayout.GetComponentsInChildren<LayoutElement>().FirstOrDefault(le => le.name == "Scroll Area");
            if (inspectLayout != null && scrollArea != null && scrollArea.transform.childCount > 0)
            {
                RectTransform description = (RectTransform)scrollArea.transform.GetChild(0);

                // Try to figure out how much extra I can work with
                float maxGrowth = (Screen.height / inspectLayout.transform.lossyScale.y) - ((RectTransform)inspectLayout.transform).rect.height;
                scrollArea.minHeight = Mathf.Max(scrollArea.minHeight, Mathf.Min(maxGrowth, description.rect.height));
            }
        }

        private class GrowInspectWindowDescriptionPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(ItemUiContext), nameof(ItemUiContext.Inspect));
            }

            [PatchPostfix]
            public static void Postfix(List<InputNode> ____children)
            {
                var inspectWindow = ____children.Last();
                if (inspectWindow != null)
                {
                    StretchDescription(inspectWindow.GetComponent<LayoutElement>());
                }
            }
        }
    }
}