ability_item_usage_dawnbreaker.lua

---@diagnostic disable: undefined-global
require(GetScriptDirectory() .. "/utility")
require(GetScriptDirectory() .. "/ability_item_usage_generic")
require(GetScriptDirectory() .. "/ability_levelup_generic")

function CourierUsageThink()
    ability_item_usage_generic.CourierUsageThink()
end

function BuybackUsageThink()
    ability_item_usage_generic.BuybackUsageThink();
end

-- Ability learn
local npcBot = GetBot();
local Abilities, Talents, AbilitiesReal = ability_levelup_generic.GetHeroAbilities(npcBot)

local AbilityToLevelUp =
{
    Abilities[1],
    Abilities[2],
    Abilities[3],
    Abilities[1],
    Abilities[1],
    Abilities[6],
    Abilities[1],
    Abilities[2],
    Abilities[2],
    Talents[1],
    Abilities[2],
    Abilities[6],
    Abilities[3],
    Abilities[3],
    Talents[3],
    Abilities[3],
    Abilities[6],
    Talents[6],
    Talents[7],
    Talents[2],
    Talents[4],
    Talents[5],
    Talents[8],
}

function AbilityLevelUpThink()
    ability_levelup_generic.AbilityLevelUpThink(AbilityToLevelUp)
end

-- Abilities
local Starbreaker = npcBot:GetAbilityByName("dawnbreaker_fire_wreath");
local CelestialHammer = npcBot:GetAbilityByName("dawnbreaker_celestial_hammer");
local Converge = npcBot:GetAbilityByName("dawnbreaker_converge");
local SolarGuardianLand = npcBot:GetAbilityByName("dawnbreaker_land");
local SolarGuardian = npcBot:GetAbilityByName("dawnbreaker_solar_guardian");

function AbilityUsageThink()
    if not utility.CanCast(npcBot) then
        return;
    end

    botMode = npcBot:GetActiveMode();
    botTarget = npcBot:GetTarget();
    HealthPercentage = npcBot:GetHealth() / npcBot:GetMaxHealth();
    ManaPercentage = npcBot:GetMana() / npcBot:GetMaxMana();

    local castStarbreakerDesire, castStarbreakerLocation = ConsiderStarbreaker();
    local castCelestialHammerDesire, castCelestialHammerLocation = ConsiderCelestialHammer();
    local castConvergeDesire = ConsiderConverge();
    local castSolarGuardianLandDesire = ConsiderSolarGuardianLand();
    local castSolarGuardianDesire, castSolarGuardianLocation = ConsiderSolarGuardian();

    if (castStarbreakerDesire > 0)
    then
        npcBot:Action_UseAbilityOnLocation(Starbreaker, castStarbreakerLocation);
        return;
    end

    if (castCelestialHammerDesire > 0)
    then
        npcBot:Action_UseAbilityOnLocation(CelestialHammer, castCelestialHammerLocation);
        return;
    end

    if (castConvergeDesire > 0)
    then
        npcBot:Action_UseAbility(Converge);
        return;
    end

    if (castSolarGuardianLandDesire > 0)
    then
        npcBot:Action_UseAbility(SolarGuardianLand);
        return;
    end

    if (castSolarGuardianDesire > 0)
    then
        npcBot:Action_UseAbilityOnLocation(SolarGuardian, castSolarGuardianLocation);
        return;
    end
end

function ConsiderStarbreaker()
    local ability = Starbreaker;
    if not utility.IsAbilityAvailable(ability) then
        return BOT_ACTION_DESIRE_NONE, 0;
    end

    if npcBot:HasModifier("modifier_starbreaker_fire_wreath_caster") or
        npcBot:HasModifier("modifier_fire_wreath_magic_immunity_tooltip")
    then
        return BOT_ACTION_DESIRE_NONE, 0;
    end

    local projectiles = GetLinearProjectiles();
    if (#projectiles > 0)
    then
        for _, project in pairs(projectiles)
        do
            if project ~= nil and project.ability:GetName() == "dawnbreaker_celestial_hammer"
            then
                return BOT_ACTION_DESIRE_NONE, 0;
            end
        end
    end

    local castRangeAbility = ability:GetSpecialValueInt("swipe_radius");
    local radiusAbility = ability:GetSpecialValueInt("smash_radius");
    local delayAbility = ability:GetSpecialValueInt("AbilityCastPoint");
    local enemyAbility = npcBot:GetNearbyHeroes(castRangeAbility, true, BOT_MODE_NONE);

    -- Cast if can interrupt cast
    if not utility.RetreatMode(npcBot)
    then
        if (#enemyAbility > 0)
        then
            for _, enemy in pairs(enemyAbility) do
                if utility.CanCastSpellOnTarget(ability, enemy) and enemy:IsChanneling()
                then
                    --npcBot:ActionImmediate_Chat("Использую Starbreaker что бы сбить заклинание!",true);
                    return BOT_ACTION_DESIRE_VERYHIGH, utility.GetTargetCastPosition(npcBot, enemy, delayAbility, 0);
                end
            end
        end
    end

    -- Cast if attack enemy
    if utility.PvPMode(npcBot) or utility.BossMode(npcBot)
    then
        if utility.IsHero(botTarget) or utility.IsBoss(botTarget)
        then
            if utility.CanCastSpellOnTarget(ability, botTarget) and GetUnitToUnitDistance(npcBot, botTarget) < castRangeAbility
            then
                return BOT_ACTION_DESIRE_VERYHIGH, utility.GetTargetCastPosition(npcBot, botTarget, delayAbility, 0);
            end
        end
    end

    -- Cast if push/defend/farm
    if utility.PvEMode(npcBot)
    then
        local locationAoE = npcBot:FindAoELocation(true, false, npcBot:GetLocation(), castRangeAbility, radiusAbility,
            0, 0);
        if locationAoE ~= nil and (ManaPercentage >= 0.5) and (locationAoE.count >= 3)
        then
            --npcBot:ActionImmediate_Chat("Использую Starbreaker по вражеским крипам!", true);
            return BOT_ACTION_DESIRE_LOW, locationAoE.targetloc;
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end

function ConsiderCelestialHammer()
    local ability = CelestialHammer;
    if not utility.IsAbilityAvailable(ability) then
        return BOT_ACTION_DESIRE_NONE, 0;
    end

    local attackRange = npcBot:GetAttackRange();
    local castRangeAbility = ability:GetSpecialValueInt("range");
    local radiusAbility = ability:GetSpecialValueInt("flare_radius");
    local damageAbility = ability:GetSpecialValueInt("hammer_damage");
    local delayAbility = ability:GetSpecialValueInt("AbilityCastPoint");
    local speedAbility = ability:GetSpecialValueInt("projectile_speed");
    local enemyAbility = npcBot:GetNearbyHeroes(castRangeAbility + 200, true, BOT_MODE_NONE);

    -- Cast if can kill somebody
    if (#enemyAbility > 0)
    then
        for _, enemy in pairs(enemyAbility) do
            if utility.CanAbilityKillTarget(enemy, damageAbility, ability:GetDamageType())
            then
                if utility.CanCastSpellOnTarget(ability, enemy)
                then
                    return BOT_ACTION_DESIRE_ABSOLUTE,
                        utility.GetTargetCastPosition(npcBot, enemy, delayAbility, speedAbility);
                end
            end
        end
    end

    -- Cast if attack enemy
    if utility.PvPMode(npcBot) or utility.BossMode(npcBot)
    then
        if utility.IsHero(botTarget) or utility.IsBoss(botTarget)
        then
            if utility.CanCastSpellOnTarget(ability, botTarget) and GetUnitToUnitDistance(npcBot, botTarget) <= castRangeAbility
                and GetUnitToUnitDistance(npcBot, botTarget) > attackRange
            then
                return BOT_ACTION_DESIRE_VERYHIGH,
                    utility.GetTargetCastPosition(npcBot, botTarget, delayAbility, speedAbility);
            end
        end
    end

    -- Cast if push/defend/farm
    if utility.PvEMode(npcBot)
    then
        local locationAoE = npcBot:FindAoELocation(true, false, npcBot:GetLocation(), castRangeAbility, radiusAbility,
            0, 0);
        if locationAoE ~= nil and (ManaPercentage >= 0.5) and (locationAoE.count >= 3)
        then
            --npcBot:ActionImmediate_Chat("Использую CelestialHammer по вражеским крипам!", true);
            return BOT_ACTION_DESIRE_LOW, locationAoE.targetloc;
        end
    end

    -- Retreat use
    if utility.RetreatMode(npcBot)
    then
        if (HealthPercentage <= 0.8) and npcBot:WasRecentlyDamagedByAnyHero(2.0) and npcBot:DistanceFromFountain() > castRangeAbility
        then
            --npcBot:ActionImmediate_Chat("Использую CelestialHammer для отхода!", true);
            return BOT_ACTION_DESIRE_HIGH, utility.GetEscapeLocation(npcBot, castRangeAbility);
        end
    end

    -- Cast when laning
    if botMode == BOT_MODE_LANING
    then
        local enemy = utility.GetWeakest(enemyAbility);
        if enemy ~= nil and utility.CanCastSpellOnTarget(ability, enemy) and (ManaPercentage >= 0.7)
        then
            --npcBot:ActionImmediate_Chat("Использую CelestialHammer по цели на ЛАЙНЕ!", true);
            return BOT_ACTION_DESIRE_VERYHIGH, utility.GetTargetCastPosition(npcBot, enemy, delayAbility, speedAbility);
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end

function ConsiderConverge()
    local ability = Converge;
    if not utility.IsAbilityAvailable(ability) then
        return BOT_ACTION_DESIRE_NONE;
    end

    local castRangeAbility = ability:GetCastRange();
    local radiusAbility = CelestialHammer:GetSpecialValueInt("flare_radius");
    local projectiles = GetLinearProjectiles();

    -- Cast if attack enemy
    if utility.PvPMode(npcBot) or utility.BossMode(npcBot)
    then
        if utility.IsHero(botTarget) or utility.IsBoss(botTarget)
        then
            if (#projectiles > 0)
            then
                for _, project in pairs(projectiles)
                do
                    if project ~= nil and project.ability:GetName() == "dawnbreaker_celestial_hammer"
                    then
                        if GetUnitToLocationDistance(botTarget, project.location) <= radiusAbility and GetUnitToUnitDistance(npcBot, botTarget) >= castRangeAbility
                        then
                            --npcBot:ActionImmediate_Chat("Использую Converge рядом с врагом!", true);
                            return BOT_ACTION_DESIRE_HIGH;
                        end
                    end
                end
            end
        end
    end

    if utility.RetreatMode(npcBot)
    then
        return BOT_ACTION_DESIRE_HIGH;
        --[[         local fountain = utility.GetFountain(npcBot);
        if fountain ~= nil and (#projectiles > 0)
        then
            for _, project in pairs(projectiles)
            do
                if project ~= nil and project.ability:GetName() == "dawnbreaker_celestial_hammer"
                then
                    if GetUnitToLocationDistance(fountain, project.location) < GetUnitToLocationDistance(npcBot, project.location)
                        and GetUnitToLocationDistance(npcBot, project.location) >= CelestialHammer:GetSpecialValueInt("range") / 2
                    then
                        --npcBot:ActionImmediate_Chat("Использую Converger для отхода!", true);
                        return BOT_ACTION_DESIRE_HIGH;
                    end
                end
            end
        end ]]
    end

    return BOT_ACTION_DESIRE_NONE;
end

function ConsiderSolarGuardianLand()
    local ability = SolarGuardianLand;
    if not utility.IsAbilityAvailable(ability) then
        return BOT_ACTION_DESIRE_NONE;
    end

    if utility.RetreatMode(npcBot)
    then
        --npcBot:ActionImmediate_Chat("Использую SolarGuardianLand!", true);
        return BOT_ACTION_DESIRE_HIGH;
    end

    return BOT_ACTION_DESIRE_NONE;
end

function ConsiderSolarGuardian()
    local ability = SolarGuardian;
    if not utility.IsAbilityAvailable(ability) then
        return BOT_ACTION_DESIRE_NONE, 0;
    end

    local castRangeAbility = ability:GetSpecialValueInt("max_offset_distance"); -- 350
    local radiusAbility = ability:GetSpecialValueInt("radius");                 -- 500
    local allyAbility = GetUnitList(UNIT_LIST_ALLIED_HEROES);

    -- Cast if attack enemy
    if utility.PvPMode(npcBot)
    then
        if utility.IsHero(botTarget) and utility.CanCastSpellOnTarget(ability, botTarget)
        then
            if (#allyAbility > 1)
            then
                for _, ally in pairs(allyAbility)
                do
                    if GetUnitToUnitDistance(ally, botTarget) < castRangeAbility
                    then
                        --npcBot:ActionImmediate_Chat("Использую SolarGuardian на союзника рядом с врагом!", true);
                        return BOT_ACTION_DESIRE_HIGH, botTarget:GetLocation();
                    end
                end
            end
        end
    end

    -- Use if need retreat
    if utility.RetreatMode(npcBot)
    then
        if (HealthPercentage <= 0.7) and utility.BotWasRecentlyDamagedByEnemyHero(2.0)
        then
            local fountainLocation = utility.GetFountainLocation();
            if (#allyAbility > 1)
            then
                for _, ally in pairs(allyAbility)
                do
                    if ally ~= npcBot and (GetUnitToLocationDistance(ally, fountainLocation) < GetUnitToLocationDistance(npcBot, fountainLocation)
                            and (GetUnitToUnitDistance(ally, npcBot) > radiusAbility))
                    then
                        --npcBot:ActionImmediate_Chat("Использую SolarGuardian для отхода!", true);
                        return BOT_ACTION_DESIRE_HIGH, ally:GetLocation() + RandomVector(castRangeAbility);
                    end
                end
            end
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end