From b4fca15aec26332c1315c8c3030169bbf342da3e Mon Sep 17 00:00:00 2001
From: ilotterytea <iltsu@alright.party>
Date: Sun, 21 Apr 2024 03:25:12 +0500
Subject: feat: command loader

---
 src/commands/command.cpp | 30 ++++++++++++++++++++++++++++++
 src/commands/command.hpp | 18 ++++++++++++++++++
 2 files changed, 48 insertions(+)
 create mode 100644 src/commands/command.cpp

(limited to 'src/commands')

diff --git a/src/commands/command.cpp b/src/commands/command.cpp
new file mode 100644
index 0000000..b6852cc
--- /dev/null
+++ b/src/commands/command.cpp
@@ -0,0 +1,30 @@
+#include "command.hpp"
+
+#include <iostream>
+#include <memory>
+#include <optional>
+
+#include "../modules/ping.hpp"
+
+namespace bot {
+  namespace command {
+    CommandLoader::CommandLoader() {
+      this->add_command(std::make_unique<mod::Ping>());
+    }
+
+    void CommandLoader::add_command(std::unique_ptr<Command> command) {
+      this->commands.push_back(std::move(command));
+    }
+
+    std::optional<std::variant<std::vector<std::string>, std::string>>
+    CommandLoader::run(const irc::Message<irc::MessageType::Privmsg> &msg) {
+      for (const std::unique_ptr<Command> &command : this->commands) {
+        if (command->get_name() == msg.message) {
+          return command->run(msg);
+        }
+      }
+
+      return std::nullopt;
+    }
+  }
+}
diff --git a/src/commands/command.hpp b/src/commands/command.hpp
index 6e05779..6efe505 100644
--- a/src/commands/command.hpp
+++ b/src/commands/command.hpp
@@ -1,3 +1,7 @@
+#pragma once
+
+#include <memory>
+#include <optional>
 #include <string>
 #include <variant>
 #include <vector>
@@ -7,9 +11,23 @@
 namespace bot {
   namespace command {
     class Command {
+      public:
         virtual std::string get_name() = 0;
         virtual std::variant<std::vector<std::string>, std::string> run(
             const irc::Message<irc::MessageType::Privmsg> &msg) = 0;
     };
+
+    class CommandLoader {
+      public:
+        CommandLoader();
+        ~CommandLoader() = default;
+
+        void add_command(std::unique_ptr<Command> cmd);
+        std::optional<std::variant<std::vector<std::string>, std::string>> run(
+            const irc::Message<irc::MessageType::Privmsg> &msg);
+
+      private:
+        std::vector<std::unique_ptr<Command>> commands;
+    };
   }
 }
-- 
cgit v1.2.3