#ifndef PKMNLIB_BATTLE_HPP
#define PKMNLIB_BATTLE_HPP

#include <CreatureLib/Battling/Models/Battle.hpp>
#include "../EventHooks/WeatherChangeEvent.hpp"
#include "../Library/BattleLibrary.hpp"
#include "../PkmnScriptCategory.hpp"

namespace PkmnLib::Battling {
    class Battle : public CreatureLib::Battling::Battle {
    private:
        CreatureLib::Battling::Script* _weatherScript = nullptr;

    public:
        Battle(const BattleLibrary* library, const ArbUt::List<CreatureLib::Battling::BattleParty*>& parties,
               bool canFlee = true, uint8_t numberOfSides = 2, uint8_t creaturesPerSide = 1)
            : CreatureLib::Battling::Battle(library, parties, canFlee, numberOfSides, creaturesPerSide) {}

        void SetWeather(const ArbUt::CaseInsensitiveConstString& name);
        void ClearWeather();
        const ArbUt::CaseInsensitiveConstString& GetWeatherName() noexcept { return _weatherScript->GetName(); }

        size_t ScriptCount() const override { return CreatureLib::Battling::Battle::ScriptCount() + 1; }
        void GetActiveScripts(ArbUt::List<CreatureLib::Battling::ScriptWrapper>& scripts) override {
            CreatureLib::Battling::Battle::GetActiveScripts(scripts);
            scripts.Append(CreatureLib::Battling::ScriptWrapper(
                CreatureLib::Battling::ScriptWrapper::FromScript(&_weatherScript)));
        }
    };
}

#endif // PKMNLIB_BATTLE_HPP