dpplogger/main.cpp

#include "sqlite_modern_cpp/sqlite_modern_cpp.h"

#include <iostream>
#include <unordered_map>
#include <variant>
#include <chrono>
#include <memory>
#include <dpp/dpp.h>



class Cache {
    using Object = std::variant<std::monostate, dpp::guild, dpp::user, dpp::guild_member>;
    std::unordered_map<dpp::snowflake, Object> cache;

public:
    void store(dpp::snowflake id, const Object& object) {
        cache[id] = object;
    }
    const Object& fetch(dpp::snowflake id) const {
        auto entry = cache.find(id);
        if (entry == cache.end()) {
            return std::monostate();
        }
        return entry->second;
    }
};

int main() {
    Cache cache;

    // Initialize Sqlite3
    sqlite::database db("log.sqlite3");

    // Create tables
    {
        db << "CREATE TABLE IF NOT EXISTS messages ("
              "    id TEXT PRIMARY KEY NOT NULL,"
              "    type INTEGER NOT NULL,"
              "    channel_id TEXT NOT NULL,"
              "    author_id TEXT NOT NULL,"
              "    replied_to_id TEXT,"
              "    is_deleted INTEGER DEFAULT 0 NOT NULL,"
              "    is_edited INTEGER DEFAULT 0 NOT NULL,"
              "    creation_timestamp TEXT NOT NULL,"
              "    update_timestamp TEXT DEFAULT '0' NOT NULL,"
              "    UNIQUE(id)"
              ");";
        db << "CREATE TABLE IF NOT EXISTS message_contents ("
              "    message_id TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    content TEXT NOT NULL"
              ");";
        db << "CREATE TABLE IF NOT EXISTS users ("
              "    id TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    full_username TEXT NOT NULL," // Username#Discriminator
              "    avatar TEXT,"
              "    bio TEXT,"
              "    has_nitro INTEGER NOT NULL,"
              "    is_bot INTEGER NOT NULL"
              ");";
        db << "CREATE TABLE IF NOT EXISTS user_statuses ("
              "    user_id TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    status TEXT," // JSON array or null if not updated: [int:online/dnd/afk/offline enum, ?:emoji (null/empty if none), string:text (empty if none)]
              "    presence TEXT" // Discord JSON presence object
              ");";
        db << "CREATE TABLE IF NOT EXISTS members ("
              "    user_id TEXT NOT NULL,"
              "    guild_id TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    nickname TEXT,"
              "    in_guild INTEGER NOT NULL,"
              "    is_untracked INTEGER NOT NULL"
              ");";
        db << "CREATE TABLE IF NOT EXISTS member_voice_connections ("
              "    channel_id TEXT NOT NULL,"
              "    user_id TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    voice_channel_id TEXT,"
              "    self_muted INTEGER NOT NULL,"
              "    self_deafened INTEGER NOT NULL,"
              "    server_muted INTEGER NOT NULL,"
              "    server_deafened INTEGER NOT NULL"
              ");";
        db << "CREATE TABLE IF NOT EXISTS channels ("
              "    id TEXT NOT NULL,"
              "    guild_id TEXT NOT NULL,"
              "    category_channel_id TEXT NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    type INTEGER NOT NULL,"
              "    name TEXT NOT NULL,"
              "    topic TEXT,"
              "    has_access INTEGER NOT NULL"
              ");";
        db << "CREATE TABLE IF NOT EXISTS guilds ("
              "    id TEXT NOT NULL,"
              "    timestamp TEXT NOT NULL,"
              "    is_initial INTEGER DEFAULT 1 NOT NULL,"
              "    name TEXT NOT NULL,"
              "    owner_user_id TEXT NOT NULL,"
              "    in_guild INTEGER DEFAULT 1 NOT NULL"
              ");";
    }

    // Initialize bot
    dpp::cluster cluster("token", dpp::i_all_intents);

    // Set up callbacks
    // Useful: https://discord.com/developers/docs/topics/gateway-events
    cluster.on_guild_create([&](const dpp::guild_create_t& event) {
        const auto guild = event.created;
        db << "INSERT OR IGNORE INTO levels (id, timestamp, name, owner_user_id)"
              "                      VALUES (?,  ?,         ?,    ?            );"
           << std::to_string(guild->id) << std::to_string(time(nullptr)) << guild->name << std::to_string(guild->owner_id);
        cache.store(guild->id, *guild);
    });
    cluster.on_guild_update([&](const dpp::guild_update_t& event) {
        const auto guild = event.updated;
        db << "INSERT OR IGNORE INTO levels (id, timestamp, is_initial, name, owner_user_id)"
              "                      VALUES (?,  ?,         0,          ?,    ?            );"
           << std::to_string(guild->id) << std::to_string(time(nullptr)) << guild->name << std::to_string(guild->owner_id);
        cache.store(guild->id, *guild);
    });
    cluster.on_guild_delete([&](const dpp::guild_delete_t& event) {
        const auto& guild = std::get<dpp::guild>(cache.fetch(event.deleted->id));
        db << "INSERT OR IGNORE INTO levels (id, timestamp, is_initial, name, owner_user_id)"
              "                      VALUES (?,  ?,         0,          ?,    ?            );"
           << std::to_string(guild.id) << std::to_string(time(nullptr)) << guild.name << std::to_string(guild.owner_id);
    });

    db << "CREATE TABLE IF NOT EXISTS message_contents ("
          "    message_id TEXT NOT NULL,"
          "    is_initial INTEGER DEFAULT 1 NOT NULL,"
          "    timestamp TEXT NOT NULL,"
          "    content TEXT NOT NULL"
          ");";
    auto on_message_content = [&](const dpp::message& msg, bool is_initial) {
        db << "INSERT OR IGNORE INTO message_contents (message_id, is_initial, timestamp, content)"
              "                                VALUES (?,          ?,          ?,         ?      );"
           << std::to_string(msg.id) << is_initial << time(nullptr) << msg.content;
    };
    cluster.on_message_create([&](const dpp::message_create_t& event) {
        const auto msg = event.msg;
        db << "INSERT OR IGNORE INTO messages (id, type, channel_id, author_id, replied_to_id, creation_timestamp)"
              "                        VALUES (?,  ?,    ?,          ?,         ?,             ?                 );"
           << std::to_string(msg.id) << std::to_string(msg.type) << std::to_string(msg.channel_id) << std::to_string(msg.author.id) << (msg.message_reference.message_id?std::optional<std::string>(std::to_string(msg.message_reference.message_id)):nullptr) << time(nullptr);
        if (!msg.content.empty()) on_message_content(msg, true);
        cache.store(msg.author.id, msg.author);
    });
}