From 604f2ffe3dc8978aebd6aa819b73374aa32d2f0e Mon Sep 17 00:00:00 2001
From: ouwou <26526779+ouwou@users.noreply.github.com>
Date: Sat, 8 Jan 2022 20:03:12 -0500
Subject: show count of unread dms in header

---
 src/discord/discord.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/discord/discord.cpp')

diff --git a/src/discord/discord.cpp b/src/discord/discord.cpp
index dce5124..5b3cdb5 100644
--- a/src/discord/discord.cpp
+++ b/src/discord/discord.cpp
@@ -1202,6 +1202,14 @@ bool DiscordClient::GetUnreadStateForGuild(Snowflake id, int &total_mentions) co
     return has_any_unread;
 }
 
+int DiscordClient::GetUnreadDMsCount() const {
+    const auto channels = GetPrivateChannels();
+    int count = 0;
+    for (const auto channel_id : channels)
+        if (GetUnreadStateForChannel(channel_id) > -1) count++;
+    return count;
+}
+
 PresenceStatus DiscordClient::GetUserStatus(Snowflake id) const {
     auto it = m_user_to_status.find(id);
     if (it != m_user_to_status.end())
-- 
cgit v1.2.3