1 files changed, 19 insertions, 0 deletions

diff --git a/recipes/telegram-react/webview.js b/recipes/telegram-react/webview.js
new file mode 100644
index 0000000..223e914
--- /dev/null
+++ b/recipes/telegram-react/webview.js
@@ -0,0 +1,19 @@
+"use strict";
+
+const path = require("path");
+
+module.exports = Franz => {
+  const getMessages = function getMessages() {
+    let count = 0;
+    const elements = document.querySelectorAll(".dialog-badge:not(.dialog-badge-muted)");
+    if (elements) {
+      for (let i = 0; i < elements.length; i += 1) {
+        if (elements[i].querySelector("span").innerHTML !== 0) {
+          count += parseInt(elements[i].querySelector("span").innerHTML);
+        }
+      }
+    }
+    Franz.setBadge(count);
+  };
+  Franz.loop(getMessages);
+};