fix: no gitter badge for mentions in channels (#752)

2 files changed, 7 insertions, 4 deletions

diff --git a/recipes/gitter/package.json b/recipes/gitter/package.json
index c543c09..e931b67 100644
--- a/recipes/gitter/package.json
+++ b/recipes/gitter/package.json
@@ -1,7 +1,7 @@
 {
   "id": "gitter",
   "name": "Gitter",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "license": "MIT",
   "config": {
     "serviceURL": "https://gitter.im",
diff --git a/recipes/gitter/webview.js b/recipes/gitter/webview.js
index 0026790..c4eee5b 100644
--- a/recipes/gitter/webview.js
+++ b/recipes/gitter/webview.js
@@ -7,13 +7,16 @@ function _interopRequireDefault(obj) {
 module.exports = Ferdi => {
   const getMessages = () => {
     // get unread messages
-    let count = 0;
+    let directCount = 0;
     for (const node of document.querySelectorAll('div.unread-indicator')) {
-      count += Ferdi.safeParseInt(node.textContent);
+      directCount += Ferdi.safeParseInt(node.textContent);
     }
 
+    const channelMentionCount =
+      document.querySelectorAll('.mention-indicator').length;
+
     // set Ferdi badge
-    Ferdi.setBadge(count);
+    Ferdi.setBadge(directCount, channelMentionCount);
   };
 
   Ferdi.loop(getMessages);