From cea769dbe833d1a8757c8017940e59f6a4e578b5 Mon Sep 17 00:00:00 2001
From: Victor B <39555268+victorbnl@users.noreply.github.com>
Date: Thu, 27 Apr 2023 23:58:21 +0200
Subject: Fix Instagram Direct Messages count (#345)

---
 recipes/instagram-direct-messages/webview.js | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

(limited to 'recipes/instagram-direct-messages/webview.js')

diff --git a/recipes/instagram-direct-messages/webview.js b/recipes/instagram-direct-messages/webview.js
index b59b9f2..290f596 100644
--- a/recipes/instagram-direct-messages/webview.js
+++ b/recipes/instagram-direct-messages/webview.js
@@ -14,12 +14,8 @@ setInterval(() => {
 
 module.exports = (Ferdium) => {
   const getMessages = () => {
-    let count = 0;
-    const elements = document.querySelectorAll('.rBNOH.soMvl');
-    if (elements.length > 0) {
-      count += 1;
-    }
-    Ferdium.setBadge(count);
+    const element = document.querySelector('a[href^="/direct/inbox"]');
+    Ferdium.setBadge(element ? Ferdium.safeParseInt(element.textContent.match(/\d+/)[0]) : 0);
   };
 
   Ferdium.loop(getMessages);
-- 
cgit v1.2.3-70-g09d2