1 files changed, 10 insertions, 6 deletions

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'));