#ifndef CREATURELIB_ATTACKTURNCHOICE_HPP
#define CREATURELIB_ATTACKTURNCHOICE_HPP

#include "../Models/Battle.hpp"
#include "../Models/CreatureIndex.hpp"
#include "../Models/LearnedAttack.hpp"
#include "../ScriptHandling/ScriptCategory.hpp"
#include "BaseTurnChoice.hpp"

namespace CreatureLib::Battling {
    class AttackTurnChoice : public BaseTurnChoice {
        LearnedAttack* _attack;
        CreatureIndex _target;
        Script* _attackScript = nullptr;

        void ResolveScript() {
            if (_attackScript != nullptr)
                return;
            auto user = GetUser();
            if (user == nullptr)
                return;
            auto battle = user->GetBattle();
            if (battle != nullptr) {
                auto library = battle->GetLibrary();
                _attackScript = library->LoadScript(ScriptCategory::Attack, _attack->GetAttack()->GetName());
            }
        }

    public:
        AttackTurnChoice(Creature* user, LearnedAttack* attack, const CreatureIndex& target)
            : BaseTurnChoice(user), _attack(attack), _target(target) {
            ResolveScript();
        }
        AttackTurnChoice(Creature* user, LearnedAttack* attack, const CreatureIndex& target, Script* script)
            : BaseTurnChoice(user), _attack(attack), _target(target), _attackScript(script) {}

        inline LearnedAttack* GetAttack() const { return _attack; }

        TurnChoiceKind GetKind() const override { return TurnChoiceKind ::Attack; }

        int8_t GetPriority() const {
            // HOOK: Change priority
            return _attack->GetAttack()->GetPriority();
        }

        const CreatureIndex& GetTarget() const { return _target; }

        Script* GetAttackScript() const { return _attackScript; }

    protected:
        void GetActiveScripts(std::vector<ScriptWrapper>& scripts) override {
            scripts.emplace_back(&_attackScript);
            GetUser()->GetActiveScripts(scripts);
        }
    };
}

#endif // CREATURELIB_ATTACKTURNCHOICE_HPP