feat: MARIADB SUPPORT!!!!

1 files changed, 170 insertions, 0 deletions

diff --git a/bot/src/commands/request.cpp b/bot/src/commands/request.cpp
index 7acc107..d7f1f83 100644
--- a/bot/src/commands/request.cpp
+++ b/bot/src/commands/request.cpp
@@ -1,6 +1,15 @@
 #include "commands/request.hpp"
 
+#include <algorithm>
+#include <optional>
 #include <sol/types.hpp>
+#include <string>
+#include <vector>
+
+#include "constants.hpp"
+#include "database.hpp"
+#include "schemas/channel.hpp"
+#include "utils/string.hpp"
 
 namespace bot::command {
   sol::table Request::as_lua_table(std::shared_ptr<sol::state> luaState) const {
@@ -25,4 +34,165 @@ namespace bot::command {
 
     return o;
   }
+
+  std::optional<Request> generate_request(
+      const command::CommandLoader &command_loader,
+      const irc::Message<irc::MessageType::Privmsg> &irc_message,
+      std::unique_ptr<db::BaseDatabase> &conn) {
+    // fetching channel
+    std::vector<schemas::Channel> chans = conn->query_all<schemas::Channel>(
+        "SELECT * FROM channels WHERE alias_id = $1",
+        {std::to_string(irc_message.source.id)});
+
+    if (chans.empty()) {
+      conn->exec(
+          "INSERT INTO channels(alias_id, alias_name) VALUES ($1, $2)",
+          {std::to_string(irc_message.source.id), irc_message.source.login});
+
+      chans = conn->query_all<schemas::Channel>(
+          "SELECT * FROM channels WHERE alias_id = $1",
+          {std::to_string(irc_message.source.id)});
+    }
+
+    schemas::Channel channel = chans[0];
+    if (channel.get_opted_out_at().has_value()) {
+      return std::nullopt;
+    }
+
+    // fetching channel preference
+    std::vector<schemas::ChannelPreferences> prefs =
+        conn->query_all<schemas::ChannelPreferences>(
+            "SELECT * FROM channel_preferences WHERE id = $1",
+            {std::to_string(channel.get_id())});
+
+    if (prefs.empty()) {
+      conn->exec(
+          "INSERT INTO channel_preferences(id, prefix, locale) VALUES ($1, "
+          "$2, $3)",
+          {std::to_string(channel.get_id()), DEFAULT_PREFIX,
+           DEFAULT_LOCALE_ID});
+
+      prefs = conn->query_all<schemas::ChannelPreferences>(
+          "SELECT * FROM channel_preferences WHERE id = $1",
+          {std::to_string(channel.get_id())});
+    }
+
+    schemas::ChannelPreferences pref = prefs[0];
+
+    // fetching channel preference
+    std::vector<schemas::User> users = conn->query_all<schemas::User>(
+        "SELECT * FROM users WHERE alias_id = $1",
+        {std::to_string(irc_message.sender.id)});
+
+    if (users.empty()) {
+      conn->exec(
+          "INSERT INTO users(alias_id, alias_name) VALUES ($1, "
+          "$2)",
+          {std::to_string(irc_message.sender.id), irc_message.sender.login});
+
+      users = conn->query_all<schemas::User>(
+          "SELECT * FROM users WHERE alias_id = $1",
+          {std::to_string(irc_message.sender.id)});
+    }
+
+    schemas::User user = users[0];
+
+    // updating username
+    if (user.get_alias_name() != irc_message.sender.login) {
+      conn->exec("UPDATE users SET alias_name = $1 WHERE id = $2",
+                 {irc_message.sender.login, std::to_string(user.get_id())});
+
+      user.set_alias_name(irc_message.sender.login);
+    }
+
+    // setting permissions
+    schemas::PermissionLevel level = schemas::PermissionLevel::USER;
+    const auto &badges = irc_message.sender.badges;
+
+    if (user.get_alias_id() == channel.get_alias_id()) {
+      level = schemas::PermissionLevel::BROADCASTER;
+    } else if (std::any_of(badges.begin(), badges.end(), [&](const auto &x) {
+                 return x.first == "moderator";
+               })) {
+      level = schemas::PermissionLevel::MODERATOR;
+    } else if (std::any_of(badges.begin(), badges.end(),
+                           [&](const auto &x) { return x.first == "vip"; })) {
+      level = schemas::PermissionLevel::VIP;
+    }
+
+    std::vector<schemas::UserRights> user_rights =
+        conn->query_all<schemas::UserRights>(
+            "SELECT * FROM user_rights WHERE user_id = $1 AND channel_id = $2",
+            {std::to_string(user.get_id()), std::to_string(channel.get_id())});
+
+    if (user_rights.empty()) {
+      conn->exec(
+          "INSERT INTO user_rights(user_id, channel_id, level) VALUES ($1, "
+          "$2, $3)",
+          {std::to_string(user.get_id()), std::to_string(channel.get_id()),
+           std::to_string(level)});
+
+      user_rights = conn->query_all<schemas::UserRights>(
+          "SELECT * FROM user_rights WHERE user_id = $1 AND channel_id = $2",
+          {std::to_string(user.get_id()), std::to_string(channel.get_id())});
+    }
+
+    schemas::UserRights user_right = user_rights[0];
+
+    if (user_right.get_level() != level) {
+      conn->exec("UPDATE user_rights SET level = $1 WHERE id = $2",
+                 {std::to_string(level), std::to_string(user_right.get_id())});
+
+      user_right.set_level(level);
+    }
+
+    // --- FETCHING MESSAGES
+    std::string fullmsg = irc_message.message;
+    const std::string &prefix = pref.get_prefix();
+
+    if (fullmsg.empty() || fullmsg.substr(0, prefix.length()) != prefix) {
+      return std::nullopt;
+    }
+
+    fullmsg = fullmsg.substr(prefix.length());
+
+    std::vector<std::string> parts = utils::string::split_text(fullmsg, ' ');
+
+    std::string command_id = parts[0];
+
+    auto cmd = std::find_if(
+        command_loader.get_commands().begin(),
+        command_loader.get_commands().end(),
+        [&command_id](const auto &c) { return c->get_name() == command_id; });
+
+    if (cmd == command_loader.get_commands().end()) {
+      return std::nullopt;
+    }
+
+    parts.erase(parts.begin());
+
+    Request req{command_id, std::nullopt, std::nullopt, irc_message,
+                channel,    pref,         user,         user_right};
+
+    if (parts.empty()) {
+      return req;
+    }
+
+    std::optional<std::string> scid = parts[0];
+    auto scids = (*cmd)->get_subcommand_ids();
+
+    if (std::any_of(scids.begin(), scids.end(),
+                    [&](const auto &x) { return x == scid.value(); })) {
+      parts.erase(parts.begin());
+    } else {
+      scid = std::nullopt;
+    }
+
+    req.subcommand_id = scid;
+
+    std::optional<std::string> message = utils::string::join_vector(parts, ' ');
+    req.message = message;
+
+    return req;
+  }
 }
\ No newline at end of file