Habitica: Add notification count.

1 files changed, 15 insertions, 1 deletions

diff --git a/recipes/habitica/webview.js b/recipes/habitica/webview.js
index 17daa35..66633cb 100755
--- a/recipes/habitica/webview.js
+++ b/recipes/habitica/webview.js
@@ -1,3 +1,17 @@
 "use strict";
 
-module.exports = Franz => Franz;
+module.exports = (Franz) => {
+  const getMessages = () => {
+    let count = 0;
+    let element = document.querySelector(".message-count");
+    if (element) {
+      count = Number(element.innerText);
+      if (Number.isNaN(count)) {
+        count = 0;
+      }
+    }
+    Franz.setBadge(count);
+  };
+
+  Franz.loop(getMessages);
+};