优化EarlyStart注入代码

Author: cyfung1031

src/app/service/content/script_executor.ts

@@ -75,10 +75,11 @@ export class ScriptExecutor {
     // 监听 脚本加载
     // 适用于此「通知环境加载完成」代码执行后的脚本加载
     window.addEventListener(`${eventNamePrefix}${messageFlags.scriptLoadComplete}`, (ev) => {
-      const scriptFlag = (ev as CustomEvent).detail?.scriptFlag;
+      const detail = (ev as CustomEvent).detail;
+      const scriptFlag = detail?.scriptFlag;
       if (typeof scriptFlag === "string") {
         ev.preventDefault(); // dispatchEvent 会回传 false -> 分离环境也能得知环境加载代码已执行
-        this.execEarlyScript(scriptFlag);
+        this.execEarlyScript(scriptFlag, detail.scriptInfo);
       }
     });
     // 通知 环境 加载完成
@@ -87,11 +88,11 @@ export class ScriptExecutor {
     window.dispatchEvent(ev);
   }
 
-  execEarlyScript(flag: string) {
+  execEarlyScript(flag: string, scriptInfo: ScriptLoadInfo) {
     const scriptFunc = (window as any)[flag] as PreScriptFunc;
     this.execScriptEntry({
-      scriptLoadInfo: scriptFunc.scriptInfo,
-      scriptFunc: scriptFunc.func,
+      scriptLoadInfo: scriptInfo,
+      scriptFunc: scriptFunc,
       scriptFlag: flag,
       envInfo: {},
     });

src/app/service/content/utils.ts

@@ -105,22 +105,21 @@ export function compilePreInjectScript(
   autoDeleteMountFunction: boolean = false
 ): string {
   const eventNamePrefix = isInjectIntoContent(script.metadata) ? messageFlags.contentFlag : messageFlags.injectFlag;
-  const autoDeleteMountCode = autoDeleteMountFunction ? `try{delete window['${script.flag}']}catch(e){}` : "";
-  return `window['${script.flag}'] = {
-  scriptInfo: ${JSON.stringify(script)},
-  func: function(){${autoDeleteMountCode}${scriptCode}}
-};
-(() => {
-  const f = () => {
-    const ev = new CustomEvent('${eventNamePrefix}${messageFlags.scriptLoadComplete}', 
-    { cancelable: true, detail: { scriptFlag: '${script.flag}' } });
-    return window.dispatchEvent(ev);
-  };
-  const noCbHooked = f();
-  if (noCbHooked) {
-    window.addEventListener('${eventNamePrefix}${messageFlags.envLoadComplete}', f, { once: true });
-  }
-})();
+  const flag = `${script.flag}`;
+  const scriptInfo = { ...script }; // clone
+  // 不需要 code
+  scriptInfo.code = "";
+  const scriptInfoJSON = `${JSON.stringify(scriptInfo)}`;
+  const autoDeleteMountCode = autoDeleteMountFunction ? `try{delete window['${flag}']}catch(e){}` : "";
+  const evScriptLoad = `${eventNamePrefix}${messageFlags.scriptLoadComplete}`;
+  const evEnvLoad = `${eventNamePrefix}${messageFlags.envLoadComplete}`;
+  return `window['${flag}'] = function(){${autoDeleteMountCode}${scriptCode}};
+{
+  let o = { cancelable: true, { scriptFlag: '${flag}', scriptInfo: ${scriptInfoJSON} } },
+  f = () => window.dispatchEvent(new CustomEvent('${evScriptLoad}', o)),
+  needWait = f();
+  if (needWait) window.addEventListener('${evEnvLoad}', f, { once: true });
+}
 `;
 }
 

src/app/service/service_worker/runtime.ts

@@ -43,6 +43,10 @@ import type { CompiledResource, ResourceType } from "@App/app/repo/resource";
 import { CompiledResourceDAO } from "@App/app/repo/resource";
 import { setOnTabURLChanged } from "./url_monitor";
 
+// 避免使用版本号控制导致代码理解混乱
+// 用来清除UserScript API里的旧缓存
+const USERSCRIPTS_REGISTER_CONTROL = "0f5b5b01-eef8-4505-9a8e-b2fc416b2f62";
+
 const ORIGINAL_URLMATCH_SUFFIX = "{ORIGINAL}"; // 用于标记原始URLPatterns的后缀
 
 const runtimeGlobal = {
@@ -241,12 +245,21 @@ export class RuntimeService {
   }
 
   async waitInit() {
-    const [cRuntimeStartFlag, compiledResources, allScripts] = await Promise.all([
+    const [registerControl, cRuntimeStartFlag, compiledResources, allScripts] = await Promise.all([
+      chrome.storage.local.get("userscripts_register_control"),
       cacheInstance.get<boolean>("runtimeStartFlag"),
       this.compiledResourceDAO.all(),
       this.scriptDAO.all(),
     ]);
 
+    if (registerControl?.userscripts_register_control !== USERSCRIPTS_REGISTER_CONTROL) {
+      await Promise.allSettled([
+        chrome.userScripts.unregister(),
+        chrome.scripting.unregisterContentScripts(),
+        chrome.storage.local.set({ userscripts_register_control: USERSCRIPTS_REGISTER_CONTROL }),
+      ]);
+    }
+
     const unregisterScriptIds = [] as string[];
     // 没有 CompiledResources 表示这是 没有启用脚本 或 代码有改变需要重新安装。
     // 这个情况会把所有有效脚本跟Inject&Content脚本先取消注册。后续载入时会重新以新代码注册。