1 files changed, 23 insertions, 0 deletions

diff --git a/recipes/redditchat/webview.js b/recipes/redditchat/webview.js
new file mode 100644
index 0000000..7e209da
--- /dev/null
+++ b/recipes/redditchat/webview.js
@@ -0,0 +1,23 @@
+'use strict';
+
+module.exports = Franz => {
+  // Regular expression for (*) or (1), will extract the asterisk or the number
+  const titleRegEx = /^\(([\*\d])\)/;
+  const getMessages = function unreadCount() {
+    var directCount = 0;
+    var indirectCount = 0;
+
+    var matchArr = document.title.match(titleRegEx);
+    if (matchArr) {
+      if (matchArr[1] === '*') {
+        indirectCount = 1;
+      } else {
+        directCount = Number(matchArr[1]);
+      }
+    }
+
+    Franz.setBadge(directCount, indirectCount);
+  }
+
+  Franz.loop(getMessages);
+};