mcp-server-diff 2.1.0 → 2.1.6

package/src/runner.ts

@@ -1,902 +0,0 @@
-/**
- * Test runner for MCP server diff
- */
-
-import * as exec from "@actions/exec";
-import * as core from "@actions/core";
-import * as path from "path";
-import * as fs from "fs";
-import { spawn, ChildProcess } from "child_process";
-import { probeServer, probeResultToFiles } from "./probe.js";
-import { createWorktree, removeWorktree, checkout, checkoutPrevious } from "./git.js";
-import type {
-  TestConfiguration,
-  ActionInputs,
-  TestResult,
-  ProbeResult,
-  CustomMessage,
-  PrimitiveCounts,
-} from "./types.js";
-
-interface RunContext {
-  workDir: string;
-  inputs: ActionInputs;
-  compareRef: string;
-}
-
-/**
- * Parse configurations from input
- */
-export function parseConfigurations(
-  input: string | undefined,
-  defaultTransport: "stdio" | "streamable-http",
-  defaultCommand: string,
-  defaultUrl: string
-): TestConfiguration[] {
-  if (!input || input.trim() === "[]" || input.trim() === "") {
-    // Return single default configuration
-    return [
-      {
-        name: "default",
-        transport: defaultTransport,
-        start_command: defaultTransport === "stdio" ? defaultCommand : undefined,
-        server_url: defaultTransport === "streamable-http" ? defaultUrl : undefined,
-      },
-    ];
-  }
-
-  try {
-    const configs = JSON.parse(input) as TestConfiguration[];
-    if (!Array.isArray(configs) || configs.length === 0) {
-      throw new Error("Configurations must be a non-empty array");
-    }
-
-    // Apply defaults to each configuration
-    return configs.map((config) => {
-      const transport = config.transport || defaultTransport;
-      return {
-        ...config,
-        transport,
-        // For stdio, use default command if not specified (appending args if needed)
-        start_command:
-          transport === "stdio" ? config.start_command || defaultCommand : config.start_command,
-        // For HTTP, use default URL if not specified
-        server_url:
-          transport === "streamable-http" ? config.server_url || defaultUrl : config.server_url,
-      };
-    });
-  } catch (error) {
-    core.warning(`Failed to parse configurations: ${error}`);
-    // Return default
-    return [
-      {
-        name: "default",
-        transport: defaultTransport,
-        start_command: defaultTransport === "stdio" ? defaultCommand : undefined,
-        server_url: defaultTransport === "streamable-http" ? defaultUrl : undefined,
-      },
-    ];
-  }
-}
-
-/**
- * Parse custom messages from input
- */
-export function parseCustomMessages(input: string | undefined): CustomMessage[] {
-  if (!input || input.trim() === "[]" || input.trim() === "") {
-    return [];
-  }
-
-  try {
-    const messages = JSON.parse(input) as CustomMessage[];
-    return Array.isArray(messages) ? messages : [];
-  } catch {
-    return [];
-  }
-}
-
-/**
- * Parse headers from input (JSON object or KEY=value format, newline separated)
- */
-export function parseHeaders(input: string | undefined): Record<string, string> {
-  if (!input || input.trim() === "" || input.trim() === "{}") {
-    return {};
-  }
-
-  // Try JSON first
-  try {
-    const parsed = JSON.parse(input);
-    if (typeof parsed === "object" && parsed !== null && !Array.isArray(parsed)) {
-      return parsed as Record<string, string>;
-    }
-  } catch {
-    // Not JSON, try KEY=value format
-  }
-
-  // Fall back to KEY=value format
-  const headers: Record<string, string> = {};
-  const lines = input.split("\n");
-  for (const line of lines) {
-    const trimmed = line.trim();
-    if (!trimmed || trimmed.startsWith("#")) continue;
-
-    const colonIndex = trimmed.indexOf(":");
-    const eqIndex = trimmed.indexOf("=");
-
-    // Support both "Header: value" and "Header=value" formats
-    const sepIndex = colonIndex > 0 ? colonIndex : eqIndex;
-    if (sepIndex > 0) {
-      const key = trimmed.substring(0, sepIndex).trim();
-      const value = trimmed.substring(sepIndex + 1).trim();
-      headers[key] = value;
-    }
-  }
-  return headers;
-}
-
-/**
- * Parse environment variables from string (KEY=value format, newline separated)
- */
-export function parseEnvVars(input?: string): Record<string, string> {
-  const env: Record<string, string> = {};
-  if (!input) return env;
-
-  const lines = input.split("\n");
-  for (const line of lines) {
-    const trimmed = line.trim();
-    if (!trimmed || trimmed.startsWith("#")) continue;
-
-    const eqIndex = trimmed.indexOf("=");
-    if (eqIndex > 0) {
-      const key = trimmed.substring(0, eqIndex);
-      const value = trimmed.substring(eqIndex + 1);
-      env[key] = value;
-    }
-  }
-  return env;
-}
-
-/**
- * Run build commands in a directory
- */
-async function runBuild(dir: string, inputs: ActionInputs): Promise<void> {
-  const options = { cwd: dir };
-
-  if (inputs.installCommand) {
-    core.info(`  Running install: ${inputs.installCommand}`);
-    await exec.exec("sh", ["-c", inputs.installCommand], options);
-  }
-
-  if (inputs.buildCommand) {
-    core.info(`  Running build: ${inputs.buildCommand}`);
-    await exec.exec("sh", ["-c", inputs.buildCommand], options);
-  }
-}
-
-/**
- * Sleep for a specified number of milliseconds
- */
-function sleep(ms: number): Promise<void> {
-  return new Promise((resolve) => setTimeout(resolve, ms));
-}
-
-/**
- * Start an HTTP server process for the given configuration.
- * Returns the spawned process which should be killed after probing.
- */
-async function startHttpServer(
-  config: TestConfiguration,
-  workDir: string,
-  envVars: Record<string, string>
-): Promise<ChildProcess | null> {
-  if (!config.start_command) {
-    return null;
-  }
-
-  core.info(`  Starting HTTP server: ${config.start_command}`);
-
-  // Merge environment variables
-  const env: Record<string, string> = {};
-  for (const [key, value] of Object.entries(process.env)) {
-    if (value !== undefined) {
-      env[key] = value;
-    }
-  }
-  for (const [key, value] of Object.entries(envVars)) {
-    env[key] = value;
-  }
-
-  const serverProcess = spawn("sh", ["-c", config.start_command], {
-    cwd: workDir,
-    env,
-    detached: true,
-    stdio: ["ignore", "pipe", "pipe"],
-  });
-
-  // Log server output for debugging
-  serverProcess.stdout?.on("data", (data) => {
-    core.debug(`  [server stdout]: ${data.toString().trim()}`);
-  });
-  serverProcess.stderr?.on("data", (data) => {
-    core.debug(`  [server stderr]: ${data.toString().trim()}`);
-  });
-
-  // Wait for server to start up
-  const waitMs = config.startup_wait_ms ?? config.pre_test_wait_ms ?? 2000;
-  core.info(`  Waiting ${waitMs}ms for server to start...`);
-  await sleep(waitMs);
-
-  // Check if process is still running
-  if (serverProcess.exitCode !== null) {
-    throw new Error(`HTTP server exited prematurely with code ${serverProcess.exitCode}`);
-  }
-
-  core.info("  HTTP server started");
-  return serverProcess;
-}
-
-/**
- * Stop an HTTP server process
- */
-function stopHttpServer(serverProcess: ChildProcess | null): void {
-  if (!serverProcess) {
-    return;
-  }
-
-  core.info("  Stopping HTTP server...");
-  try {
-    // Kill the process group (negative PID kills the group)
-    if (serverProcess.pid) {
-      process.kill(-serverProcess.pid, "SIGTERM");
-    }
-  } catch (error) {
-    // Process might already be dead
-    core.debug(`  Error stopping server: ${error}`);
-  }
-}
-
-/**
- * Run pre-test command if specified
- */
-async function runPreTestCommand(config: TestConfiguration, workDir: string): Promise<void> {
-  if (config.pre_test_command) {
-    core.info(`  Running pre-test command: ${config.pre_test_command}`);
-    await exec.exec("sh", ["-c", config.pre_test_command], { cwd: workDir });
-
-    if (config.pre_test_wait_ms && config.pre_test_wait_ms > 0) {
-      core.info(`  Waiting ${config.pre_test_wait_ms}ms for service to be ready...`);
-      await sleep(config.pre_test_wait_ms);
-    }
-  }
-}
-
-/**
- * Run post-test command if specified (cleanup)
- */
-async function runPostTestCommand(config: TestConfiguration, workDir: string): Promise<void> {
-  if (config.post_test_command) {
-    core.info(`  Running post-test command: ${config.post_test_command}`);
-    try {
-      await exec.exec("sh", ["-c", config.post_test_command], { cwd: workDir });
-    } catch (error) {
-      // Log but don't fail - cleanup errors shouldn't break the test
-      core.warning(`  Post-test command failed: ${error}`);
-    }
-  }
-}
-
-/**
- * Probe a server with a specific configuration
- * @param useSharedServer - If true, skip starting per-config HTTP server (shared server is already running)
- */
-async function probeWithConfig(
-  config: TestConfiguration,
-  workDir: string,
-  globalEnvVars: Record<string, string>,
-  globalHeaders: Record<string, string>,
-  globalCustomMessages: CustomMessage[],
-  useSharedServer: boolean = false
-): Promise<ProbeResult> {
-  const configEnvVars = parseEnvVars(config.env_vars);
-  const envVars = { ...globalEnvVars, ...configEnvVars };
-  const headers = { ...globalHeaders, ...config.headers };
-  const customMessages = config.custom_messages || globalCustomMessages;
-
-  // Run pre-test command before probing
-  await runPreTestCommand(config, workDir);
-
-  if (config.transport === "stdio") {
-    const command = config.start_command || "";
-    const parts = command.split(/\s+/);
-    const cmd = parts[0];
-    const args = parts.slice(1);
-
-    // Also parse additional args if provided
-    if (config.args) {
-      args.push(...config.args.split(/\s+/));
-    }
-
-    return await probeServer({
-      transport: "stdio",
-      command: cmd,
-      args,
-      workingDir: workDir,
-      envVars,
-      customMessages,
-    });
-  } else {
-    // For HTTP transport, optionally start the server if start_command is provided
-    // Skip if using shared server
-    let serverProcess: ChildProcess | null = null;
-    try {
-      if (config.start_command && !useSharedServer) {
-        serverProcess = await startHttpServer(config, workDir, envVars);
-      }
-
-      return await probeServer({
-        transport: "streamable-http",
-        url: config.server_url,
-        headers,
-        envVars,
-        customMessages,
-      });
-    } finally {
-      // Always stop the server if we started it
-      stopHttpServer(serverProcess);
-    }
-  }
-}
-
-/**
- * Compare two sets of probe result files and return diffs
- */
-function compareResults(
-  branchFiles: Map<string, string>,
-  baseFiles: Map<string, string>
-): Map<string, string> {
-  const diffs = new Map<string, string>();
-
-  // Check all endpoints
-  const allEndpoints = new Set([...branchFiles.keys(), ...baseFiles.keys()]);
-
-  for (const endpoint of allEndpoints) {
-    const branchContent = branchFiles.get(endpoint);
-    const baseContent = baseFiles.get(endpoint);
-
-    if (!branchContent && baseContent) {
-      diffs.set(endpoint, `Endpoint removed in current branch (was present in base)`);
-    } else if (branchContent && !baseContent) {
-      diffs.set(endpoint, `Endpoint added in current branch (not present in base)`);
-    } else if (branchContent !== baseContent) {
-      // Generate a semantic JSON diff
-      const diff = generateJsonDiff(endpoint, baseContent || "", branchContent || "");
-      if (diff) {
-        diffs.set(endpoint, diff);
-      }
-    }
-  }
-
-  return diffs;
-}
-
-/**
- * Generate a semantic JSON diff that shows actual changes
- * rather than line-by-line text comparison
- */
-function generateJsonDiff(name: string, base: string, branch: string): string | null {
-  try {
-    const baseObj = JSON.parse(base);
-    const branchObj = JSON.parse(branch);
-
-    const differences = findJsonDifferences(baseObj, branchObj, "");
-
-    if (differences.length === 0) {
-      return null;
-    }
-
-    const diffLines = [`--- base/${name}.json`, `+++ branch/${name}.json`, ""];
-    diffLines.push(...differences);
-
-    return diffLines.join("\n");
-  } catch {
-    // Fallback to simple diff if JSON parsing fails
-    return generateSimpleTextDiff(name, base, branch);
-  }
-}
-
-/**
- * Recursively find differences between two JSON objects
- */
-function findJsonDifferences(base: unknown, branch: unknown, path: string): string[] {
-  const diffs: string[] = [];
-
-  // Handle null/undefined
-  if (base === null || base === undefined) {
-    if (branch !== null && branch !== undefined) {
-      diffs.push(`+ ${path || "root"}: ${formatValue(branch)}`);
-    }
-    return diffs;
-  }
-
-  if (branch === null || branch === undefined) {
-    diffs.push(`- ${path || "root"}: ${formatValue(base)}`);
-    return diffs;
-  }
-
-  // Handle type mismatch
-  if (typeof base !== typeof branch) {
-    diffs.push(`- ${path || "root"}: ${formatValue(base)}`);
-    diffs.push(`+ ${path || "root"}: ${formatValue(branch)}`);
-    return diffs;
-  }
-
-  // Handle arrays
-  if (Array.isArray(base) && Array.isArray(branch)) {
-    return compareArrays(base, branch, path);
-  }
-
-  // Handle objects
-  if (typeof base === "object" && typeof branch === "object") {
-    const baseObj = base as Record<string, unknown>;
-    const branchObj = branch as Record<string, unknown>;
-    const allKeys = new Set([...Object.keys(baseObj), ...Object.keys(branchObj)]);
-
-    for (const key of allKeys) {
-      const newPath = path ? `${path}.${key}` : key;
-
-      if (!(key in baseObj)) {
-        diffs.push(`+ ${newPath}: ${formatValue(branchObj[key])}`);
-      } else if (!(key in branchObj)) {
-        diffs.push(`- ${newPath}: ${formatValue(baseObj[key])}`);
-      } else {
-        diffs.push(...findJsonDifferences(baseObj[key], branchObj[key], newPath));
-      }
-    }
-    return diffs;
-  }
-
-  // Handle primitives
-  if (base !== branch) {
-    diffs.push(`- ${path}: ${formatValue(base)}`);
-    diffs.push(`+ ${path}: ${formatValue(branch)}`);
-  }
-
-  return diffs;
-}
-
-/**
- * Compare arrays by finding items by their identity (name, uri, etc.)
- */
-function compareArrays(base: unknown[], branch: unknown[], path: string): string[] {
-  const diffs: string[] = [];
-
-  // Try to identify items by name/uri for better diff
-  const baseItems = new Map<string, { item: unknown; index: number }>();
-  const branchItems = new Map<string, { item: unknown; index: number }>();
-
-  base.forEach((item, index) => {
-    const key = getItemKey(item, index);
-    baseItems.set(key, { item, index });
-  });
-
-  branch.forEach((item, index) => {
-    const key = getItemKey(item, index);
-    branchItems.set(key, { item, index });
-  });
-
-  // Find removed items
-  for (const [key, { item }] of baseItems) {
-    if (!branchItems.has(key)) {
-      const itemPath = `${path}[${key}]`;
-      diffs.push(`- ${itemPath}: ${formatValue(item)}`);
-    }
-  }
-
-  // Find added items
-  for (const [key, { item }] of branchItems) {
-    if (!baseItems.has(key)) {
-      const itemPath = `${path}[${key}]`;
-      diffs.push(`+ ${itemPath}: ${formatValue(item)}`);
-    }
-  }
-
-  // Find modified items
-  for (const [key, { item: baseItem }] of baseItems) {
-    const branchEntry = branchItems.get(key);
-    if (branchEntry) {
-      const itemPath = `${path}[${key}]`;
-      diffs.push(...findJsonDifferences(baseItem, branchEntry.item, itemPath));
-    }
-  }
-
-  return diffs;
-}
-
-/**
- * Get a unique key for an array item based on common identifiers
- */
-function getItemKey(item: unknown, index: number): string {
-  if (item === null || item === undefined || typeof item !== "object") {
-    return `#${index}`;
-  }
-
-  const obj = item as Record<string, unknown>;
-
-  // Try common identifier fields
-  if (typeof obj.name === "string") return obj.name;
-  if (typeof obj.uri === "string") return obj.uri;
-  if (typeof obj.uriTemplate === "string") return obj.uriTemplate;
-  if (typeof obj.type === "string" && typeof obj.text === "string") {
-    return `${obj.type}:${String(obj.text).slice(0, 50)}`;
-  }
-  if (typeof obj.method === "string") return obj.method;
-
-  return `#${index}`;
-}
-
-/**
- * Format a value for display in the diff
- */
-function formatValue(value: unknown): string {
-  if (value === null) return "null";
-  if (value === undefined) return "undefined";
-
-  if (typeof value === "string") {
-    // Truncate long strings
-    if (value.length > 100) {
-      return JSON.stringify(value.slice(0, 100) + "...");
-    }
-    return JSON.stringify(value);
-  }
-
-  if (typeof value === "object") {
-    const json = JSON.stringify(value);
-    // Truncate long objects
-    if (json.length > 200) {
-      return json.slice(0, 200) + "...";
-    }
-    return json;
-  }
-
-  return String(value);
-}
-
-/**
- * Extract primitive counts from a probe result
- */
-function extractCounts(result: ProbeResult): PrimitiveCounts {
-  return {
-    tools: result.tools?.tools?.length || 0,
-    prompts: result.prompts?.prompts?.length || 0,
-    resources: result.resources?.resources?.length || 0,
-    resourceTemplates: result.resourceTemplates?.resourceTemplates?.length || 0,
-  };
-}
-
-/**
- * Generate a simple line-by-line diff (fallback for non-JSON)
- */
-function generateSimpleTextDiff(name: string, base: string, branch: string): string | null {
-  const baseLines = base.split("\n");
-  const branchLines = branch.split("\n");
-
-  const diffLines: string[] = [];
-  const maxLines = Math.max(baseLines.length, branchLines.length);
-
-  for (let i = 0; i < maxLines; i++) {
-    const baseLine = baseLines[i];
-    const branchLine = branchLines[i];
-
-    if (baseLine !== branchLine) {
-      if (baseLine !== undefined) {
-        diffLines.push(`- ${baseLine}`);
-      }
-      if (branchLine !== undefined) {
-        diffLines.push(`+ ${branchLine}`);
-      }
-    }
-  }
-
-  if (diffLines.length === 0) {
-    return null;
-  }
-
-  return `--- base/${name}.json\n+++ branch/${name}.json\n${diffLines.join("\n")}`;
-}
-
-/**
- * Start a shared HTTP server for all HTTP transport configurations
- */
-async function startSharedHttpServer(
-  command: string,
-  workDir: string,
-  waitMs: number,
-  envVars: Record<string, string>
-): Promise<ChildProcess> {
-  core.info(`🚀 Starting HTTP server: ${command}`);
-
-  // Merge environment variables
-  const env: Record<string, string> = {};
-  for (const [key, value] of Object.entries(process.env)) {
-    if (value !== undefined) {
-      env[key] = value;
-    }
-  }
-  for (const [key, value] of Object.entries(envVars)) {
-    env[key] = value;
-  }
-
-  const serverProcess = spawn("sh", ["-c", command], {
-    cwd: workDir,
-    env,
-    detached: true,
-    stdio: ["ignore", "pipe", "pipe"],
-  });
-
-  // Log server output for debugging
-  serverProcess.stdout?.on("data", (data) => {
-    core.debug(`  [HTTP server stdout]: ${data.toString().trim()}`);
-  });
-  serverProcess.stderr?.on("data", (data) => {
-    core.debug(`  [HTTP server stderr]: ${data.toString().trim()}`);
-  });
-
-  core.info(`  Waiting ${waitMs}ms for HTTP server to start...`);
-  await sleep(waitMs);
-
-  // Check if process is still running
-  if (serverProcess.exitCode !== null) {
-    throw new Error(`HTTP server exited prematurely with code ${serverProcess.exitCode}`);
-  }
-
-  core.info("  ✅ HTTP server started");
-  return serverProcess;
-}
-
-/**
- * Probe a single configuration (without comparison)
- */
-async function probeConfig(
-  config: TestConfiguration,
-  workDir: string,
-  envVars: Record<string, string>,
-  headers: Record<string, string>,
-  customMessages: CustomMessage[],
-  useSharedServer: boolean
-): Promise<{ result: ProbeResult; time: number }> {
-  core.info(`  📋 Probing: ${config.name} (${config.transport})`);
-  const start = Date.now();
-  let result: ProbeResult;
-  try {
-    result = await probeWithConfig(
-      config,
-      workDir,
-      envVars,
-      headers,
-      customMessages,
-      useSharedServer
-    );
-  } finally {
-    await runPostTestCommand(config, workDir);
-  }
-  return { result, time: Date.now() - start };
-}
-
-/**
- * Run all diff tests using the "probe all, then compare" approach
- */
-export async function runAllTests(ctx: RunContext): Promise<TestResult[]> {
-  const globalEnvVars = parseEnvVars(ctx.inputs.envVars);
-  const globalHeaders = ctx.inputs.headers || {};
-  const globalCustomMessages = ctx.inputs.customMessages || [];
-
-  // Check if we have a shared HTTP server to manage
-  const httpStartCommand = ctx.inputs.httpStartCommand;
-  const httpStartupWaitMs = ctx.inputs.httpStartupWaitMs || 2000;
-  const hasHttpConfigs = ctx.inputs.configurations.some((c) => c.transport === "streamable-http");
-  const useSharedServer = !!httpStartCommand && hasHttpConfigs;
-
-  // Store probe results for comparison
-  const branchResults: Map<string, { result: ProbeResult; time: number }> = new Map();
-  const baseResults: Map<string, { result: ProbeResult; time: number }> = new Map();
-
-  // ========================================
-  // PHASE 1: Probe all configs on current branch
-  // ========================================
-  core.info("\n🔄 Phase 1: Testing current branch...");
-
-  let sharedServerProcess: ChildProcess | null = null;
-  try {
-    // Start shared HTTP server if configured
-    if (useSharedServer) {
-      sharedServerProcess = await startSharedHttpServer(
-        httpStartCommand,
-        ctx.workDir,
-        httpStartupWaitMs,
-        globalEnvVars
-      );
-    }
-
-    for (const config of ctx.inputs.configurations) {
-      const configUsesSharedServer = useSharedServer && config.transport === "streamable-http";
-      try {
-        const probeData = await probeConfig(
-          config,
-          ctx.workDir,
-          globalEnvVars,
-          globalHeaders,
-          globalCustomMessages,
-          configUsesSharedServer
-        );
-        branchResults.set(config.name, probeData);
-      } catch (error) {
-        core.error(`Failed to probe ${config.name} on current branch: ${error}`);
-        branchResults.set(config.name, {
-          result: {
-            initialize: null,
-            tools: null,
-            prompts: null,
-            resources: null,
-            resourceTemplates: null,
-            customResponses: new Map(),
-            error: String(error),
-          },
-          time: 0,
-        });
-      }
-    }
-  } finally {
-    if (sharedServerProcess) {
-      core.info("🛑 Stopping current branch HTTP server...");
-      stopHttpServer(sharedServerProcess);
-      await sleep(500); // Give time for port to be released
-    }
-  }
-
-  // ========================================
-  // PHASE 2: Probe all configs on base ref
-  // ========================================
-  core.info(`\n🔄 Phase 2: Testing comparison ref: ${ctx.compareRef}...`);
-
-  const worktreePath = path.join(ctx.workDir, ".mcp-diff-base");
-  let useWorktree = false;
-
-  try {
-    // Set up comparison ref
-    useWorktree = await createWorktree(ctx.compareRef, worktreePath);
-    if (!useWorktree) {
-      core.info("  Worktree not available, using checkout");
-      await checkout(ctx.compareRef);
-    }
-
-    const baseWorkDir = useWorktree ? worktreePath : ctx.workDir;
-
-    // Build on base
-    core.info("🔨 Building on comparison ref...");
-    await runBuild(baseWorkDir, ctx.inputs);
-
-    let baseServerProcess: ChildProcess | null = null;
-    try {
-      // Start HTTP server for base ref if needed
-      if (useSharedServer) {
-        baseServerProcess = await startSharedHttpServer(
-          httpStartCommand,
-          baseWorkDir,
-          httpStartupWaitMs,
-          globalEnvVars
-        );
-      }
-
-      for (const config of ctx.inputs.configurations) {
-        const configUsesSharedServer = useSharedServer && config.transport === "streamable-http";
-        try {
-          const probeData = await probeConfig(
-            config,
-            baseWorkDir,
-            globalEnvVars,
-            globalHeaders,
-            globalCustomMessages,
-            configUsesSharedServer
-          );
-          baseResults.set(config.name, probeData);
-        } catch (error) {
-          core.error(`Failed to probe ${config.name} on base ref: ${error}`);
-          baseResults.set(config.name, {
-            result: {
-              initialize: null,
-              tools: null,
-              prompts: null,
-              resources: null,
-              resourceTemplates: null,
-              customResponses: new Map(),
-              error: String(error),
-            },
-            time: 0,
-          });
-        }
-      }
-    } finally {
-      if (baseServerProcess) {
-        core.info("🛑 Stopping base ref HTTP server...");
-        stopHttpServer(baseServerProcess);
-      }
-    }
-  } finally {
-    // Clean up
-    if (useWorktree) {
-      await removeWorktree(worktreePath);
-    } else {
-      await checkoutPrevious();
-    }
-  }
-
-  // ========================================
-  // PHASE 3: Compare all results
-  // ========================================
-  core.info("\n📊 Phase 3: Comparing results...");
-
-  const results: TestResult[] = [];
-
-  for (const config of ctx.inputs.configurations) {
-    const branchData = branchResults.get(config.name);
-    const baseData = baseResults.get(config.name);
-
-    const result: TestResult = {
-      configName: config.name,
-      transport: config.transport,
-      branchTime: branchData?.time || 0,
-      baseTime: baseData?.time || 0,
-      hasDifferences: false,
-      diffs: new Map(),
-      branchCounts: branchData?.result ? extractCounts(branchData.result) : undefined,
-      baseCounts: baseData?.result ? extractCounts(baseData.result) : undefined,
-    };
-
-    // Handle errors
-    if (branchData?.result.error) {
-      result.hasDifferences = true;
-      result.diffs.set("error", `Current branch probe failed: ${branchData.result.error}`);
-      results.push(result);
-      continue;
-    }
-
-    if (baseData?.result.error) {
-      result.hasDifferences = true;
-      result.diffs.set("error", `Base ref probe failed: ${baseData.result.error}`);
-      results.push(result);
-      continue;
-    }
-
-    // Compare results
-    const branchFiles = probeResultToFiles(branchData!.result);
-    const baseFiles = probeResultToFiles(baseData!.result);
-
-    result.diffs = compareResults(branchFiles, baseFiles);
-    result.hasDifferences = result.diffs.size > 0;
-
-    if (result.hasDifferences) {
-      core.info(`📋 Configuration ${config.name}: ${result.diffs.size} change(s) found`);
-    } else {
-      core.info(`✅ Configuration ${config.name}: no changes`);
-    }
-
-    // Save individual result
-    const resultPath = path.join(ctx.workDir, ".mcp-diff-results", `${config.name}.json`);
-    fs.mkdirSync(path.dirname(resultPath), { recursive: true });
-    fs.writeFileSync(
-      resultPath,
-      JSON.stringify(
-        {
-          ...result,
-          diffs: Object.fromEntries(result.diffs),
-        },
-        null,
-        2
-      )
-    );
-
-    results.push(result);
-  }
-
-  return results;
-}