handle change of mute state for guilds

2 files changed, 34 insertions, 0 deletions

diff --git a/src/discord/discord.cpp b/src/discord/discord.cpp
index 7b0db38..0536a4e 100644
--- a/src/discord/discord.cpp
+++ b/src/discord/discord.cpp
@@ -1919,6 +1919,26 @@ void DiscordClient::HandleGatewayUserGuildSettingsUpdate(const GatewayMessage &m
     const auto channels = GetChannelsInGuild(data.Settings.GuildID);
     std::set<Snowflake> now_muted_channels;
     const auto now = Snowflake::FromNow();
+
+    const bool was_muted = IsGuildMuted(data.Settings.GuildID);
+    bool now_muted = false;
+    if (data.Settings.Muted) {
+        if (data.Settings.MuteConfig.EndTime.has_value()) {
+            const auto end = Snowflake::FromISO8601(*data.Settings.MuteConfig.EndTime);
+            if (end.IsValid() && end > now)
+                now_muted = true;
+        } else {
+            now_muted = true;
+        }
+    }
+    if (was_muted && !now_muted) {
+        m_muted_guilds.erase(data.Settings.GuildID);
+        m_signal_guild_unmuted.emit(data.Settings.GuildID);
+    } else if (!was_muted && now_muted) {
+        m_muted_guilds.insert(data.Settings.GuildID);
+        m_signal_guild_muted.emit(data.Settings.GuildID);
+    }
+
     for (const auto &override : data.Settings.ChannelOverrides) {
         if (override.Muted) {
             if (override.MuteConfig.EndTime.has_value()) {
@@ -2575,6 +2595,14 @@ DiscordClient::type_signal_channel_unmuted DiscordClient::signal_channel_unmuted
     return m_signal_channel_unmuted;
 }
 
+DiscordClient::type_signal_guild_muted DiscordClient::signal_guild_muted() {
+    return m_signal_guild_muted;
+}
+
+DiscordClient::type_signal_guild_unmuted DiscordClient::signal_guild_unmuted() {
+    return m_signal_guild_unmuted;
+}
+
 DiscordClient::type_signal_message_send_fail DiscordClient::signal_message_send_fail() {
     return m_signal_message_send_fail;
 }
diff --git a/src/discord/discord.hpp b/src/discord/discord.hpp
index 4374396..1f8daa5 100644
--- a/src/discord/discord.hpp
+++ b/src/discord/discord.hpp
@@ -372,6 +372,8 @@ public:
     typedef sigc::signal<void, Message> type_signal_message_sent;
     typedef sigc::signal<void, Snowflake> type_signal_channel_muted;
     typedef sigc::signal<void, Snowflake> type_signal_channel_unmuted;
+    typedef sigc::signal<void, Snowflake> type_signal_guild_muted;
+    typedef sigc::signal<void, Snowflake> type_signal_guild_unmuted;
 
     typedef sigc::signal<void, std::string /* nonce */, float /* retry_after */> type_signal_message_send_fail; // retry after param will be 0 if it failed for a reason that isnt slowmode
     typedef sigc::signal<void, bool, GatewayCloseCode> type_signal_disconnected;                                // bool true if reconnecting
@@ -422,6 +424,8 @@ public:
     type_signal_message_sent signal_message_sent();
     type_signal_channel_muted signal_channel_muted();
     type_signal_channel_unmuted signal_channel_unmuted();
+    type_signal_guild_muted signal_guild_muted();
+    type_signal_guild_unmuted signal_guild_unmuted();
     type_signal_message_send_fail signal_message_send_fail();
     type_signal_disconnected signal_disconnected();
     type_signal_connected signal_connected();
@@ -472,6 +476,8 @@ protected:
     type_signal_message_sent m_signal_message_sent;
     type_signal_channel_muted m_signal_channel_muted;
     type_signal_channel_unmuted m_signal_channel_unmuted;
+    type_signal_guild_muted m_signal_guild_muted;
+    type_signal_guild_unmuted m_signal_guild_unmuted;
     type_signal_message_send_fail m_signal_message_send_fail;
     type_signal_disconnected m_signal_disconnected;
     type_signal_connected m_signal_connected;