feat: force-copy setup scripts

Author: WindRunnerMax

packages/force-copy/src/content/index.ts

@@ -4,15 +4,15 @@ import { onPopupMessage } from "./channel/popup";
 import { LOG_LEVEL, logger } from "@/utils/logger";
 import { initializeWorker } from "./runtime/initialize";
 import { isInIframe } from "@/utils/is";
-import { implantScript } from "./runtime/script";
+import { importScript } from "./runtime/script";
 
 (() => {
   if (__DEV__) {
     !isInIframe && onReceiveReloadMsg();
     logger.setLevel(LOG_LEVEL.INFO);
   }
   logger.info("Content Script Loaded");
-  implantScript();
+  importScript();
   !isInIframe && initializeWorker();
   PCBridge.onPopupMessage(onPopupMessage);
 })();

packages/force-copy/src/manifest/index.ts

@@ -37,7 +37,7 @@ const __MANIFEST__: Record<string, unknown> = {
     service_worker: "worker.js",
   },
   host_permissions: [...__URL_MATCH__],
-  permissions: ["activeTab", "tabs", "scripting"],
+  permissions: ["activeTab", "tabs", "scripting", "management"],
   minimum_chrome_version: "88.0",
 };
 
@@ -61,6 +61,7 @@ if (process.env.PLATFORM === "gecko") {
     "tabs",
     "webRequest",
     "webRequestBlocking",
+    "management",
     ...__URL_MATCH__,
   ];
   __MANIFEST__.browser_specific_settings = {

packages/force-copy/src/worker/index.ts

@@ -1,12 +1,14 @@
 import { LOG_LEVEL, logger } from "@/utils/logger";
-import { implantScript } from "./runtime/script";
+import { importScript } from "./runtime/script";
 import { CWBridge } from "@/bridge/content-worker";
 import { onContentMessage } from "./channel/content";
+import { initializeEvents } from "./runtime/initialize";
 
 (() => {
   if (__DEV__) {
     logger.setLevel(LOG_LEVEL.INFO);
   }
-  implantScript();
+  importScript();
+  initializeEvents();
   CWBridge.onContentMessage(onContentMessage);
 })();

packages/force-copy/src/worker/runtime/initialize.ts

@@ -0,0 +1,53 @@
+import { URL_MATCH } from "@/utils/constant";
+import { cross } from "@/utils/global";
+import { logger } from "@/utils/logger";
+import { NOOP } from "../utils/constant";
+
+export const initializeEvents = () => {
+  const onSetup = () => {
+    logger.info("Register Inject Scripts On Setup");
+    cross.tabs.query({}).then(tabs => {
+      for (const tab of tabs) {
+        const tabId = tab && tab.id;
+        const tabURL = tab && tab.url;
+        if (!tabId || !tabURL) continue;
+        if (!URL_MATCH.some(match => new RegExp(match).test(tabURL))) {
+          continue;
+        }
+        // #IFDEF CHROMIUM
+        // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/scripting/executeScript
+        cross.scripting.executeScript({
+          target: { tabId: tabId, allFrames: true },
+          files: ["content.js"],
+          injectImmediately: true,
+        });
+        cross.scripting.executeScript({
+          target: { tabId: tabId, allFrames: true },
+          files: [process.env.INJECT_FILE + ".js"],
+          injectImmediately: true,
+          world: "MAIN",
+        });
+        // #ENDIF
+        // #IFDEF GECKO
+        // https://developer.mozilla.org/zh-CN/docs/Mozilla/Add-ons/WebExtensions/API/tabs/executeScript
+        cross.tabs
+          .executeScript(tabId, {
+            allFrames: true,
+            file: process.env.INJECT_FILE + ".js",
+          })
+          .catch(NOOP);
+        cross.tabs
+          .executeScript(tabId, {
+            allFrames: true,
+            file: "content.js",
+          })
+          .catch(NOOP);
+        // #ENDIF
+      }
+    });
+  };
+  // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/runtime/onInstalled
+  cross.runtime.onInstalled.addListener(onSetup);
+  // https://developer.mozilla.org/en-US/docs/Mozilla/Add-ons/WebExtensions/API/management/onEnabled
+  cross.management.onEnabled.addListener(onSetup);
+};

packages/force-copy/src/worker/utils/constant.ts

@@ -10,3 +10,5 @@ export const CODE_SUFFIX = [
   `  script.remove();`,
   `}`,
 ].join("\n");
+
+export const NOOP = () => null;