1 files changed, 24 insertions, 0 deletions

diff --git a/recipes/slite/webview.js b/recipes/slite/webview.js
new file mode 100644
index 0000000..6e0fa46
--- /dev/null
+++ b/recipes/slite/webview.js
@@ -0,0 +1,24 @@
+"use strict";
+
+var _path = _interopRequireDefault(require("path"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+module.exports = Ferdi => {
+  const getMessages = function getMessages() {
+    let direct = 0;
+
+    const notificationButton = document.querySelector(
+      "#app button[data-test-id='notificationsCount'"
+    );
+
+    if (notificationButton) {
+      const notificationCount = parseInt(notificationButton.innerText);
+      direct = isNaN(notificationCount) ? 0 : notificationCount;
+    }
+
+    Ferdi.setBadge(direct);
+  };
+
+  Ferdi.loop(getMessages);
+};