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
|
#pragma once
#include <string>
#include <vector>
#include "../bundle.hpp"
#include "../commands/command.hpp"
#include "../commands/response_error.hpp"
namespace bot {
namespace mod {
class CustomCommand : public command::Command {
std::string get_name() const override { return "scmd"; }
schemas::PermissionLevel get_permission_level() const override {
return schemas::PermissionLevel::MODERATOR;
}
std::vector<std::string> get_subcommand_ids() const override {
return {"new", "remove"};
}
command::Response 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 cmds = work.exec(
"SELECT id FROM custom_commands WHERE name = '" + name +
"' AND channel_id = " + std::to_string(request.channel.get_id()));
if (subcommand_id == "new") {
if (!cmds.empty()) {
throw ResponseException<ResponseError::NAMESAKE_CREATION>(
request, bundle.localization, name);
}
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 custom_commands(channel_id, name, message) VALUES "
"(" +
std::to_string(request.channel.get_id()) + ", '" + name +
"', '" + m + "')");
work.commit();
return command::Response(
bundle.localization
.get_formatted_line(request, loc::LineId::CustomcommandNew,
{name})
.value());
} else if (subcommand_id == "remove") {
if (cmds.empty()) {
throw ResponseException<ResponseError::NOT_FOUND>(
request, bundle.localization, name);
}
work.exec("DELETE FROM custom_commands WHERE id = " +
std::to_string(cmds[0][0].as<int>()));
work.commit();
return command::Response(
bundle.localization
.get_formatted_line(
request, loc::LineId::CustomcommandDelete, {name})
.value());
}
throw ResponseException<ResponseError::SOMETHING_WENT_WRONG>(
request, bundle.localization);
}
};
}
}
|
