Fixing unread count in telegram-react

2 files changed, 4 insertions, 6 deletions

diff --git a/recipes/telegram-react/package.json b/recipes/telegram-react/package.json
index 1b40ab7..f6f11f3 100644
--- a/recipes/telegram-react/package.json
+++ b/recipes/telegram-react/package.json
@@ -1,7 +1,7 @@
 {
   "id": "telegram-react",
   "name": "Telegram React",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Telegram React",
   "main": "index.js",
   "author": "Amine Mouafik <amine@mouafik.fr>",
diff --git a/recipes/telegram-react/webview.js b/recipes/telegram-react/webview.js
index 223e914..8bc3839 100644
--- a/recipes/telegram-react/webview.js
+++ b/recipes/telegram-react/webview.js
@@ -1,15 +1,13 @@
 "use strict";
 
-const path = require("path");
-
 module.exports = Franz => {
   const getMessages = function getMessages() {
     let count = 0;
-    const elements = document.querySelectorAll(".dialog-badge:not(.dialog-badge-muted)");
+    const elements = document.querySelectorAll(".chatlist > li:not(.is-muted)");
     if (elements) {
       for (let i = 0; i < elements.length; i += 1) {
-        if (elements[i].querySelector("span").innerHTML !== 0) {
-          count += parseInt(elements[i].querySelector("span").innerHTML);
+        if (elements[i].querySelector(".unread") && elements[i].querySelector(".unread").innerHTML !== 0) {
+          count += parseInt(elements[i].querySelector(".unread").innerHTML);
         }
       }
     }