using Aki.Reflection.Patching;
using EFT.InventoryLogic;
using EFT.UI;
using EFT.UI.WeaponModding;
using HarmonyLib;
using System;
using System.Reflection;
using UnityEngine.EventSystems;

namespace UIFixes
{
    public class WeaponZoomPatch
    {
        public static void Enable()
        {
            new EditBuildScreenZoomPatch().Enable();
            new WeaponModdingScreenZoomPatch().Enable();
        }

        public class EditBuildScreenZoomPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(EditBuildScreen), nameof(EditBuildScreen.Show), [typeof(Item), typeof(Item), typeof(InventoryControllerClass), typeof(ISession)]);
            }

            [PatchPrefix]
            public static void Prefix(EditBuildScreen __instance, WeaponPreview ____weaponPreview)
            {
                var scrollTrigger = __instance.gameObject.AddComponent<ScrollTrigger>();
                scrollTrigger.OnOnScroll += (PointerEventData eventData) =>
                {
                    if (____weaponPreview != null && __instance != null)
                    {
                        ____weaponPreview.Zoom(eventData.scrollDelta.y * 0.12f);
                        __instance.UpdatePositions();
                    }
                };
            }
        }

        public class WeaponModdingScreenZoomPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(WeaponModdingScreen), nameof(WeaponModdingScreen.Show), [typeof(Item), typeof(InventoryControllerClass), typeof(LootItemClass[])]);
            }

            [PatchPrefix]
            public static void Prefix(WeaponModdingScreen __instance, WeaponPreview ____weaponPreview)
            {
                var scrollTrigger = __instance.gameObject.AddComponent<ScrollTrigger>();
                scrollTrigger.OnOnScroll += (PointerEventData eventData) =>
                {
                    if (____weaponPreview != null && __instance != null)
                    {
                        ____weaponPreview.Zoom(eventData.scrollDelta.y * 0.12f);
                        __instance.UpdatePositions();
                    }
                };
            }
        }
    }
}