From e39d90442dfd6fb136cda23b0d013da98fb089e9 Mon Sep 17 00:00:00 2001
From: ilotterytea <iltsu@alright.party>
Date: Sat, 5 Jul 2025 18:12:43 +0500
Subject: feat: parse irc messages fully

---
 bot/src/irc/message.cpp | 78 +++++++++++++++++++++++++++++++++++++------------
 1 file changed, 59 insertions(+), 19 deletions(-)

(limited to 'bot/src/irc/message.cpp')

diff --git a/bot/src/irc/message.cpp b/bot/src/irc/message.cpp
index 038df83..43590e5 100644
--- a/bot/src/irc/message.cpp
+++ b/bot/src/irc/message.cpp
@@ -4,29 +4,69 @@
 #include <string>
 #include <vector>
 
-namespace bot {
-  namespace irc {
-    std::optional<MessageType> define_message_type(const std::string &msg) {
-      std::vector<std::string> parts = utils::string::split_text(msg, ' ');
-      int i;
-
-      if (msg[0] == '@') {
-        i = 2;
-      } else if (msg[0] == ':') {
-        i = 1;
-      } else {
-        return std::nullopt;
+namespace bot::irc {
+  std::optional<MessageType> define_message_type(const std::string &msg) {
+    std::vector<std::string> parts = utils::string::split_text(msg, ' ');
+    int i;
+
+    if (msg[0] == '@') {
+      i = 2;
+    } else if (msg[0] == ':') {
+      i = 1;
+    } else {
+      return std::nullopt;
+    }
+
+    if (parts[i] == "NOTICE") {
+      return MessageType::Notice;
+    } else if (parts[i] == "PRIVMSG") {
+      return MessageType::Privmsg;
+    } else if (parts[i] == "PING") {
+      return MessageType::Ping;
+    }
+
+    return std::nullopt;
+  }
+
+  std::optional<IRCMessage> IRCMessage::from_string(std::string msg) {
+    IRCMessage m;
+
+    if (msg[0] == '@') {
+      int end = msg.find(' ');
+      std::string raw_tags = msg.substr(1, end);
+      for (const std::string tag : utils::string::split_text(raw_tags, ';')) {
+        auto splitted_tag = utils::string::split_text_n(tag, "=", 1);
+        m.tags.insert_or_assign(splitted_tag[0], splitted_tag[1]);
       }
+      msg = msg.substr(end + 1);
+    }
 
-      if (parts[i] == "NOTICE") {
-        return MessageType::Notice;
-      } else if (parts[i] == "PRIVMSG") {
-        return MessageType::Privmsg;
-      } else if (parts[i] == "PING") {
-        return MessageType::Ping;
+    if (msg[0] == ':') {
+      int end = msg.find(' ');
+      m.prefix = msg.substr(1, end);
+      int bang = msg.find('!');
+      if (bang != std::string::npos) {
+        m.nick = m.prefix.substr(0, bang - 1);
       }
+      msg = msg.substr(end + 1);
+    }
 
-      return std::nullopt;
+    std::vector<std::string> parts = utils::string::split_text(msg, ' ');
+
+    m.command = parts.front();
+    parts.erase(parts.begin());
+
+    for (int i = 0; i < parts.size(); i++) {
+      if (parts[i][0] == ':') {
+        parts = std::vector<std::string>(parts.begin() + i, parts.end());
+        m.params.push_back(utils::string::join_vector(parts, ' ').substr(1));
+        break;
+      }
+      if (!parts[i].empty()) {
+        m.params.push_back(parts[i]);
+      }
     }
+
+    return m;
   }
 }
-- 
cgit v1.2.3