celisej567/abaddon
1
0
Fork 0
mirror of https://github.com/celisej567/abaddon.git synced 2026-09-13 20:18:50 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'master' into msys

Browse Source
This commit is contained in:
ouwou
2022-03-26 02:58:59 -04:00
parent 481685b3bb a2a45757e9
commit a0b3c9f8a4
25 changed files with 528 additions and 91 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
src/discord/store.cpp
View File

@@ -13,18 +13,6 @@ Store::Store(bool mem_store)
return;
}
m_db.Execute(R"(
PRAGMA writable_schema = 1;
DELETE FROM sqlite_master WHERE TYPE IN ("view", "table", "index", "trigger");
PRAGMA writable_schema = 0;
VACUUM;
PRAGMA integrity_check;
)");
if (!m_db.OK()) {
fprintf(stderr, "failed to clear database: %s\n", m_db.ErrStr());
return;
}
if (m_db.Execute("PRAGMA journal_mode = WAL") != SQLITE_OK) {
fprintf(stderr, "enabling write-ahead-log failed: %s\n", m_db.ErrStr());
return;
@@ -588,6 +576,23 @@ std::vector<Snowflake> Store::GetChannelIDsWithParentID(Snowflake channel_id) co
return ret;
}
std::unordered_set<Snowflake> Store::GetMembersInGuild(Snowflake guild_id) const {
auto &s = m_stmt_get_guild_member_ids;
s->Bind(1, guild_id);
std::unordered_set<Snowflake> ret;
while (s->FetchOne()) {
Snowflake x;
s->Get(0, x);
ret.insert(x);
}
s->Reset();
return ret;
}
void Store::AddReaction(const MessageReactionAddObject &data, bool byself) {
auto &s = m_stmt_add_reaction;
@@ -646,6 +651,7 @@ std::optional<ChannelData> Store::GetChannel(Snowflake id) const {
s->Get(6, r.IsNSFW);
s->Get(7, r.LastMessageID);
s->Get(10, r.RateLimitPerUser);
s->Get(11, r.Icon);
s->Get(12, r.OwnerID);
s->Get(14, r.ParentID);
if (!s->IsNull(16)) {
@@ -2145,10 +2151,25 @@ bool Store::CreateStatements() {
return false;
}
m_stmt_get_guild_member_ids = std::make_unique<Statement>(m_db, R"(
SELECT user_id FROM members WHERE guild_id = ?
)");
if (!m_stmt_get_guild_member_ids->OK()) {
fprintf(stderr, "failed to prepare get guild member ids statement: %s\n", m_db.ErrStr());
return false;
}
return true;
}
Store::Database::Database(const char *path) {
if (path != ":memory:"s) {
std::error_code ec;
if (std::filesystem::exists(path, ec) && !std::filesystem::remove(path, ec)) {
fprintf(stderr, "the database could not be removed. the database may be corrupted as a result\n");
}
}
m_err = sqlite3_open(path, &m_db);
}