Fix unread message count for groupme and hostnet recipes

2 files changed, 11 insertions, 3 deletions

diff --git a/recipes/groupme/package.json b/recipes/groupme/package.json
index 983c679..0fa12ad 100644
--- a/recipes/groupme/package.json
+++ b/recipes/groupme/package.json
@@ -1,7 +1,7 @@
 {
   "id": "groupme",
   "name": "GroupMe",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "license": "MIT",
   "config": {
     "serviceURL": "https://web.groupme.com",
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);