CypherCore/Source/Scripts/Pets/Mage.cs

/*
 * Copyright (C) 2012-2020 CypherCore <http://github.com/CypherCore>
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

using Framework.Constants;
using Game.AI;
using Game.Entities;
using Game.Maps;
using Game.Scripting;
using System.Collections.Generic;
using System;

namespace Scripts.Pets
{
    namespace Mage
    {
        struct SpellIds
        {
            public const uint CloneMe = 45204;
            public const uint MastersThreatList = 58838;
            public const uint MageFrostBolt = 59638;
            public const uint MageFireBlast = 59637;
        }

        struct MiscConst
        {
            public const uint TimerMirrorImageInit = 0;
            public const uint TimerMirrorImageFrostBolt = 4000;
            public const uint TimerMirrorImageFireBlast = 6000;
        }

        [Script]
        class npc_pet_mage_mirror_image : ScriptedAI
        {
            const float CHASE_DISTANCE = 35.0f;

            uint _fireBlastTimer = 0;

            public npc_pet_mage_mirror_image(Creature creature) : base(creature) { }

            public override void InitializeAI()
            {
                Unit owner = me.GetOwner();
                if (owner == null)
                    return;

                // here mirror image casts on summoner spell (not present in client dbc) 49866
                // here should be auras (not present in client dbc): 35657, 35658, 35659, 35660 selfcast by mirror images (stats related?)
                // Clone Me!
                owner.CastSpell(me, SpellIds.CloneMe, true);
            }

            // custom UpdateVictim implementation to handle special target selection
            // we prioritize between things that are in combat with owner based on the owner's threat to them
            new bool UpdateVictim()
            {
                Unit owner = me.GetOwner();
                if (owner == null)
                    return false;

                if (!me.HasUnitState(UnitState.Casting) && !me.IsInCombat() && !owner.IsInCombat())
                    return false;

                Unit currentTarget = me.GetVictim();
                if (currentTarget && !CanAIAttack(currentTarget))
                {
                    me.InterruptNonMeleeSpells(true); // do not finish casting on invalid targets
                    me.AttackStop();
                    currentTarget = null;
                }

                // don't reselect if we're currently casting anyway
                if (currentTarget && me.HasUnitState(UnitState.Casting))
                    return true;

                Unit selectedTarget = null;
                var mgr = owner.GetCombatManager();
                if (mgr.HasPvPCombat())
                { // select pvp target
                    float minDistance = 0.0f;
                    foreach (var pair in mgr.GetPvPCombatRefs())
                    {
                        Unit target = pair.Value.GetOther(owner);
                        if (!target.IsPlayer())
                            continue;

                        if (!CanAIAttack(target))
                            continue;

                        float dist = owner.GetDistance(target);
                        if (!selectedTarget || dist < minDistance)
                        {
                            selectedTarget = target;
                            minDistance = dist;
                        }
                    }
                }

                if (!selectedTarget)
                { // select pve target
                    float maxThreat = 0.0f;
                    foreach (var pair in mgr.GetPvECombatRefs())
                    {
                        Unit target = pair.Value.GetOther(owner);
                        if (!CanAIAttack(target))
                            continue;

                        float threat = target.GetThreatManager().GetThreat(owner);
                        if (threat >= maxThreat)
                        {
                            selectedTarget = target;
                            maxThreat = threat;
                        }
                    }
                }

                if (!selectedTarget)
                {
                    EnterEvadeMode(EvadeReason.NoHostiles);
                    return false;
                }

                if (selectedTarget != me.GetVictim())
                    AttackStartCaster(selectedTarget, CHASE_DISTANCE);
                return true;
            }

            public override void UpdateAI(uint diff)
            {
                Unit owner = me.GetOwner();
                if (owner == null)
                {
                    me.DespawnOrUnsummon();
                    return;
                }

                if (_fireBlastTimer != 0)
                {
                    if (_fireBlastTimer <= diff)
                        _fireBlastTimer = 0;
                    else
                        _fireBlastTimer -= diff;
                }

                if (!UpdateVictim())
                    return;

                if (me.HasUnitState(UnitState.Casting))
                    return;

                if (_fireBlastTimer == 0)
                {
                    DoCastVictim(SpellIds.MageFireBlast);
                    _fireBlastTimer = MiscConst.TimerMirrorImageFireBlast;
                }
                else
                    DoCastVictim(SpellIds.MageFrostBolt);
            }

            public override bool CanAIAttack(Unit who)
            {
                Unit owner = me.GetOwner();
                return owner && who.IsAlive() &&
                    !who.HasBreakableByDamageCrowdControlAura() &&
                    who.IsInCombatWith(owner) && CanAIAttack(who);
            }

            // Do not reload Creature templates on evade mode enter - prevent visual lost
            public override void EnterEvadeMode(EvadeReason why)
            {
                if (me.IsInEvadeMode() || !me.IsAlive())
                    return;

                Unit owner = me.GetCharmerOrOwner();

                me.CombatStop(true);
                if (owner && !me.HasUnitState(UnitState.Follow))
                {
                    me.GetMotionMaster().Clear();
                    me.GetMotionMaster().MoveFollow(owner, SharedConst.PetFollowDist, me.GetFollowAngle());
                }
            }
        }
    }
}