Merge pull request #230 from CrEaK/mattermost-fix

Mattermost: Fix badge for unread channels when in single team

2 files changed, 4 insertions, 3 deletions

diff --git a/uncompressed/mattermost/package.json b/uncompressed/mattermost/package.json
index 48f6860..52a1908 100644
--- a/uncompressed/mattermost/package.json
+++ b/uncompressed/mattermost/package.json
@@ -1,7 +1,7 @@
 {
   "id": "mattermost",
   "name": "Mattermost",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "description": "Mattermost",
   "main": "index.js",
   "author": "Stefan Malzner <stefan@adlk.io>",
diff --git a/uncompressed/mattermost/webview.js b/uncompressed/mattermost/webview.js
index 03258e0..f348da4 100644
--- a/uncompressed/mattermost/webview.js
+++ b/uncompressed/mattermost/webview.js
@@ -4,10 +4,11 @@ module.exports = Franz => {
   const getMessages = function getMessages() {
     const directMessages = document.querySelectorAll('.sidebar--left .has-badge .badge').length;
     const allMessages = document.querySelectorAll('.sidebar--left .has-badge').length - directMessages;
+    const channelMessages = document.querySelectorAll('.sidebar--left .unread-title').length - allMessages;
     const teamDirectMessages = document.querySelectorAll('.team-wrapper .team-container .badge').length;
     const teamMessages = document.querySelectorAll('.team-wrapper .unread').length - teamDirectMessages;
-    Franz.setBadge(directMessages + teamDirectMessages, allMessages + teamMessages);
+    Franz.setBadge(directMessages + teamDirectMessages, allMessages + channelMessages + teamMessages);
   };
 
   Franz.loop(getMessages);
-};
\ No newline at end of file
+};