#ifndef PKMNLIB_POKEMON_HPP
#define PKMNLIB_POKEMON_HPP

#include <Battling/Models/Creature.hpp>
#include <utility>
#include "../../Library/Statistic.hpp"
#include "../Library/BattleLibrary.hpp"
#include "LearnedMove.hpp"

namespace PkmnLib::Battling {
    class Pokemon : public CreatureLib::Battling::Creature {
    private:
        CreatureLib::Core::StatisticSet<uint8_t> _individualValues;
        CreatureLib::Core::StatisticSet<uint8_t> _effortValues;

        uint8_t _nature;
        const Library::Nature* _natureCache = nullptr;

        const BattleLibrary* GetLibrary() const { return reinterpret_cast<const BattleLibrary*>(_library); }

    public:
        Pokemon(const BattleLibrary* library, const Library::PokemonSpecies* species,
                const Library::PokemonForme* forme, uint8_t level, uint32_t experience, uint32_t uid,
                CreatureLib::Library::Gender gender, uint8_t coloring, const Library::Item* heldItem,
                const std::string& nickname, int8_t talent, std::vector<CreatureLib::Battling::LearnedAttack*> moves,
                CreatureLib::Core::StatisticSet<uint8_t> individualValues,
                CreatureLib::Core::StatisticSet<uint8_t> effortValues, uint8_t nature)
            : CreatureLib::Battling::Creature(library, species, forme, level, experience, uid, gender, coloring,
                                              heldItem, nickname, talent, std::move(moves)),
              _individualValues(individualValues), _effortValues(effortValues), _nature(nature) {}

        const Library::PokemonForme* GetForme() const {
            return dynamic_cast<const Library::PokemonForme*>(GetVariant());
        }

        const bool IsShiny() const { return _coloring == 1; }

        const std::vector<LearnedMove*>& GetMoves() const {
            return reinterpret_cast<const std::vector<LearnedMove*>&>(_attacks);
        }

        const Library::Nature& GetNature() const;
        uint8_t GetIndividualValue(CreatureLib::Core::Statistic stat) const { return _individualValues.GetStat(stat); }
        uint8_t GetEffortValue(CreatureLib::Core::Statistic stat) const { return _effortValues.GetStat(stat); }
    };
}

#endif // PKMNLIB_POKEMON_HPP