1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
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", "remove"};
}
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 == "remove") {
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);
}
};
}
}
|
