1 files changed, 10 insertions, 2 deletions

diff --git a/recipes/groupme/webview.js b/recipes/groupme/webview.js
index e2d6954..0d5afc3 100644
--- a/recipes/groupme/webview.js
+++ b/recipes/groupme/webview.js
@@ -1,9 +1,17 @@
 module.exports = Ferdium => {
   const getMessages = () => {
-    const directMessages = document.querySelectorAll('.badge-count:not(.ng-hide)').length;
+
+    // array-ify the list of conversations
+    const allConversations = [...document.querySelectorAll('#tray .tray-list .list-item')]
+    // for each conversation on the list...
+    const filteredConversations = allConversations.filter(e => {
+      // keep it on the list if [1] it has unread messages (not .ng-hide), and [2] it isn't muted (not .overlay)
+      return (!e.innerHTML.includes('ng-hide') && !e.innerHTML.includes('overlay'))
+    });
+    const unreadConversations = filteredConversations.length;
 
     // set Ferdium badge
-    Ferdium.setBadge(directMessages);
+    Ferdium.setBadge(unreadConversations);
   };
 
   Ferdium.loop(getMessages);