/*
 * This file is part of the TrinityCore Project. See AUTHORS file for Copyright information
 *
 * 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 2 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/>.
 */

#include "GuardAI.h"
#include "ObjectAccessor.h"
#include "Player.h"
#include "Random.h"
#include "ScriptMgr.h"
#include "SpellInfo.h"
#include "CreatureAIImpl.h"

enum GuardMisc
{
    SAY_GUARD_SIL_AGGRO = 0,

    NPC_CENARION_HOLD_INFANTRY = 15184,
    NPC_STORMWIND_CITY_GUARD = 68,
    NPC_STORMWIND_CITY_PATROLLER = 1976,
    NPC_ORGRIMMAR_GRUNT = 3296,
    NPC_ALDOR_VINDICATOR = 18549,

    SPELL_BANISHED_SHATTRATH_A = 36642,
    SPELL_BANISHED_SHATTRATH_S = 36671,
    SPELL_BANISH_TELEPORT = 36643,
    SPELL_EXILE = 39533,
};

struct npc_guard_generic : public GuardAI
{
    npc_guard_generic(Creature* creature) : GuardAI(creature)
    {
        _scheduler.SetValidator([this]
        {
            return !me->HasUnitState(UNIT_STATE_CASTING) && !me->IsInEvadeMode() && me->IsAlive();
        });
        _combatScheduler.SetValidator([this]
        {
            return !me->HasUnitState(UNIT_STATE_CASTING);
        });
    }

    void Reset() override
    {
        _scheduler.CancelAll();
        _combatScheduler.CancelAll();
        _scheduler.Schedule(Seconds(1), [this](TaskContext context)
        {
            // Find a spell that targets friendly and applies an aura (these are generally buffs)
            if (SpellInfo const* spellInfo = SelectSpell(me, 0, 0, SELECT_TARGET_ANY_FRIEND, 0, 0, SELECT_EFFECT_AURA))
                DoCast(me, spellInfo->Id);

            context.Repeat(Minutes(10));
        });
    }

    void DoReplyToTextEmote(uint32 emote)
    {
        switch (emote)
        {
            case TEXT_EMOTE_KISS:
                me->HandleEmoteCommand(EMOTE_ONESHOT_BOW);
                break;
            case TEXT_EMOTE_WAVE:
                me->HandleEmoteCommand(EMOTE_ONESHOT_WAVE);
                break;
            case TEXT_EMOTE_SALUTE:
                me->HandleEmoteCommand(EMOTE_ONESHOT_SALUTE);
                break;
            case TEXT_EMOTE_SHY:
                me->HandleEmoteCommand(EMOTE_ONESHOT_FLEX);
                break;
            case TEXT_EMOTE_RUDE:
            case TEXT_EMOTE_CHICKEN:
                me->HandleEmoteCommand(EMOTE_ONESHOT_POINT);
                break;
            default:
                break;
        }
    }

    void ReceiveEmote(Player* player, uint32 textEmote) override
    {
        switch (me->GetEntry())
        {
            case NPC_STORMWIND_CITY_GUARD:
            case NPC_STORMWIND_CITY_PATROLLER:
            case NPC_ORGRIMMAR_GRUNT:
                break;
            default:
                return;
        }

        if (!me->IsFriendlyTo(player))
            return;

        DoReplyToTextEmote(textEmote);
    }

    void JustEngagedWith(Unit* who) override
    {
        if (me->GetEntry() == NPC_CENARION_HOLD_INFANTRY)
            Talk(SAY_GUARD_SIL_AGGRO, who);

        _combatScheduler.Schedule(Seconds(1), [this](TaskContext meleeContext)
        {
            Unit* victim = me->GetVictim();
            if (!me->isAttackReady() || !me->IsWithinMeleeRange(victim))
            {
                meleeContext.Repeat();
                return;
            }
            if (roll_chance_i(20))
            {
                if (SpellInfo const* spellInfo = SelectSpell(me->GetVictim(), 0, 0, SELECT_TARGET_ANY_ENEMY, 0, NOMINAL_MELEE_RANGE, SELECT_EFFECT_DONTCARE))
                {
                    me->resetAttackTimer();
                    DoCastVictim(spellInfo->Id);
                }
            }
            meleeContext.Repeat();
        }).Schedule(Seconds(5), [this](TaskContext spellContext)
        {
            bool healing = false;
            SpellInfo const* spellInfo = nullptr;

            // Select a healing spell if less than 30% hp and ONLY 33% of the time
            if (me->HealthBelowPct(30) && roll_chance_i(33))
                spellInfo = SelectSpell(me, 0, 0, SELECT_TARGET_ANY_FRIEND, 0, 0, SELECT_EFFECT_HEALING);

            // No healing spell available, check if we can cast a ranged spell
            if (spellInfo)
                healing = true;
            else
                spellInfo = SelectSpell(me->GetVictim(), 0, 0, SELECT_TARGET_ANY_ENEMY, NOMINAL_MELEE_RANGE, 0, SELECT_EFFECT_DONTCARE);

            // Found a spell
            if (spellInfo)
            {
                if (healing)
                    DoCast(me, spellInfo->Id);
                else
                    DoCastVictim(spellInfo->Id);
                spellContext.Repeat(Seconds(5));
            }
            else
                spellContext.Repeat(Seconds(1));
        });
    }

    void UpdateAI(uint32 diff) override
    {
        _scheduler.Update(diff);

        if (!UpdateVictim())
            return;

        _combatScheduler.Update(diff);
    }

private:
    TaskScheduler _scheduler;
    TaskScheduler _combatScheduler;
};

struct npc_guard_shattrath_faction : public GuardAI
{
    npc_guard_shattrath_faction(Creature* creature) : GuardAI(creature)
    {
        _scheduler.SetValidator([this]
        {
            return !me->HasUnitState(UNIT_STATE_CASTING);
        });
    }

    void Reset() override
    {
        _scheduler.CancelAll();
    }

    void JustEngagedWith(Unit* /*who*/) override
    {
        ScheduleVanish();
    }

    void UpdateAI(uint32 diff) override
    {
        if (!UpdateVictim())
            return;

        _scheduler.Update(diff);
    }

    void ScheduleVanish()
    {
        _scheduler.Schedule(Seconds(5), [this](TaskContext banishContext)
        {
            Unit* temp = me->GetVictim();
            if (temp && temp->GetTypeId() == TYPEID_PLAYER)
            {
                DoCast(temp, me->GetEntry() == NPC_ALDOR_VINDICATOR ? SPELL_BANISHED_SHATTRATH_S : SPELL_BANISHED_SHATTRATH_A);
                ObjectGuid playerGUID = temp->GetGUID();
                banishContext.Schedule(Seconds(9), [this, playerGUID](TaskContext /*exileContext*/)
                {
                    if (Unit* temp = ObjectAccessor::GetUnit(*me, playerGUID))
                    {
                        temp->CastSpell(temp, SPELL_EXILE, true);
                        temp->CastSpell(temp, SPELL_BANISH_TELEPORT, true);
                    }
                    ScheduleVanish();
                });
            }
            else
                banishContext.Repeat();
        });
    }

private:
    TaskScheduler _scheduler;
};

void AddSC_npc_guard()
{
    RegisterCreatureAI(npc_guard_generic);
    RegisterCreatureAI(npc_guard_shattrath_faction);
}