Add setDialogTitle feature to api, WhatsApp and Telegram (#750)

Co-authored-by: Vijay A <vraravam@users.noreply.github.com>

2 files changed, 18 insertions, 7 deletions

diff --git a/recipes/telegram/package.json b/recipes/telegram/package.json
index adb6de6..beda3c3 100644
--- a/recipes/telegram/package.json
+++ b/recipes/telegram/package.json
@@ -1,7 +1,7 @@
 {
   "id": "telegram",
   "name": "Telegram",
-  "version": "3.2.2",
+  "version": "3.2.3",
   "license": "MIT",
   "config": {
     "serviceURL": "https://web.telegram.org",
diff --git a/recipes/telegram/webview.js b/recipes/telegram/webview.js
index 358bdaa..7d2c8e3 100644
--- a/recipes/telegram/webview.js
+++ b/recipes/telegram/webview.js
@@ -8,25 +8,36 @@ function _interopRequireDefault(obj) {
 
 module.exports = Ferdi => {
   const getMessages = () => {
-    let count = 0;
-    let count_sec = 0;
+    let direct = 0;
+    let indirect = 0;
     const elements = document.querySelectorAll('.rp');
     for (const element of elements) {
       const subtitleBadge = element.querySelector('.dialog-subtitle-badge');
       if (subtitleBadge) {
         const parsedValue = Ferdi.safeParseInt(subtitleBadge.textContent);
         if (element.dataset.peerId > 0) {
-          count += parsedValue;
+          direct += parsedValue;
         } else {
-          count_sec += parsedValue;
+          indirect += parsedValue;
         }
       }
     }
 
-    Ferdi.setBadge(count, count_sec);
+    Ferdi.setBadge(direct, indirect);
   };
 
-  Ferdi.loop(getMessages);
+  const getActiveDialogTitle = () => {
+    const element = document.querySelector('.top .peer-title');
+
+    Ferdi.setDialogTitle(element ? element.textContent : '');
+  };
+
+  const loopFunc = () => {
+    getMessages();
+    getActiveDialogTitle();
+  };
+
+  Ferdi.loop(loopFunc);
 
   Ferdi.injectCSS(_path.default.join(__dirname, 'service.css'));
 };