#include "../../sc_defines.h"
#include "../../../../../game/Player.h"

#define SAY_FREED   "Thank you for freeing me!"

struct MANGOS_DLL_DECL mob_enslaved_netherwing_drakeAI : public ScriptedAI
{
    mob_enslaved_netherwing_drakeAI(Creature *c) : ScriptedAI(c) {EnterEvadeMode();}

    Player* pUnit;
    uint32 Invisible_Timer;
    bool Invisible;
    bool InCombat;
    bool Casted;

    void EnterEvadeMode()
    {
        pUnit = NULL;
        Invisible_Timer = NULL;
        Invisible = false;
        InCombat = false;
        Casted = false;
        m_creature->setFaction(62);
        m_creature->RemoveAllAuras();
        m_creature->DeleteThreatList();
        m_creature->CombatStop();
        DoGoHome();
    }

    void KilledUnit(Unit* Victim)
    {
        if (Victim->GetEntry() == 21718)
        {
            m_creature->Say(SAY_FREED, LANG_UNIVERSAL, pUnit->GetGUID());
            m_creature->InterruptSpell(CURRENT_GENERIC_SPELL);
            m_creature->setFaction(35);
            m_creature->SetFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_NOT_SELECTABLE);
            m_creature->SetUInt32Value(UNIT_FIELD_DISPLAYID,11686);
            pUnit->CastedCreatureOrGO(21722, m_creature->GetGUID(), 38762);
            InCombat = false;
            Invisible = true;
            Invisible_Timer = 120000;
        }
    }

    void AttackStart(Unit *who)
    {
        if (!who || Invisible)
            return;

        if (who->isTargetableForAttack() && who!= m_creature)
        {
            DoStartMeleeAttack(who);
        }
    }

    void MoveInLineOfSight(Unit *who)
    {
        if (!who || m_creature->getVictim() || Invisible)
            return;

        if (who->isTargetableForAttack() && who->isInAccessablePlaceFor(m_creature) && m_creature->IsHostileTo(who))
        {
            float attackRadius = m_creature->GetAttackDistance(who);
            if (m_creature->IsWithinDistInMap(who, attackRadius) && m_creature->GetDistanceZ(who) <= CREATURE_Z_ATTACK_RANGE && m_creature->IsWithinLOSInMap(who))
            {
                if(who->HasStealthAura())
                    who->RemoveSpellsCausingAura(SPELL_AURA_MOD_STEALTH);

                DoStartMeleeAttack(who);
                InCombat = true;
            }
        }
    }

    void SpellHit(Unit *caster, const SpellEntry *spell)
    {
        if ( m_creature->getFaction() == 62 && m_creature->isAlive() )
        {
            if (caster->GetTypeId() == TYPEID_PLAYER)
            {
                if ( !pUnit = ((Player*)caster) )
                {
                    if( ((Player*)caster)->GetQuestStatus(10854) == QUEST_STATUS_INCOMPLETE && spell->Id == 38762)
                    {
                        pUnit = ((Player*)caster);
                        m_creature->setFaction(((Player*)caster)->getFaction());
                    }
                }
            }
        }
        return;
    }
    void UpdateAI(const uint32 diff)
    {
        if (Invisible && Invisible_Timer < diff && !InCombat)
        {
            m_creature->setFaction(62);
            m_creature->RemoveFlag(UNIT_FIELD_FLAGS, UNIT_FLAG_NOT_SELECTABLE);
            m_creature->SetUInt32Value(UNIT_FIELD_DISPLAYID,19980);
            Invisible = false;
        }else Invisible_Timer -= diff;

        if (!m_creature->SelectHostilTarget() || !m_creature->getVictim() )
            return;

        DoMeleeAttackIfReady();
    }
};

CreatureAI* GetAI_mob_enslaved_netherwing_drake(Creature *_Creature)
{
    return new mob_enslaved_netherwing_drakeAI (_Creature);
}

void AddSC_mob_enslaved_netherwing_drake()
{
    Script *newscript;
    newscript = new Script;
    newscript->Name="mob_enslaved_netherwing_drake";
    newscript->GetAI = GetAI_mob_enslaved_netherwing_drake;
    m_scripts[nrscripts++] = newscript;
}