summaryrefslogtreecommitdiff
path: root/discord/store.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'discord/store.hpp')
-rw-r--r--discord/store.hpp172
1 files changed, 0 insertions, 172 deletions
diff --git a/discord/store.hpp b/discord/store.hpp
deleted file mode 100644
index f84d13e..0000000
--- a/discord/store.hpp
+++ /dev/null
@@ -1,172 +0,0 @@
-#pragma once
-#include "../util.hpp"
-#include "objects.hpp"
-#include <unordered_map>
-#include <unordered_set>
-#include <mutex>
-#include <filesystem>
-#include <sqlite3.h>
-
-#ifdef GetMessage // fuck you windows.h
- #undef GetMessage
-#endif
-
-class Store {
-public:
- Store(bool mem_store = false);
- ~Store();
-
- bool IsValid() const;
-
- void SetUser(Snowflake id, const UserData &user);
- void SetChannel(Snowflake id, const ChannelData &chan);
- void SetGuild(Snowflake id, const GuildData &guild);
- void SetRole(Snowflake id, const RoleData &role);
- void SetMessage(Snowflake id, const Message &message);
- void SetGuildMember(Snowflake guild_id, Snowflake user_id, const GuildMember &data);
- void SetPermissionOverwrite(Snowflake channel_id, Snowflake id, const PermissionOverwrite &perm);
- void SetEmoji(Snowflake id, const EmojiData &emoji);
- void SetBan(Snowflake guild_id, Snowflake user_id, const BanData &ban);
-
- // slap const on everything even tho its not *really* const
-
- std::optional<ChannelData> GetChannel(Snowflake id) const;
- std::optional<EmojiData> GetEmoji(Snowflake id) const;
- std::optional<GuildData> GetGuild(Snowflake id) const;
- std::optional<GuildMember> GetGuildMember(Snowflake guild_id, Snowflake user_id) const;
- std::optional<Message> GetMessage(Snowflake id) const;
- std::optional<PermissionOverwrite> GetPermissionOverwrite(Snowflake channel_id, Snowflake id) const;
- std::optional<RoleData> GetRole(Snowflake id) const;
- std::optional<UserData> GetUser(Snowflake id) const;
- std::optional<BanData> GetBan(Snowflake guild_id, Snowflake user_id) const;
- std::vector<BanData> GetBans(Snowflake guild_id) const;
-
- std::vector<Message> GetLastMessages(Snowflake id, size_t num) const;
- std::vector<Snowflake> GetChannelMessageIDs(Snowflake id) const;
- std::vector<Message> GetPinnedMessages(Snowflake channel_id) const;
- std::vector<ChannelData> GetActiveThreads(Snowflake channel_id) const; // public
-
- void ClearGuild(Snowflake id);
- void ClearChannel(Snowflake id);
- void ClearBan(Snowflake guild_id, Snowflake user_id);
-
- using users_type = std::unordered_map<Snowflake, UserData>;
- using channels_type = std::unordered_map<Snowflake, ChannelData>;
- using guilds_type = std::unordered_map<Snowflake, GuildData>;
- using roles_type = std::unordered_map<Snowflake, RoleData>;
- using messages_type = std::unordered_map<Snowflake, Message>;
- using members_type = std::unordered_map<Snowflake, std::unordered_map<Snowflake, GuildMember>>; // [guild][user]
- using permission_overwrites_type = std::unordered_map<Snowflake, std::unordered_map<Snowflake, PermissionOverwrite>>; // [channel][user/role]
- using emojis_type = std::unordered_map<Snowflake, EmojiData>;
-
- const std::unordered_set<Snowflake> &GetChannels() const;
- const std::unordered_set<Snowflake> &GetGuilds() const;
-
- void ClearAll();
-
- void BeginTransaction();
- void EndTransaction();
-
-private:
- Message GetMessageBound(sqlite3_stmt *stmt) const;
-
- void SetMessageInteractionPair(Snowflake message_id, const MessageInteractionData &interaction);
-
- std::unordered_set<Snowflake> m_channels;
- std::unordered_set<Snowflake> m_guilds;
-
- bool CreateTables();
- bool CreateStatements();
- void Cleanup();
-
- template<typename T>
- void Bind(sqlite3_stmt *stmt, int index, const std::optional<T> &opt) const;
-
- template<typename T>
- typename std::enable_if<std::is_enum<T>::value, void>::type
- Bind(sqlite3_stmt *stmt, int index, T val) const;
-
- void Bind(sqlite3_stmt *stmt, int index, int num) const;
- void Bind(sqlite3_stmt *stmt, int index, uint64_t num) const;
- void Bind(sqlite3_stmt *stmt, int index, const std::string &str) const;
- void Bind(sqlite3_stmt *stmt, int index, bool val) const;
- void Bind(sqlite3_stmt *stmt, int index, std::nullptr_t) const;
- bool RunInsert(sqlite3_stmt *stmt);
- bool FetchOne(sqlite3_stmt *stmt) const;
-
- template<typename T>
- void Get(sqlite3_stmt *stmt, int index, std::optional<T> &out) const;
-
- template<typename T>
- typename std::enable_if<std::is_enum<T>::value, void>::type
- Get(sqlite3_stmt *stmt, int index, T &out) const;
-
- void Get(sqlite3_stmt *stmt, int index, int &out) const;
- void Get(sqlite3_stmt *stmt, int index, uint64_t &out) const;
- void Get(sqlite3_stmt *stmt, int index, std::string &out) const;
- void Get(sqlite3_stmt *stmt, int index, bool &out) const;
- void Get(sqlite3_stmt *stmt, int index, Snowflake &out) const;
- bool IsNull(sqlite3_stmt *stmt, int index) const;
- void Reset(sqlite3_stmt *stmt) const;
-
- std::filesystem::path m_db_path;
- mutable sqlite3 *m_db;
- mutable int m_db_err;
- mutable sqlite3_stmt *m_set_user_stmt;
- mutable sqlite3_stmt *m_get_user_stmt;
- mutable sqlite3_stmt *m_set_perm_stmt;
- mutable sqlite3_stmt *m_get_perm_stmt;
- mutable sqlite3_stmt *m_set_msg_stmt;
- mutable sqlite3_stmt *m_get_msg_stmt;
- mutable sqlite3_stmt *m_set_role_stmt;
- mutable sqlite3_stmt *m_get_role_stmt;
- mutable sqlite3_stmt *m_set_emote_stmt;
- mutable sqlite3_stmt *m_get_emote_stmt;
- mutable sqlite3_stmt *m_set_member_stmt;
- mutable sqlite3_stmt *m_get_member_stmt;
- mutable sqlite3_stmt *m_set_guild_stmt;
- mutable sqlite3_stmt *m_get_guild_stmt;
- mutable sqlite3_stmt *m_set_chan_stmt;
- mutable sqlite3_stmt *m_get_chan_stmt;
- mutable sqlite3_stmt *m_set_ban_stmt;
- mutable sqlite3_stmt *m_get_ban_stmt;
- mutable sqlite3_stmt *m_clear_ban_stmt;
- mutable sqlite3_stmt *m_get_bans_stmt;
- mutable sqlite3_stmt *m_set_msg_interaction_stmt;
- mutable sqlite3_stmt *m_get_last_msgs_stmt;
- mutable sqlite3_stmt *m_get_msg_ids_stmt;
- mutable sqlite3_stmt *m_get_pins_stmt;
- mutable sqlite3_stmt *m_get_threads_stmt;
- mutable sqlite3_stmt *m_clear_chan_stmt;
-};
-
-template<typename T>
-inline void Store::Bind(sqlite3_stmt *stmt, int index, const std::optional<T> &opt) const {
- if (opt.has_value())
- Bind(stmt, index, *opt);
- else
- sqlite3_bind_null(stmt, index);
-}
-
-template<typename T>
-inline typename std::enable_if<std::is_enum<T>::value, void>::type
-Store::Bind(sqlite3_stmt *stmt, int index, T val) const {
- Bind(stmt, index, static_cast<typename std::underlying_type<T>::type>(val));
-}
-
-template<typename T>
-inline void Store::Get(sqlite3_stmt *stmt, int index, std::optional<T> &out) const {
- if (sqlite3_column_type(stmt, index) == SQLITE_NULL)
- out = std::nullopt;
- else {
- T v;
- Get(stmt, index, v);
- out = std::optional<T>(v);
- }
-}
-
-template<typename T>
-inline typename std::enable_if<std::is_enum<T>::value, void>::type
-Store::Get(sqlite3_stmt *stmt, int index, T &out) const {
- out = static_cast<T>(sqlite3_column_int(stmt, index));
-}
generated by cgit v1.2.3 (git 2.39.1) at 2025-08-23 15:12:43 +0000