fix unread count for msteams service

2 files changed, 7 insertions, 9 deletions

diff --git a/recipes/msteams/package.json b/recipes/msteams/package.json
index abefdd0..a67616a 100644
--- a/recipes/msteams/package.json
+++ b/recipes/msteams/package.json
@@ -1,7 +1,7 @@
 {
   "id": "msteams",
   "name": "Microsoft Teams",
-  "version": "3.2.0",
+  "version": "3.2.1",
   "license": "MIT",
   "aliases": [
     "teamsChat"
diff --git a/recipes/msteams/webview.js b/recipes/msteams/webview.js
index c5e32e9..149cc76 100644
--- a/recipes/msteams/webview.js
+++ b/recipes/msteams/webview.js
@@ -7,16 +7,14 @@ function _interopRequireDefault(obj) {
 module.exports = Ferdium => {
   const getMessages = () => {
     let messages = 0;
-    const badge = document.querySelector(
-      '.activity-badge.dot-activity-badge .activity-badge',
-    );
-    if (badge) {
-      messages = Ferdium.safeParseInt(badge.textContent);
+    const badges = document.querySelectorAll('.activity-badge.dot-activity-badge .activity-badge');
+    if (badges) {
+      Array.prototype.forEach.call(badges, badge => {
+        messages += Ferdium.safeParseInt(badge.textContent);
+      });
     }
 
-    const indirectMessages = document.querySelectorAll(
-      '[class*=channel-anchor][class*=ts-unread-channel]',
-    ).length;
+    const indirectMessages = document.querySelectorAll('.app-bar-mention').length;
 
     Ferdium.setBadge(messages, indirectMessages);
   };