tri11paragon
/
discord-bot
mirror of https://github.com/tri11paragon/discord-bot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

lovely servers

main
Brett 2024-03-02 15:18:07 -05:00
parent 802962a2fa
commit 0de6e2a439
2 changed files with 78 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

106
include/data_structs.h
View File

@ -97,14 +97,31 @@ namespace db
std::string content; std::string content;
}; };
auto make_server_info_table()
{
using namespace sqlite_orm;
return make_table("server_info",
make_column("member_count", &server_info_t::member_count),
make_column("name", &server_info_t::name),
make_column("description", &server_info_t::description),
make_column("icon", &server_info_t::icon),
make_column("splash", &server_info_t::splash),
make_column("discovery_splash", &server_info_t::discovery_splash),
make_column("banner", &server_info_t::banner),
primary_key(&server_info_t::member_count, &server_info_t::name, &server_info_t::description, &server_info_t::icon,
&server_info_t::splash, &server_info_t::discovery_splash, &server_info_t::banner));
}
using server_info_table_t = decltype(make_server_info_table());
auto make_user_table() auto make_user_table()
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("users", return make_table("users",
make_column("userID", &user_info_t::userID, primary_key()), make_column("userID", &user_info_t::userID, primary_key()),
make_column("username", &user_info_t::username), make_column("username", &user_info_t::username),
make_column("global_nickname", &user_info_t::global_nickname), make_column("global_nickname", &user_info_t::global_nickname),
make_column("server_name", &user_info_t::server_name)); make_column("server_name", &user_info_t::server_name));
} }
using user_table_t = decltype(make_user_table()); using user_table_t = decltype(make_user_table());
@ -113,13 +130,13 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("user_history", return make_table("user_history",
make_column("userID", &user_history_t::userID), make_column("userID", &user_history_t::userID),
make_column("time_changed", &user_history_t::time_changed), make_column("time_changed", &user_history_t::time_changed),
make_column("old_username", &user_history_t::old_username), make_column("old_username", &user_history_t::old_username),
make_column("old_global_nickname", &user_history_t::old_global_nickname), make_column("old_global_nickname", &user_history_t::old_global_nickname),
make_column("old_server_name", &user_history_t::old_server_name), make_column("old_server_name", &user_history_t::old_server_name),
foreign_key(&user_history_t::userID).references(&user_info_t::userID), foreign_key(&user_history_t::userID).references(&user_info_t::userID),
primary_key(&user_history_t::userID, &user_history_t::time_changed)); primary_key(&user_history_t::userID, &user_history_t::time_changed));
} }
using user_history_table_t = decltype(make_user_history_table()); using user_history_table_t = decltype(make_user_history_table());
@ -128,9 +145,9 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("channels", return make_table("channels",
make_column("channelID", &channel_info_t::channelID, primary_key()), make_column("channelID", &channel_info_t::channelID, primary_key()),
make_column("channel_topic", &channel_info_t::channel_topic), make_column("channel_topic", &channel_info_t::channel_topic),
make_column("channel_name", &channel_info_t::channel_name)); make_column("channel_name", &channel_info_t::channel_name));
} }
using channel_table_t = decltype(make_channel_table()); using channel_table_t = decltype(make_channel_table());
@ -139,11 +156,11 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("channel_history", return make_table("channel_history",
make_column("channelID", &channel_history_t::channelID), make_column("channelID", &channel_history_t::channelID),
make_column("time_changed", &channel_history_t::time_changed), make_column("time_changed", &channel_history_t::time_changed),
make_column("old_channel_name", &channel_history_t::old_channel_name), make_column("old_channel_name", &channel_history_t::old_channel_name),
foreign_key(&channel_history_t::channelID).references(&channel_info_t::channelID), foreign_key(&channel_history_t::channelID).references(&channel_info_t::channelID),
primary_key(&channel_history_t::channelID, &channel_history_t::time_changed)); primary_key(&channel_history_t::channelID, &channel_history_t::time_changed));
} }
using channel_history_table_t = decltype(make_channel_history_table()); using channel_history_table_t = decltype(make_channel_history_table());
@ -152,12 +169,12 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("messages", return make_table("messages",
make_column("messageID", &message_t::messageID, primary_key()), make_column("messageID", &message_t::messageID, primary_key()),
make_column("channelID", &message_t::channelID), make_column("channelID", &message_t::channelID),
make_column("userID", &message_t::userID), make_column("userID", &message_t::userID),
make_column("content", &message_t::content), make_column("content", &message_t::content),
foreign_key(&message_t::channelID).references(&channel_info_t::channelID), foreign_key(&message_t::channelID).references(&channel_info_t::channelID),
foreign_key(&message_t::userID).references(&user_info_t::userID)); foreign_key(&message_t::userID).references(&user_info_t::userID));
} }
using message_table_t = decltype(make_message_table()); using message_table_t = decltype(make_message_table());
@ -166,13 +183,13 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("attachments", return make_table("attachments",
make_column("messageID", &attachment_t::messageID), make_column("messageID", &attachment_t::messageID),
make_column("attachmentID", &attachment_t::attachmentID), make_column("attachmentID", &attachment_t::attachmentID),
make_column("filename", &attachment_t::filename), make_column("filename", &attachment_t::filename),
make_column("description", &attachment_t::description), make_column("description", &attachment_t::description),
make_column("url", &attachment_t::url), make_column("url", &attachment_t::url),
foreign_key(&attachment_t::messageID).references(&message_t::messageID), foreign_key(&attachment_t::messageID).references(&message_t::messageID),
primary_key(&attachment_t::messageID, &attachment_t::attachmentID)); primary_key(&attachment_t::messageID, &attachment_t::attachmentID));
} }
using attachment_table_t = decltype(make_attachment_table()); using attachment_table_t = decltype(make_attachment_table());
@ -181,11 +198,11 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("message_edits", return make_table("message_edits",
make_column("messageID", &message_edits_t::messageID), make_column("messageID", &message_edits_t::messageID),
make_column("old_content", &message_edits_t::old_content), make_column("old_content", &message_edits_t::old_content),
make_column("new_content", &message_edits_t::new_content), make_column("new_content", &message_edits_t::new_content),
foreign_key(&message_edits_t::messageID).references(&message_t::messageID), foreign_key(&message_edits_t::messageID).references(&message_t::messageID),
primary_key(&message_edits_t::messageID, &message_edits_t::old_content, &message_edits_t::new_content)); primary_key(&message_edits_t::messageID, &message_edits_t::old_content, &message_edits_t::new_content));
} }
using message_edits_table_t = decltype(make_message_edits_table()); using message_edits_table_t = decltype(make_message_edits_table());
@ -194,12 +211,12 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_table("message_deletes", return make_table("message_deletes",
make_column("messageID", &message_deletes_t::messageID), make_column("messageID", &message_deletes_t::messageID),
make_column("channelID", &message_deletes_t::channelID), make_column("channelID", &message_deletes_t::channelID),
make_column("content", &message_deletes_t::content), make_column("content", &message_deletes_t::content),
foreign_key(&message_deletes_t::messageID).references(&message_t::messageID), foreign_key(&message_deletes_t::messageID).references(&message_t::messageID),
foreign_key(&message_deletes_t::channelID).references(&channel_info_t::channelID), foreign_key(&message_deletes_t::channelID).references(&channel_info_t::channelID),
primary_key(&message_deletes_t::messageID, &message_deletes_t::channelID)); primary_key(&message_deletes_t::messageID, &message_deletes_t::channelID));
} }
using message_deletes_table_t = decltype(make_message_deletes_table()); using message_deletes_table_t = decltype(make_message_deletes_table());
@ -208,7 +225,8 @@ namespace db
{ {
using namespace sqlite_orm; using namespace sqlite_orm;
return make_storage(std::move(path), make_user_table(), make_user_history_table(), make_channel_table(), make_channel_history_table(), return make_storage(std::move(path), make_user_table(), make_user_history_table(), make_channel_table(), make_channel_history_table(),
make_message_table(), make_attachment_table(), make_message_edits_table(), make_message_deletes_table()); make_message_table(), make_attachment_table(), make_message_edits_table(), make_message_deletes_table(),
make_server_info_table());
} }
using database_type = decltype(make_database("")); using database_type = decltype(make_database(""));

21
src/main.cpp
View File

@ -103,7 +103,8 @@ struct db_obj
void commit(const user_info_t& edited) void commit(const user_info_t& edited)
{ {
auto existing_user = db.select(sql::object<user_info_t>(), sql::where(sql::c(&user_info_t::userID) == edited.userID)); auto existing_user = db.select(sql::object<user_info_t>(), sql::from<user_info_t>(),
sql::where(sql::c(&user_info_t::userID) == edited.userID));
if (!existing_user.empty()) if (!existing_user.empty())
{ {
@ -149,17 +150,27 @@ struct db_obj
void commit(const message_edits_t& edited) void commit(const message_edits_t& edited)
{ {
db.insert(edited);
} }
void commit(const message_deletes_t& deleted) void commit(const message_deletes_t& deleted)
{ {
db.insert(deleted);
} }
void commit(const server_info_t& deleted) void commit(const server_info_t& server)
{ {
using namespace sql;
auto data = db.select(columns(&server_info_t::name), from<server_info_t>(),
where((c(&server_info_t::member_count) == server.member_count) &&
(c(&server_info_t::name) == server.name) &&
(c(&server_info_t::description) == server.description) &&
(c(&server_info_t::icon) == server.icon) &&
(c(&server_info_t::splash) == server.splash) &&
(c(&server_info_t::discovery_splash) == server.discovery_splash) &&
(c(&server_info_t::banner) == server.banner)));
if (data.empty())
db.insert(server);
} }
void process_queue(dpp::cluster& bot) void process_queue(dpp::cluster& bot)