1 files changed, 155 insertions, 0 deletions
diff --git a/bot/src/irc/client.cpp b/bot/src/irc/client.cpp
new file mode 100644
index 0000000..018736e
--- /dev/null
+++ b/bot/src/irc/client.cpp
@@ -0,0 +1,155 @@
+#include "client.hpp"
+
+#include <ixwebsocket/IXWebSocketMessage.h>
+#include <ixwebsocket/IXWebSocketMessageType.h>
+
+#include <algorithm>
+#include <optional>
+#include <string>
+#include <vector>
+
+#include "../logger.hpp"
+#include "cpr/api.h"
+#include "cpr/cprtypes.h"
+#include "cpr/response.h"
+#include "message.hpp"
+#include "nlohmann/json.hpp"
+
+using namespace bot::irc;
+
+Client::Client(std::string client_id, std::string token) {
+  this->client_id = client_id;
+  this->token = token;
+
+  this->host = "wss://irc-ws.chat.twitch.tv";
+  this->port = "443";
+
+  this->websocket.setUrl(this->host + ":" + this->port);
+
+  // getting token owner
+  cpr::Response response = cpr::Get(
+      cpr::Url{"https://api.twitch.tv/helix/users"}, cpr::Bearer{this->token},
+      cpr::Header{{"Client-Id", this->client_id}});
+
+  if (response.status_code != 200) {
+    log::warn("IRC", "Failed to get bot username from Twitch API: " +
+                         std::to_string(response.status_code) + " " +
+                         response.status_line);
+  } else {
+    nlohmann::json j = nlohmann::json::parse(response.text);
+
+    auto d = j["data"][0];
+    this->id = std::stoi(d["id"].get<std::string>());
+    this->username = d["login"];
+  }
+}
+
+void Client::run() {
+  this->websocket.setOnMessageCallback(
+      [this](const ix::WebSocketMessagePtr &msg) {
+        switch (msg->type) {
+          case ix::WebSocketMessageType::Message: {
+            log::debug("IRC", "Received message: " + msg->str);
+
+            std::vector<std::string> lines =
+                utils::string::split_text(msg->str, '\n');
+
+            for (std::string &line : lines) {
+              line.erase(std::remove_if(line.begin(), line.end(),
+                                        [](char c) {
+                                          return c == '\n' || c == '\r' ||
+                                                 c == '\t';
+                                        }),
+                         line.end());
+
+              std::optional<MessageType> type = define_message_type(line);
+
+              if (!type.has_value()) {
+                break;
+              }
+
+              MessageType m_type = type.value();
+
+              if (m_type == MessageType::Privmsg) {
+                std::optional<Message<MessageType::Privmsg>> message =
+                    parse_message<MessageType::Privmsg>(line);
+
+                if (message.has_value()) {
+                  this->onPrivmsg(message.value());
+                }
+              }
+            }
+
+            break;
+          }
+          case ix::WebSocketMessageType::Open: {
+            log::info("IRC", "Connected to Twitch IRC");
+            this->is_connected = true;
+            this->authorize();
+            for (const auto &msg : this->pool) {
+              this->websocket.send(msg);
+            }
+            this->pool.clear();
+            break;
+          }
+          case ix::WebSocketMessageType::Close: {
+            log::info("IRC", "Twitch IRC connection closed");
+            this->is_connected = false;
+
+            for (const auto &x : this->joined_channels) {
+              this->raw("JOIN #" + x);
+            }
+
+            break;
+          }
+          default: {
+            break;
+          }
+        }
+      });
+
+  this->websocket.start();
+}
+
+void Client::say(const std::string &channel_login, const std::string &message) {
+  this->raw("PRIVMSG #" + channel_login + " :" + message);
+  log::debug("IRC", "Sent '" + message + "' in #" + channel_login);
+}
+
+bool Client::join(const std::string &channel_login) {
+  auto already_joined =
+      std::any_of(this->joined_channels.begin(), this->joined_channels.end(),
+                  [&](const auto &x) { return x == channel_login; });
+
+  if (!already_joined) {
+    this->raw("JOIN #" + channel_login);
+    this->joined_channels.push_back(channel_login);
+    log::info("IRC", "Joined #" + channel_login);
+  }
+
+  return !already_joined;
+}
+
+void Client::raw(const std::string &raw_message) {
+  std::string msg = raw_message + "\r\n";
+  if (this->is_connected) {
+    this->websocket.send(msg);
+  } else {
+    this->pool.push_back(msg);
+  }
+}
+
+void Client::authorize() {
+  if (this->username.empty() || this->token.empty()) {
+    log::error("IRC", "Bot username and token must be set for authorization!");
+    return;
+  }
+
+  log::info("IRC", "Authorizing on Twitch IRC servers...");
+
+  this->raw("PASS oauth:" + this->token);
+  this->raw("NICK " + this->username);
+  this->raw("CAP REQ :twitch.tv/membership");
+  this->raw("CAP REQ :twitch.tv/commands");
+  this->raw("CAP REQ :twitch.tv/tags");
+}