CreatureLib/src/Battling/ScriptHandling/ScriptAggregator.hpp

#ifndef CREATURELIB_SCRIPTAGGREGATOR_HPP
#define CREATURELIB_SCRIPTAGGREGATOR_HPP

#include <any>
#include <queue>
#include "Script.hpp"
#include "ScriptSet.hpp"
#include "../../Core/Exceptions/NotReachableException.hpp"
#include "ScriptWrapper.hpp"

namespace CreatureLib::Battling{
    class ScriptAggregator{
        std::vector<ScriptWrapper> _scripts;
        int _index = 0;
        bool _isSetSet = false;
        std::__detail::_Node_const_iterator<std::pair<const std::string, Script *>, false, true> _setIterator;
        std::__detail::_Node_const_iterator<std::pair<const std::string, Script *>, false, true> _setEnd;
    public:
        ScriptAggregator(std::vector<ScriptWrapper> scripts) : _scripts(std::move(scripts)){
        };

        bool HasNext(){
            return _index < _scripts.size() || (_isSetSet && _setIterator != _setEnd);
        }

        Script* GetNext(){
            // We can probably do this in a cleaner version once C++ 20 drops with Coroutine support.
            if (_isSetSet){
                if (_setIterator == _setEnd){
                    _isSetSet = false;
                    return GetNext();
                }
                auto s = _setIterator->second;
                _setIterator.operator++();
                if (_setIterator == _setEnd){
                    _isSetSet = false;
                }
                return s;
            }
            if (_index >= _scripts.size())
                return nullptr;
            auto next = _scripts[_index];
            _index++;
            if (!next.IsSet()){
                auto scriptPtr = next.GetScript();
                if (scriptPtr == nullptr)
                    return GetNext();
                return *scriptPtr;
            }
            else{
                auto set = next.GetScriptSet();
                if (set->Count() == 0)
                    return GetNext();
                auto it = set->GetIterator();
                _setIterator = it->begin();
                _setEnd = it->end();
                _isSetSet = true;
                return GetNext();
            }
            throw NotReachableException();
        }
    };
}

#endif //CREATURELIB_SCRIPTAGGREGATOR_HPP