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

#ifndef _PLAYERBOT_PARTYMEMBERVALUE_H
#define _PLAYERBOT_PARTYMEMBERVALUE_H

#include "Player.h"
#include "Value.h"

class PlayerbotAI;

class FindPlayerPredicate
{
public:
    virtual ~FindPlayerPredicate() = default;
    virtual bool Check(Unit* /*unit*/) = 0;
};

class SpellEntryPredicate
{
public:
    virtual bool Check(SpellInfo const* /*spellInfo*/) = 0;
};

class PartyMemberValue : public UnitCalculatedValue
{
public:
    PartyMemberValue(PlayerbotAI* botAI, std::string const name = "party member", int checkInterval = 1)
        : UnitCalculatedValue(botAI, name, checkInterval)
    {
    }

    bool IsTargetOfSpellCast(Player* target, SpellEntryPredicate& predicate);

protected:
    Unit* FindPartyMember(FindPlayerPredicate& predicate, bool ignoreOutOfGroup = false);
    Unit* FindPartyMember(std::vector<Player*>* party, FindPlayerPredicate& predicate);
    virtual bool Check(Unit* player);
};

class PartyMemberMainTankValue : public PartyMemberValue
{
public:
    PartyMemberMainTankValue(PlayerbotAI* botAI) : PartyMemberValue(botAI, "main tank member", 2 * 1000) {}
    virtual Unit* Calculate();
};

#endif
