support pattern lock website

Author: HoangTran0410

scripts/_test.js

@@ -10,6 +10,19 @@ export default {
   },
 
   popupScript: {
+    onClick: async () => {
+      const { openWebAndRunScript } = await import("./helpers/utils.js");
+      openWebAndRunScript({
+        url: "https://www.whitescreen.online/blue-screen-of-death-windows/",
+        func: () => {
+          setTimeout(() => {
+            document.querySelector(".full-screen").click();
+          }, 1000);
+        },
+        focusImmediately: true,
+        waitUntilLoadEnd: true,
+      });
+    },
     _onClick: async () => {
       const { getCurrentTab, showLoading } = await import("./helpers/utils.js");
       const tab = await getCurrentTab();

scripts/auto_lockWebsite.js

@@ -80,6 +80,19 @@ export async function checkPass(reason) {
   return false;
 }
 
+function matchOneOfPatterns(url, patterns) {
+  for (let pattern of patterns) {
+    const regex = new RegExp(
+      pattern
+        .split("*")
+        .map((part) => part.replace(/[.*+?^${}()|[\]\\]/g, "\\$&"))
+        .join(".*")
+    );
+    if (regex.test(url)) return pattern;
+  }
+  return false;
+}
+
 const locker = {
   getLockedWebsites: async () => {
     let key = lockedWebsiteKey;
@@ -109,7 +122,7 @@ const locker = {
     let pass = data?.[key];
     return pass;
   },
-  lock: (pass) => {
+  lock: (pass, matchedPattern) => {
     const id = "ufs_auto_lock_website_overlay";
     const idStyle = id + "-style";
 
@@ -235,7 +248,7 @@ const locker = {
         inputPass.value = "";
 
         if (!unlockTemporarly.checked) {
-          locker.removeLockedWebsite(location.hostname);
+          locker.removeLockedWebsite(matchedPattern);
         }
       }
     });
@@ -301,9 +314,8 @@ export default {
         if (pass != null) {
           locker.getLockedWebsites().then((websites) => {
             let hostname = location.hostname;
-            if (websites.includes(hostname)) {
-              locker.lock(pass);
-            }
+            let matchedPattern = matchOneOfPatterns(hostname, websites);
+            if (matchedPattern) locker.lock(pass, matchedPattern);
           });
         }
       });

scripts/auto_lockWebsite_main.js

@@ -76,8 +76,8 @@ function initAddSite() {
     Swal.fire({
       title: t({ vi: "Khoá trang web", en: "Lock website" }),
       inputLabel: t({
-        vi: "Nhập link trang web muốn khoá",
-        en: "Enter website url want to lock",
+        vi: "Nhập link/tên trang web muốn khoá\nVí dụ:\n+ facebook\n+ *.facebook.com\n+ https://www.facebook.com",
+        en: "Enter website url/name want to lock\nE.g:\n+ facebook\n+ *.facebook.com\n+ https://www.facebook.com",
       }),
       input: "text",
       inputAttributes: {
@@ -99,12 +99,12 @@ function initAddSite() {
       if (!isConfirmed) return;
 
       let hostname = value;
-      try {
-        let url = new URL(value);
-        hostname = url.hostname;
-      } catch (error) {
-        console.log(error);
-      }
+      // try {
+      //   let url = new URL(value);
+      //   hostname = url.hostname;
+      // } catch (error) {
+      //   console.log(error);
+      // }
       addSite(hostname);
     });
   });

scripts/helpers/utils.js

@@ -390,8 +390,8 @@ export const runScriptFileInCurrentTab = async (scriptFile, world = "MAIN") => {
 export async function openWebAndRunScript({
   url,
   func,
-  args,
-  world,
+  args = [],
+  world = "MAIN",
   waitUntilLoadEnd = true,
   focusAfterRunScript = true,
   closeAfterRunScript = false,