Class MockUserScripts

packages/chrome-extension-mock/user_scripts.ts:1–120

export default class MockUserScripts {
  // ---- Types ----
  public static ExecutionWorld = {
    MAIN: "MAIN",
    USER_SCRIPT: "USER_SCRIPT",
  } as const;

  public scripts: any[] = [];
  public worlds: any[] = [];

  // ---- configureWorld ----
  configureWorld(properties: any): Promise<void>;
  configureWorld(properties: any, callback: () => void): void;
  configureWorld(properties: any, callback?: () => void): Promise<void> | void {
    // console.log("configureWorld called with:", properties);
    this.worlds.push(properties);
    if (callback) callback();
    else return Promise.resolve();
  }

  // ---- getScripts ----
  getScripts(filter?: any): Promise<any[]>;
  getScripts(filter: any, callback: (scripts: any[]) => void): void;
  getScripts(filter?: any, callback?: (scripts: any[]) => void): Promise<any[]> | void {
    // console.log("getScripts called with:", filter);
    let result = this.scripts;
    if (filter?.ids) {
      result = this.scripts.filter((s) => filter.ids.includes(s.id));
    }
    if (callback) callback(result);
    else return Promise.resolve(result);
  }

  // ---- getWorldConfigurations ----
  getWorldConfigurations(): Promise<any[]>;
  getWorldConfigurations(callback: (worlds: any[]) => void): void;
  getWorldConfigurations(callback?: (worlds: any[]) => void): Promise<any[]> | void {
    // console.log("getWorldConfigurations called");
    if (callback) callback(this.worlds);
    else return Promise.resolve(this.worlds);
  }

  // ---- execute ----
  execute(injection: any): Promise<any[]>;
  execute(injection: any, callback: (result: any[]) => void): void;
  execute(injection: any, callback?: (result: any[]) => void): Promise<any[]> | void {
    // console.log("execute called with:", injection);
    const result = {
      documentId: "dummy-doc-id",
      frameId: injection.target.frameIds?.[0] ?? 0,
      result: "dummy-result",
    };
    if (callback) callback([result]);
    else return Promise.resolve([result]);
  }

  // ---- register ----
  register(scripts: any[]): Promise<void>;
  register(scripts: any[], callback: () => void): void;
  register(scripts: any[], callback?: () => void): Promise<void> | void {