1 files changed, 17 insertions, 0 deletions

diff --git a/recipes/notion/webview.js b/recipes/notion/webview.js
new file mode 100644
index 0000000..5308626
--- /dev/null
+++ b/recipes/notion/webview.js
@@ -0,0 +1,17 @@
+"use strict";
+
+module.exports = (Franz) => {
+  function getMessages() {
+    let direct = 0;
+    let indirect = 0;
+    const badgeDiv = document.querySelector('.notion-sidebar-container > div > div > div > :nth-child(4) > :nth-child(2) > div > :nth-child(3) > div > div');
+    if (badgeDiv) {
+      const count = parseInt(badgeDiv.innerText);
+      direct = isNaN(count) ? 0 : count;
+    }
+
+    Franz.setBadge(direct, indirect);
+  }
+
+  Franz.loop(getMessages);
+}