Merge pull request #307 from kris7t/gmail-unread-count

More careful Gmail unread count detection

1 files changed, 6 insertions, 3 deletions

diff --git a/recipes/gmail/webview.js b/recipes/gmail/webview.js
index bc6e4a0..a2c712e 100644
--- a/recipes/gmail/webview.js
+++ b/recipes/gmail/webview.js
@@ -9,9 +9,12 @@ module.exports = (Franz) => {
   const getMessages = function getMessages() {
     let count = 0;
 
-    if (document.getElementsByClassName('bsU').length > 0) {
-      if (document.getElementsByClassName('bsU')[0].innerHTML != null) {
-        count = parseInt(document.getElementsByClassName('bsU')[0].innerHTML.trim(), 10);
+    const inboxLinks = document.getElementsByClassName('J-Ke n0');
+    if (inboxLinks.length > 0) {
+      const inboxLink = inboxLinks[0];
+      const unreadCounts = inboxLink.parentNode.parentNode.getElementsByClassName('bsU');
+      if (unreadCounts.length > 0) {
+        count = parseInt(unreadCounts[0].innerHTML.trim(), 10);
       }
     }