refactor: defensive programming to avoid javascript error for unread badges

1 files changed, 6 insertions, 6 deletions

diff --git a/src/webview/badge.ts b/src/webview/badge.ts
index 753e90fef..024e29b3f 100644
--- a/src/webview/badge.ts
+++ b/src/webview/badge.ts
@@ -26,17 +26,17 @@ export class BadgeHandler {
     return Math.max(adjustedNumber, 0);
   }
 
-  setBadge(direct: string | number, indirect: string | number) {
-    if (this.countCache.direct.toString() === direct.toString()
-        && this.countCache.indirect.toString() === indirect.toString()) {
-      return;
-    }
-
+  setBadge(direct: string | number | undefined | null, indirect: string | number | undefined | null) {
     const count = {
       direct: this.safeParseInt(direct),
       indirect: this.safeParseInt(indirect),
     };
 
+    if (this.countCache.direct.toString() === count.direct.toString()
+        && this.countCache.indirect.toString() === count.indirect.toString()) {
+      return;
+    }
+
     debug('Sending badge count to host', count);
     ipcRenderer.sendToHost('message-counts', count);