@skyramp/mcp 0.4.4-rc.1 → 0.5.0-rc.1

package/build/index.js

@@ -1,15 +1,17 @@
 #!/usr/bin/env node
-import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { serveStdio } from "@modelcontextprotocol/server/stdio";
+import { McpServer } from "@modelcontextprotocol/server";
 import { runnableLanguagesText } from "./types/TestTypes.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import { registerStartTraceCollectionPrompt } from "./prompts/startTraceCollectionPrompts.js";
 import { registerTraceTool } from "./tools/trace/startTraceCollectionTool.js";
 import { registerTraceStopTool } from "./tools/trace/stopTraceCollectionTool.js";
 import { registerExecuteSkyrampTestTool } from "./tools/executeSkyrampTestTool.js";
+import { registerExecuteFleetTool } from "./tools/executeFleetTool.js";
 import { AUTH_PLACEHOLDER_TOKEN } from "./types/TestTypes.js";
 import { AUTH_CONFLICT_ERROR_MSG } from "./prompts/test-recommendation/recommendationSections.js";
 import { logger } from "./utils/logger.js";
 import { registerUITestTool } from "./tools/generate-tests/generateUIRestTool.js";
+import { registerUILoadTestTool } from "./tools/generate-tests/generateUILoadRestTool.js";
 import { registerSmokeTestTool } from "./tools/generate-tests/generateSmokeRestTool.js";
 import { registerFuzzTestTool } from "./tools/generate-tests/generateFuzzRestTool.js";
 import { registerContractTestTool } from "./tools/generate-tests/generateContractRestTool.js";
@@ -31,7 +33,6 @@ import { registerMockTool } from "./tools/generate-tests/generateMockRestTool.js
 import { registerBatchMockTool } from "./tools/generate-tests/batchMockGenerationTool.js";
 import { registerAnalyzeChangesTool, registerAnalyzeTestHealthTool, registerActionsTool, registerRegisterTestPlanTool, registerResolveScreenTool, } from "./tools/test-management/index.js";
 import { registerTestbotPrompt } from "./prompts/testbot/testbot-prompts.js";
-import { registerSutSetupResource } from "./resources/sutSetupResource.js";
 import { registerTestbotResource } from "./resources/testbotResource.js";
 import { registerLocalDevPrompt } from "./prompts/local-dev/local-dev-prompts.js";
 import { registerSubmitReportTool } from "./tools/submitReportTool.js";
@@ -72,7 +73,7 @@ else if (localDevEnabled) {
     oneClickInstructions = localDevInstructions;
 }
 // Global workspace-init nudge for bare MCP usage. Suppressed in testbot mode,
-// which runs its own INIT_WORKSPACE step (buildWorkspaceInitSection()) instead.
+// which initializes the workspace in its own steps instead.
 const workspaceInitInstructions = isTestbotEnabled() || oneClickEnabled || localDevEnabled
     ? ""
     : `
@@ -83,220 +84,247 @@ If the workspace root is a git repo AND \`.skyramp/workspace.yml\` does not exis
 3. Proceed with the originally requested tool, if it is not one of the two workspace-init tools above.
 Skip only if: not a git repo, \`.skyramp/workspace.yml\` already exists, or user explicitly declines.
 `;
-const server = new McpServer({
-    name: "Skyramp MCP Server",
-    version: "1.0.0",
-}, {
-    capabilities: {
-        tools: {
-            listChanged: true,
+/**
+ * Builds one fully-registered server. `serveStdio` calls this per connection,
+ * so every registration must happen here rather than at module load.
+ */
+async function createServer({ era }) {
+    const server = new McpServer({
+        name: "Skyramp MCP Server",
+        version: "1.0.0",
+    }, {
+        capabilities: {
+            tools: {
+                listChanged: true,
+            },
+            prompts: {
+                listChanged: true,
+            },
+            resources: {
+                listChanged: true,
+            },
         },
-        prompts: {
-            listChanged: true,
-        },
-        resources: {
-            listChanged: true,
-        },
-    },
-    instructions: `Skyramp MCP Server — generates and executes API tests (fuzz, contract, integration, E2E, UI).
-${workspaceInitInstructions}
-## Rules
-- NEVER show CLI commands. NEVER attempt to install or configure the Skyramp CLI. ALWAYS use the MCP tools provided.
-- For UI and E2E tests, there are TWO recording modes:
-  1. **AI-driven recording** (default): Use the browser_* tools (browser_navigate, browser_click, etc.) to record interactions, then call skyramp_export_zip to export the trace, then call skyramp_ui_test_generation with the zip path.
-  2. **Manual recording**: ONLY when the user explicitly says "manual recording", "record myself", "I will interact", or "Docker trace" — use skyramp_start_trace_collection / skyramp_stop_trace_collection to let the user interact with the browser themselves.
-- To capture an authentication session for re-use (e.g. "save session", "store login", "record auth state"): call \`skyramp_start_trace_collection\` (\`playwrightSaveStoragePath\` defaults to saving the session in \`outputDir\` — no need to set it). Tell the user to log in once and then call \`skyramp_stop_trace_collection\` IMMEDIATELY after login completes. The saved file can be passed back as \`playwrightStoragePath\` on later trace runs, or referenced as \`storageState\` in generated tests.${oneClickInstructions}
+        instructions: `Skyramp MCP Server — generates and executes API tests (fuzz, contract, integration, E2E, UI).
+  ${workspaceInitInstructions}
+  ## Rules
+  - NEVER show CLI commands. NEVER attempt to install or configure the Skyramp CLI. ALWAYS use the MCP tools provided.
+  - For UI and E2E tests, there are TWO recording modes:
+    1. **AI-driven recording** (default): Use the browser_* tools (browser_navigate, browser_click, etc.) to record interactions, then call skyramp_export_zip to export the trace, then call skyramp_ui_test_generation with the zip path.
+    2. **Manual recording**: ONLY when the user explicitly says "manual recording", "record myself", "I will interact", or "Docker trace" — use skyramp_start_trace_collection / skyramp_stop_trace_collection to let the user interact with the browser themselves.
+  - To capture an authentication session for re-use (e.g. "save session", "store login", "record auth state"): call \`skyramp_start_trace_collection\` (\`playwrightSaveStoragePath\` defaults to saving the session in \`outputDir\` — no need to set it). Tell the user to log in once and then call \`skyramp_stop_trace_collection\` IMMEDIATELY after login completes. The saved file can be passed back as \`playwrightStoragePath\` on later trace runs, or referenced as \`storageState\` in generated tests.${oneClickInstructions}
 
-## Test Management Flow
-Use \`skyramp_analyze_changes\` as the single entry point for both test recommendations and test health analysis.
+  ## Test Management Flow
+  Use \`skyramp_analyze_changes\` as the single entry point for both test recommendations and test health analysis.
 
-### Recommendations (2-step)
-1. Call \`skyramp_analyze_changes\` with \`repositoryPath\` and \`scope\` → scans endpoints, computes branch diff, discovers existing tests → returns a \`stateFile\` **and** inline ranked test recommendations.
-2. (Optional) Use the \`skyramp_recommend_tests\` prompt with \`stateFile\` in Cursor/IDE chat for a focused recommendation view.
+  ### Recommendations (2-step)
+  1. Call \`skyramp_analyze_changes\` with \`repositoryPath\` and \`scope\` → scans endpoints, computes branch diff, discovers existing tests → returns a \`stateFile\` **and** inline ranked test recommendations.
+  2. (Optional) Use the \`skyramp_recommend_tests\` prompt with \`stateFile\` in Cursor/IDE chat for a focused recommendation view.
 
-### Health Analysis (5-step)
-1. Call \`skyramp_analyze_changes\` with \`repositoryPath\` and \`scope\` → returns a \`stateFile\`.
-2. Run each external test in ${runnableLanguagesText()} with \`skyramp_execute_test\`, \`phase: "before"\`, and \`stateFile\` → the server runs your command and writes the verdict back to stateFile. The tool takes no other language, and the gate in step 3 does not ask for those.
-3. Call \`skyramp_analyze_test_health\` with \`stateFile\` → runs drift analysis + health scoring + LLM semantic assessment. It refuses until step 2 has run.
-4. (Optional) Run Skyramp-generated tests with \`skyramp_execute_test\` and \`stateFile\` → writes each verdict back for health scoring.
-5. Call \`skyramp_actions\` with \`stateFile\` → executes UPDATE/REGENERATE/ADD recommendations (with execution-aware prioritization if step 4 ran).
+  ### Health Analysis (5-step)
+  1. Call \`skyramp_analyze_changes\` with \`repositoryPath\` and \`scope\` → returns a \`stateFile\`.
+  2. Run each external test in ${runnableLanguagesText()} with \`skyramp_execute_test\`, \`phase: "before"\`, and \`stateFile\` → the server runs your command and writes the verdict back to stateFile. The tool takes no other language, and the gate in step 3 does not ask for those.
+  3. Call \`skyramp_analyze_test_health\` with \`stateFile\` → runs drift analysis + health scoring + LLM semantic assessment. It refuses until step 2 has run.
+  4. (Optional) Run Skyramp-generated tests with \`skyramp_execute_test\` and \`stateFile\` → writes each verdict back for health scoring.
+  5. Call \`skyramp_actions\` with \`stateFile\` → executes UPDATE/REGENERATE/ADD recommendations (with execution-aware prioritization if step 4 ran).
 
-After \`skyramp_analyze_changes\`, inspect enriched data via MCP Resources (use the \`sessionId\` returned in the output):
-- \`skyramp://analysis/{sessionId}/summary\` — high-level overview
-- \`skyramp://analysis/{sessionId}/endpoints\` — compact endpoint listing
-- \`skyramp://analysis/{sessionId}/endpoints/{path}\` — full path detail
-- \`skyramp://analysis/{sessionId}/endpoints/{path}/{method}\` — single method detail
-- \`skyramp://analysis/{sessionId}/scenarios\` — drafted scenarios
-- \`skyramp://analysis/{sessionId}/diff\` — branch diff context
+  After \`skyramp_analyze_changes\`, inspect enriched data via MCP Resources (use the \`sessionId\` returned in the output):
+  - \`skyramp://analysis/{sessionId}/summary\` — high-level overview
+  - \`skyramp://analysis/{sessionId}/endpoints\` — compact endpoint listing
+  - \`skyramp://analysis/{sessionId}/endpoints/{path}\` — full path detail
+  - \`skyramp://analysis/{sessionId}/endpoints/{path}/{method}\` — single method detail
+  - \`skyramp://analysis/{sessionId}/scenarios\` — drafted scenarios
+  - \`skyramp://analysis/{sessionId}/diff\` — branch diff context
 
-## Workspace Defaults for Test Generation (MANDATORY)
-Before calling ANY test generation tool, you MUST follow this flow:
+  ## Workspace Defaults for Test Generation (MANDATORY)
+  Before calling ANY test generation tool, you MUST follow this flow:
 
-1. **Read** the .skyramp/workspace.yml file to get the configured defaults.
-2. **Extract** the \`language\`, \`framework\`, \`testDirectory\`, \`api.baseUrl\`, \`api.authHeader\`, and \`api.authType\` from the matching service in the services section.
-3. **Use those values** as defaults for the test generation tool call. Pass the service \`testDirectory\` as the generation tool \`outputDir\`. Do NOT ask the user for these values if they are already configured in the workspace file.
-4. **CRITICAL — endpointURL**: The \`endpointURL\` parameter MUST be the full URL to the specific endpoint being tested, NOT just the base URL. Construct it by combining \`api.baseUrl\` with the endpoint path. Example: if \`api.baseUrl\` is \`http://localhost:8000\` and the endpoint is \`/api/v1/products\`, pass \`endpointURL: "http://localhost:8000/api/v1/products"\`. NEVER pass just the base URL (e.g. \`http://localhost:8000\`) as \`endpointURL\`.
-5. **CRITICAL — scenario generation**: When calling \`skyramp_batch_scenario_test_generation\`, ALWAYS pass:
-   - \`baseURL\`: The full base URL from \`api.baseUrl\` (e.g., \`http://localhost:3000\`). This determines the scheme, host, and port in the generated trace. Without it, the trace defaults to https:443 which is almost always wrong for local development.
-   - \`authHeader\`: Which HTTP header carries the auth credential. Get it from \`api.authHeader\` in workspace config. Examples: \`Authorization\` (Bearer/Token auth), \`X-Api-Key\` (API key auth), \`Cookie\` (session/cookie auth like NextAuth). Pass \`""\` to document the endpoint as unauthenticated in the generated test's assertions (unauthenticated endpoints or \`api.authType: "none"\`).
-   - \`authScheme\`: Only when \`authHeader\` is \`Authorization\`. The prefix before the token (e.g., \`"Bearer"\` → \`Authorization: Bearer <token>\`). **Derive from**: (1) OpenAPI spec \`securitySchemes\`/\`securityDefinitions\`, (2) source code auth middleware, (3) workspace \`api.authType\`. **Do NOT guess.**
-   - \`authToken\`: The full header value, used verbatim. When omitted, \`SKYRAMP_PLACEHOLDER_TOKEN\` is auto-generated. Only provide when the header needs a specific format (e.g., \`"session=${AUTH_PLACEHOLDER_TOKEN}"\` for Cookie). **Do NOT fabricate token values.**
-   - \`apiSchema\` is OPTIONAL — omit it for code-first apps without OpenAPI specs.
-6. **CRITICAL — integration test from scenario**: When calling \`skyramp_integration_test_generation\` with a \`scenarioFile\`:
-   - If workspace has \`api.authType\` set: omit auth params entirely — passing auth here alongside workspace \`authType\` causes "${AUTH_CONFLICT_ERROR_MSG}".
-   - If workspace has no \`api.authType\`: pass \`authHeader\` only (no \`authScheme\`).
-7. **If the workspace file does not exist**, or the needed values (language, framework, testDirectory) are missing from the workspace config, ASK the user which language, framework, and outputDir they want before calling the tool.
-8. The user can always override workspace defaults by explicitly specifying values in their request.
-`,
-});
-// Register prompts
-logger.info("Starting prompt registration process");
-const prompts = [
-    registerStartTraceCollectionPrompt,
-    registerRecommendTestsPrompt,
-    registerTraceRecordingPrompt,
-];
-if (isTestbotEnabled()) {
-    prompts.push(registerTestbotPrompt);
-    registerTestbotResource(server);
-    logger.info("Testbot prompt enabled via SKYRAMP_FEATURE_TESTBOT");
-    registerSutSetupResource(server);
-    logger.info("Testbot SUT setup resource enabled via SKYRAMP_FEATURE_TESTBOT");
-}
-if (isLocalDevEnabled()) {
-    prompts.push(registerLocalDevPrompt);
-    logger.info("Local-dev prompt enabled via SKYRAMP_FEATURE_LOCAL_DEV");
-}
-prompts.forEach((registerPrompt) => registerPrompt(server));
-logger.info("All prompts registered successfully");
-// Register test generation tools
-const testGenerationTools = [
-    registerSmokeTestTool,
-    registerFuzzTestTool,
-    registerContractTestTool,
-    registerLoadTestTool,
-    registerIntegrationTestTool,
-    registerE2ETestTool,
-    registerUITestTool,
-    registerBatchScenarioTestTool,
-    registerMockTool,
-    registerBatchMockTool,
-];
-testGenerationTools.forEach((registerTool) => registerTool(server));
-// Register modularization and code quality tools
-const codeQualityTools = [
-    registerModularizationTool,
-    registerCodeReuseTool,
-    registerEnhanceAssertionsTool,
-    registerAssertionVerifyTool,
-    registerVerifyTestDependenciesTool,
-];
-// With the Skyramp plugin loaded the agent already has the
-// fix-test-import-errors skill; the tool would be a second copy of
-// the same text (SKYR-4296).
-if (!isSkillsLoaded()) {
-    // Same reasoning as fix-test-import-errors: with the plugin loaded the agent
-    // has the validate-assertion-alignment-post-execution skill directly; this
-    // tool exists so an agent without skill support still gets the same
-    // procedure, rendered from the same markdown with real arguments filled in.
-    codeQualityTools.push(registerFixErrorTool, registerAlignAssertionsTool);
-}
-else {
-    // The enhance-assertions skill drives its own snapshot step itself (SKYR-4310).
-    codeQualityTools.push(registerAssertionSnapshotTool);
-}
-codeQualityTools.forEach((registerTool) => registerTool(server));
-// Register analysis resources (MCP Resources for enriched data access)
-registerAnalysisResources(server);
-registerProgressResource(server);
-// Register unified test-management tools (replaces separate test-maintenance tools)
-registerAnalyzeChangesTool(server);
-registerAnalyzeTestHealthTool(server);
-registerActionsTool(server);
-registerRegisterTestPlanTool(server);
-registerResolveScreenTool(server);
-// Register workspace management tools
-registerInitScanWorkspaceTool(server);
-registerInitializeWorkspaceTool(server);
-// Register one-click orchestrated workflows
-if (oneClickEnabled || localDevEnabled) {
-    registerOneClickTool(server);
-    logger.info("One-click tools enabled", {
-        enabledBy: {
-            SKYRAMP_FEATURE_ONE_CLICK: oneClickEnabled,
-            SKYRAMP_FEATURE_LOCAL_DEV: localDevEnabled,
-        },
+  1. **Read** the .skyramp/workspace.yml file to get the configured defaults.
+  2. **Extract** the \`language\`, \`framework\`, \`testDirectory\`, \`api.baseUrl\`, \`api.authHeader\`, and \`api.authType\` from the matching service in the services section.
+  3. **Use those values** as defaults for the test generation tool call. Pass the service \`testDirectory\` as the generation tool \`outputDir\`. Do NOT ask the user for these values if they are already configured in the workspace file.
+  4. **CRITICAL — endpointURL**: The \`endpointURL\` parameter MUST be the full URL to the specific endpoint being tested, NOT just the base URL. Construct it by combining \`api.baseUrl\` with the endpoint path. Example: if \`api.baseUrl\` is \`http://localhost:8000\` and the endpoint is \`/api/v1/products\`, pass \`endpointURL: "http://localhost:8000/api/v1/products"\`. NEVER pass just the base URL (e.g. \`http://localhost:8000\`) as \`endpointURL\`.
+  5. **CRITICAL — scenario generation**: When calling \`skyramp_batch_scenario_test_generation\`, ALWAYS pass:
+     - \`baseURL\`: The full base URL from \`api.baseUrl\` (e.g., \`http://localhost:3000\`). This determines the scheme, host, and port in the generated trace. Without it, the trace defaults to https:443 which is almost always wrong for local development.
+     - \`authHeader\`: Which HTTP header carries the auth credential. Get it from \`api.authHeader\` in workspace config. Examples: \`Authorization\` (Bearer/Token auth), \`X-Api-Key\` (API key auth), \`Cookie\` (session/cookie auth like NextAuth). Pass \`""\` to document the endpoint as unauthenticated in the generated test's assertions (unauthenticated endpoints or \`api.authType: "none"\`).
+     - \`authScheme\`: Only when \`authHeader\` is \`Authorization\`. The prefix before the token (e.g., \`"Bearer"\` → \`Authorization: Bearer <token>\`). **Derive from**: (1) OpenAPI spec \`securitySchemes\`/\`securityDefinitions\`, (2) source code auth middleware, (3) workspace \`api.authType\`. **Do NOT guess.**
+     - \`authToken\`: The full header value, used verbatim. When omitted, \`SKYRAMP_PLACEHOLDER_TOKEN\` is auto-generated. Only provide when the header needs a specific format (e.g., \`"session=${AUTH_PLACEHOLDER_TOKEN}"\` for Cookie). **Do NOT fabricate token values.**
+     - \`apiSchema\` is OPTIONAL — omit it for code-first apps without OpenAPI specs.
+  6. **CRITICAL — integration test from scenario**: When calling \`skyramp_integration_test_generation\` with a \`scenarioFile\`:
+     - If workspace has \`api.authType\` set: omit auth params entirely — passing auth here alongside workspace \`authType\` causes "${AUTH_CONFLICT_ERROR_MSG}".
+     - If workspace has no \`api.authType\`: pass \`authHeader\` only (no \`authScheme\`).
+  7. **If the workspace file does not exist**, or the needed values (language, framework, testDirectory) are missing from the workspace config, ASK the user which language, framework, and outputDir they want before calling the tool.
+  8. The user can always override workspace defaults by explicitly specifying values in their request.
+  `,
     });
-}
-// Register other Skyramp tools
-const infrastructureTools = [
-    registerLoginTool,
-    registerLogoutTool,
-    registerExecuteSkyrampTestTool,
-    registerTraceTool,
-    registerTraceStopTool,
-];
-if (isLocalDevEnabled()) {
-    infrastructureTools.push(registerEnrichTestWithMocksTool, registerPreflightMockCheckTool, registerGenerateEnrichedIntegrationTestTool, registerLocalDevWorkerComposeTool, registerQueryProxyMocksTool);
-    logger.info("Local-dev tools enabled via SKYRAMP_FEATURE_LOCAL_DEV");
-}
-if (isTestbotEnabled()) {
-    infrastructureTools.push(registerSubmitReportTool);
-    logger.info("Testbot tools enabled via SKYRAMP_FEATURE_TESTBOT");
-}
-infrastructureTools.forEach((registerTool) => registerTool(server));
-// Register Playwright browser tools (trace recording via browser automation).
-//
-// Tell the capture which attribute this repository uses for test ids. Without
-// it the capture assumes Playwright's `data-testid`, and a repo using anything
-// else gets no test-id anchor and a positional xpath for every element — which
-// resolves at capture time and breaks later, so the miss never announces itself
-// (SKYR-4136). This server is the only component that knows which repository it
-// is running against.
-//
-// A declared `use.testIdAttribute` is authoritative. Failing that, the
-// repository's own tests name the attribute whenever they select on it, so read
-// it out of their selectors rather than guessing from a list of popular names.
-// Never fatal: no answer leaves the capture on Playwright's default.
-(async () => {
-    let testIdAttribute;
-    try {
-        const root = await findRoot(process.cwd());
-        testIdAttribute = await readTestIdAttribute(root);
-        if (testIdAttribute) {
-            logger.info(`testIdAttribute declared by the Playwright config: ${testIdAttribute}`);
-        }
-        else {
-            const discovered = await discoverTestIdAttribute(root);
-            testIdAttribute = discovered.attribute;
+    // Register prompts
+    logger.info("Starting prompt registration process");
+    const prompts = [
+        registerStartTraceCollectionPrompt,
+        registerRecommendTestsPrompt,
+        registerTraceRecordingPrompt,
+    ];
+    if (isTestbotEnabled()) {
+        prompts.push(registerTestbotPrompt);
+        registerTestbotResource(server);
+        logger.info("Testbot prompt enabled via SKYRAMP_FEATURE_TESTBOT");
+    }
+    if (isLocalDevEnabled()) {
+        prompts.push(registerLocalDevPrompt);
+        logger.info("Local-dev prompt enabled via SKYRAMP_FEATURE_LOCAL_DEV");
+    }
+    prompts.forEach((registerPrompt) => registerPrompt(server));
+    logger.info("All prompts registered successfully");
+    // Register test generation tools
+    const testGenerationTools = [
+        registerSmokeTestTool,
+        registerFuzzTestTool,
+        registerContractTestTool,
+        registerLoadTestTool,
+        registerIntegrationTestTool,
+        registerE2ETestTool,
+        registerUITestTool,
+        registerUILoadTestTool,
+        registerBatchScenarioTestTool,
+        registerMockTool,
+        registerBatchMockTool,
+    ];
+    testGenerationTools.forEach((registerTool) => registerTool(server));
+    // Register modularization and code quality tools
+    const codeQualityTools = [
+        registerModularizationTool,
+        registerCodeReuseTool,
+        registerEnhanceAssertionsTool,
+        registerAssertionVerifyTool,
+        registerVerifyTestDependenciesTool,
+    ];
+    // With the Skyramp plugin loaded the agent already has the
+    // fix-test-import-errors skill; the tool would be a second copy of
+    // the same text (SKYR-4296).
+    if (!isSkillsLoaded()) {
+        // Same reasoning as fix-test-import-errors: with the plugin loaded the agent
+        // has the validate-assertion-alignment-post-execution skill directly; this
+        // tool exists so an agent without skill support still gets the same
+        // procedure, rendered from the same markdown with real arguments filled in.
+        codeQualityTools.push(registerFixErrorTool, registerAlignAssertionsTool);
+    }
+    else {
+        // The enhance-assertions skill drives its own snapshot step itself (SKYR-4310).
+        codeQualityTools.push(registerAssertionSnapshotTool);
+    }
+    codeQualityTools.forEach((registerTool) => registerTool(server));
+    // Register analysis resources (MCP Resources for enriched data access)
+    registerAnalysisResources(server);
+    registerProgressResource(server);
+    // Register unified test-management tools (replaces separate test-maintenance tools)
+    registerAnalyzeChangesTool(server);
+    registerAnalyzeTestHealthTool(server);
+    registerActionsTool(server);
+    registerRegisterTestPlanTool(server);
+    registerResolveScreenTool(server);
+    // Register workspace management tools
+    registerInitScanWorkspaceTool(server);
+    registerInitializeWorkspaceTool(server);
+    // Register one-click orchestrated workflows
+    if (oneClickEnabled || localDevEnabled) {
+        registerOneClickTool(server);
+        logger.info("One-click tools enabled", {
+            enabledBy: {
+                SKYRAMP_FEATURE_ONE_CLICK: oneClickEnabled,
+                SKYRAMP_FEATURE_LOCAL_DEV: localDevEnabled,
+            },
+        });
+    }
+    // Register other Skyramp tools
+    const infrastructureTools = [
+        registerLoginTool,
+        registerLogoutTool,
+        registerExecuteSkyrampTestTool,
+        registerExecuteFleetTool,
+        registerTraceTool,
+        registerTraceStopTool,
+    ];
+    if (isLocalDevEnabled()) {
+        infrastructureTools.push(registerEnrichTestWithMocksTool, registerPreflightMockCheckTool, registerGenerateEnrichedIntegrationTestTool, registerLocalDevWorkerComposeTool, registerQueryProxyMocksTool);
+        logger.info("Local-dev tools enabled via SKYRAMP_FEATURE_LOCAL_DEV");
+    }
+    if (isTestbotEnabled()) {
+        infrastructureTools.push(registerSubmitReportTool);
+        logger.info("Testbot tools enabled via SKYRAMP_FEATURE_TESTBOT");
+    }
+    infrastructureTools.forEach((registerTool) => registerTool(server));
+    // Register Playwright browser tools (trace recording via browser automation).
+    //
+    // Tell the capture which attribute this repository uses for test ids. Without
+    // it the capture assumes Playwright's `data-testid`, and a repo using anything
+    // else gets no test-id anchor and a positional xpath for every element — which
+    // resolves at capture time and breaks later, so the miss never announces itself
+    // (SKYR-4136). This server is the only component that knows which repository it
+    // is running against.
+    //
+    // A declared `use.testIdAttribute` is authoritative. Failing that, the
+    // repository's own tests name the attribute whenever they select on it, so read
+    // it out of their selectors rather than guessing from a list of popular names.
+    // Never fatal: no answer leaves the capture on Playwright's default.
+    //
+    // Awaited, and worth the wait: `PlaywrightTraceService.initialize()` is 206ms
+    // of it and the browser tools come out of that service, so nothing can
+    // register them earlier. Running it in the background instead delays every
+    // connection's first response by nothing but leaves the client holding 27
+    // tools of 52 — Claude Code does not re-list on `tools/list_changed`, so all
+    // 23 `browser_*` tools stay invisible for the whole run.
+    await (async () => {
+        let testIdAttribute;
+        try {
+            const root = await findRoot(process.cwd());
+            testIdAttribute = await readTestIdAttribute(root);
             if (testIdAttribute) {
-                // Log the runner-up counts: on a part-migrated repo the most-used
-                // attribute wins, and a wrong pick should be visible here rather than
-                // showing up later as brittle selectors.
-                logger.info(`testIdAttribute discovered from test selectors: ${testIdAttribute}`, { counts: discovered.counts, filesScanned: discovered.filesScanned });
+                logger.info(`testIdAttribute declared by the Playwright config: ${testIdAttribute}`);
             }
             else {
-                // No name rule can reach an arbitrary convention like `xyz-id`, so name
-                // what the selectors did contain and the variable that overrides it.
-                // Editing the repo's own Playwright config is the wrong instruction: a
-                // suite may have several, and none of them at the root.
-                logger.info("No testIdAttribute declared or discovered — capture stays on Playwright's data-testid. " +
-                    "Set PLAYWRIGHT_MCP_TEST_ID_ATTRIBUTE to override.", {
-                    attributesSeenInSelectors: discovered.observed,
-                    filesScanned: discovered.filesScanned,
-                });
+                const discovered = await discoverTestIdAttribute(root);
+                testIdAttribute = discovered.attribute;
+                if (testIdAttribute) {
+                    // Log the runner-up counts: on a part-migrated repo the most-used
+                    // attribute wins, and a wrong pick should be visible here rather than
+                    // showing up later as brittle selectors.
+                    logger.info(`testIdAttribute discovered from test selectors: ${testIdAttribute}`, {
+                        counts: discovered.counts,
+                        filesScanned: discovered.filesScanned,
+                    });
+                }
+                else {
+                    // No name rule can reach an arbitrary convention like `xyz-id`, so name
+                    // what the selectors did contain and the variable that overrides it.
+                    // Editing the repo's own Playwright config is the wrong instruction: a
+                    // suite may have several, and none of them at the root.
+                    logger.info("No testIdAttribute declared or discovered — capture stays on Playwright's data-testid. " +
+                        "Set PLAYWRIGHT_MCP_TEST_ID_ATTRIBUTE to override.", {
+                        attributesSeenInSelectors: discovered.observed,
+                        filesScanned: discovered.filesScanned,
+                    });
+                }
             }
         }
-    }
-    catch (err) {
-        logger.warning("Could not determine this repository's testIdAttribute", {
-            error: err,
+        catch (err) {
+            logger.warning("Could not determine this repository's testIdAttribute", {
+                error: err,
+            });
+        }
+        await registerPlaywrightTools(server, { testIdAttribute });
+    })().catch((err) => {
+        logger.error("Failed to register Playwright tools", { error: err });
+    });
+    const runInitTrigger = () => {
+        registerInitTriggerOnMCPInitialized().catch((err) => {
+            logger.error("Failed to run MCP initialized trigger", { error: err });
         });
-    }
-    await registerPlaywrightTools(server, { testIdAttribute });
-})().catch((err) => {
-    logger.error("Failed to register Playwright tools", { error: err });
-});
+    };
+    server.server.oninitialized = runInitTrigger;
+    // A 2026-07-28 connection has no `initialize` exchange, so `oninitialized`
+    // never fires on it.
+    if (era === "modern")
+        runInitTrigger();
+    return server;
+}
 // Global error handlers for crash telemetry
 process.on("uncaughtException", async (error) => {
     // Don't send analytics for EPIPE errors
@@ -350,13 +378,11 @@ process.on("unhandledRejection", async (reason) => {
 });
 // Start MCP server
 async function main() {
-    const transport = new StdioServerTransport();
-    server.server.oninitialized = () => {
-        registerInitTriggerOnMCPInitialized().catch((err) => {
-            logger.error("Failed to run MCP initialized trigger", { error: err });
-        });
-    };
-    await server.connect(transport);
+    // `serveStdio` owns the era decision: it reads the opening exchange and pins
+    // one instance from the factory for the connection. `StdioServerTransport`
+    // serves the 2025 handshake only, so the 2026-07-28 revision is unreachable
+    // through it however the client asks.
+    const handle = serveStdio(createServer);
     logger.info("MCP Server started successfully");
     // Listen for stdin closure (parent process disconnected)
     // Using multiple events for robustness across different disconnection scenarios
@@ -372,7 +398,7 @@ async function main() {
             ? playwrightService.shutdown().catch(() => { })
             : Promise.resolve();
         cleanupPlaywright.finally(() => {
-            server.close().finally(() => {
+            handle.close().finally(() => {
                 process.exit(0);
             });
         });

package/build/playwright/PlaywrightTraceService.d.ts

@@ -5,7 +5,7 @@
  * The Playwright fork is CJS; this project is ESM. We use createRequire
  * to bridge the module systems.
  */
-import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import type { CallToolResult } from "@modelcontextprotocol/server";
 interface McpTool {
     name: string;
     description?: string;

package/build/playwright/blueprintDigestStore.d.ts

@@ -1,4 +1,4 @@
-import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import type { CallToolResult } from "@modelcontextprotocol/server";
 import type { BlueprintCaptureDigest } from "../types/BlueprintDigest.js";
 /** A run that takes more captures than this has a browsing problem of its own;
  *  the cap is here so a runaway loop cannot grow the process without bound. */

package/build/playwright/registerPlaywrightTools.d.ts

@@ -3,6 +3,6 @@
  * tools in the Skyramp MCP server, delegating calls to PlaywrightTraceService.
  */
 import { PlaywrightTraceService, type PlaywrightTraceOptions } from "./PlaywrightTraceService.js";
-import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { McpServer } from "@modelcontextprotocol/server";
 export declare function getPlaywrightTraceService(): PlaywrightTraceService | null;
 export declare function registerPlaywrightTools(server: McpServer, options?: PlaywrightTraceOptions): Promise<void>;

package/build/playwright/traceRecordingPrompt.d.ts

@@ -2,7 +2,7 @@
  * MCP prompt that guides the LLM through the Playwright-based trace recording
  * and Skyramp test generation flow.
  */
-import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { McpServer } from "@modelcontextprotocol/server";
 export interface TraceRecordingPromptOpts {
     /** Directory where zip files should be written. Defaults to same dir as test file. */
     outputDir?: string;

package/build/playwright/traceRecordingPrompt.js

@@ -82,7 +82,7 @@ export function registerTraceRecordingPrompt(server) {
     logger.info("Registering trace recording prompt");
     server.registerPrompt("skyramp_trace_recording_prompt", {
         description: "Guide for recording browser interactions as a Skyramp trace and generating UI tests",
-        argsSchema: {
+        argsSchema: z.object({
             outputDir: z
                 .string()
                 .optional()
@@ -92,7 +92,7 @@ export function registerTraceRecordingPrompt(server) {
                 .default(true)
                 .optional()
                 .describe("Whether to run skyramp_modularization after generation. Default: true. Set to false in CI."),
-        },
+        }),
     }, (args) => ({
         messages: [
             {

package/build/prompts/initialize-workspace/initializeWorkspacePrompt.d.ts

@@ -1 +1,18 @@
 export declare const INIT_WORKSPACE_INSTRUCTIONS: string;
+/** A repository of the Target other than the workspace's own, scanned into the same workspace.yml. */
+export interface ScanRelatedRepository {
+    /** owner/repo, set verbatim as each of its services' `repository`. */
+    repository: string;
+    repositoryPath: string;
+}
+/**
+ * The scan instructions for one workspace. With related repositories, every
+ * repository is scanned into the one workspace.yml at `workspacePath`, each
+ * related repository's services carrying its `repository`. With a test
+ * repository, testDirectory is read from, and relative to, that checkout.
+ */
+export declare function buildInitWorkspaceInstructions(target: {
+    workspacePath: string;
+    relatedRepositories?: ScanRelatedRepository[];
+    testRepositoryPath?: string;
+}): string;