Change isImage function to link instead of URL (#470)

* Change isImage function

* Fix comments

* Fix last comments and lint

1 files changed, 21 insertions, 4 deletions

diff --git a/src/webview/lib/RecipeWebview.ts b/src/webview/lib/RecipeWebview.ts
index 5701ca9c5..a896f1b6e 100644
--- a/src/webview/lib/RecipeWebview.ts
+++ b/src/webview/lib/RecipeWebview.ts
@@ -2,7 +2,9 @@ import { ipcRenderer } from 'electron';
 import { BrowserWindow } from '@electron/remote';
 import { pathExistsSync, readFileSync, existsSync } from 'fs-extra';
 
-const debug = require('../../preload-safe-debug')('Ferdium:Plugin:RecipeWebview');
+const debug = require('../../preload-safe-debug')(
+  'Ferdium:Plugin:RecipeWebview',
+);
 
 class RecipeWebview {
   badgeHandler: any;
@@ -93,12 +95,27 @@ class RecipeWebview {
   }
 
   /**
-   * Find if url contains image
+   * Find if link contains image
    *
    * @param  {string | number | undefined | null} text to be parsed
    */
-  isImage(url): boolean {
-    return /\.(jpg|jpeg|png|webp|avif|gif|svg)$/.test(url.split(/[#?]/)[0]);
+  isImage(link): boolean {
+    if (typeof link === 'undefined') {
+      return false;
+    }
+
+    const { role } = link.dataset;
+
+    if (typeof role !== 'undefined') {
+      const roles = ['img'];
+      return roles.includes(role);
+    }
+
+    const url = link.getAttribute('href');
+
+    const regex = /\.(jpg|jpeg|png|webp|avif|gif|svg)($|\?|:)/;
+
+    return regex.test(url.split(/[#?]/)[0]);
   }
 
   /**