DefenseAndFlaskTask.cs

using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using DreamPoeBot.Loki.Bot;
using DreamPoeBot.Loki.Game;
using DreamPoeBot.Loki.Game.NativeWrappers;
using DreamPoeBot.Loki.Game.Objects;
using DreamPoeBot.Loki.RemoteMemoryObjects;
using FollowBot.Class;
using FollowBot.Helpers;
using FollowBot.SimpleEXtensions;
using FlaskHud = DreamPoeBot.Loki.Game.LokiPoe.InGameState.QuickFlaskHud;

namespace FollowBot
{
    public class DefenseAndFlaskTask : ITask
    {
        public const string LifeFlaskEffect = "flask_effect_life";
        public const string ManaFlaskEffect = "flask_effect_mana";
        public const string EnduringManaFlaskEffect = "flask_effect_mana_not_removed_when_full";
        public const string QsilverEffect = "flask_utility_sprint";
        public static bool ShouldTeleport = false;
        public static bool ShouldOpenPortal = false;
        private static readonly Dictionary<string, string> FlaskEffects = new Dictionary<string, string>
        {
            [FlaskNames.Diamond] = "flask_utility_critical_strike_chance",
            [FlaskNames.Ruby] = "flask_utility_resist_fire",
            [FlaskNames.Sapphire] = "flask_utility_resist_cold",
            [FlaskNames.Topaz] = "flask_utility_resist_lightning",
            [FlaskNames.Granite] = "flask_utility_ironskin",
            [FlaskNames.Quicksilver] = "flask_utility_sprint",
            [FlaskNames.Amethyst] = "flask_utility_resist_chaos",
            [FlaskNames.Quartz] = "flask_utility_phase",
            [FlaskNames.Jade] = "flask_utility_evasion",
            [FlaskNames.Basalt] = "flask_utility_stone",
            [FlaskNames.Aquamarine] = "flask_utility_aquamarine",
            [FlaskNames.Stibnite] = "flask_utility_smoke",
            [FlaskNames.Sulphur] = "flask_utility_consecrate",
            [FlaskNames.Silver] = "flask_utility_haste",
            [FlaskNames.Bismuth] = "flask_utility_prismatic",


            [FlaskNames.BloodOfKarui] = "unique_flask_blood_of_the_karui",
            [FlaskNames.ZerphiLastBreath] = "unique_flask_zerphis_last_breath",
            [FlaskNames.DivinationDistillate] = "unique_flask_divination_distillate",
            [FlaskNames.CoruscatingElixir] = "unique_flask_chaos_damage_damages_es",
            [FlaskNames.TasteOfHate] = "unique_flask_taste_of_hate",
            [FlaskNames.KiaraDetermination] = "kiaras_determination",
            [FlaskNames.ForbiddenTaste] = "unique_flask_forbidden_taste",
            [FlaskNames.LionRoar] = "unique_flask_lions_roar",
            [FlaskNames.WitchfireBrew] = "unique_flask_witchfire_brew",
            [FlaskNames.AtziriPromise] = "unique_flask_atziris_promise",
            [FlaskNames.DyingSun] = "unique_flask_dying_sun",
            [FlaskNames.RumiConcoction] = "unique_flask_rumis_concoction",
            [FlaskNames.LaviangaSpirit] = "unique_flask_laviangas_cup",
            [FlaskNames.SorrowOfDivine] = "unique_flask_zealots_oath",
            [FlaskNames.OverflowingChalice] = "overflowing_chalice",
            [FlaskNames.SinRebirth] = "unholy_might_from_flask",
            [FlaskNames.VesselOfVinktar] = "lightning_flask",
            [FlaskNames.WiseOak] = "unique_flask_the_basics",
            [FlaskNames.CoralitoSignature] = "unique_flask_gorgon_poison",

            // No unique effects
            [FlaskNames.Rotgut] = " flask_utility_sprint",
            [FlaskNames.SoulCatcher] = "unique_flask_soul_catcher",
            [FlaskNames.DoedreElixir] = string.Empty,
            [FlaskNames.WrithingJar] = string.Empty,
        };
        private static readonly Dictionary<string, int> LinkRotationDictionary = new Dictionary<string, int>();

        public static class FlaskNames
        {
            public const string Diamond = "Diamond Flask";
            public const string Ruby = "Ruby Flask";
            public const string Sapphire = "Sapphire Flask";
            public const string Topaz = "Topaz Flask";
            public const string Granite = "Granite Flask";
            public const string Quicksilver = "Quicksilver Flask";
            public const string Amethyst = "Amethyst Flask";
            public const string Quartz = "Quartz Flask";
            public const string Jade = "Jade Flask";
            public const string Basalt = "Basalt Flask";
            public const string Aquamarine = "Aquamarine Flask";
            public const string Stibnite = "Stibnite Flask";
            public const string Sulphur = "Sulphur Flask";
            public const string Silver = "Silver Flask";
            public const string Bismuth = "Bismuth Flask";


            public const string BloodOfKarui = "Blood of the Karui";
            public const string DoedreElixir = "Doedre's Elixir";
            public const string ZerphiLastBreath = "Zerphi's Last Breath";
            public const string LaviangaSpirit = "Lavianga's Spirit";
            public const string DivinationDistillate = "Divination Distillate";
            public const string WrithingJar = "The Writhing Jar";
            public const string WiseOak = "The Wise Oak";
            public const string SinRebirth = "Sin's Rebirth";
            public const string CoruscatingElixir = "Coruscating Elixir";
            public const string TasteOfHate = "Taste of Hate";
            public const string KiaraDetermination = "Kiara's Determination";
            public const string ForbiddenTaste = "Forbidden Taste";
            public const string LionRoar = "Lion's Roar";
            public const string OverflowingChalice = "The Overflowing Chalice";
            public const string SorrowOfDivine = "The Sorrow of the Divine";
            public const string Rotgut = "Rotgut";
            public const string WitchfireBrew = "Witchfire Brew";
            public const string AtziriPromise = "Atziri's Promise";
            public const string DyingSun = "Dying Sun";
            public const string RumiConcoction = "Rumi's Concoction";
            public const string VesselOfVinktar = "Vessel of Vinktar";
            public const string CoralitoSignature = "Coralito's Signature";
            public const string SoulCatcher = "Soul Catcher";
        }

        public async Task<bool> Run()
        {
            if (!LokiPoe.IsInGame) return false;
            if (ShouldTeleport)
            {
                ShouldTeleport = false;
                if (FollowBot._leaderPartyEntry != null && FollowBot._leaderPartyEntry.PlayerEntry != null)
                {
                    var leadername = FollowBot._leaderPartyEntry.PlayerEntry.Name;
                    if (!string.IsNullOrEmpty(leadername))
                    {
                        await PartyHelper.FastGotoPartyZone(leadername);
                    }
                }
            }
            if (ShouldOpenPortal)
            {
                if (!World.CurrentArea.IsTown && !World.CurrentArea.IsHideoutArea)
                {
                    await PlayerAction.CreateTownPortal();
                }
                ShouldOpenPortal = false;
                return true;
            }
            if (LokiPoe.CurrentWorldArea.IsTown) return false;
            if (LokiPoe.CurrentWorldArea.Id == "HeistHub") return false;
            if (!LokiPoe.CurrentWorldArea.IsCombatArea) return false;
            
            if(LokiPoe.Me.HasAura("Grace Period"))
            {
                //await PlayerAction.MoveAway(15, 20);
            }

            var hpPct = LokiPoe.Me.HealthPercent;
            var esPct = LokiPoe.Me.EnergyShieldPercent;
            var manaPct = LokiPoe.Me.ManaPercent;

            #region Flasks
            foreach (var flask in FollowBotSettings.Instance.Flasks)
            {
                if (!flask.Enabled) continue;
                var postUseDelay = flask.PostUseDelay.ElapsedMilliseconds;
                if (postUseDelay < 100) continue;
                if (postUseDelay < flask.Cooldown) continue;
                var thisflask = FlaskHud.InventoryControl.Inventory.Items.FirstOrDefault(x => x.LocationTopLeft.X == flask.Slot - 1);
                if (thisflask == null) continue;
                if (thisflask.Name == Class.FlaskNames.Quicksilver && LokiPoe.Me.HasAura(QsilverEffect) && !flask.IgnoreEffect) continue;
                if (!flask.IgnoreEffect && !thisflask.Components.FlaskComponent.IsInstantRecovery && flask.PostUseDelay.ElapsedMilliseconds < thisflask.Components.FlaskComponent.RecoveryTime.TotalMilliseconds) continue;
                if (!flask.IgnoreEffect && HasFlaskEffect(thisflask)) continue;
                var threshold = flask.UseEs ? esPct : flask.UseMana ? manaPct : hpPct;
                if (threshold < flask.Threshold)
                {
                    if (UseFlask(thisflask, flask.Slot))
                    {
                        flask.PostUseDelay.Restart();
                        return true;                        
                    }
                }
            }
            #endregion

            #region Defense
            foreach (var skill in FollowBotSettings.Instance.DefensiveSkills)
            {
                if (!skill.Enabled) continue;
                if (!skill.IsReadyToCast) continue;
                var threshold = skill.UseEs ? esPct : hpPct;
                if (threshold < skill.Threshold)
                {
                    CastDefensiveSkill(skill);
                }
            }
            #endregion

            return false;
        }

        private bool HasFlaskEffect(Item thisflask)
        {
            string name = thisflask.ProperName();
            string effect = Flasks.GetEffect(name);
            if (effect == null) return false;
            return LokiPoe.Me.HasAura(effect);
        }

        private static void CastDefensiveSkill(DefensiveSkillsClass skillClass)
        {
            if (skillClass == null) return;
            var skills = LokiPoe.InGameState.SkillBarHud.Skills;
            if (skills == null) return;
            Skill skill = skills.FirstOrDefault(s => s.Name == skillClass.Name);
            if (skill != null && skill.CanUse())
            {
                if (skillClass.CastOnLeader)
                {
                    var leader = GetNextOnRotation(skillClass.LinkWhitelist);// FollowBot.Leader;
                    if (leader != null && leader.Distance < 50)
                    {
                        LokiPoe.InGameState.SkillBarHud.UseAt(skill.Slot, false, leader.Position, false);
                        skillClass.Casted();
                    }
                }
                else
                {
                    LokiPoe.InGameState.SkillBarHud.Use(skill.Slot, false, false);
                    skillClass.Casted();
                }
            }
        }

        private static bool UseFlask(Item thisflask, int slot)
        {
            if (!thisflask.CanUse) return false;
            //if (thisflask.Name == FlaskNames.Quicksilver)
            //{
            //    if (LokiPoe.Me.HasAura(QsilverEffect)) return false;
            //}
            //string name = thisflask.ProperName();
            //string effect = Flasks.GetEffect(name);
            //if (effect != null)
            //{
            //    if (LokiPoe.Me.HasAura(effect)) return false;
            //}

            if (!FlaskHud.UseFlaskInSlot(slot)) return false;
            return true;
        }

        private static Player GetNextOnRotation(string linkWhitelist)
        {
            List<PartyMember> party;
            if (string.IsNullOrEmpty(linkWhitelist))
            {
                return FollowBot.Leader;
            }
            else
            {
                var listSplit = linkWhitelist.Split(',');
                if (listSplit.Length < 1) 
                {
                    GlobalLog.Error("--------------------------------------------------------------------------------------------------------------------------------");
                    GlobalLog.Error(" ");
                    GlobalLog.Error("[DefenseAndFlaskTask] CastOnParty is selected, but the LinkWhitelist have some problems, pls check the Defensive skill setting.!");
                    GlobalLog.Error(" ");
                    GlobalLog.Error("--------------------------------------------------------------------------------------------------------------------------------");
                }
                var partyMembers = LokiPoe.InstanceInfo.PartyMembers;
                party = partyMembers.Where(x => x != null && x.PlayerEntry != null && listSplit.Contains(x.PlayerEntry.Name) && x.PlayerEntry.Name != LokiPoe.Me.Name && LokiPoe.InGameState.PartyHud.IsInSameZone(x.PlayerEntry.Name)).ToList();
                if (party.Count < 1)
                {
                    GlobalLog.Error("--------------------------------------------------------------------------------------------------------------------------------");
                    GlobalLog.Error(" ");
                    GlobalLog.Error($"[DefenseAndFlaskTask] CastOnParty is selected, but the LinkWhitelist have some problems (the name in the LinkWithelist do not match the party names [partyMembers = {partyMembers.Count}]), pls check the Defensive skill setting.!");
                    GlobalLog.Error(" ");
                    GlobalLog.Error("--------------------------------------------------------------------------------------------------------------------------------");
                }
                foreach (var partyMember in partyMembers)
                {
                    var name = partyMember?.PlayerEntry?.Name;
                    if (string.IsNullOrEmpty(name)) continue;
                    if (!LokiPoe.InGameState.PartyHud.IsInSameZone(name)) continue;                   
                    if (!LinkRotationDictionary.ContainsKey(name))
                    {
                        LinkRotationDictionary.Add(name, 0);
                    }                  
                }
                var nam = "";
                int weight = int.MaxValue;
                var keys = LinkRotationDictionary.Keys;
                foreach (var key in keys)
                {
                    if (LinkRotationDictionary[key] < weight) 
                    { 
                        weight = LinkRotationDictionary[key]; 
                        nam = key;
                    }
                }
                //var selected = LinkRotationDictionary.OrderBy(x => x.Value).First();
                LinkRotationDictionary[nam]++;
                var p = LokiPoe.ObjectManager.GetObjectsByType<Player>().FirstOrDefault(x => x.Name == nam);
                return p;
            }

        }

        #region Unused interface methods

        public MessageResult Message(Message message)
        {
            if (message.Id == Events.Messages.AreaChanged)
            {
                LinkRotationDictionary.Clear();
            }
            return MessageResult.Unprocessed;
        }

        public async Task<LogicResult> Logic(Logic logic)
        {
            return LogicResult.Unprovided;
        }

        public void Start()
        {

        }

        public void Tick()
        {
        }

        public void Stop()
        {
        }

        public string Name => "DefenseAndFlaskTask";
        public string Description => "This task provides a logic to use defensive skills and flasks.";
        public string Author => "Alcor75";
        public string Version => "1.0";

        #endregion
    }
}