1 files changed, 12 insertions, 0 deletions

diff --git a/recipes/gmail/webview.js b/recipes/gmail/webview.js
index dead877..6db2eec 100644
--- a/recipes/gmail/webview.js
+++ b/recipes/gmail/webview.js
@@ -18,6 +18,8 @@ module.exports = Ferdium => {
     let countImportant = 0;
     let countNonImportant = 0;
     const inboxLinks = document.querySelectorAll('.J-Ke.n0');
+    const spaceAndChatBadges = document.querySelectorAll('span.XU.aH6');
+
     if (inboxLinks.length > 0) {
       const { parentNode } = inboxLinks[0];
       if (parentNode) {
@@ -41,6 +43,16 @@ module.exports = Ferdium => {
         }
       }
     }
+
+    if (spaceAndChatBadges.length > 0) {
+      const arr = [...spaceAndChatBadges];
+      const spaceAndChatCount = arr.reduce(
+        (acc, e) => Ferdium.safeParseInt(e.getInnerHTML()) + acc,
+        0,
+      );
+      countImportant += spaceAndChatCount;
+    }
+
     Ferdium.setBadge(countImportant, countNonImportant);
   };