#ifndef PKMNLIB_POKEMONSPECIES_HPP
#define PKMNLIB_POKEMONSPECIES_HPP
#include <Arbutils/Memory/UniquePtrList.hpp>
#include <CreatureLib/Battling/Models/Creature.hpp>
#include "../Evolutions/EvolutionData.hpp"
#include "PokemonForme.hpp"

namespace PkmnLib::Library {
    class PokemonSpecies : public CreatureLib::Library::CreatureSpecies {
    private:
        uint8_t _baseHappiness;
        ArbUt::UniquePtrList<const EvolutionData> _evolutions;
        ArbUt::List<ArbUt::StringView> _eggGroups;

    public:
        PokemonSpecies(uint16_t id, const ArbUt::StringView& name, const PokemonForme* defaultForme, float genderRatio,
                       const ArbUt::StringView& growthRate, uint8_t captureRate, uint8_t baseHappiness,
                       const ArbUt::List<ArbUt::StringView>& eggGroups, std::unordered_set<uint32_t> flags = {}) noexcept
            : CreatureSpecies(id, name, defaultForme, genderRatio, growthRate, captureRate, flags),
              _baseHappiness(baseHappiness), _eggGroups(eggGroups) {}

        ~PokemonSpecies() override = default;

        inline uint8_t GetBaseHappiness() const { return _baseHappiness; }

        inline ArbUt::BorrowedPtr<const PokemonForme> GetDefaultForme() const {
            return CreatureSpecies::GetVariant("default"_cnc.GetHash()).As<const PokemonForme>();
        }

        inline bool HasForme(const ArbUt::BasicStringView& key) const { return HasVariant(key); }

        inline bool TryGetForme(const ArbUt::BasicStringView& key,
                                ArbUt::BorrowedPtr<const PokemonForme>& forme) const {
            auto v = forme.As<const PokemonForme::SpeciesVariant>();
            auto res = TryGetVariant(key, v);
            forme = v.As<const PokemonForme>();
            return res;
        }

        inline ArbUt::BorrowedPtr<const PokemonForme> GetForme(const ArbUt::BasicStringView& key) const {
            return CreatureSpecies::GetVariant(key).As<const PokemonForme>();
        }

        inline void AddEvolution(const EvolutionData* data) noexcept { _evolutions.Append(data); }
        const ArbUt::UniquePtrList<const EvolutionData>& GetEvolutions() const noexcept { return _evolutions; }

        bool HasEggGroup(const ArbUt::StringView& sv) const noexcept { return _eggGroups.Contains(sv); }
        const ArbUt::List<ArbUt::StringView>& GetEggGroups() const noexcept { return _eggGroups; }
    };
}

#endif // PKMNLIB_POKEMONSPECIES_HPP