ability_item_usage_bounty_hunter.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[3],
    Abilities[2],
    Abilities[1],
    Abilities[1],
    Abilities[6],
    Abilities[1],
    Abilities[2],
    Abilities[2],
    Talents[2],
    Abilities[2],
    Abilities[6],
    Abilities[3],
    Abilities[3],
    Talents[3],
    Abilities[3],
    Abilities[6],
    Talents[6],
    Talents[7],
    Talents[1],
    Talents[4],
    Talents[5],
    Talents[8],
}

function AbilityLevelUpThink()
    ability_levelup_generic.AbilityLevelUpThink(AbilityToLevelUp)
end

-- Abilities
local ShurikenToss = npcBot:GetAbilityByName("bounty_hunter_shuriken_toss");
local Jinada = npcBot:GetAbilityByName("bounty_hunter_jinada");
local ShadowWalk = npcBot:GetAbilityByName("bounty_hunter_wind_walk");
local FriendlyShadow = npcBot:GetAbilityByName("bounty_hunter_wind_walk_ally");
local Track = npcBot:GetAbilityByName("bounty_hunter_track");

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 castShurikenTossDesire, castShurikenTossTarget = ConsiderShurikenToss();
    --local castJinadaDesire, castJinadaTarget = ConsiderJinada();
    ConsiderJinada();
    local castShadowWalkDesire = ConsiderShadowWalk();
    local castFriendlyShadowDesire, castFriendlyShadowTarget = ConsiderFriendlyShadow();
    local castTrackDesire, castTrackTarget = ConsiderTrack();

    if (castShurikenTossDesire > 0)
    then
        npcBot:Action_UseAbilityOnEntity(ShurikenToss, castShurikenTossTarget);
        return;
    end

    if (castShadowWalkDesire > 0)
    then
        npcBot:Action_UseAbility(ShadowWalk);
        return;
    end

    if (castFriendlyShadowDesire > 0)
    then
        npcBot:Action_UseAbilityOnEntity(FriendlyShadow, castFriendlyShadowTarget);
        return;
    end

    if (castTrackDesire > 0)
    then
        npcBot:Action_UseAbilityOnEntity(Track, castTrackTarget);
        return;
    end

    --[[     if (castJinadaDesire ~= nil)
    then
        npcBot:Action_UseAbilityOnEntity(Jinada, castJinadaTarget);
        return;
    end ]]
end

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

    local castRangeAbility = ability:GetSpecialValueInt("cast_range");
    local radiusAbility = ability:GetSpecialValueInt("bounce_aoe");
    local damageAbility = ability:GetSpecialValueInt("bonus_damage");
    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
                    --npcBot:ActionImmediate_Chat("Использую ShurikenToss что бы убить цель!", true);
                    return BOT_ACTION_DESIRE_VERYHIGH, enemy;
                end
            end
        end
    end

    if npcBot:IsInvisible()
    then
        return BOT_ACTION_DESIRE_NONE, 0;
    end

    -- Attack use
    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
                if botTarget:HasModifier("modifier_bounty_hunter_track")
                then
                    local enemyHeroes = npcBot:GetNearbyHeroes(castRangeAbility, true, BOT_MODE_NONE);
                    local enemyCreeps = npcBot:GetNearbyCreeps(castRangeAbility, true);
                    if (#enemyHeroes > 1)
                    then
                        for _, enemy in pairs(enemyHeroes) do
                            if enemy ~= botTarget and utility.CanCastSpellOnTarget(ability, enemy) and GetUnitToUnitDistance(enemy, botTarget) <= radiusAbility
                            then
                                --npcBot:ActionImmediate_Chat("Использую ShurikenToss по помеченному врагу (на героя)!", true);
                                return BOT_ACTION_DESIRE_HIGH, enemy;
                            end
                        end
                    end
                    if (#enemyCreeps > 0)
                    then
                        for _, enemy in pairs(enemyCreeps) do
                            if utility.CanCastSpellOnTarget(ability, enemy) and GetUnitToUnitDistance(enemy, botTarget) <= radiusAbility
                            then
                                --npcBot:ActionImmediate_Chat("Использую ShurikenToss по помеченному врагу (на крипа)!",true);
                                return BOT_ACTION_DESIRE_HIGH, enemy;
                            end
                        end
                    end
                end
                --npcBot:ActionImmediate_Chat("Использую ShurikenToss по не помеченному врагу!", true);
                return BOT_ACTION_DESIRE_HIGH, botTarget;
            end
        end
    end

    -- Retreat use
    if utility.RetreatMode(npcBot)
    then
        if (#enemyAbility > 0)
        then
            for _, enemy in pairs(enemyAbility) do
                if utility.CanCastSpellOnTarget(ability, enemy) and not utility.IsDisabled(enemy)
                then
                    --npcBot:ActionImmediate_Chat("Использую ShurikenToss что бы оторваться от врага", true);
                    return BOT_ACTION_DESIRE_HIGH, enemy;
                end
            end
        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("Использую ShurikenToss по цели на ЛАЙНЕ!", true);
            return BOT_ACTION_DESIRE_HIGH, enemy;
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end

function ConsiderJinada()
    local ability = Jinada;
    if not utility.IsAbilityAvailable(ability) then
        return;
    end

    if utility.IsNeedTurnOnAttackModifier()
    then
        if not ability:GetAutoCastState() then
            ability:ToggleAutoCast()
        end
    else
        if ability:GetAutoCastState() then
            ability:ToggleAutoCast()
        end
    end

    --local castRangeAbility = ability:GetCastRange();
    --local damageAbility = npcBot:GetAttackDamage() + ability:GetSpecialValueInt("bonus_damage");
    --local enemyAbility = npcBot:GetNearbyHeroes(castRangeAbility, 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
                    --npcBot:ActionImmediate_Chat("Использую Jinada что бы убить цель!", true);
                    return BOT_ACTION_DESIRE_VERYHIGH, enemy;
                end
            end
        end
    end

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

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

    if npcBot:IsInvisible()
    then
        return BOT_ACTION_DESIRE_NONE;
    end

    local attackTarget = npcBot:GetAttackTarget();
    local attackRange = npcBot:GetAttackRange();

    -- Try to interrupt enemy cast
    if attackTarget ~= nil and attackTarget:IsChanneling()
    then
        --npcBot:ActionImmediate_Chat("Использую ShadowWalk против кастующей цели!", true);
        return BOT_ACTION_DESIRE_VERYHIGH;
    end

    -- Attack use
    if utility.PvPMode(npcBot)
    then
        if utility.IsHero(botTarget) and (GetUnitToUnitDistance(npcBot, botTarget) > attackRange and GetUnitToUnitDistance(npcBot, botTarget) <= 3000)
        then
            --npcBot:ActionImmediate_Chat("Использую ShadowWalk для нападения!", true);
            return BOT_ACTION_DESIRE_VERYHIGH;
        end
    end

    -- Retreat use
    if utility.RetreatMode(npcBot)
    then
        --npcBot:ActionImmediate_Chat("Использую ShadowWalk для отхода!", true);
        return BOT_ACTION_DESIRE_VERYHIGH;
    end

    -- General use
    if utility.WanderMode(npcBot)
    then
        local enemyTowers = npcBot:GetNearbyTowers(1000, true);
        if (#enemyTowers == 0) and npcBot:GetCurrentActionType() ~= BOT_ACTION_TYPE_ATTACK
        then
            return BOT_ACTION_DESIRE_MODERATE;
        end
    end

    return BOT_ACTION_DESIRE_NONE;
end

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

    local castRangeAbility = ability:GetCastRange();
    local allyAbility = npcBot:GetNearbyHeroes(castRangeAbility + 200, false, BOT_MODE_NONE);

    if (#allyAbility > 1)
    then
        -- Safe ally hero
        for _, ally in pairs(allyAbility)
        do
            if ally ~= npcBot and utility.IsHero(ally) and not ally:IsInvisible()
            then
                if ally:GetHealth() / ally:GetMaxHealth() <= 0.8 and ((ally:WasRecentlyDamagedByAnyHero(2.0) or ally:WasRecentlyDamagedByTower(2.0))
                        or ally:IsChanneling())
                then
                    --npcBot:ActionImmediate_Chat("Использую FriendlyShadow на союзника!", true);
                    return BOT_ACTION_DESIRE_VERYHIGH, ally;
                end
            end
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end

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

    local castRangeAbility = ability:GetCastRange();
    local enemyAbility = npcBot:GetNearbyHeroes(castRangeAbility + 200, true, BOT_MODE_NONE);

    -- Attack use
    if utility.PvPMode(npcBot)
    then
        if utility.IsHero(botTarget)
        then
            if utility.CanCastSpellOnTarget(ability, botTarget) and GetUnitToUnitDistance(npcBot, botTarget) <= castRangeAbility + 200
            then
                if not botTarget:HasModifier("modifier_bounty_hunter_track")
                then
                    --npcBot:ActionImmediate_Chat("Использую Track по основной цели!", true);
                    return BOT_ACTION_DESIRE_VERYHIGH, botTarget;
                else
                    if (#enemyAbility > 1)
                    then
                        for _, enemy in pairs(enemyAbility) do
                            if enemy ~= botTarget and utility.CanCastSpellOnTarget(ability, enemy) and not enemy:HasModifier("modifier_bounty_hunter_track")
                            then
                                --npcBot:ActionImmediate_Chat("Использую Track по 2 цели!", true);
                                return BOT_ACTION_DESIRE_VERYHIGH, enemy;
                            end
                        end
                    end
                end
            end
        end
    end

    -- General use
    if (#enemyAbility > 0) and (ManaPercentage >= 0.3)
    then
        for _, enemy in pairs(enemyAbility) do
            if utility.CanCastSpellOnTarget(ability, enemy) and not enemy:HasModifier("modifier_bounty_hunter_track")
            then
                --npcBot:ActionImmediate_Chat("Использую Track для отметки врага!", true);
                return BOT_ACTION_DESIRE_VERYHIGH, enemy;
            end
        end
    end

    return BOT_ACTION_DESIRE_NONE, 0;
end