1 files changed, 31 insertions, 0 deletions

diff --git a/recipes/discord/webview.js b/recipes/discord/webview.js
new file mode 100644
index 0000000..2d68a44
--- /dev/null
+++ b/recipes/discord/webview.js
@@ -0,0 +1,31 @@
+"use strict";
+
+var _path = _interopRequireDefault(require("path"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+module.exports = Franz => {
+  const getMessages = function getMessages() {
+    const direct = document.querySelector('[class*="guilds-"]').querySelectorAll('[class^="numberBadge-"]').length;
+        
+        var indirect = 0;
+        var guilds = document.querySelector("[data-ref-id=guildsnav]");
+        if(guilds != null) {
+                var channelPills = [].slice.call(guilds.querySelectorAll("[class*=item-2hkk8m]"));
+                indirect += channelPills.filter(y => y.clientHeight == 8).length;
+        
+                var activeWindow = channelPills.find(y => y.clientHeight == 40);
+                if(activeWindow != null) {
+                        var unreadChannels = document.querySelector("[class*=modeUnread]");
+                        
+                        if(unreadChannels != null)
+                                indirect++;
+                }
+        }
+        
+    Franz.setBadge(direct, indirect);
+  };
+
+  Franz.loop(getMessages);
+  Franz.injectCSS(_path.default.join(__dirname, 'service.css'));
+};