make dm member lists a bit better

2 files changed, 7 insertions, 1 deletions

diff --git a/components/memberlist.cpp b/components/memberlist.cpp
index 5578e17..e04c188 100644
--- a/components/memberlist.cpp
+++ b/components/memberlist.cpp
@@ -105,6 +105,7 @@ void MemberList::UpdateMemberListInternal() {
     if (chan->Type == ChannelType::DM || chan->Type == ChannelType::GROUP_DM) {
         for (const auto &user : chan->Recipients)
             ids.insert(user.ID);
+        ids.insert(discord.GetUserData().ID);
     } else {
         ids = discord.GetUsersInGuild(m_guild_id);
     }
@@ -203,7 +204,10 @@ void MemberList::UpdateMemberListInternal() {
             add_user(data);
     }
 
-    add_role("@everyone");
+    if (chan->Type == ChannelType::DM || chan->Type == ChannelType::GROUP_DM)
+        add_role("Users");
+    else
+        add_role("@everyone");
     for (const auto &id : roleless_users) {
         const auto *user = discord.GetUser(id);
         if (user != nullptr)
diff --git a/discord/discord.cpp b/discord/discord.cpp
index 5f3f6e4..ffd701d 100644
--- a/discord/discord.cpp
+++ b/discord/discord.cpp
@@ -591,6 +591,8 @@ void DiscordClient::HandleGatewayReady(const GatewayMessage &msg) {
 
     for (const auto &dm : data.PrivateChannels) {
         m_store.SetChannel(dm.ID, dm);
+        for (const auto &recipient : dm.Recipients)
+            m_store.SetUser(recipient.ID, recipient);
     }
 
     m_signal_gateway_ready.emit();