#include "HallsOfLightningMultipliers.h"
#include "HallsOfLightningActions.h"
#include "GenericSpellActions.h"
#include "ChooseTargetActions.h"
#include "MovementActions.h"
#include "HallsOfLightningTriggers.h"
#include "Action.h"

#include "WarriorActions.h"

float BjarngrimMultiplier::GetValue(Action* action)
{
    Unit* boss = AI_VALUE2(Unit*, "find target", "general bjarngrim");
    if (!boss || botAI->IsHeal(bot)) { return 1.0f; }

    if (boss->HasUnitState(UNIT_STATE_CASTING) && boss->FindCurrentSpellBySpellId(SPELL_WHIRLWIND_BJARNGRIM))
    {
        if (dynamic_cast<MovementAction*>(action) && !dynamic_cast<AvoidWhirlwindAction*>(action))
        {
            return 0.0f;
        }
    }

    // Detect boss adds this way as sometimes they don't get added to threat table on dps bots,
    // and some dps just stand at range and don't engage the boss at all as they can't find the adds
    // Unit* boss_add = AI_VALUE2(Unit*, "find target", "stormforged lieutenant");
    Unit* boss_add = nullptr;
    GuidVector targets = AI_VALUE(GuidVector, "possible targets no los");

    for (auto i = targets.begin(); i != targets.end(); ++i)
    {
        Unit* unit = botAI->GetUnit(*i);
        if (unit && unit->GetEntry() == NPC_STORMFORGED_LIEUTENANT)
        {
            boss_add = unit;
            break;
        }
    }

    if (!boss_add || botAI->IsTank(bot)) { return 1.0f; }

    if (dynamic_cast<DpsAssistAction*>(action))
    {
        return 0.0f;
    }

    if (action->getThreatType() == Action::ActionThreatType::Aoe)
    {
        return 0.0f;
    }

    return 1.0f;
}

float VolkhanMultiplier::GetValue(Action* action)
{
    Unit* boss = AI_VALUE2(Unit*, "find target", "volkhan");
    if (!boss || botAI->IsTank(bot) || botAI->IsHeal(bot)) { return 1.0f; }

    if (dynamic_cast<DpsAssistAction*>(action))
    {
        return 0.0f;
    }

    if (action->getThreatType() == Action::ActionThreatType::Aoe)
    {
        return 0.0f;
    }

    return 1.0f;
}

float IonarMultiplier::GetValue(Action* action)
{
    Unit* boss = AI_VALUE2(Unit*, "find target", "ionar");
    if (!boss) { return 1.0f; }

    // Check if the boss has dispersed into Sparks (not visible).
    if (!bot->CanSeeOrDetect(boss))
    {
        // Block MovementActions except for specific exceptions.
        if (dynamic_cast<MovementAction*>(action)
            && !dynamic_cast<DispersePositionAction*>(action)
            && !dynamic_cast<StaticOverloadSpreadAction*>(action))
        {
            return 0.0f;
        }
    }

    if (boss->FindCurrentSpellBySpellId(SPELL_DISPERSE))
    {
        // Explicitly block the CastChargeAction during dispersal.
        if (dynamic_cast<CastChargeAction*>(action))
        {
            return 0.0f;
        }
    }
    return 1.0f;
}
float LokenMultiplier::GetValue(Action* action)
{
    Unit* boss = AI_VALUE2(Unit*, "find target", "loken");
    if (!boss) { return 1.0f; }

    // Prevent FleeAction from being executed.
    if (dynamic_cast<FleeAction*>(action)) { return 0.0f; }

    // Prevent MovementActions during Lightning Nova unless it's AvoidLightningNovaAction.
    if (boss->FindCurrentSpellBySpellId(SPELL_LIGHTNING_NOVA))
    {
        if (dynamic_cast<MovementAction*>(action)
            && !dynamic_cast<AvoidLightningNovaAction*>(action))
        {
            return 0.0f;
        }

        // Specifically prevent Charge during Lightning Nova.
        if (dynamic_cast<CastChargeAction*>(action))
        {
            return 0.0f;
        }
    }

    return 1.0f; // Default multiplier value for other cases.
}