feat: message handlers

2 files changed, 25 insertions, 0 deletions

diff --git a/src/irc/client.hpp b/src/irc/client.hpp
index e6846d2..6f0791f 100644
--- a/src/irc/client.hpp
+++ b/src/irc/client.hpp
@@ -4,6 +4,8 @@
 
 #include <string>
 
+#include "message.hpp"
+
 namespace RedpilledBot {
   namespace IRC {
     class Client {
@@ -13,6 +15,17 @@ namespace RedpilledBot {
 
         void run();
 
+        template <MessageType T>
+        void on(typename MessageHandler<T>::fn function) {
+          switch (T) {
+            case Privmsg:
+              this->onPrivmsg = function;
+              break;
+            default:
+              break;
+          }
+        }
+
       private:
         void authorize();
 
@@ -23,6 +36,9 @@ namespace RedpilledBot {
         std::string port;
 
         ix::WebSocket websocket;
+
+        // Message handlers
+        typename MessageHandler<MessageType::Privmsg>::fn onPrivmsg;
     };
   }
 }
diff --git a/src/irc/message.hpp b/src/irc/message.hpp
index b0ae353..23bbe29 100644
--- a/src/irc/message.hpp
+++ b/src/irc/message.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include <functional>
 #include <optional>
 #include <sstream>
 #include <string>
@@ -94,5 +95,13 @@ namespace RedpilledBot {
       return std::nullopt;
     }
 
+    template <MessageType T>
+    struct MessageHandler;
+
+    template <>
+    struct MessageHandler<MessageType::Privmsg> {
+        using fn = std::function<void(Message<Privmsg> message)>;
+    };
+
   }
 }