1 files changed, 14 insertions, 0 deletions

diff --git a/recipes/mattermost/webview.js b/recipes/mattermost/webview.js
new file mode 100644
index 0000000..f348da4
--- /dev/null
+++ b/recipes/mattermost/webview.js
@@ -0,0 +1,14 @@
+"use strict";
+
+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 + channelMessages + teamMessages);
+  };
+
+  Franz.loop(getMessages);
+};