CreatureLib/src/Battling/ScriptHandling/ScriptSet.hpp

#ifndef CREATURELIB_SCRIPTSET_HPP
#define CREATURELIB_SCRIPTSET_HPP

#include <any>
#include <unordered_map>
#include "Script.hpp"

namespace CreatureLib::Battling {
    class ScriptSet {
        std::vector<Script*> _scripts;
        std::unordered_map<uint32_t, size_t> _lookup;

    public:
        ~ScriptSet() {
            for (auto s : _scripts) {
                delete s;
            }
        }

        void Add(Script* script) {
            auto f = _lookup.find(script->GetName());
            if (f != _lookup.end()) {
                _scripts[f.operator*().second]->Stack();
                delete script;
                return;
            }
            _scripts.push_back(script);
            _lookup.insert({script->GetName(), _scripts.size() - 1});
        }

        Script* Get(const ConstString& key) const { return Get(key.GetHash()); }

        Script* Get(uint32_t keyHash) const {
            auto f = _lookup.find(keyHash);
            if (f != _lookup.end()) {
                return _scripts[f->second];
            }
            return nullptr;
        }

        void Remove(const ConstString& key) { Remove(key.GetHash()); }

        void Remove(uint32_t keyHash) {
            auto find = _lookup.find(keyHash);
            if (find != _lookup.end()) {
                auto script = _scripts[find->second];
                script->OnRemove();
                delete script;
                _scripts.erase(_scripts.begin() + find.operator*().second);
                _lookup.erase(keyHash);
            }
        }

        void Clear() {
            for (auto s : _scripts) {
                delete s;
            }
            _scripts.clear();
            _lookup.clear();
        }

        bool Has(const ConstString& key) const { return _lookup.find(key) != _lookup.end(); }

        bool Has(uint32_t keyHash) const { return _lookup.find(keyHash) != _lookup.end(); }

        size_t Count() const { return _scripts.size(); }

        const std::vector<Script*>* GetIterator() const { return &_scripts; }
    };
}

#endif // CREATURELIB_SCRIPTSET_HPP