import { vi } from "vitest";
import * as parentCoreApiModule from "../core-api.js";
import * as browserCliSharedModule from "./browser-cli-shared.js";
import * as cliCoreApiModule from "./core-api.js";

type BrowserRequest = { path?: string };
type BrowserRuntimeOptions = { timeoutMs?: number };

export type BrowserManageCall = [unknown, BrowserRequest, BrowserRuntimeOptions | undefined];

const browserManageMocks = vi.hoisted(() => ({
  callBrowserRequest: vi.fn<
    (
      opts: unknown,
      req: BrowserRequest,
      runtimeOpts?: BrowserRuntimeOptions,
    ) => Promise<Record<string, unknown>>
  >(async (_opts: unknown, req: BrowserRequest) =>
    req.path === "/"
      ? {
          enabled: true,
          running: true,
          pid: 1,
          cdpPort: 18800,
          chosenBrowser: "chrome",
          userDataDir: "/tmp/openclaw",
          color: "blue",
          headless: true,
          attachOnly: false,
        }
      : {},
  ),
}));

vi.spyOn(browserCliSharedModule, "callBrowserRequest").mockImplementation(
  browserManageMocks.callBrowserRequest,
);
vi.spyOn(parentCoreApiModule, "runCommandWithRuntime").mockImplementation(
  async (_runtime, action, onError) => {
    try {
      await action();
    } catch (err) {
      onError?.(err);
    }
  },
);
const { createBrowserProgram, getBrowserCliRuntime } =
  await import("./browser-cli.test-support.js");
const browserCliRuntime = getBrowserCliRuntime();
vi.spyOn(cliCoreApiModule.defaultRuntime, "log").mockImplementation(browserCliRuntime.log);
vi.spyOn(cliCoreApiModule.defaultRuntime, "writeJson").mockImplementation(
  browserCliRuntime.writeJson,
);
vi.spyOn(cliCoreApiModule.defaultRuntime, "error").mockImplementation(browserCliRuntime.error);
vi.spyOn(cliCoreApiModule.defaultRuntime, "exit").mockImplementation(browserCliRuntime.exit);

const { registerBrowserManageCommands } = await import("./browser-cli-manage.js");

export function createBrowserManageProgram(params?: { withParentTimeout?: boolean }) {
  const { program, browser, parentOpts } = createBrowserProgram();
  if (params?.withParentTimeout) {
    browser.option("--timeout <ms>", "Timeout in ms", "30000");
  }
  registerBrowserManageCommands(browser, parentOpts);
  return program;
}

export function getBrowserManageCallBrowserRequestMock() {
  return browserManageMocks.callBrowserRequest;
}

export function findBrowserManageCall(path: string): BrowserManageCall | undefined {
  return browserManageMocks.callBrowserRequest.mock.calls.find(
    (call) => (call[1] ?? {}).path === path,
  ) as BrowserManageCall | undefined;
}