rocketchat: fix unread counter disappearing (#489)

2 files changed, 13 insertions, 13 deletions

diff --git a/recipes/rocketchat/package.json b/recipes/rocketchat/package.json
index 164cd5f..92c245a 100644
--- a/recipes/rocketchat/package.json
+++ b/recipes/rocketchat/package.json
@@ -1,7 +1,7 @@
 {
   "id": "rocketchat",
   "name": "Rocket.Chat",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "license": "MIT",
   "config": {
     "serviceURL": "https://{teamId}.rocket.chat",
diff --git a/recipes/rocketchat/webview.js b/recipes/rocketchat/webview.js
index 74556de..67f50f1 100644
--- a/recipes/rocketchat/webview.js
+++ b/recipes/rocketchat/webview.js
@@ -6,19 +6,19 @@ const _path = _interopRequireDefault(require('path'));
 
 module.exports = Ferdium => {
   const getMessages = () => {
-    const directMessages = document.querySelectorAll('.rcx-badge');
-
-    let directMessagesCount = 0;
-
-    for (const directMessage of directMessages) {
-      directMessagesCount += Ferdium.safeParseInt(directMessage.textContent);
+    const title = document.querySelector('title');
+    const matches = title.textContent.match(/^\((\S*?)\)/);
+
+    if (matches) {
+      const count = Ferdium.safeParseInt(matches[1], 10);
+      if (count) {
+        Ferdium.setBadge(count);
+      } else {
+        Ferdium.setBadge(0, 1);
+      }
+    } else {
+      Ferdium.setBadge(0);
     }
-
-    const indirectMessagesCount = Math.round(
-      document.querySelectorAll('.rcx-sidebar-item--highlighted').length,
-    );
-
-    Ferdium.setBadge(directMessagesCount, indirectMessagesCount);
   };
 
   Ferdium.loop(getMessages);