refactor: expose safeParseInt so that all recipes can use the same for consistent error handling

3 files changed, 20 insertions, 4 deletions

diff --git a/src/webview/badge.ts b/src/webview/badge.ts
index b1da36271..753e90fef 100644
--- a/src/webview/badge.ts
+++ b/src/webview/badge.ts
@@ -12,11 +12,16 @@ export class BadgeHandler {
     };
   }
 
-  _normalizeNumber(count: string | number) {
+  // TODO: Need to extract this into a utility class and reuse outside of the recipes
+  safeParseInt(text: string | number | undefined | null) {
+    if (text === undefined || text === null) {
+      return 0;
+    }
+
     // Parse number to integer
     // This will correct errors that recipes may introduce, e.g.
     // by sending a String instead of an integer
-    const parsedNumber = parseInt(count.toString(), 10);
+    const parsedNumber = parseInt(text.toString(), 10);
     const adjustedNumber = Number.isNaN(parsedNumber) ? 0 : parsedNumber;
     return Math.max(adjustedNumber, 0);
   }
@@ -28,8 +33,8 @@ export class BadgeHandler {
     }
 
     const count = {
-      direct: this._normalizeNumber(direct),
-      indirect: this._normalizeNumber(indirect),
+      direct: this.safeParseInt(direct),
+      indirect: this.safeParseInt(indirect),
     };
 
     debug('Sending badge count to host', count);
diff --git a/src/webview/lib/RecipeWebview.js b/src/webview/lib/RecipeWebview.js
index 96caa125e..2bd6bad8d 100644
--- a/src/webview/lib/RecipeWebview.js
+++ b/src/webview/lib/RecipeWebview.js
@@ -47,6 +47,15 @@ class RecipeWebview {
   }
 
   /**
+   * Safely parse the given text into an integer
+   *
+   * @param  {string | number | undefined | null} text to be parsed
+   */
+  safeParseInt(text) {
+    return this.badgeHandler.safeParseInt(text);
+  }
+
+  /**
    * Injects the contents of a CSS file into the current webview
    *
    * @param {Array} files     CSS files that should be injected. This must
diff --git a/src/webview/recipe.js b/src/webview/recipe.js
index f88b94cfe..1981c8731 100644
--- a/src/webview/recipe.js
+++ b/src/webview/recipe.js
@@ -102,6 +102,8 @@ contextBridge.exposeInMainWorld('ferdi', {
   open: window.open,
   setBadge: (direct, indirect) =>
     badgeHandler.setBadge(direct, indirect),
+  safeParseInt: (text) =>
+    badgeHandler.safeParseInt(text),
   displayNotification: (title, options) =>
     notificationsHandler.displayNotification(title, options),
   getDisplayMediaSelector,