From 2b273a31f5bde30c26b87ebe958014936932b467 Mon Sep 17 00:00:00 2001
From: Kenneth Hendricks <50819541+kenhendricks00@users.noreply.github.com>
Date: Fri, 8 Nov 2024 16:58:53 -0500
Subject: [PATCH] Update background.js

---
 src/js/background.js | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/js/background.js b/src/js/background.js
index 31421f2..cbd9777 100644
--- a/src/js/background.js
+++ b/src/js/background.js
@@ -195,6 +195,10 @@ function updatePageAction(status, tabId) {
       19: "../res/icons/fmhy_19.png",
       38: "../res/icons/fmhy_38.png",
     },
+    extension_page: {
+      19: "../res/ext_icon_144.png",
+      38: "../res/ext_icon_144.png",
+    },
     default: {
       19: "../res/icons/default_19.png",
       38: "../res/icons/default_38.png",
@@ -230,6 +234,17 @@ function checkSiteAndUpdatePageAction(tabId, url) {
   const normalizedUrl = normalizeUrl(url.trim());
   const rootUrl = extractRootUrl(normalizedUrl);
 
+  // Check if URL is an extension page
+  const isExtensionPage =
+    url.includes(browserAPI.runtime.getURL("pub/warning-page.html")) ||
+    url.includes(browserAPI.runtime.getURL("pub/settings-page.html")) ||
+    url.includes(browserAPI.runtime.getURL("pub/welcome-page.html"));
+
+  if (isExtensionPage) {
+    updatePageAction("extension_page", tabId);
+    return;
+  }
+
   const isUnsafe =
     unsafeSitesRegex?.test(rootUrl) || unsafeSitesRegex?.test(normalizedUrl);
   const isPotentiallyUnsafe =