#ifndef CREATURELIB_BATTLECREATURE_HPP
#define CREATURELIB_BATTLECREATURE_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;

        const Library::CreatureSpecies* _species;
        const Library::SpeciesVariant* _variant;

        const Library::CreatureSpecies* _displaySpecies = nullptr;
        const Library::SpeciesVariant* _displayVariant = nullptr;

        uint8_t _level;
        uint32_t _experience;
        uint32_t _uniqueIdentifier;
        Library::Gender _gender;
        uint8_t _coloring;
        const Library::Item* _heldItem;
        uint32_t _currentHealth;

        Library::StatisticSet<int8_t> _statBoost;
        Library::StatisticSet<uint32_t> _flatStats;
        Library::StatisticSet<uint32_t> _boostedStats;

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

        std::string _nickname = "";
        CreatureLib::Library::TalentIndex _talentIndex;
        Script* _activeTalent = nullptr;

        bool _hasOverridenTalent;
        ConstString _overridenTalentName = ""_cnc;
        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,
                 const Library::TalentIndex& talent, std::vector<LearnedAttack*> attacks);

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

        virtual void Initialize() {
            RecalculateFlatStats();
            _currentHealth = GetBoostedStat(Library::Statistic::Health);
        }

        inline const Library::CreatureSpecies* GetSpecies() const { return _species; }
        inline const Library::SpeciesVariant* GetVariant() const { return _variant; }
        inline uint8_t GetLevel() const { return _level; }
        inline uint32_t GetExperience() const { return _experience; }
        inline Library::Gender GetGender() const { return _gender; }
        inline uint8_t GetColoring() const { return _coloring; }
        inline bool HasHeldItem(const ConstString& name) const {
            return _heldItem != nullptr && _heldItem->GetName() == name;
        }
        inline bool HasHeldItem(uint32_t nameHash) const {
            return _heldItem != nullptr && _heldItem->GetName() == nameHash;
        }
        inline const Library::Item* GetHeldItem() const { return _heldItem; }
        void SetHeldItem(const ConstString& itemName);
        void SetHeldItem(uint32_t itemNameHash);
        inline void SetHeldItem(const Library::Item* item) { _heldItem = item; };

        inline uint32_t GetCurrentHealth() const { return _currentHealth; }

        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 { return _nickname; }
        const ConstString& GetActiveTalent() const;

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

        uint32_t GetMaxHealth() const { return _boostedStats.GetHealth(); }
        void ChangeLevelBy(int8_t amount);
        void Damage(uint32_t damage, DamageSource source);
        void Heal(uint32_t amount, bool canRevive = false);
        void OverrideActiveTalent(const ConstString& 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();
        void AddVolatileScript(const ConstString& name);
        void AddVolatileScript(Script* script);
        void RemoveVolatileScript(const ConstString& name);
        void RemoveVolatileScript(Script* script);
        bool HasVolatileScript(const ConstString& name) const;

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

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

        const void SetDisplaySpecies(const Library::CreatureSpecies* species) { _displaySpecies = species; }
        const void SetDisplayVariant(const Library::SpeciesVariant* variant) { _displayVariant = variant; };

        // region Stat APIs

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

        // endregion
    };
}

#endif // CREATURELIB_CREATURE_HPP