1 files changed, 71 insertions, 0 deletions
diff --git a/src/command/commandHandler.hpp b/src/command/commandHandler.hpp
new file mode 100644
index 0000000..7fc20d5
--- /dev/null
+++ b/src/command/commandHandler.hpp
@@ -0,0 +1,71 @@
+#pragma once
+
+#include "../core/session.hpp"
+#include <any>
+#include <functional>
+#include <memory>
+#include <netinet/in.h>
+#include <string>
+#include <vector>
+
+class CCommandHandler {
+  public:
+	CCommandHandler();
+	~CCommandHandler();
+
+	struct SResult {
+		std::string result;
+		bool		good = false;
+	};
+
+	struct SParseResult {
+		std::any data;
+		bool	 good = false;
+	};
+
+	struct SCommand {
+		bool												 requireAdmin = true;
+		std::string											 name;
+		std::function<SParseResult(const std::string &args)> parser;
+		std::function<SResult(const std::any &parsed)>		 exe;
+	};
+
+	bool isCommand(const std::string &str) const;
+	bool validCommand(const std::string &command) const;
+	// return false if command struct is invalid
+	bool	registerCommand(const SCommand &command);
+	SResult newCommand(const std::string &command, const std::string &args, std::shared_ptr<CSession> session) const;
+	void	handleCommand(std::string input, std::shared_ptr<CSession> user = nullptr) const;
+
+	template <typename T>
+	struct SMakeCommandArgs {
+		std::string												   name;
+		std::function<std::pair<T, bool>(const std::string &args)> parser;
+		std::function<SResult(const T &parsed)>					   exe;
+	};
+
+	template <typename T>
+	static SCommand makeCommand(SMakeCommandArgs<T> args) {
+		return {
+			.name	= std::move(args.name),
+			.parser = [parser = std::move(args.parser)](const std::string &args) -> SParseResult {
+				auto [data, good] = parser(args);
+				return {.data = std::any(std::move(data)), .good = good};
+			},
+			.exe = [exe = std::move(args.exe)](const std::any &parsed) -> SResult {
+				try {
+					return exe(std::any_cast<const T &>(parsed));
+				} catch (const std::bad_any_cast &) {
+					return {.result = "Invalid data type", .good = false};
+				}
+			}
+		};
+	}
+
+  private:
+	std::shared_ptr<CCommandHandler::SCommand> getCommand(const std::string &command) const;
+	SResult									   exeCommand(const std::shared_ptr<SCommand> command, const std::string &args) const;
+
+	std::vector<std::shared_ptr<SCommand>> m_vCommands;
+};
+inline std::unique_ptr<CCommandHandler> g_pCommandHandler;