From 88f70a8355b5eb7e7041ba4690d9709a1a5e5d75 Mon Sep 17 00:00:00 2001
From: mmso <mmso@protonmail.com>
Date: Tue, 2 Feb 2021 21:46:05 +0100
Subject: Fix null element for proton-mail (#432)

---
 recipes/proton-mail/webview.js | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/recipes/proton-mail/webview.js b/recipes/proton-mail/webview.js
index 8d85afd..3d8a30e 100644
--- a/recipes/proton-mail/webview.js
+++ b/recipes/proton-mail/webview.js
@@ -1,7 +1,16 @@
 module.exports = Franz => {
     function getMessages() {
-        const count = document.querySelector('.navigationItem-counter').innerText
-        Franz.setBadge(count ? Number(count.substring(1, count.length - 1)) : 0)
+        const element = document.querySelector('.navigationItem-counter')
+        if (!element) {
+            return
+        }
+        const text = element.innerText
+        const count = Number(text.substring(1, text.length - 1))
+        if (Number.isNaN(count)) {
+            return
+        }
+        Franz.setBadge(count)
     }
+
     Franz.loop(getMessages)
 }
-- 
cgit v1.2.3-70-g09d2