PkmnLib/src/Library/Species/SpeciesLibrary.hpp

#ifndef PKMNLIB_SPECIESLIBRARY_HPP
#define PKMNLIB_SPECIESLIBRARY_HPP

#include <CreatureLib/Library/SpeciesLibrary.hpp>
#include "PokemonSpecies.hpp"

namespace PkmnLib::Library {
    class SpeciesLibrary : public CreatureLib::Library::SpeciesLibrary {
    public:
        inline bool TryGet(const ArbUt::BasicStringView& name,
                           ArbUt::BorrowedPtr<const PokemonSpecies>& outSpecies) const {
            auto v = outSpecies.As<const PokemonSpecies::CreatureSpecies>();
            auto res = CreatureLib::Library::SpeciesLibrary::TryGet(name, v);
            outSpecies = v.As<const PokemonSpecies>();
            return res;
        }

        inline ArbUt::BorrowedPtr<const PokemonSpecies> Get(const ArbUt::BasicStringView& name) const {
            return CreatureLib::Library::SpeciesLibrary::Get(name).As<const PokemonSpecies>();
        }

        ArbUt::BorrowedPtr<const PokemonSpecies> operator[](const ArbUt::BasicStringView& name) const {
            return Get(name);
        }

        void Insert(const ArbUt::StringView& name, const PokemonSpecies* species) {
            CreatureLib::Library::SpeciesLibrary::Insert(name, species);
        }
    };
}

#endif // PKMNLIB_SPECIESLIBRARY_HPP