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
|
#include "request_util.hpp"
#include <optional>
#include <pqxx/pqxx>
#include <string>
#include "../constants.hpp"
#include "../irc/message.hpp"
#include "../schemas/channel.hpp"
#include "command.hpp"
#include "request.hpp"
namespace bot::command {
std::optional<Request> generate_request(
const command::CommandLoader &command_loader,
const irc::Message<irc::MessageType::Privmsg> &irc_message,
pqxx::connection &conn) {
pqxx::work *work;
work = new pqxx::work(conn);
std::vector<std::string> parts =
utils::string::split_text(irc_message.message, ' ');
std::string command_id = parts[0];
if (command_id.substr(0, DEFAULT_PREFIX.length()) != DEFAULT_PREFIX) {
return std::nullopt;
}
command_id =
command_id.substr(DEFAULT_PREFIX.length(), command_id.length());
bool found = std::any_of(
command_loader.get_commands().begin(),
command_loader.get_commands().end(),
[&](const auto &command) { return command->get_name() == command_id; });
if (!found) {
return std::nullopt;
}
parts.erase(parts.begin());
pqxx::result channel_query =
work->exec("SELECT * FROM channels WHERE alias_id = " +
std::to_string(irc_message.source.id));
// Create new channel data in the database if it didn't exist b4
if (channel_query.empty()) {
work->exec("INSERT INTO channels (alias_id, alias_name) VALUES (" +
std::to_string(irc_message.source.id) + ", '" +
irc_message.source.login + "')");
work->commit();
delete work;
work = new pqxx::work(conn);
channel_query = work->exec("SELECT * FROM channels WHERE alias_id = " +
std::to_string(irc_message.source.id));
}
schemas::Channel channel(channel_query[0]);
pqxx::result user_query =
work->exec("SELECT * FROM users WHERE alias_id = " +
std::to_string(irc_message.sender.id));
// Create new user data in the database if it didn't exist before
if (user_query.empty()) {
work->exec("INSERT INTO users (alias_id, alias_name) VALUES (" +
std::to_string(irc_message.sender.id) + ", '" +
irc_message.sender.login + "')");
work->commit();
delete work;
work = new pqxx::work(conn);
user_query = work->exec("SELECT * FROM users WHERE alias_id = " +
std::to_string(irc_message.sender.id));
}
schemas::User user(user_query[0]);
delete work;
if (parts.empty()) {
Request req{command_id, std::nullopt, std::nullopt, irc_message,
channel, user, conn};
return req;
}
std::optional<std::string> subcommand_id = parts[0];
if (subcommand_id->empty()) {
subcommand_id = std::nullopt;
}
parts.erase(parts.begin());
std::optional<std::string> message = utils::string::join_vector(parts, ' ');
if (message->empty()) {
message = std::nullopt;
}
Request req{command_id, subcommand_id, message, irc_message,
channel, user, conn};
return req;
}
}
|
