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/package.json | 2 +-
 recipes/instagram-direct-messages/webview.js   | 8 ++------
 2 files changed, 3 insertions(+), 7 deletions(-)

(limited to 'recipes')

diff --git a/recipes/instagram-direct-messages/package.json b/recipes/instagram-direct-messages/package.json
index 674f8b3..04de3fa 100644
--- a/recipes/instagram-direct-messages/package.json
+++ b/recipes/instagram-direct-messages/package.json
@@ -1,7 +1,7 @@
 {
   "id": "instagram-direct-messages",
   "name": "Instagram Direct Messages",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "license": "MIT",
   "config": {
       "serviceURL": "https://www.instagram.com/direct/inbox/",
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-54-g00ecf