Merge pull request #356 from iansealy/habitica

Habitica: Add notification count.

2 files changed, 16 insertions, 2 deletions

diff --git a/recipes/habitica/package.json b/recipes/habitica/package.json
index 8fd53a1..688d3f2 100755
--- a/recipes/habitica/package.json
+++ b/recipes/habitica/package.json
@@ -1,7 +1,7 @@
 {
   "id": "habitica",
   "name": "Habitica",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Habitica",
   "main": "index.js",
   "license": "MIT",
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);
+};