Update src/background.js

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
ai-shifu · Oct 7, 2024 · d2ed80d · d2ed80d
1 parent ec7b9a2
commit d2ed80d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/background.js b/src/background.js
@@ -210,8 +210,11 @@ async function createWindow() {
       "aliyun.com",
       "qianwen.aliyun.com"
     ];
-    const parsedHost = urlLib.parse(`https://${domain.startsWith(".") ? domain.substring(1) : domain}`).host;
-    return allowedDomains.includes(parsedHost);
+    const parsedHost = new URL(`https://${domain.startsWith(".") ? domain.substring(1) : domain}`).host;
+    return allowedDomains.some(
+      (allowedDomain) =>
+        parsedHost === allowedDomain || parsedHost.endsWith(`.${allowedDomain}`)
+    );
   }
 
   // Modify the Referer header for each request and special patch for some sites.