Minor refactoring

3 files changed, 15 insertions, 22 deletions

diff --git a/src/webview/badge.ts b/src/webview/badge.ts
index afecd22d4..b33d05255 100644
--- a/src/webview/badge.ts
+++ b/src/webview/badge.ts
@@ -1,29 +1,16 @@
 import { ipcRenderer } from 'electron';
+import { safeParseInt } from '../jsUtils';
 
 const debug = require('../preload-safe-debug')('Ferdium:Plugin:BadgeHandler');
 
 export default class BadgeHandler {
-  // 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 = Number.parseInt(text.toString(), 10);
-    const adjustedNumber = Number.isNaN(parsedNumber) ? 0 : parsedNumber;
-    return Math.max(adjustedNumber, 0);
-  }
-
   setBadge(
     direct: string | number | undefined | null,
     indirect: string | number | undefined | null,
   ) {
     const count = {
-      direct: this.safeParseInt(direct),
-      indirect: this.safeParseInt(indirect),
+      direct: safeParseInt(direct),
+      indirect: safeParseInt(indirect),
     };
 
     debug('Sending badge count to host: %j', count);
diff --git a/src/webview/lib/RecipeWebview.ts b/src/webview/lib/RecipeWebview.ts
index 20be3f866..436525b9e 100644
--- a/src/webview/lib/RecipeWebview.ts
+++ b/src/webview/lib/RecipeWebview.ts
@@ -1,6 +1,7 @@
 import { ipcRenderer } from 'electron';
 import { BrowserWindow } from '@electron/remote';
 import { pathExistsSync, readFileSync, existsSync } from 'fs-extra';
+import { safeParseInt } from '../../jsUtils';
 
 const debug = require('../../preload-safe-debug')(
   'Ferdium:Plugin:RecipeWebview',
@@ -91,7 +92,7 @@ class RecipeWebview {
    * @param  {string | number | undefined | null} text to be parsed
    */
   safeParseInt(text) {
-    return this.badgeHandler.safeParseInt(text);
+    return safeParseInt(text);
   }
 
   /**
diff --git a/src/webview/recipe.ts b/src/webview/recipe.ts
index dc3f39401..ed45192d3 100644
--- a/src/webview/recipe.ts
+++ b/src/webview/recipe.ts
@@ -44,7 +44,7 @@ import {
 } from './spellchecker';
 
 import { DEFAULT_APP_SETTINGS } from '../config';
-import { ifUndefined } from '../jsUtils';
+import { ifUndefined, safeParseInt } from '../jsUtils';
 import { AppStore } from '../@types/stores.types';
 import Service from '../models/Service';
 
@@ -116,10 +116,15 @@ window.open = (url, frameName, features): WindowProxy | null => {
 // then overwrite the corresponding field of the window object by injected JS.
 contextBridge.exposeInMainWorld('ferdium', {
   open: window.open,
-  setBadge: (direct, indirect) => badgeHandler.setBadge(direct, indirect),
-  safeParseInt: text => badgeHandler.safeParseInt(text),
-  setDialogTitle: title => dialogTitleHandler.setDialogTitle(title),
-  displayNotification: (title, options) =>
+  setBadge: (
+    direct: string | number | null | undefined,
+    indirect: string | number | null | undefined,
+  ) => badgeHandler.setBadge(direct, indirect),
+  safeParseInt: (text: string | number | null | undefined) =>
+    safeParseInt(text),
+  setDialogTitle: (title: string | null | undefined) =>
+    dialogTitleHandler.setDialogTitle(title),
+  displayNotification: (title: string, options: any) =>
     notificationsHandler.displayNotification(title, options),
   getDisplayMediaSelector,
 });