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

namespace UIFixes
{
    public static class StackFirItemsPatches
    {
        public static void Enable()
        {
            new ContainerStackPatch().Enable();
            new TopUpStackPatch().Enable();
        }

        public class ContainerStackPatch : ModulePatch
        {
            private static Type MergeableItemType;

            protected override MethodBase GetTargetMethod()
            {
                MethodInfo method = AccessTools.Method(typeof(InteractionsHandlerClass), nameof(InteractionsHandlerClass.smethod_0));
                MergeableItemType = method.GetParameters()[2].ParameterType.GetElementType(); // parameter is a ref type, get underlying type
                return method;
            }

            // Reimplementing this entire method to ignore SpawnedInSession for certain types
            [PatchPrefix]
            public static bool Prefix(IEnumerable<StashGridClass> gridsToPut, Item itemToMerge, ref object mergeableItem, int overrideCount, ref bool __result)
            {
                if (!MergeableItemType.IsInstanceOfType(itemToMerge))
                {
                    mergeableItem = null;
                    __result = false;
                }

                if (overrideCount <= 0)
                {
                    overrideCount = itemToMerge.StackObjectsCount;
                }

                bool ignoreSpawnedInSession;
                if (itemToMerge.Template is MoneyClass)
                {
                    ignoreSpawnedInSession = Settings.MergeFIRMoney.Value;
                }
                else if (itemToMerge.Template is AmmoTemplate)
                {
                    ignoreSpawnedInSession = Settings.MergeFIRAmmo.Value;
                }
                else
                {
                    ignoreSpawnedInSession = Settings.MergeFIROther.Value;
                }

                mergeableItem = gridsToPut.SelectMany(x => x.Items).Where(x => MergeableItemType.IsInstanceOfType(x))
                    .Where(x => x != itemToMerge)
                    .Where(x => x.TemplateId == itemToMerge.TemplateId)
                    .Where(x => ignoreSpawnedInSession || x.SpawnedInSession == itemToMerge.SpawnedInSession)
                    .Where(x => x.StackObjectsCount < x.StackMaxSize)
                    .FirstOrDefault(x => overrideCount <= x.StackMaxSize - x.StackObjectsCount);

                __result = mergeableItem != null;
                return false;
            }
        }

        public class TopUpStackPatch : ModulePatch
        {
            protected override MethodBase GetTargetMethod()
            {
                return AccessTools.Method(typeof(Item), nameof(Item.IsSameItem));
            }

            [PatchPrefix]
            public static bool Prefix(Item __instance, Item other, ref bool __result)
            {
                bool ignoreSpawnedInSession;
                if (__instance.Template is MoneyClass)
                {
                    ignoreSpawnedInSession = Settings.MergeFIRMoney.Value;
                }
                else if (__instance.Template is AmmoTemplate)
                {
                    ignoreSpawnedInSession = Settings.MergeFIRAmmo.Value;
                }
                else
                {
                    ignoreSpawnedInSession = Settings.MergeFIROther.Value;
                }

                __result = __instance.TemplateId == other.TemplateId && __instance.Id != other.Id && (ignoreSpawnedInSession || __instance.SpawnedInSession == other.SpawnedInSession);
                return false;
            }
        }
    }
}