/*
 * Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU GPL v2 license, you may redistribute it and/or modify it under version 2 of the License, or (at your option), any later version.
 */

#include "PartyMemberToDispel.h"
#include "Playerbots.h"

class PartyMemberToDispelPredicate : public FindPlayerPredicate, public PlayerbotAIAware
{
    public:
        PartyMemberToDispelPredicate(PlayerbotAI* botAI, uint32 dispelType) : PlayerbotAIAware(botAI), FindPlayerPredicate(), dispelType(dispelType) { }

        bool Check(Unit* unit) override
        {
            return unit->IsAlive() && botAI->HasAuraToDispel(unit, dispelType);
        }

    private:
        uint32 dispelType;
};

Unit* PartyMemberToDispel::Calculate()
{
    uint32 dispelType = atoi(qualifier.c_str());

    PartyMemberToDispelPredicate predicate(botAI, dispelType);
    return FindPartyMember(predicate);
}