Fixing group unread count in telegram recipe.

2 files changed, 11 insertions, 7 deletions

diff --git a/recipes/telegram/package.json b/recipes/telegram/package.json
index 15fcaf7..c5f7043 100644
--- a/recipes/telegram/package.json
+++ b/recipes/telegram/package.json
@@ -1,7 +1,7 @@
 {
   "id": "telegram",
   "name": "Telegram",
-  "version": "3.0.1",
+  "version": "3.0.2",
   "description": "Telegram",
   "main": "index.js",
   "author": "Stefan Malzner <stefan@adlk.io>",
diff --git a/recipes/telegram/webview.js b/recipes/telegram/webview.js
index 1d33d2f..a8c2ddd 100644
--- a/recipes/telegram/webview.js
+++ b/recipes/telegram/webview.js
@@ -7,16 +7,20 @@ const path = require('path');
 module.exports = Franz => {
   const getMessages = function getMessages() {
     let count = 0;
-    const elements = document.querySelectorAll('.unread');
-    if (elements) {
-      for (let i = 0; i < elements.length; i += 1) {
-        if (parseInt(elements[i].innerText)) {
-          count += parseInt(elements[i].innerText)
+    let count_sec = 0
+    const elements = document.querySelectorAll('.rp');
+    for (let i = 0; i < elements.length; i += 1) {
+      if (elements[i].querySelector('.dialog-subtitle-badge') && (!isNaN(parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText)) )) {
+        if (parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText) != "" && (elements[i].dataset.peerId > 0)) {
+          count = +count + parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText)
+        }
+        else {
+          count_sec = +count_sec + parseInt(elements[i].querySelector('.dialog-subtitle-badge').innerText)
         }
       }
     }
 
-    Franz.setBadge(count);
+    Franz.setBadge(count, count_sec);
   };
 
   Franz.injectCSS(path.join(__dirname, 'service.css'));