Class MCPClientWrapper

benchmark/helpers.ts:259–335

 * ```
 */
export class MCPClientWrapper {
  private client: Client | null = null;

  /**
   * Initializes the Chrome DevTools MCP server and establishes connection.
   *
   * @param options - Configuration options
   * @param options.headless - Run browser in headless mode (default: true)
   * @returns This wrapper instance for chaining
   * @throws Error if connection fails
   */
  async connect(options?: { headless?: boolean }): Promise<this> {
    const headless = options?.headless ?? true;

    cli.info("Starting Chrome DevTools MCP server...");

    const transport = new StdioClientTransport({
      command: "npx",
      args: [
        "@mcp-b/chrome-devtools-mcp@latest",
        ...(headless ? ["--headless"] : []),
      ],
    });

    this.client = new Client({
      name: "benchmark-client",
      version: "1.0.0",
    });

    await this.client.connect(transport);
    cli.success("MCP client connected");

    const tools = await this.client.listTools();
    cli.info(`Available MCP tools: ${tools.tools.length}`);

    // Set viewport to 14-inch MacBook size for realistic screenshots
    await this.callTool("resize_page", { width: 1512, height: 982 });
    cli.info("Browser viewport set to 1512x982");

    return this;
  }

  /**
   * Executes an MCP tool call with type-safe arguments.
   *
   * @param name - Tool name to call
   * @param args - Arguments to pass to the tool
   * @returns The tool result
   * @throws Error if client is not connected
   */
  async callTool(
    name: string,
    args: Record<string, unknown> = {}
  ): Promise<CompatibilityCallToolResult> {
    if (!this.client) {
      throw new Error("MCP client not initialized. Call connect() first.");
    }
    return this.client.callTool({ name, arguments: args });