modelstudioai__cli/packages/commands/tests/e2e/helpers.ts

import { dirname, join } from "path";
import { fileURLToPath } from "url";
import {
  cliTimeoutPrefix,
  cliTimeoutSeconds,
  e2eLabelFromMetaUrl,
  isConsoleAuthFailure,
  makeE2eOutputDir,
  parseStdoutJson,
} from "e2e/output";
import { captureRegistryHelp } from "e2e/registry-smoke";
import { runNodeMain, type RunCliResult } from "e2e/runner";
import type { AnyCommand } from "bailian-cli-core";
import { CommandRegistry, resolve } from "bailian-cli-runtime";
import type { E2eRouteExports } from "./topic-routes.ts";

export {
  cliTimeoutPrefix,
  cliTimeoutSeconds,
  e2eLabelFromMetaUrl,
  isConsoleAuthFailure,
  makeE2eOutputDir,
  parseStdoutJson,
};
export type { RunCliResult };

export {
  isBailianE2EEnabled,
  isBailianE2EMediaEnabled,
  isBailianE2EVideoEnabled,
  isChatE2EReady,
  isConnectorE2EReady,
  isConsoleE2EReady,
  isDashScopeE2EReady,
  isImageKbE2EReady,
  isKbAdminE2EReady,
  isMemoryE2EReady,
  isMultimodalChatE2EReady,
  isMultimodalSearchE2EReady,
  isOpenApiE2EReady,
  isOssImportE2EReady,
  isSearchE2EReady,
  isTableKbE2EReady,
  isTableSearchE2EReady,
} from "e2e/gating";

const e2eDir = dirname(fileURLToPath(import.meta.url));
const harnessMainTs = join(e2eDir, "harness", "main.ts");
const registryCache = new WeakMap<E2eRouteExports, Promise<CommandRegistry>>();

/** `packages/commands` 根目录 */
export const commandsPackageRoot = join(e2eDir, "..", "..");

/** E2E fixtures 目录 */
export const e2eFixturesDir = join(e2eDir, "fixtures");

function serializeRoutes(routes: E2eRouteExports): string {
  return JSON.stringify(
    Object.entries(routes).map(([path, exportName]) => ({ path, export: exportName })),
  );
}

function getCommandRegistry(routes: E2eRouteExports): Promise<CommandRegistry> {
  const cachedRegistry = registryCache.get(routes);
  if (cachedRegistry) return cachedRegistry;

  const registryPromise = import("bailian-cli-commands").then((commandExports) => {
    const commandLibrary = commandExports as Record<string, unknown>;
    const commands: Record<string, AnyCommand> = {};
    for (const [path, exportName] of Object.entries(routes)) {
      const command = commandLibrary[exportName];
      if (typeof command !== "object" || command === null || !("run" in command)) {
        throw new Error(`Unknown bailian-cli-commands export: ${exportName}`);
      }
      commands[path] = command as AnyCommand;
    }
    return new CommandRegistry(commands, "bl");
  });
  registryCache.set(routes, registryPromise);
  return registryPromise;
}

/** 使用 topic 最小路由在当前 Vitest worker 中渲染命令 help，不启动 CLI 子进程。 */
export async function captureCommandHelp(
  routes: E2eRouteExports,
  commandPath: string[],
): Promise<string> {
  const registry = await getCommandRegistry(routes);
  const resolution = resolve([...commandPath, "--help"], registry);
  if (resolution.kind !== "help" || resolution.path.join(" ") !== commandPath.join(" ")) {
    throw new Error(`Command help did not resolve: ${commandPath.join(" ")}`);
  }
  return captureRegistryHelp(registry, commandPath);
}

/** 与 runCommandE2e 的结果结构兼容，但 help 在当前 Vitest worker 中渲染。 */
export async function runCommandHelp(
  routes: E2eRouteExports,
  args: string[],
): Promise<RunCliResult> {
  if (args.at(-1) !== "--help") {
    throw new Error("runCommandHelp requires --help as the final argument");
  }
  const stderr = await captureCommandHelp(routes, args.slice(0, -1));
  return { stdout: "", stderr, exitCode: 0 };
}

/**
 * 通过 harness 子进程执行命令 E2E。
 * `routes` 为本用例所需的最小 path → export 映射，不维护全量产品 map。
 */
export async function runCommandE2e(
  routes: E2eRouteExports,
  args: string[],
  envOverrides: NodeJS.ProcessEnv = {},
): Promise<RunCliResult> {
  return runNodeMain(harnessMainTs, args, {
    cwd: commandsPackageRoot,
    env: {
      BAILIAN_E2E_ROUTES: serializeRoutes(routes),
      ...envOverrides,
    },
  });
}