1 files changed, 67 insertions, 0 deletions
diff --git a/src/state/basic_state.hpp b/src/state/basic_state.hpp
new file mode 100644
index 0000000..4529338
--- /dev/null
+++ b/src/state/basic_state.hpp
@@ -0,0 +1,67 @@
+#pragma once
+
+#include "events.hpp"
+#include "interfaces/mount_point_state_machine.hpp"
+
+struct BasicState
+{
+    BasicState(interfaces::MountPointStateMachine& machine) : machine{machine}
+    {
+    }
+    virtual ~BasicState() = default;
+
+    BasicState(const BasicState& state) = delete;
+    BasicState(BasicState&&) = delete;
+
+    BasicState& operator=(const BasicState&) = delete;
+    BasicState& operator=(BasicState&& state) = delete;
+
+    virtual std::unique_ptr<BasicState> handleEvent(Event event) = 0;
+    virtual std::unique_ptr<BasicState> onEnter() = 0;
+    virtual std::string_view getStateName() const = 0;
+
+    template <class T>
+    T* get_if()
+    {
+        if (getStateName() == T::stateName())
+        {
+            return static_cast<T*>(this);
+        }
+        return nullptr;
+    }
+
+    interfaces::MountPointStateMachine& machine;
+};
+
+template <class T>
+struct BasicStateT : public BasicState
+{
+    BasicStateT(interfaces::MountPointStateMachine& machine) :
+        BasicState(machine)
+    {
+    }
+
+    ~BasicStateT()
+    {
+        LogMsg(Logger::Debug, "cleaning state: ", T::stateName());
+    }
+
+    std::unique_ptr<BasicState> onEnter() override
+    {
+        return nullptr;
+    }
+
+    std::unique_ptr<BasicState> handleEvent(Event event) override final
+    {
+        return std::visit(
+            [this](auto e) {
+                return static_cast<T*>(this)->handleEvent(std::move(e));
+            },
+            std::move(event));
+    }
+
+    std::string_view getStateName() const override final
+    {
+        return T::stateName();
+    }
+};