using TOHE.Roles.AddOns;
using static TOHE.Options;
using static TOHE.Translator;

namespace TOHE.Roles.Crewmate;

internal class Merchant : RoleBase
{
    //===========================SETUP================================\\
    private const int Id = 8800;
    private static readonly HashSet<byte> playerIdList = [];
    public static bool HasEnabled => playerIdList.Any();
    
    public override CustomRoles ThisRoleBase => CustomRoles.Crewmate;
    public override Custom_RoleType ThisRoleType => Custom_RoleType.CrewmateSupport;
    //==================================================================\\

    private static List<CustomRoles> addons = [];
    private static readonly Dictionary<byte, int> addonsSold = [];
    private static readonly Dictionary<byte, HashSet<byte>> bribedKiller = [];

    private static OptionItem OptionMaxSell;
    private static OptionItem OptionMoneyPerSell;
    private static OptionItem OptionMoneyRequiredToBribe;
    private static OptionItem OptionNotifyBribery;
    private static OptionItem OptionCanTargetCrew;
    private static OptionItem OptionCanTargetImpostor;
    private static OptionItem OptionCanTargetNeutral;
    private static OptionItem OptionCanSellHelpful;
    private static OptionItem OptionCanSellHarmful;
    private static OptionItem OptionCanSellNeutral;
    private static OptionItem OptionSellOnlyHarmfulToEvil;
    private static OptionItem OptionSellOnlyHelpfulToCrew;
    private static OptionItem OptionSellOnlyEnabledAddons;

    private static int GetCurrentAmountOfMoney(byte playerId)
    {
        return (addonsSold[playerId] * OptionMoneyPerSell.GetInt()) - (bribedKiller[playerId].Count * OptionMoneyRequiredToBribe.GetInt());
    }

    public override void SetupCustomOption()
    {
        SetupRoleOptions(Id, TabGroup.CrewmateRoles, CustomRoles.Merchant);
        OptionMaxSell = IntegerOptionItem.Create(Id + 2, "MerchantMaxSell", new(1, 99, 1), 5, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]).SetValueFormat(OptionFormat.Times);
        OptionMoneyPerSell = IntegerOptionItem.Create(Id + 3, "MerchantMoneyPerSell", new(1, 99, 1), 1, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]).SetValueFormat(OptionFormat.Times);
        OptionMoneyRequiredToBribe = IntegerOptionItem.Create(Id + 4, "MerchantMoneyRequiredToBribe", new(1, 99, 1), 5, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]).SetValueFormat(OptionFormat.Times);
        OptionNotifyBribery = BooleanOptionItem.Create(Id + 5, "MerchantNotifyBribery", false, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanTargetCrew = BooleanOptionItem.Create(Id + 6, "MerchantTargetCrew", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanTargetImpostor = BooleanOptionItem.Create(Id + 7, "MerchantTargetImpostor", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanTargetNeutral = BooleanOptionItem.Create(Id + 8, "MerchantTargetNeutral", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanSellHelpful = BooleanOptionItem.Create(Id + 9, "MerchantSellHelpful", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanSellHarmful = BooleanOptionItem.Create(Id + 10, "MerchantSellHarmful", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionCanSellNeutral = BooleanOptionItem.Create(Id + 11, "MerchantSellMixed", true, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionSellOnlyHarmfulToEvil = BooleanOptionItem.Create(Id + 13, "MerchantSellHarmfulToEvil", false, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionSellOnlyHelpfulToCrew = BooleanOptionItem.Create(Id + 14, "MerchantSellHelpfulToCrew", false, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);
        OptionSellOnlyEnabledAddons = BooleanOptionItem.Create(Id + 15, "MerchantSellOnlyEnabledAddons",false, TabGroup.CrewmateRoles, false).SetParent(CustomRoleSpawnChances[CustomRoles.Merchant]);

        OverrideTasksData.Create(Id + 16, TabGroup.CrewmateRoles, CustomRoles.Merchant);
    }
    public override void Init()
    {
        playerIdList.Clear();

        addons.Clear();
        addonsSold.Clear();
        bribedKiller.Clear();

        if (OptionCanSellHelpful.GetBool())
        {
            addons.AddRange(GroupedAddons[AddonTypes.Helpful]);
        }

        if (OptionCanSellHarmful.GetBool())
        {
            addons.AddRange(GroupedAddons[AddonTypes.Harmful]);
        }

        if (OptionCanSellNeutral.GetBool())
        {
            addons.AddRange(GroupedAddons[AddonTypes.Mixed]);
        }
        if (OptionSellOnlyEnabledAddons.GetBool())
        { 
            addons = addons.Where(role => role.GetMode() != 0).ToList();
        }
    }

    public override void Add(byte playerId)
    {
        playerIdList.Add(playerId);
        addonsSold[playerId] = 0;
        bribedKiller.TryAdd(playerId, []);
    }
    public override void Remove(byte playerId)
    {
        playerIdList.Remove(playerId);
        addonsSold.Remove(playerId);
        bribedKiller.Remove(playerId);
    }
    public override bool OnCheckMurderAsTarget(PlayerControl killer, PlayerControl target)
    {
        return !OnClientMurder(killer, target);
    }
    public override bool OnTaskComplete(PlayerControl player, int completedTaskCount, int totalTaskCount)
    {
        if (!player.IsAlive()) return true;

        if (addonsSold[player.PlayerId] >= OptionMaxSell.GetInt())
        {
            return true;
        }

        if (addons.Count == 0)
        {
            player.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("MerchantAddonSellFail")));
            Logger.Info("No addons to sell.", "Merchant");
            return true;
        }

        var rd = IRandom.Instance;
        CustomRoles addon = addons.RandomElement();

        List<PlayerControl> AllAlivePlayer =
            Main.AllAlivePlayerControls.Where(x =>
                x.PlayerId != player.PlayerId
                &&
                (!x.Is(CustomRoles.Stubborn))
                &&
                !addon.IsConverted()
                &&
                CustomRolesHelper.CheckAddonConfilct(addon, x, checkLimitAddons: false)
                &&
                (!Cleanser.CantGetAddon() || (Cleanser.CantGetAddon() && !x.Is(CustomRoles.Cleansed)))
                &&
                (
                    (OptionCanTargetCrew.GetBool() && x.GetCustomRole().IsCrewmate()) 
                    ||
                    (OptionCanTargetImpostor.GetBool() && x.GetCustomRole().IsImpostor())
                    ||
                    (OptionCanTargetNeutral.GetBool() && x.GetCustomRole().IsNeutral())
                )
            ).ToList();

        if (AllAlivePlayer.Any())
        {
            bool helpfulAddon = GroupedAddons[AddonTypes.Helpful].Contains(addon);
            bool harmfulAddon = GroupedAddons[AddonTypes.Harmful].Contains(addon);

            if (helpfulAddon && OptionSellOnlyHarmfulToEvil.GetBool())
            {
                AllAlivePlayer = AllAlivePlayer.Where(a => a.GetCustomRole().IsCrewmate()).ToList();
            }

            if (harmfulAddon && OptionSellOnlyHelpfulToCrew.GetBool())
            {
                AllAlivePlayer = AllAlivePlayer.Where(a =>
                    a.GetCustomRole().IsImpostor()
                    ||
                    a.GetCustomRole().IsNeutral()
                    
                ).ToList();
            }

            PlayerControl target = AllAlivePlayer.RandomElement();

            target.RpcSetCustomRole(addon);
            target.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("MerchantAddonSell")));
            player.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("MerchantAddonDelivered")));

            target.AddInSwitchAddons(target, addon);
            
            addonsSold[player.PlayerId] += 1;
        }
        else
        {
            player.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("MerchantAddonSellFail")));
            Logger.Info("All Alive Player Count = 0", "Merchant");
            return true;
        }

        return true;
    }
    public override bool OnRoleGuess(bool isUI, PlayerControl target, PlayerControl pc, CustomRoles role, ref bool guesserSuicide)
    {
        if (IsBribedKiller(pc, target))
        {
            pc.ShowInfoMessage(isUI, GetString("BribedByMerchant2"));
            return true;
        }
        return false;
    }

    public static bool OnClientMurder(PlayerControl killer, PlayerControl target)
    {
        if (IsBribedKiller(killer, target))
        {
            NotifyBribery(killer, target);
            return true;
        }

        if (GetCurrentAmountOfMoney(target.PlayerId) >= OptionMoneyRequiredToBribe.GetInt())
        {
            NotifyBribery(killer, target);
            bribedKiller[target.PlayerId].Add(killer.PlayerId);
            return true;
        }

        return false;
    }

    public static bool IsBribedKiller(PlayerControl killer, PlayerControl target) => bribedKiller.TryGetValue(target.PlayerId, out var targets) && targets.Contains(killer.PlayerId);

    private static void NotifyBribery(PlayerControl killer, PlayerControl target)
    {
        killer.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("BribedByMerchant")));

        if (OptionNotifyBribery.GetBool())
        {
            target.Notify(Utils.ColorString(Utils.GetRoleColor(CustomRoles.Merchant), GetString("MerchantKillAttemptBribed")));
        }
    }
}
