stateMachine.h

#pragma once
#include<string>
#include<unordered_map>
 
namespace Aire {
    struct State {
    public:
        std::string name;
        std::unordered_map<std::string, State*> linkedStates;
 
        State(const std::string& name) : name(name) {}
 
        virtual void onActive()=0;
 
        void linkStateAsDst(State* dstState) {
            if (dstState) {
                linkedStates.insert({ dstState->name, dstState });
            }
        }
 
        virtual ~State() {}
    };
 
    struct StateMachine {
        State* currentState;
 
        StateMachine() : currentState(nullptr) {}
        StateMachine(State* currentState) : currentState(currentState) {}
 
        void transitionTo(std::string dstState) {
            if (currentState) {
                auto it = currentState->linkedStates.find(dstState);
                if (it != currentState->linkedStates.end()) {
                    currentState = it->second;
                    currentState->onActive();
                }
            }
        }
    };
}