1 files changed, 22 insertions, 0 deletions

diff --git a/uncompressed/element/webview.js b/uncompressed/element/webview.js
new file mode 100644
index 0000000..6201f27
--- /dev/null
+++ b/uncompressed/element/webview.js
@@ -0,0 +1,22 @@
+'use strict';
+
+module.exports = Franz => {
+  function getMessages() {
+    const indirect_badge = document.getElementsByClassName('mx_RoomSubList_badge');
+    let indirect_count = 0;
+    if (indirect_badge.length !== 0) {
+      indirect_count = indirect_badge.length;
+    }
+
+    const direct_badge = document.getElementsByClassName('mx_RoomSubList_badgeHighlight');
+    let direct_count = 0;
+    if (direct_badge.length !== 0) {
+      direct_count = direct_badge.length;
+    }
+    // set Franz badge
+    Franz.setBadge(direct_count, indirect_count);
+  }
+
+  // check for new messages every second and update Franz badge
+  Franz.loop(getMessages);
+};