1 files changed, 38 insertions, 0 deletions

diff --git a/recipes/chatwork/webview.js b/recipes/chatwork/webview.js
new file mode 100644
index 0000000..48a3fbf
--- /dev/null
+++ b/recipes/chatwork/webview.js
@@ -0,0 +1,38 @@
+'use strict';
+
+module.exports = (Franz, options) => {
+  function getMessages() {
+    let groupCount = 0;
+    let individualCount = 0;
+    let directCount = 0;
+    let indirectCount = 0;
+    let roomInfoContainer = document.querySelectorAll('li.sc-dnqmqq');
+    Array.prototype.forEach.call(roomInfoContainer, function (room) {
+      let count = 0;
+      let unreadBadge = room.querySelector("span.sc-kAzzGY");
+      let unreadBadgeHasMention = room.querySelector("li._unreadBadge.sc-cSHVUG");
+
+      if (unreadBadge && unreadBadge.innerText) {
+        count = parseInt(unreadBadge.innerText);
+      }
+
+      if (0 < count) {
+        if (room.querySelector("img.sc-gqjmRU").getAttribute('src').indexOf('avatar') < 0) {
+          groupCount += count;
+
+          if (unreadBadgeHasMention) {
+            directCount++;
+          } else {
+            indirectCount++;
+          }
+        } else {
+          individualCount += count;
+          directCount++;
+        }
+      }
+    });
+    Franz.setBadge(directCount, indirectCount);
+  }
+
+  Franz.loop(getMessages);
+};
\ No newline at end of file