/*
 * Author:          Patrick-Christopher Mattulat
 * Company:         Lynar Studios
 * E-Mail:          webmaster@lynarstudios.com
 * Created:         2020-09-05
 * Changed:         2022-05-05
 *
 * */

#ifndef LS_STD_STATE_MACHINE_HPP
#define LS_STD_STATE_MACHINE_HPP

#include <memory>
#include <unordered_map>
#include <string>
#include <vector>
#include <ls_std/core/Class.hpp>
#include "State.hpp"
#include "StateMachineTypes.hpp"

namespace ls
{
  class StateMachine : public ls::Class
  {
    public:

      explicit StateMachine(const ::std::string& _name);
      ~StateMachine() override = default;

      bool addState(const ::std::shared_ptr<ls::State> &_state);
      ::std::shared_ptr<ls::State> getCurrentState();
      ::std::vector<ls::StateId> getMemory();
      ::std::string getName();
      ::std::unordered_map<StateId, ::std::shared_ptr<ls::State>> getStates();
      bool hasState(const ls::StateId &_id);
      bool proceed();
      void setMemory(const ::std::vector<ls::StateId>& _memory);
      void setName(const ::std::string& _name);
      bool setStartState(const ls::StateId &_id);

    private:

      ::std::shared_ptr<State> currentState{};
      ::std::vector<ls::StateId> memory{};
      ::std::string name{};
      ::std::unordered_map<ls::StateId, ::std::shared_ptr<ls::State>> states{};

      void _assignMemory(const ::std::vector<ls::StateId>& _memory);
      void _assignName(const ::std::string& _name);
      ::std::vector<ls::StateId> _getNextValidStates();
      void _remember(const ls::StateId &_id);
      bool _hasState(const ls::StateId &_id);
  };
}

#endif