1 files changed, 32 insertions, 0 deletions

diff --git a/src/FSM.hpp b/src/FSM.hpp
new file mode 100644
index 0000000..9570e26
--- /dev/null
+++ b/src/FSM.hpp
@@ -0,0 +1,32 @@
+#pragma once
+
+#include <utility>
+#include <functional>
+#include <map>
+
+template<class T>
+class FSM {
+	T previousState;
+	T &state;
+	std::map<Transaction, std::function<void()>> handlers;
+public:
+	using Transaction = std::pair<T, T>;
+
+	FSM(T &value) : state(value), previousState(value) {}
+
+	~FSM() = default;
+
+	void Update() {
+		auto handler = handlers[Transaction{previousState, state}];
+		if (handler)
+			handler();
+	}
+
+	void RegisterHandler(T state, std::function<void()> handler) {
+		handlers[Transaction{state, state}] = handler;
+	}
+
+	void RegisterTransactionHandler(Transaction transaction, std::function<void()> handler) {
+		handlers[transaction] = handler;
+	}
+};
\ No newline at end of file