Fixes notifications for WhatsApp (#120)

- Fixes an issue where whatsapp doesn't properly detect notifications.
- Added support for indirect notifications (muted chats)

2 files changed, 16 insertions, 8 deletions

diff --git a/uncompressed/whatsapp/package.json b/uncompressed/whatsapp/package.json
index 210e8d1..49afab1 100644
--- a/uncompressed/whatsapp/package.json
+++ b/uncompressed/whatsapp/package.json
@@ -1,7 +1,7 @@
 {
   "id": "whatsapp",
   "name": "WhatsApp",
-  "version": "2.0.0",
+  "version": "2.0.1",
   "description": "WhatsApp",
   "main": "index.js",
   "author": "Stefan Malzner <stefan@adlk.io>",
@@ -9,6 +9,8 @@
   "config": {
     "serviceURL": "https://web.whatsapp.com",
     "hasNotificationSound": true,
+    "hasDirectMessages": true,
+    "hasIndirectMessages": true,
     "disablewebsecurity": true
   }
 }
diff --git a/uncompressed/whatsapp/webview.js b/uncompressed/whatsapp/webview.js
index 989680a..6cc173a 100644
--- a/uncompressed/whatsapp/webview.js
+++ b/uncompressed/whatsapp/webview.js
@@ -28,16 +28,22 @@ window.addEventListener('beforeunload', async () => {
 
 module.exports = Franz => {
   const getMessages = function getMessages() {
-    const elements = document.querySelectorAll('.CxUIE, .unread, ._0LqQ');
-    let count = 0;
+    const elements = document.querySelectorAll("[class=OUeyt]");
+    var count = 0;
+        var indirectCount = 0;
 
-    for (let i = 0; i < elements.length; i += 1) {
-      if (elements[i].querySelectorAll('*[data-icon="muted"]').length === 0) {
-        count += 1;
-      }
+    for (var i = 0; i < elements.length; i += 1) {
+          var countValue = parseInt(elements[i].textContent, 10);
+          
+      if (elements[i].parentNode.previousElementSibling === null || elements[i].parentNode.previousElementSibling.querySelectorAll("[data-icon=muted]").length === 0) {
+        count += countValue;
+      } 
+          else {
+                indirectCount += countValue;
+          }
     }
 
-    Franz.setBadge(count);
+    Franz.setBadge(count, indirectCount);
   };
 
   Franz.injectCSS(path.join(__dirname, 'service.css'));