using Aki.Reflection.Patching;
using Aki.Reflection.Utils;
using EFT;
using System;
using System.Linq;
using System.Reflection;

namespace Aki.Custom.Patches
{
    public class AddEnemyToAllGroupsInBotZonePatch : ModulePatch
    {
        private static Type _targetType;
        private const string methodName = "AddEnemyToAllGroupsInBotZone";

        public AddEnemyToAllGroupsInBotZonePatch()
        {
            _targetType = PatchConstants.EftTypes.Single(IsTargetType);
        }

        private bool IsTargetType(Type type)
        {
            if (type.Name == nameof(BotControllerClass) && type.GetMethod(methodName) != null)
            {
                return true;
            }

            return false;
        }

        protected override MethodBase GetTargetMethod()
        {
            Logger.LogDebug($"{this.GetType().Name} Type: {_targetType.Name}");
            Logger.LogDebug($"{this.GetType().Name} Method: {methodName}");

            return _targetType.GetMethod(methodName);
        }

        /// <summary>
        /// AddEnemyToAllGroupsInBotZone()
        /// Goal: by default, AddEnemyToAllGroupsInBotZone doesn't check if the bot group is on the same side as the player.
        /// The effect of this is that when you are a Scav and kill a Usec, every bot group in the zone will aggro you including other Scavs.
        /// This should fix that.
        /// </summary>
        [PatchPrefix]
        private static bool PatchPrefix(BotControllerClass __instance, IAIDetails aggressor, IAIDetails groupOwner, IAIDetails target)
        {
            BotZone botZone = groupOwner.AIData.BotOwner.BotsGroup.BotZone;
            foreach (var item in __instance.Groups())
            {
                if (item.Key != botZone)
                {
                    continue;
                }

                foreach (var group in item.Value.GetGroups(notNull: true))
                {
                    bool differentSide = aggressor.Side != group.Side;
                    bool sameSide = aggressor.Side == target.Side;

                    if (!group.Enemies.ContainsKey(aggressor)
                        && (differentSide || !sameSide)
                        && !group.HaveMemberWithRole(WildSpawnType.gifter)
                        && group.ShallRevengeFor(target)
                        )
                    {
                        group.AddEnemy(aggressor);
                    }
                }
            }

            return false;
        }
    }
}