diff options
Diffstat (limited to 'discord/discord.cpp')
| -rw-r--r-- | discord/discord.cpp | 216 |
1 files changed, 212 insertions, 4 deletions
diff --git a/discord/discord.cpp b/discord/discord.cpp index 42c6550..3523989 100644 --- a/discord/discord.cpp +++ b/discord/discord.cpp @@ -20,6 +20,7 @@ void DiscordClient::Start() { } void DiscordClient::Stop() { + std::scoped_lock<std::mutex> guard(m_mutex); if (!m_client_connected) return; m_heartbeat_waiter.kill(); @@ -32,6 +33,17 @@ bool DiscordClient::IsStarted() const { return m_client_connected; } +const DiscordClient::Guilds_t &DiscordClient::GetGuilds() const { + std::scoped_lock<std::mutex> guard(m_mutex); + return m_guilds; +} + +const UserSettingsData &DiscordClient::GetUserSettings() const { + std::scoped_lock<std::mutex> guard(m_mutex); + assert(m_ready_received); + return m_user_settings; +} + void DiscordClient::HandleGatewayMessage(nlohmann::json j) { GatewayMessage m; try { @@ -70,7 +82,19 @@ void DiscordClient::HandleGatewayMessage(nlohmann::json j) { } void DiscordClient::HandleGatewayReady(const GatewayMessage &msg) { + m_ready_received = true; + ReadyEventData data = msg.Data; + for (const auto &g : data.Guilds) { + StoreGuild(g.ID, g); + } + m_abaddon->DiscordNotifyReady(); + m_user_settings = data.UserSettings; + return; +} +void DiscordClient::StoreGuild(Snowflake id, const GuildData &g) { + assert(id.IsValid() && id == g.ID); + m_guilds[id] = g; } void DiscordClient::HeartbeatThread() { @@ -106,16 +130,169 @@ void DiscordClient::LoadEventMap() { m_event_map["READY"] = GatewayEvent::READY; } +#define JS_D(k, t) \ + do { \ + j.at(k).get_to(t); \ + } while (0) + +#define JS_O(k, t) \ + do { \ + if (j.contains(k)) j.at(k).get_to(t); \ + } while (0) + +#define JS_N(k, t) \ + do { \ + if (!j.at(k).is_null()) j.at(k).get_to(t); \ + } while (0) + +#define JS_ON(k, t) \ + do { \ + if (j.contains(k) && !j.at(k).is_null()) \ + j.at(k).get_to(t); \ + } while (0) + void from_json(const nlohmann::json &j, GatewayMessage &m) { - j.at("op").get_to(m.Opcode); + JS_D("op", m.Opcode); m.Data = j.at("d"); - if (j.contains("t") && !j.at("t").is_null()) - j.at("t").get_to(m.Type); + JS_ON("t", m.Type); } void from_json(const nlohmann::json &j, HelloMessageData &m) { - j.at("heartbeat_interval").get_to(m.HeartbeatInterval); + JS_D("heartbeat_interval", m.HeartbeatInterval); +} + +void from_json(const nlohmann::json &j, UserData &m) { + JS_D("id", m.ID); + JS_D("username", m.Username); + JS_D("discriminator", m.Discriminator); + JS_N("avatar", m.Avatar); + JS_O("bot", m.IsBot); + JS_O("system", m.IsSystem); + JS_O("mfa_enabled", m.IsMFAEnabled); + JS_O("locale", m.Locale); + JS_O("verified", m.IsVerified); + JS_O("email", m.Email); + JS_O("flags", m.Flags); + JS_O("premium_type", m.PremiumType); + JS_O("public_flags", m.PublicFlags); + JS_O("desktop", m.IsDesktop); + JS_O("mobile", m.IsMobile); + JS_ON("nsfw_allowed", m.IsNSFWAllowed); + JS_ON("phone", m.Phone); +} + +void from_json(const nlohmann::json &j, GuildData &m) { + JS_D("id", m.ID); + JS_D("name", m.Name); + JS_N("icon", m.Icon); + JS_N("splash", m.Splash); + JS_N("discovery_splash", m.DiscoverySplash); + JS_O("owner", m.IsOwner); + JS_D("owner_id", m.OwnerID); + JS_O("permissions", m.Permissions); + JS_O("permissions_new", m.PermissionsNew); + JS_D("region", m.VoiceRegion); + JS_N("afk_channel_id", m.AFKChannelID); + JS_D("afk_timeout", m.AFKTimeout); + JS_O("embed_enabled", m.IsEmbedEnabled); + JS_ON("embed_channel_id", m.EmbedChannelID); + JS_D("verification_level", m.VerificationLevel); + JS_D("default_message_notifications", m.DefaultMessageNotifications); + JS_D("explicit_content_filter", m.ExplicitContentFilter); + // JS_D("roles", m.Roles); + // JS_D("emojis", m.Emojis); + JS_D("features", m.Features); + JS_D("mfa_level", m.MFALevel); + JS_N("application_id", m.ApplicationID); + JS_O("widget_enabled", m.IsWidgetEnabled); + JS_ON("widget_channel_id", m.WidgetChannelID); + JS_N("system_channel_id", m.SystemChannelID); + JS_D("system_channel_flags", m.SystemChannelFlags); + JS_N("rules_channel_id", m.RulesChannelID); + JS_O("joined_at", m.JoinedAt); + JS_O("large", m.IsLarge); + JS_O("unavailable", m.IsUnavailable); + JS_O("member_count", m.MemberCount); + // JS_O("voice_states", m.VoiceStates); + // JS_O("members", m.Members); + JS_O("channels", m.Channels); + // JS_O("presences", m.Presences); + JS_ON("max_presences", m.MaxPresences); + JS_O("max_members", m.MaxMembers); + JS_N("vanity_url_code", m.VanityURL); + JS_N("description", m.Description); + JS_N("banner", m.BannerHash); + JS_D("premium_tier", m.PremiumTier); + JS_O("premium_subscription_count", m.PremiumSubscriptionCount); + JS_D("preferred_locale", m.PreferredLocale); + JS_N("public_updates_channel_id", m.PublicUpdatesChannelID); + JS_O("max_video_channel_users", m.MaxVideoChannelUsers); + JS_O("approximate_member_count", m.ApproximateMemberCount); + JS_O("approximate_presence_count", m.ApproximatePresenceCount); +} + +void from_json(const nlohmann::json &j, ChannelData &m) { + JS_D("id", m.ID); + JS_D("type", m.Type); + JS_O("guild_id", m.GuildID); + JS_O("position", m.Position); + // JS_O("permission_overwrites", m.PermissionOverwrites); + JS_O("name", m.Name); + JS_ON("topic", m.Topic); + JS_O("nsfw", m.IsNSFW); + JS_ON("last_message_id", m.LastMessageID); + JS_O("bitrate", m.Bitrate); + JS_O("user_limit", m.UserLimit); + JS_O("rate_limit_per_user", m.RateLimitPerUser); + JS_O("recipients", m.Recipients); + JS_ON("icon", m.Icon); + JS_O("owner_id", m.OwnerID); + JS_O("application_id", m.ApplicationID); + JS_ON("parent_id", m.ParentID); + JS_ON("last_pin_timestamp", m.LastPinTimestamp); +} + +void from_json(const nlohmann::json &j, ReadyEventData &m) { + JS_D("v", m.GatewayVersion); + JS_D("user", m.User); + JS_D("guilds", m.Guilds); + JS_D("session_id", m.SessionID); + JS_D("analytics_token", m.AnalyticsToken); + JS_D("friend_suggestion_count", m.FriendSuggestionCount); + JS_D("user_settings", m.UserSettings); +} + +void from_json(const nlohmann::json &j, UserSettingsData &m) { + JS_D("timezone_offset", m.TimezoneOffset); + JS_D("theme", m.Theme); + JS_D("stream_notifications_enabled", m.AreStreamNotificationsEnabled); + JS_D("status", m.Status); + JS_D("show_current_game", m.ShouldShowCurrentGame); + // JS_D("restricted_guilds", m.RestrictedGuilds); + JS_D("render_reactions", m.ShouldRenderReactions); + JS_D("render_embeds", m.ShouldRenderEmbeds); + JS_D("native_phone_integration_enabled", m.IsNativePhoneIntegrationEnabled); + JS_D("message_display_compact", m.ShouldMessageDisplayCompact); + JS_D("locale", m.Locale); + JS_D("inline_embed_media", m.ShouldInlineEmbedMedia); + JS_D("inline_attachment_media", m.ShouldInlineAttachmentMedia); + JS_D("guild_positions", m.GuildPositions); + // JS_D("guild_folders", m.GuildFolders); + JS_D("gif_auto_play", m.ShouldGIFAutoplay); + // JS_D("friend_source_flags", m.FriendSourceFlags); + JS_D("explicit_content_filter", m.ExplicitContentFilter); + JS_D("enable_tts_command", m.IsTTSCommandEnabled); + JS_D("disable_games_tab", m.ShouldDisableGamesTab); + JS_D("developer_mode", m.DeveloperMode); + JS_D("detect_platform_accounts", m.ShouldDetectPlatformAccounts); + JS_D("default_guilds_restricted", m.AreDefaultGuildsRestricted); + // JS_N("custom_status", m.CustomStatus); + JS_D("convert_emoticons", m.ShouldConvertEmoticons); + JS_D("contact_sync_enabled", m.IsContactSyncEnabled); + JS_D("animate_emoji", m.ShouldAnimateEmojis); + JS_D("allow_accessibility_detection", m.IsAccessibilityDetectionAllowed); + JS_D("afk_timeout", m.AFKTimeout); } void to_json(nlohmann::json &j, const IdentifyProperties &m) { @@ -141,3 +318,34 @@ void to_json(nlohmann::json &j, const HeartbeatMessage &m) { else j["d"] = m.Sequence; } + +Snowflake::Snowflake() + : m_num(Invalid) {} + +Snowflake::Snowflake(const Snowflake &s) + : m_num(s.m_num) {} + +Snowflake::Snowflake(uint64_t n) + : m_num(n) {} + +Snowflake::Snowflake(const std::string &str) { + if (str.size()) + m_num = std::stoull(str); + else + m_num = Invalid; +}; + +bool Snowflake::IsValid() const { + return m_num != Invalid; +} + +void from_json(const nlohmann::json &j, Snowflake &s) { + std::string tmp; + j.get_to(tmp); + s.m_num = std::stoull(tmp); +} + +#undef JS_O +#undef JS_D +#undef JS_N +#undef JS_ON |