CreatureLib/src/Battling/Models/Creature.hpp

#ifndef CREATURELIB_BATTLECREATURE_HPP
#define CREATURELIB_BATTLECREATURE_HPP

#include "../../GenericTemplates.hpp"
#include "../../Library/CreatureData/CreatureSpecies.hpp"
#include "../../Library/Items/Item.hpp"
#include "../ScriptHandling/ScriptAggregator.hpp"
#include "../ScriptHandling/ScriptSet.hpp"
#include "../ScriptHandling/ScriptSource.hpp"
#include "DamageSource.hpp"
#include "LearnedAttack.hpp"

namespace CreatureLib::Battling {
    // Forward declare battle class
    class Battle;
    class BattleSide;
    class BattleLibrary;

    class Creature : public ScriptSource {
    protected:
        const BattleLibrary* _library;

        GetProperty(const Library::CreatureSpecies*, Species);
        GetProperty(const Library::SpeciesVariant*, Variant);

        GetProperty(uint8_t, Level);
        GetProperty(uint32_t, Experience);
        GetProperty(uint32_t, UniqueIdentifier);
        GetProperty(Library::Gender, Gender);
        GetProperty(uint8_t, Coloring);
        GetProperty(const Library::Item*, HeldItem);
        GetProperty(uint32_t, CurrentHealth);

    protected:
        Core::StatisticSet<int8_t> _statBoost;
        Core::StatisticSet<uint32_t> _flatStats;
        Core::StatisticSet<uint32_t> _boostedStats;

        Battle* _battle;
        BattleSide* _side;
        bool _onBattleField = false;

        std::string _nickname = "";
        int8_t _talentIndex;
        Script* _activeTalent = nullptr;

        bool _hasOverridenTalent;
        std::string _overridenTalentName = "";
        std::unordered_set<Creature*> _seenOpponents = {};

        std::vector<LearnedAttack*> _attacks;

        Script* _status = nullptr;
        ScriptSet _volatile = {};

    private
        void OnFaint();

    public:
        Creature(const BattleLibrary* library, const Library::CreatureSpecies* species,
                 const Library::SpeciesVariant* variant, uint8_t level, uint32_t experience, uint32_t uid,
                 Library::Gender gender, uint8_t coloring, const Library::Item* heldItem, std::string nickname,
                 int8_t talent, std::vector<LearnedAttack*> attacks);

        virtual ~Creature() {
            for (auto attack : _attacks) {
                delete attack;
            }
        };

        void SetBattleData(Battle* battle, BattleSide* side);
        Battle* GetBattle() const;
        BattleSide* GetBattleSide() const;
        void SetOnBattleField(bool value) { _onBattleField = value; }
        bool IsOnBattleField() const { return _onBattleField; }

        const std::string& GetNickname() const;
        const std::string& GetActiveTalent() const;

        [[nodiscard]] bool IsFainted() const;
        [[nodiscard]] const std::vector<uint8_t>& GetTypes() const;
        [[nodiscard]] bool HasType(uint8_t type) const;

        void ChangeLevel(int8_t amount);
        void Damage(uint32_t damage, DamageSource source);
        void OverrideActiveTalent(const std::string& talent);
        void AddExperience(uint32_t amount);

        void MarkOpponentAsSeen(Creature* creature) { _seenOpponents.insert(creature); }
        const std::unordered_set<Creature*>& GetSeenOpponents() const { return _seenOpponents; }

        void GetActiveScripts(std::vector<ScriptWrapper>& scripts) override;
        void ClearVolatileScripts();

        std::vector<LearnedAttack*>& GetAttacks() { return _attacks; }

        const Library::CreatureSpecies* GetDisplaySpecies();
        const Library::SpeciesVariant* GetDisplayVariant();

        // region Stat APIs

        void ChangeStatBoost(Core::Statistic stat, int8_t diffAmount);
        [[nodiscard]] uint32_t GetFlatStat(Core::Statistic stat) const;
        [[nodiscard]] uint32_t GetBoostedStat(Core::Statistic stat) const;
        [[nodiscard]] uint32_t GetBaseStat(Core::Statistic stat) const;
        [[nodiscard]] int8_t GetStatBoost(Core::Statistic stat) const;
        void RecalculateFlatStats();
        void RecalculateBoostedStats();
        void RecalculateFlatStat(Core::Statistic);
        void RecalculateBoostedStat(Core::Statistic);

        // endregion
    };
}

#endif // CREATURELIB_CREATURE_HPP