feat: new command - !timer

6 files changed, 132 insertions, 3 deletions

diff --git a/localization/english.json b/localization/english.json
index 8cec9cb..97b74d7 100644
--- a/localization/english.json
+++ b/localization/english.json
@@ -35,5 +35,8 @@
   "join.not_allowed": "{sender.alias_name}: The bot cannot join chat rooms!%s",
 
   "custom_command.new": "{sender.alias_name}: New command \"%s\" has been successfully created!",
-  "custom_command.delete": "{sender.alias_name}: The command \"%s\" has been deleted!"
+  "custom_command.delete": "{sender.alias_name}: The command \"%s\" has been deleted!",
+
+  "timer.new": "{sender.alias_name}: New timer \"%s\" has been successfully created!",
+  "timer.delete": "{sender.alias_name}: The timer \"%s\" has been deleted!"
 }
diff --git a/localization/russian.json b/localization/russian.json
index 3fca425..ab0ebff 100644
--- a/localization/russian.json
+++ b/localization/russian.json
@@ -35,5 +35,8 @@
   "join.not_allowed": "{sender.alias_name}: Бот не может заходить в чужие чаты!%s",
 
   "custom_command.new": "{sender.alias_name}: Новая команда \"%s\" была успешно создана!",
-  "custom_command.delete": "{sender.alias_name}: Команда \"%s\" была удалена!"
+  "custom_command.delete": "{sender.alias_name}: Команда \"%s\" была удалена!",
+
+  "timer.new": "{sender.alias_name}: Новый таймер \"%s\" был успешно создан!",
+  "timer.delete": "{sender.alias_name}: Таймер \"%s\" был удален!"
 }
diff --git a/src/commands/command.cpp b/src/commands/command.cpp
index 4cfe959..0bc0f15 100644
--- a/src/commands/command.cpp
+++ b/src/commands/command.cpp
@@ -15,6 +15,7 @@
 #include "../modules/massping.hpp"
 #include "../modules/notify.hpp"
 #include "../modules/ping.hpp"
+#include "../modules/timer.hpp"
 #include "../utils/chrono.hpp"
 #include "request.hpp"
 
@@ -27,6 +28,7 @@ namespace bot {
       this->add_command(std::make_unique<mod::Notify>());
       this->add_command(std::make_unique<mod::Join>());
       this->add_command(std::make_unique<mod::CustomCommand>());
+      this->add_command(std::make_unique<mod::Timer>());
     }
 
     void CommandLoader::add_command(std::unique_ptr<Command> command) {
diff --git a/src/localization/line_id.cpp b/src/localization/line_id.cpp
index f4ea6f1..82441fc 100644
--- a/src/localization/line_id.cpp
+++ b/src/localization/line_id.cpp
@@ -80,6 +80,12 @@ namespace bot {
         return LineId::CustomcommandDelete;
       }
 
+      else if (str == "timer.new") {
+        return LineId::TimerNew;
+      } else if (str == "timer.delete") {
+        return LineId::TimerDelete;
+      }
+
       else {
         return std::nullopt;
       }
diff --git a/src/localization/line_id.hpp b/src/localization/line_id.hpp
index e9796bb..e820559 100644
--- a/src/localization/line_id.hpp
+++ b/src/localization/line_id.hpp
@@ -42,7 +42,10 @@ namespace bot {
       JoinNotAllowed,
 
       CustomcommandNew,
-      CustomcommandDelete
+      CustomcommandDelete,
+
+      TimerNew,
+      TimerDelete
     };
 
     std::optional<LineId> string_to_line_id(const std::string &str);
diff --git a/src/modules/timer.hpp b/src/modules/timer.hpp
new file mode 100644
index 0000000..03e319e
--- /dev/null
+++ b/src/modules/timer.hpp
@@ -0,0 +1,112 @@
+#pragma once
+
+#include <string>
+#include <variant>
+#include <vector>
+
+#include "../bundle.hpp"
+#include "../commands/command.hpp"
+#include "../commands/response_error.hpp"
+
+namespace bot {
+  namespace mod {
+    class Timer : public command::Command {
+        std::string get_name() const override { return "timer"; }
+
+        schemas::PermissionLevel get_permission_level() const override {
+          return schemas::PermissionLevel::MODERATOR;
+        }
+
+        std::vector<std::string> get_subcommand_ids() const override {
+          return {"new", "delete"};
+        }
+
+        std::variant<std::vector<std::string>, std::string> run(
+            const InstanceBundle &bundle,
+            const command::Request &request) const override {
+          if (!request.subcommand_id.has_value()) {
+            throw ResponseException<NOT_ENOUGH_ARGUMENTS>(
+                request, bundle.localization, command::SUBCOMMAND);
+          }
+
+          const std::string &subcommand_id = request.subcommand_id.value();
+
+          if (!request.message.has_value()) {
+            throw ResponseException<ResponseError::NOT_ENOUGH_ARGUMENTS>(
+                request, bundle.localization, command::CommandArgument::NAME);
+          }
+
+          const std::string &message = request.message.value();
+          std::vector<std::string> s = utils::string::split_text(message, ' ');
+
+          std::string name = s[0];
+          s.erase(s.begin());
+
+          pqxx::work work(request.conn);
+          pqxx::result timers = work.exec(
+              "SELECT id FROM timers WHERE name = '" + name +
+              "' AND channel_id = " + std::to_string(request.channel.get_id()));
+
+          if (subcommand_id == "new") {
+            if (!timers.empty()) {
+              throw ResponseException<ResponseError::NAMESAKE_CREATION>(
+                  request, bundle.localization, name);
+            }
+
+            if (s.empty()) {
+              throw ResponseException<ResponseError::NOT_ENOUGH_ARGUMENTS>(
+                  request, bundle.localization,
+                  command::CommandArgument::INTERVAL);
+            }
+
+            int interval_s;
+
+            try {
+              interval_s = std::stoi(s[0]);
+            } catch (std::exception e) {
+              throw ResponseException<ResponseError::INCORRECT_ARGUMENT>(
+                  request, bundle.localization, s[0]);
+            }
+
+            s.erase(s.begin());
+
+            if (s.empty()) {
+              throw ResponseException<ResponseError::NOT_ENOUGH_ARGUMENTS>(
+                  request, bundle.localization,
+                  command::CommandArgument::MESSAGE);
+            }
+
+            std::string m = utils::string::str(s.begin(), s.end(), ' ');
+
+            work.exec(
+                "INSERT INTO timers(channel_id, name, message, interval_sec) "
+                "VALUES "
+                "(" +
+                std::to_string(request.channel.get_id()) + ", '" + name +
+                "', '" + m + "', " + std::to_string(interval_s) + ")");
+            work.commit();
+
+            return bundle.localization
+                .get_formatted_line(request, loc::LineId::TimerNew, {name})
+                .value();
+          } else if (subcommand_id == "delete") {
+            if (timers.empty()) {
+              throw ResponseException<ResponseError::NOT_FOUND>(
+                  request, bundle.localization, name);
+            }
+
+            work.exec("DELETE FROM timers WHERE id = " +
+                      std::to_string(timers[0][0].as<int>()));
+            work.commit();
+
+            return bundle.localization
+                .get_formatted_line(request, loc::LineId::TimerDelete, {name})
+                .value();
+          }
+
+          throw ResponseException<ResponseError::SOMETHING_WENT_WRONG>(
+              request, bundle.localization);
+        }
+    };
+  }
+}