🐛 兼容TM的GM_info #478

Author: CodFrm

src/app/service/content/content.ts

@@ -2,6 +2,7 @@ import { ScriptRunResouce } from "@App/app/repo/scripts";
 import { Client, sendMessage } from "@Packages/message/client";
 import { CustomEventMessage } from "@Packages/message/custom_event_message";
 import { forwardMessage, Message, MessageSend, Server } from "@Packages/message/server";
+import { GMInfoEnv } from "./gm_api";
 
 // content页的处理
 export default class ContentRuntime {
@@ -12,7 +13,7 @@ export default class ContentRuntime {
     private msg: Message
   ) {}
 
-  start(scripts: ScriptRunResouce[]) {
+  start(scripts: ScriptRunResouce[], envInfo: GMInfoEnv) {
     this.extServer.on("runtime/emitEvent", (data) => {
       // 转发给inject
       return sendMessage(this.msg, "inject/runtime/emitEvent", data);
@@ -101,6 +102,6 @@ export default class ContentRuntime {
       }
     );
     const client = new Client(this.msg, "inject");
-    client.do("pageLoad", { scripts });
+    client.do("pageLoad", { scripts, envInfo });
   }
 }

src/app/service/content/exec_script.test.ts

@@ -4,6 +4,7 @@ import { compileScript, compileScriptCode } from "./utils";
 import { ExtVersion } from "@App/app/const";
 import { initTestEnv } from "@Tests/utils";
 import { describe, expect, it } from "vitest";
+import { GMInfoEnv } from "./gm_api";
 
 initTestEnv();
 
@@ -20,9 +21,18 @@ const scriptRes = {
     none: true,
   },
 } as unknown as ScriptRunResouce;
+const envInfo: GMInfoEnv = {
+  sandboxMode: "raw",
+  userAgentData: {
+    brands: [],
+    mobile: false,
+    platform: "",
+  },
+  isIncognito: false,
+};
 
 // @ts-ignore
-const noneExec = new ExecScript(scriptRes);
+const noneExec = new ExecScript(scriptRes, undefined, undefined, undefined, envInfo, undefined);
 
 const scriptRes2 = {
   id: 0,
@@ -37,7 +47,7 @@ const scriptRes2 = {
 } as unknown as ScriptRunResouce;
 
 // @ts-ignore
-const sandboxExec = new ExecScript(scriptRes2);
+const sandboxExec = new ExecScript(scriptRes2, undefined, undefined, undefined, envInfo, undefined);
 
 describe("GM_info", () => {
   it("none", async () => {

src/app/service/content/exec_script.ts

@@ -1,6 +1,6 @@
 import LoggerCore from "@App/app/logger/core";
 import Logger from "@App/app/logger/logger";
-import GMApi from "./gm_api";
+import GMApi, { GMInfoEnv } from "./gm_api";
 import { compileScript, createContext, proxyContext, ScriptFunc } from "./utils";
 import { Message } from "@Packages/message/server";
 import { ScriptLoadInfo } from "../service_worker/runtime";
@@ -40,6 +40,7 @@ export default class ExecScript {
     envPrefix: "content" | "offscreen",
     message: Message,
     code: string | ScriptFunc,
+    envInfo: GMInfoEnv,
     thisContext?: { [key: string]: any }
   ) {
     this.scriptRes = scriptRes;
@@ -48,7 +49,7 @@ export default class ExecScript {
       uuid: this.scriptRes.uuid,
       name: this.scriptRes.name,
     });
-    this.GM_info = GMApi.GM_info(this.scriptRes);
+    this.GM_info = GMApi.GM_info(envInfo, this.scriptRes);
     // 构建脚本资源
     if (typeof code === "string") {
       this.scriptFunc = compileScript(code);

src/app/service/content/exec_warp.ts

@@ -1,7 +1,7 @@
-import { ScriptRunResouce } from "@App/app/repo/scripts";
 import ExecScript from "./exec_script";
 import { Message } from "@Packages/message/server";
 import { ScriptLoadInfo } from "../service_worker/runtime";
+import { GMInfoEnv } from "./gm_api";
 
 export class CATRetryError {
   msg: string;
@@ -64,7 +64,16 @@ export class BgExecScriptWarp extends ExecScript {
     };
     // @ts-ignore
     thisContext.CATRetryError = CATRetryError;
-    super(scriptRes, "offscreen", message, scriptRes.code, thisContext);
+    let envInfo: GMInfoEnv = {
+      sandboxMode: "raw",
+      userAgentData: {
+        brands: [],
+        mobile: false,
+        platform: "",
+      },
+      isIncognito: false,
+    };
+    super(scriptRes, "offscreen", message, scriptRes.code, envInfo, thisContext);
     this.setTimeout = setTimeout;
     this.setInterval = setInterval;
   }

src/app/service/content/gm_api.ts

@@ -21,6 +21,12 @@ export interface ApiValue {
   param: ApiParam;
 }
 
+export interface GMInfoEnv {
+  userAgentData: typeof GM_info.userAgentData;
+  sandboxMode: typeof GM_info.sandboxMode;
+  isIncognito: typeof GM_info.isIncognito;
+}
+
 export class GMContext {
   static apis: Map<string, ApiValue> = new Map();
 
@@ -119,7 +125,7 @@ export default class GMApi {
   }
 
   // 获取脚本信息和管理器信息
-  static GM_info(script: ScriptLoadInfo) {
+  static GM_info(envInfo: GMInfoEnv, script: ScriptLoadInfo) {
     const options = {
       description: script.metadata.description?.[0] || null,
       matches: script.metadata.match || [],
@@ -133,12 +139,13 @@ export default class GMApi {
       connects: script.metadata.connect || [],
     };
     return {
-      downloadMode: "browser",
-      // isIncognito
+      downloadMode: "native",
+      isIncognito: envInfo.isIncognito,
       // relaxedCsp
-      // sandboxMode
+      sandboxMode: envInfo.sandboxMode,
       scriptWillUpdate: true,
       scriptHandler: "ScriptCat",
+      userAgentData: envInfo.userAgentData,
       // "" => null
       scriptUpdateURL: script.downloadUrl || null,
       scriptMetaStr: script.metadataStr,

src/app/service/content/inject.ts

@@ -5,14 +5,16 @@ import { getStorageName } from "@App/pkg/utils/utils";
 import { EmitEventRequest, ScriptLoadInfo } from "../service_worker/runtime";
 import { ExternalWhitelist } from "@App/app/const";
 import { sendMessage } from "@Packages/message/client";
+import { GMInfoEnv } from "./gm_api";
 
 export class InjectRuntime {
   execList: ExecScript[] = [];
 
   constructor(
     private server: Server,
     private msg: Message,
-    private scripts: ScriptLoadInfo[]
+    private scripts: ScriptLoadInfo[],
+    private envInfo: GMInfoEnv
   ) {}
 
   start() {
@@ -84,7 +86,7 @@ export class InjectRuntime {
   execScript(script: ScriptLoadInfo, scriptFunc: ScriptFunc) {
     // @ts-ignore
     delete window[script.flag];
-    const exec = new ExecScript(script, "content", this.msg, scriptFunc);
+    const exec = new ExecScript(script, "content", this.msg, scriptFunc, this.envInfo);
     this.execList.push(exec);
     // 注入css
     if (script.metadata["require-css"]) {

src/app/service/service_worker/client.ts

@@ -13,6 +13,7 @@ import { Subscribe } from "@App/app/repo/subscribe";
 import { Permission } from "@App/app/repo/permission";
 import { ResourceBackup } from "@App/pkg/backup/struct";
 import { VSCodeConnect } from "../offscreen/vscode-connect";
+import { GMInfoEnv } from "../content/gm_api";
 
 export class ServiceWorkerClient extends Client {
   constructor(msg: MessageSend) {
@@ -129,7 +130,7 @@ export class ScriptClient extends Client {
     if (urls.length == 0) {
       return;
     }
-    const results = await Promise.allSettled(
+    const results = (await Promise.allSettled(
       urls.map(async (url) => {
         const formattedResult = await this.formatUrl(url);
         if (formattedResult instanceof Object) {
@@ -139,7 +140,7 @@ export class ScriptClient extends Client {
         }
       })
       // this.do 只会resolve 不会reject
-    ) as PromiseFulfilledResult<{ success: boolean; msg: string }>[];
+    )) as PromiseFulfilledResult<{ success: boolean; msg: string }>[];
     const stat = results.reduce(
       (obj, result, index) => {
         if (result.value.success) {
@@ -201,7 +202,7 @@ export class RuntimeClient extends Client {
     return this.do("stopScript", uuid);
   }
 
-  pageLoad(): Promise<{ flag: string; scripts: ScriptRunResouce[] }> {
+  pageLoad(): Promise<{ flag: string; scripts: ScriptRunResouce[]; envInfo: GMInfoEnv }> {
     return this.do("pageLoad");
   }
 

src/app/service/service_worker/runtime.ts

@@ -28,6 +28,7 @@ import { ResourceService } from "./resource";
 import { LocalStorageDAO } from "@App/app/repo/localStorage";
 import Logger from "@App/app/logger/logger";
 import { getMetadataStr, getUserConfigStr } from "@App/pkg/utils/script";
+import { GMInfoEnv } from "../content/gm_api";
 
 // 为了优化性能，存储到缓存时删除了code、value与resource
 export interface ScriptMatchInfo extends ScriptRunResouce {
@@ -58,6 +59,7 @@ export class RuntimeService {
 
   isEnableDeveloperMode = false;
   isEnableUserscribe = true;
+  userAgentData: typeof GM_info.userAgentData = {};
 
   constructor(
     private systemConfig: SystemConfig,
@@ -72,6 +74,25 @@ export class RuntimeService {
     this.logger = LoggerCore.logger({ component: "runtime" });
   }
 
+  async initUserAgentData() {
+    // @ts-ignore
+    if (navigator.userAgentData) {
+      // @ts-ignore
+      this.userAgentData = {
+        // @ts-ignore
+        brands: navigator.userAgentData.brands,
+        // @ts-ignore
+        mobile: navigator.userAgentData.mobile,
+        // @ts-ignore
+        platform: navigator.userAgentData.platform,
+      };
+      // 处理architecture和bitness
+      const platformInfo = await chrome.runtime.getPlatformInfo();
+      this.userAgentData.architecture = platformInfo.nacl_arch;
+      this.userAgentData.bitness = platformInfo.arch.includes("64") ? "64" : "32";
+    }
+  }
+
   async init() {
     // 启动gm api
     const permission = new PermissionVerify(this.group.group("permission"), this.mq);
@@ -206,6 +227,8 @@ export class RuntimeService {
     });
     // 加载黑名单
     this.loadBlacklist(await this.systemConfig.getBlacklist());
+    // 初始化一下userAgentData
+    this.initUserAgentData();
   }
 
   private loadBlacklist(blacklist: string) {
@@ -515,7 +538,15 @@ export class RuntimeService {
       scripts: enableScript,
     });
 
-    return { flag: scriptFlag, scripts: enableScript };
+    return {
+      flag: scriptFlag,
+      scripts: enableScript,
+      envInfo: {
+        sandboxMode: "raw",
+        isIncognito: chrome.extension.inIncognitoContext,
+        userAgentData: this.userAgentData,
+      } as GMInfoEnv,
+    };
   }
 
   // 停止脚本

src/content.ts

@@ -24,5 +24,5 @@ client.pageLoad().then((data) => {
   const extServer = new Server("content", extMsg);
   // 初始化运行环境
   const runtime = new ContentRuntime(extServer, server, send, msg);
-  runtime.start(data.scripts);
+  runtime.start(data.scripts, data.envInfo);
 });

src/inject.ts

@@ -4,6 +4,7 @@ import { CustomEventMessage } from "@Packages/message/custom_event_message";
 import { Server } from "@Packages/message/server";
 import { InjectRuntime } from "./app/service/content/inject";
 import { ScriptLoadInfo } from "./app/service/service_worker/runtime";
+import { GMInfoEnv } from "./app/service/content/gm_api";
 
 const msg = new CustomEventMessage(MessageFlag, false);
 
@@ -15,9 +16,9 @@ const logger = new LoggerCore({
 
 const server = new Server("inject", msg);
 
-server.on("pageLoad", (data: { scripts: ScriptLoadInfo[] }) => {
+server.on("pageLoad", (data: { scripts: ScriptLoadInfo[]; envInfo: GMInfoEnv }) => {
   logger.logger().debug("inject start");
   // 监听事件
-  const runtime = new InjectRuntime(server, msg, data.scripts);
+  const runtime = new InjectRuntime(server, msg, data.scripts, data.envInfo);
   runtime.start();
 });