#23 Add Write

1 files changed, 16 insertions, 0 deletions

diff --git a/uncompressed/wrike/webview.js b/uncompressed/wrike/webview.js
new file mode 100644
index 0000000..113b8ae
--- /dev/null
+++ b/uncompressed/wrike/webview.js
@@ -0,0 +1,16 @@
+'use strict';
+
+module.exports = (Franz, options) => {
+  function getMessages() {
+    let directCount = 0;
+    let element = document.querySelector('span.notification-indicator');
+
+    if (element && element.innerText) {
+      directCount = parseInt(element.innerText);
+    }
+
+    Franz.setBadge(directCount);
+  }
+
+  Franz.loop(getMessages);
+}