VisorBot/ModLogModule.h at master · VisorModStudio/VisorBot · GitHub

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
#pragma once
#include "BotClient.h"
#include "EventModule.h"
#include <dpp/dpp.h>
#include <string>
#include "sqlite3.h"
#include "AuditLogEvents.h"
#include "GeminiClient.h"
#include "KnowledgeCache.h"

enum class LogType
{
    Info,
    Warning,
    Alert,
    Success
};

enum class ObjectType
{
    Channel,
    Member,
    Role,
    Emoji

};

class ModLogModule : public EventModule {
    private:
    BotClient& bot;
    sqlite3* db;
    GeminiClient& gemini;
    KnowledgeCache& knowledgeCache;


    void sendLogWithAudit(const dpp::snowflake guild_id, AuditLogEvent action_type, const dpp::snowflake& object_id,
                       std::string title, ObjectType eventObject, LogType logType,
                       std::string precomputed_display = "");

    void sendLog(dpp::snowflake guild_id, LogType type, const std::string& title, const std::string& description);


    void onButtonClick(const dpp::button_click_t& event);

    void onMemberJoin(const dpp::guild_member_add_t& event);
    void onMemberBan(const dpp::guild_ban_add_t& event);
    void onMemberUnban(const dpp::guild_ban_remove_t& event);
    void onRoleCreate(const dpp::guild_role_create_t& event);
    void onRoleDelete(const dpp::guild_role_delete_t& event);
    void onChannelCreate(const dpp::channel_create_t& event);
    void onChannelDelete(const dpp::channel_delete_t& event);
    void onChannelUpdate(const dpp::channel_update_t& event);
    void onMemberRemove(const dpp::guild_member_remove_t& event);
    void onInviteCreate(const dpp::invite_create_t& event);
    void onInviteDelete(const dpp::invite_delete_t& event);
    //TODO Webhook
    void onGuildEmojisUpdate(const dpp::guild_emojis_update_t& event);
    void onMessageDelete(const dpp::message_delete_t& event);
    void onMessageCreate(const dpp::message_create_t& event);
    void onMessageBulkDelete(const dpp::message_delete_bulk_t& event);
    //

    void onThreadCreate(const dpp::thread_create_t& event);

    std::unordered_map<dpp::snowflake, std::map<dpp::snowflake, dpp::emoji>> emoji_cache;
    std::unordered_map<dpp::snowflake, std::pair<dpp::snowflake, std::string>> message_cache;

    public:


    dpp::snowflake getColumnFromServerConfig(dpp::snowflake guild_id, std::string column) const;
    std::string getStringColumnFromServerConfig(dpp::snowflake guild_id, std::string column) const;


    ModLogModule(BotClient& bot_ref, sqlite3* db_ref, GeminiClient& gemini_ref, KnowledgeCache& knowledgeCache_ref)
        : bot(bot_ref), db(db_ref), gemini(gemini_ref), knowledgeCache(knowledgeCache_ref)
    {
    }


    void registerHandlers() override;

};