@documentation-robotics/cli 0.3.0 → 0.4.0

package/dist/commands/init.js

@@ -1,13 +1,9 @@
 import { intro, text, outro } from "@clack/prompts";
 import ansis from "ansis";
-import { Model } from "../core/model.js";
-import { fileExists } from "../utils/file-io.js";
-import { logVerbose, logDebug } from "../utils/globals.js";
-import { installSpecReference } from "../utils/spec-installer.js";
+import { hasModel, initProject } from "../core/project-init.js";
 import { getCliBundledSpecVersion } from "../utils/spec-version.js";
 import { isTelemetryEnabled, startSpan, endSpan } from "../telemetry/index.js";
 import { getErrorMessage } from "../utils/errors.js";
-import { regenerateLayerReports } from "./reports.js";
 async function initCommand(options) {
   intro(ansis.bold(ansis.blue("\u2699\uFE0F  Initialize Documentation Robotics Model")));
   const span = isTelemetryEnabled ? startSpan("model.init", {
@@ -16,8 +12,8 @@ async function initCommand(options) {
     "init.hasAuthor": !!options.author
   }) : null;
   try {
-    const rootPath = process.cwd(), manifestPath = `${`${rootPath}/documentation-robotics/model`}/manifest.yaml`;
-    await fileExists(manifestPath) && (console.error(ansis.red("Error: Model already initialized in this directory")), isTelemetryEnabled && span && span.setStatus({ code: 2, message: "Model already exists" }), endSpan(span), process.exit(1));
+    const rootPath = process.cwd();
+    await hasModel(rootPath) && (console.error(ansis.red("Error: Model already initialized in this directory")), isTelemetryEnabled && span && span.setStatus({ code: 2, message: "Model already exists" }), endSpan(span), process.exit(1));
     const isInteractive = process.stdin.isTTY;
     isTelemetryEnabled && span && span.setAttribute("cli.interactive", isInteractive);
     const skipPrompts = !!options.name, name = options.name || (isInteractive ? await text({
@@ -32,28 +28,15 @@ async function initCommand(options) {
       message: "Author (optional):",
       defaultValue: ""
     }) : "");
-    isTelemetryEnabled && span && (span.setAttribute("model.name", name), span.setAttribute("model.specVersion", getCliBundledSpecVersion())), logDebug(`Creating model directory at ${rootPath}/documentation-robotics/model`);
-    const model = await Model.init(rootPath, {
+    isTelemetryEnabled && span && (span.setAttribute("model.name", name), span.setAttribute("model.specVersion", getCliBundledSpecVersion()));
+    const result = await initProject(rootPath, {
       name,
-      version: "0.1.0",
       description: description || void 0,
-      author: author || void 0,
-      specVersion: getCliBundledSpecVersion(),
-      created: (/* @__PURE__ */ new Date()).toISOString()
-    }, { lazyLoad: !1 });
-    logVerbose(`Model saved with:
-  - Version: 0.1.0
-  - Spec Version: ${getCliBundledSpecVersion()}
-  - Location: ${rootPath}/documentation-robotics/model`), logDebug("Installing spec reference (.dr/ folder)..."), await installSpecReference(rootPath, !1), logVerbose("Spec reference installed"), logDebug("Generating reports and README...");
-    try {
-      await regenerateLayerReports(rootPath), logVerbose("Reports and README generated");
-    } catch (reportError) {
-      const reportMessage = getErrorMessage(reportError);
-      console.warn(ansis.yellow(`\u26A0 Warning: Failed to generate reports and README: ${reportMessage}
+      author: author || void 0
+    });
+    result.reportError !== void 0 && console.warn(ansis.yellow(`\u26A0 Warning: Failed to generate reports and README: ${result.reportError}
   Model initialized successfully. To regenerate reports later, run:
-  dr reports regenerate`)), logDebug(`Report generation failed (non-fatal): ${reportMessage}`);
-    }
-    isTelemetryEnabled && span && (span.setAttribute("model.path", rootPath), span.setStatus({ code: 0 })), outro(ansis.green(`\u2713 Model initialized: ${ansis.bold(model.manifest.name)}`));
+  dr reports regenerate`)), isTelemetryEnabled && span && (span.setAttribute("model.path", rootPath), span.setStatus({ code: 0 })), outro(ansis.green(`\u2713 Model initialized: ${ansis.bold(result.name)}`));
   } catch (error) {
     isTelemetryEnabled && span && (span.recordException(error), span.setStatus({
       code: 2,

package/dist/commands/init.js.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
   "sources": ["../../src/commands/init.ts"],
-  "sourcesContent": ["/**\n * Initialize a new Documentation Robotics model\n */\n\nimport { intro, text, outro } from \"@clack/prompts\";\nimport ansis from \"ansis\";\nimport { Model } from \"../core/model.js\";\nimport { fileExists } from \"../utils/file-io.js\";\nimport { logVerbose, logDebug } from \"../utils/globals.js\";\nimport { installSpecReference } from \"../utils/spec-installer.js\";\nimport { getCliBundledSpecVersion } from \"../utils/spec-version.js\";\nimport { isTelemetryEnabled, startSpan, endSpan } from \"../telemetry/index.js\";\nimport { getErrorMessage } from \"../utils/errors.js\";\nimport { regenerateLayerReports } from \"./reports.js\";\n\nexport interface InitOptions {\n  name?: string;\n  author?: string;\n  description?: string;\n  verbose?: boolean;\n  debug?: boolean;\n}\n\nexport async function initCommand(options: InitOptions): Promise<void> {\n  intro(ansis.bold(ansis.blue(\"\u2699\uFE0F  Initialize Documentation Robotics Model\")));\n\n  const span = isTelemetryEnabled\n    ? startSpan(\"model.init\", {\n        \"init.hasName\": !!options.name,\n        \"init.hasDescription\": !!options.description,\n        \"init.hasAuthor\": !!options.author,\n      })\n    : null;\n\n  try {\n    const rootPath = process.cwd();\n    const modelPath = `${rootPath}/documentation-robotics/model`;\n    const manifestPath = `${modelPath}/manifest.yaml`;\n\n    // Check if model already exists\n    if (await fileExists(manifestPath)) {\n      console.error(ansis.red(\"Error: Model already initialized in this directory\"));\n      if (isTelemetryEnabled && span) {\n        (span as any).setStatus({ code: 2, message: \"Model already exists\" });\n      }\n      endSpan(span);\n      process.exit(1);\n    }\n\n    // Gather model metadata\n    // Check if stdin is a TTY to determine if we should prompt\n    const isInteractive = process.stdin.isTTY;\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"cli.interactive\", isInteractive);\n    }\n\n    // If name is provided, skip all prompts\n    const skipPrompts = !!options.name;\n\n    const name =\n      options.name ||\n      (isInteractive\n        ? await text({\n            message: \"Model name:\",\n            validate: (value) => (value.length === 0 ? \"Name is required\" : undefined),\n          })\n        : (() => {\n            console.error(ansis.red(\"Error: Model name is required (use --name option)\"));\n            if (isTelemetryEnabled && span) {\n              (span as any).setStatus({ code: 2, message: \"Model name required\" });\n            }\n            endSpan(span);\n            process.exit(1);\n          })());\n\n    const description =\n      options.description ||\n      (skipPrompts\n        ? \"\"\n        : isInteractive\n          ? await text({\n              message: \"Description (optional):\",\n              defaultValue: \"\",\n            })\n          : \"\");\n\n    const author =\n      options.author ||\n      (skipPrompts\n        ? \"\"\n        : isInteractive\n          ? await text({\n              message: \"Author (optional):\",\n              defaultValue: \"\",\n            })\n          : \"\");\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"model.name\", name as string);\n      (span as any).setAttribute(\"model.specVersion\", getCliBundledSpecVersion());\n    }\n\n    // Initialize model\n    logDebug(`Creating model directory at ${rootPath}/documentation-robotics/model`);\n    const model = await Model.init(\n      rootPath,\n      {\n        name: name as string,\n        version: \"0.1.0\",\n        description: (description as string) || undefined,\n        author: (author as string) || undefined,\n        specVersion: getCliBundledSpecVersion(),\n        created: new Date().toISOString(),\n      },\n      { lazyLoad: false }\n    );\n\n    logVerbose(`Model saved with:\n  - Version: 0.1.0\n  - Spec Version: ${getCliBundledSpecVersion()}\n  - Location: ${rootPath}/documentation-robotics/model`);\n\n    // Install spec reference (.dr/ folder)\n    logDebug(\"Installing spec reference (.dr/ folder)...\");\n    await installSpecReference(rootPath, false);\n    logVerbose(\"Spec reference installed\");\n\n    // Generate README and layer reports\n    logDebug(\"Generating reports and README...\");\n    try {\n      await regenerateLayerReports(rootPath);\n      logVerbose(\"Reports and README generated\");\n    } catch (reportError) {\n      const reportMessage = getErrorMessage(reportError);\n      console.warn(\n        ansis.yellow(\n          `\u26A0 Warning: Failed to generate reports and README: ${reportMessage}\\n` +\n          `  Model initialized successfully. To regenerate reports later, run:\\n` +\n          `  dr reports regenerate`\n        )\n      );\n      logDebug(`Report generation failed (non-fatal): ${reportMessage}`);\n    }\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"model.path\", rootPath);\n      (span as any).setStatus({ code: 0 });\n    }\n\n    outro(ansis.green(`\u2713 Model initialized: ${ansis.bold(model.manifest.name)}`));\n  } catch (error) {\n    if (isTelemetryEnabled && span) {\n      (span as any).recordException(error as Error);\n      (span as any).setStatus({\n        code: 2,\n        message: getErrorMessage(error),\n      });\n    }\n    const message = getErrorMessage(error);\n    console.error(ansis.red(`Error: ${message}`));\n    endSpan(span);\n    process.exit(1);\n  } finally {\n    endSpan(span);\n  }\n}\n"],
-  "mappings": "AAIA,SAAS,OAAO,MAAM,aAAa;AACnC,OAAO,WAAW;AAClB,SAAS,aAAa;AACtB,SAAS,kBAAkB;AAC3B,SAAS,YAAY,gBAAgB;AACrC,SAAS,4BAA4B;AACrC,SAAS,gCAAgC;AACzC,SAAS,oBAAoB,WAAW,eAAe;AACvD,SAAS,uBAAuB;AAChC,SAAS,8BAA8B;AAUvC,eAAsB,YAAY,SAAoB;AACpD,QAAM,MAAM,KAAK,MAAM,KAAK,uDAA6C,CAAC,CAAC;AAE3E,QAAM,OAAO,qBACT,UAAU,cAAc;IACtB,gBAAgB,CAAC,CAAC,QAAQ;IAC1B,uBAAuB,CAAC,CAAC,QAAQ;IACjC,kBAAkB,CAAC,CAAC,QAAQ;GAC7B,IACD;AAEJ,MAAI;AACF,UAAM,WAAW,QAAQ,IAAG,GAEtB,eAAe,GADH,GAAG,QAAQ,+BACI;AAGjC,IAAI,MAAM,WAAW,YAAY,MAC/B,QAAQ,MAAM,MAAM,IAAI,oDAAoD,CAAC,GACzE,sBAAsB,QACvB,KAAa,UAAU,EAAE,MAAM,GAAG,SAAS,uBAAsB,CAAE,GAEtE,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;AAKhB,UAAM,gBAAgB,QAAQ,MAAM;AAEpC,IAAI,sBAAsB,QACvB,KAAa,aAAa,mBAAmB,aAAa;AAI7D,UAAM,cAAc,CAAC,CAAC,QAAQ,MAExB,OACJ,QAAQ,SACP,gBACG,MAAM,KAAK;MACT,SAAS;MACT,UAAU,CAAC,UAAW,MAAM,WAAW,IAAI,qBAAqB;KACjE,KACA,MAAK;AACJ,cAAQ,MAAM,MAAM,IAAI,mDAAmD,CAAC,GACxE,sBAAsB,QACvB,KAAa,UAAU,EAAE,MAAM,GAAG,SAAS,sBAAqB,CAAE,GAErE,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;IAChB,GAAE,IAEF,cACJ,QAAQ,gBACP,cACG,KACA,gBACE,MAAM,KAAK;MACT,SAAS;MACT,cAAc;KACf,IACD,KAEF,SACJ,QAAQ,WACP,cACG,KACA,gBACE,MAAM,KAAK;MACT,SAAS;MACT,cAAc;KACf,IACD;AAER,IAAI,sBAAsB,SACvB,KAAa,aAAa,cAAc,IAAc,GACtD,KAAa,aAAa,qBAAqB,yBAAwB,CAAE,IAI5E,SAAS,+BAA+B,QAAQ,+BAA+B;AAC/E,UAAM,QAAQ,MAAM,MAAM,KACxB,UACA;MACE;MACA,SAAS;MACT,aAAc,eAA0B;MACxC,QAAS,UAAqB;MAC9B,aAAa,yBAAwB;MACrC,UAAS,oBAAI,KAAI,GAAG,YAAW;OAEjC,EAAE,UAAU,GAAK,CAAE;AAGrB,eAAW;;oBAEK,yBAAwB,CAAE;gBAC9B,QAAQ,+BAA+B,GAGnD,SAAS,4CAA4C,GACrD,MAAM,qBAAqB,UAAU,EAAK,GAC1C,WAAW,0BAA0B,GAGrC,SAAS,kCAAkC;AAC3C,QAAI;AACF,YAAM,uBAAuB,QAAQ,GACrC,WAAW,8BAA8B;IAC3C,SAAS,aAAa;AACpB,YAAM,gBAAgB,gBAAgB,WAAW;AACjD,cAAQ,KACN,MAAM,OACJ,0DAAqD,aAAa;;wBAEzC,CAC1B,GAEH,SAAS,yCAAyC,aAAa,EAAE;IACnE;AAEA,IAAI,sBAAsB,SACvB,KAAa,aAAa,cAAc,QAAQ,GAChD,KAAa,UAAU,EAAE,MAAM,EAAC,CAAE,IAGrC,MAAM,MAAM,MAAM,6BAAwB,MAAM,KAAK,MAAM,SAAS,IAAI,CAAC,EAAE,CAAC;EAC9E,SAAS,OAAO;AACd,IAAI,sBAAsB,SACvB,KAAa,gBAAgB,KAAc,GAC3C,KAAa,UAAU;MACtB,MAAM;MACN,SAAS,gBAAgB,KAAK;KAC/B;AAEH,UAAM,UAAU,gBAAgB,KAAK;AACrC,YAAQ,MAAM,MAAM,IAAI,UAAU,OAAO,EAAE,CAAC,GAC5C,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;EAChB;AACE,YAAQ,IAAI;EACd;AACF;",
+  "sourcesContent": ["/**\n * Initialize a new Documentation Robotics model\n */\n\nimport { intro, text, outro } from \"@clack/prompts\";\nimport ansis from \"ansis\";\nimport { hasModel, initProject } from \"../core/project-init.js\";\nimport { getCliBundledSpecVersion } from \"../utils/spec-version.js\";\nimport { isTelemetryEnabled, startSpan, endSpan } from \"../telemetry/index.js\";\nimport { getErrorMessage } from \"../utils/errors.js\";\n\nexport interface InitOptions {\n  name?: string;\n  author?: string;\n  description?: string;\n  verbose?: boolean;\n  debug?: boolean;\n}\n\nexport async function initCommand(options: InitOptions): Promise<void> {\n  intro(ansis.bold(ansis.blue(\"\u2699\uFE0F  Initialize Documentation Robotics Model\")));\n\n  const span = isTelemetryEnabled\n    ? startSpan(\"model.init\", {\n        \"init.hasName\": !!options.name,\n        \"init.hasDescription\": !!options.description,\n        \"init.hasAuthor\": !!options.author,\n      })\n    : null;\n\n  try {\n    const rootPath = process.cwd();\n    // Check if model already exists\n    if (await hasModel(rootPath)) {\n      console.error(ansis.red(\"Error: Model already initialized in this directory\"));\n      if (isTelemetryEnabled && span) {\n        (span as any).setStatus({ code: 2, message: \"Model already exists\" });\n      }\n      endSpan(span);\n      process.exit(1);\n    }\n\n    // Gather model metadata\n    // Check if stdin is a TTY to determine if we should prompt\n    const isInteractive = process.stdin.isTTY;\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"cli.interactive\", isInteractive);\n    }\n\n    // If name is provided, skip all prompts\n    const skipPrompts = !!options.name;\n\n    const name =\n      options.name ||\n      (isInteractive\n        ? await text({\n            message: \"Model name:\",\n            validate: (value) => (value.length === 0 ? \"Name is required\" : undefined),\n          })\n        : (() => {\n            console.error(ansis.red(\"Error: Model name is required (use --name option)\"));\n            if (isTelemetryEnabled && span) {\n              (span as any).setStatus({ code: 2, message: \"Model name required\" });\n            }\n            endSpan(span);\n            process.exit(1);\n          })());\n\n    const description =\n      options.description ||\n      (skipPrompts\n        ? \"\"\n        : isInteractive\n          ? await text({\n              message: \"Description (optional):\",\n              defaultValue: \"\",\n            })\n          : \"\");\n\n    const author =\n      options.author ||\n      (skipPrompts\n        ? \"\"\n        : isInteractive\n          ? await text({\n              message: \"Author (optional):\",\n              defaultValue: \"\",\n            })\n          : \"\");\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"model.name\", name as string);\n      (span as any).setAttribute(\"model.specVersion\", getCliBundledSpecVersion());\n    }\n\n    // Create the model, spec reference and reports (shared with the project_init MCP tool)\n    const result = await initProject(rootPath, {\n      name: name as string,\n      description: (description as string) || undefined,\n      author: (author as string) || undefined,\n    });\n\n    if (result.reportError !== undefined) {\n      console.warn(\n        ansis.yellow(\n          `\u26A0 Warning: Failed to generate reports and README: ${result.reportError}\\n` +\n          `  Model initialized successfully. To regenerate reports later, run:\\n` +\n          `  dr reports regenerate`\n        )\n      );\n    }\n\n    if (isTelemetryEnabled && span) {\n      (span as any).setAttribute(\"model.path\", rootPath);\n      (span as any).setStatus({ code: 0 });\n    }\n\n    outro(ansis.green(`\u2713 Model initialized: ${ansis.bold(result.name)}`));\n  } catch (error) {\n    if (isTelemetryEnabled && span) {\n      (span as any).recordException(error as Error);\n      (span as any).setStatus({\n        code: 2,\n        message: getErrorMessage(error),\n      });\n    }\n    const message = getErrorMessage(error);\n    console.error(ansis.red(`Error: ${message}`));\n    endSpan(span);\n    process.exit(1);\n  } finally {\n    endSpan(span);\n  }\n}\n"],
+  "mappings": "AAIA,SAAS,OAAO,MAAM,aAAa;AACnC,OAAO,WAAW;AAClB,SAAS,UAAU,mBAAmB;AACtC,SAAS,gCAAgC;AACzC,SAAS,oBAAoB,WAAW,eAAe;AACvD,SAAS,uBAAuB;AAUhC,eAAsB,YAAY,SAAoB;AACpD,QAAM,MAAM,KAAK,MAAM,KAAK,uDAA6C,CAAC,CAAC;AAE3E,QAAM,OAAO,qBACT,UAAU,cAAc;IACtB,gBAAgB,CAAC,CAAC,QAAQ;IAC1B,uBAAuB,CAAC,CAAC,QAAQ;IACjC,kBAAkB,CAAC,CAAC,QAAQ;GAC7B,IACD;AAEJ,MAAI;AACF,UAAM,WAAW,QAAQ,IAAG;AAE5B,IAAI,MAAM,SAAS,QAAQ,MACzB,QAAQ,MAAM,MAAM,IAAI,oDAAoD,CAAC,GACzE,sBAAsB,QACvB,KAAa,UAAU,EAAE,MAAM,GAAG,SAAS,uBAAsB,CAAE,GAEtE,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;AAKhB,UAAM,gBAAgB,QAAQ,MAAM;AAEpC,IAAI,sBAAsB,QACvB,KAAa,aAAa,mBAAmB,aAAa;AAI7D,UAAM,cAAc,CAAC,CAAC,QAAQ,MAExB,OACJ,QAAQ,SACP,gBACG,MAAM,KAAK;MACT,SAAS;MACT,UAAU,CAAC,UAAW,MAAM,WAAW,IAAI,qBAAqB;KACjE,KACA,MAAK;AACJ,cAAQ,MAAM,MAAM,IAAI,mDAAmD,CAAC,GACxE,sBAAsB,QACvB,KAAa,UAAU,EAAE,MAAM,GAAG,SAAS,sBAAqB,CAAE,GAErE,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;IAChB,GAAE,IAEF,cACJ,QAAQ,gBACP,cACG,KACA,gBACE,MAAM,KAAK;MACT,SAAS;MACT,cAAc;KACf,IACD,KAEF,SACJ,QAAQ,WACP,cACG,KACA,gBACE,MAAM,KAAK;MACT,SAAS;MACT,cAAc;KACf,IACD;AAER,IAAI,sBAAsB,SACvB,KAAa,aAAa,cAAc,IAAc,GACtD,KAAa,aAAa,qBAAqB,yBAAwB,CAAE;AAI5E,UAAM,SAAS,MAAM,YAAY,UAAU;MACzC;MACA,aAAc,eAA0B;MACxC,QAAS,UAAqB;KAC/B;AAED,IAAI,OAAO,gBAAgB,UACzB,QAAQ,KACN,MAAM,OACJ,0DAAqD,OAAO,WAAW;;wBAE9C,CAC1B,GAID,sBAAsB,SACvB,KAAa,aAAa,cAAc,QAAQ,GAChD,KAAa,UAAU,EAAE,MAAM,EAAC,CAAE,IAGrC,MAAM,MAAM,MAAM,6BAAwB,MAAM,KAAK,OAAO,IAAI,CAAC,EAAE,CAAC;EACtE,SAAS,OAAO;AACd,IAAI,sBAAsB,SACvB,KAAa,gBAAgB,KAAc,GAC3C,KAAa,UAAU;MACtB,MAAM;MACN,SAAS,gBAAgB,KAAK;KAC/B;AAEH,UAAM,UAAU,gBAAgB,KAAK;AACrC,YAAQ,MAAM,MAAM,IAAI,UAAU,OAAO,EAAE,CAAC,GAC5C,QAAQ,IAAI,GACZ,QAAQ,KAAK,CAAC;EAChB;AACE,YAAQ,IAAI;EACd;AACF;",
   "names": []
 }

package/dist/commands/mcp.js

@@ -1,8 +1,11 @@
 import { isCancel, text } from "@clack/prompts";
+import { RootsListChangedNotificationSchema } from "@modelcontextprotocol/sdk/types.js";
 import { ApiKeyManager } from "../mcp/api-key-manager.js";
+import { followClientRoots, markEnvProjectRoot } from "../mcp/project-root.js";
 import { McpResourceRegistry } from "../mcp/resource-registry.js";
+import { SERVER_INSTRUCTIONS } from "../mcp/server-instructions.js";
 import { McpToolRegistry } from "../mcp/tool-registry.js";
-import { loadModel } from "../mcp/tools/shared.js";
+import { clearModelCache, loadModel } from "../mcp/tools/shared.js";
 import { startActiveSpan } from "../telemetry/index.js";
 import { CLIError, getErrorMessage } from "../utils/errors.js";
 import { getCliVersion } from "../utils/spec-version.js";
@@ -15,48 +18,55 @@ const cliVersion = getCliVersion(), promptForKeyPath = async (defaultPath) => {
   return isCancel(result) || typeof result != "string" || result.trim().length === 0 ? defaultPath : result.trim();
 };
 async function createConfiguredServer() {
-  const { McpServer } = await import("@modelcontextprotocol/sdk/server/mcp.js"), server = new McpServer({
-    name: "documentation-robotics",
-    version: cliVersion
-  });
-  new McpToolRegistry().registerAll(server), await new McpResourceRegistry().registerAll(server);
+  const { McpServer } = await import("@modelcontextprotocol/sdk/server/mcp.js"), server = new McpServer({ name: "documentation-robotics", version: cliVersion }, { instructions: SERVER_INSTRUCTIONS });
+  return new McpToolRegistry().registerAll(server), await new McpResourceRegistry().registerAll(server), server;
+}
+async function warmModel() {
   try {
     await loadModel();
   } catch (error) {
     process.stderr.write(`[mcp] model warmup skipped: ${getErrorMessage(error)}
 `);
   }
-  return server;
+}
+function serveClientRoots(server) {
+  const inner = server.server, follow = () => followClientRoots(() => inner.listRoots(), { onChange: () => clearModelCache() }).then(warmModel);
+  inner.setNotificationHandler(RootsListChangedNotificationSchema, async () => {
+    await follow();
+  }), inner.oninitialized = () => {
+    inner.getClientCapabilities()?.roots ? follow() : warmModel();
+  };
 }
 function serveFromRootPathEnv(env = process.env) {
   const rootPath = env.DR_MCP_ROOT_PATH;
-  if (rootPath)
-    try {
-      process.chdir(rootPath);
-    } catch (error) {
-      process.stderr.write(`[mcp] DR_MCP_ROOT_PATH=${rootPath} is not usable (${getErrorMessage(error)}); serving from ${process.cwd()}
-`);
-    }
+  if (!rootPath)
+    return !1;
+  try {
+    process.chdir(rootPath);
+  } catch (error) {
+    return process.stderr.write(`[mcp] DR_MCP_ROOT_PATH=${rootPath} is not usable (${getErrorMessage(error)}); serving from ${process.cwd()}
+`), !1;
+  }
+  return markEnvProjectRoot(process.cwd()), !0;
 }
 async function mcpCommand(options = {}) {
   const keyManager = new ApiKeyManager();
   try {
     if (options.regenerateKey) {
       await startActiveSpan("mcp.key.rotate", async (span) => {
-        const isInteractive2 = process.stdin.isTTY === !0, { key: key2, path: path2 } = await keyManager.rotate(isInteractive2 ? promptForKeyPath : void 0);
-        span.setAttribute("mcp.key.path", path2), process.stderr.write(`Generated new MCP API key, stored at ${path2}
-`), process.stderr.write(`MCP API key: ${key2}
+        const isInteractive = process.stdin.isTTY === !0, { key, path } = await keyManager.rotate(isInteractive ? promptForKeyPath : void 0);
+        span.setAttribute("mcp.key.path", path), process.stderr.write(`Generated new MCP API key, stored at ${path}
+`), process.stderr.write(`MCP API key: ${key}
 `);
       });
       return;
     }
-    serveFromRootPathEnv();
-    const isInteractive = process.stdin.isTTY === !0, { key, path, isNew } = await keyManager.ensureKey(isInteractive ? promptForKeyPath : void 0);
-    isNew && process.stderr.write(`Generated new MCP API key, stored at ${path}
-`);
-    const quietKey = /^(1|true|yes|on)$/i.test(process.env.DR_MCP_QUIET_KEY ?? "");
-    if (quietKey || process.stderr.write(`MCP API key: ${key}
-`), options.transport === "http") {
+    const servesEnvRoot = serveFromRootPathEnv();
+    if (options.transport === "http") {
+      const isInteractive = process.stdin.isTTY === !0, { key, path, isNew } = await keyManager.ensureKey(isInteractive ? promptForKeyPath : void 0);
+      isNew && process.stderr.write(`Generated new MCP API key, stored at ${path}
+`), /^(1|true|yes|on)$/i.test(process.env.DR_MCP_QUIET_KEY ?? "") || process.stderr.write(`MCP API key: ${key}
+`), await warmModel();
       const { createMcpHttpApp, startHttpServer, closeHttpServer } = await import("../mcp/http-transport.js"), host = options.host || "127.0.0.1", port = options.port || 3100;
       let httpApp;
       const httpServer = await startActiveSpan("mcp.server.start", async (span) => {
@@ -90,15 +100,9 @@ Shutting down HTTP server...
       };
       process.on("SIGINT", shutdown), process.on("SIGTERM", shutdown), await keepAlive;
     } else {
-      if (!keyManager.validate(process.env.DR_MCP_API_KEY, key))
-        throw quietKey && process.stderr.write(`MCP API key: ${key}
-`), process.stderr.write(`Error: DR_MCP_API_KEY environment variable is missing or does not match the stored API key.
-`), process.stderr.write(`Set DR_MCP_API_KEY to the key above when configuring the MCP client, e.g.:
-`), process.stderr.write(`  { "command": "dr", "args": ["mcp"], "env": { "DR_MCP_API_KEY": "<key>" } }
-`), new CLIError("MCP authentication failed: invalid or missing DR_MCP_API_KEY", 1);
       const transport = await startActiveSpan("mcp.server.start", async (span) => {
         const { StdioServerTransport } = await import("@modelcontextprotocol/sdk/server/stdio.js"), server = await createConfiguredServer();
-        span.setAttribute("mcp.server.name", "documentation-robotics"), span.setAttribute("mcp.server.version", cliVersion), span.setAttribute("mcp.server.transport", "stdio");
+        servesEnvRoot ? await warmModel() : serveClientRoots(server), span.setAttribute("mcp.server.name", "documentation-robotics"), span.setAttribute("mcp.server.version", cliVersion), span.setAttribute("mcp.server.transport", "stdio");
         const serverTransport = new StdioServerTransport();
         return await server.connect(serverTransport), serverTransport;
       });

package/dist/commands/mcp.js.map

@@ -1,7 +1,7 @@
 {
   "version": 3,
   "sources": ["../../src/commands/mcp.ts"],
-  "sourcesContent": ["/**\n * MCP server command - Starts an MCP server for AI assistant integration\n *\n * Supports two transport modes:\n * - stdio (default): In-process over stdio; stdout carries the MCP JSON-RPC protocol\n * - http: HTTP server with Express, exposing MCP at /mcp endpoint with bearer token auth\n *\n * All non-protocol output goes to stderr.\n */\n\nimport { isCancel, text } from \"@clack/prompts\";\nimport { ApiKeyManager, type ApiKeyStoragePrompt } from \"../mcp/api-key-manager.js\";\nimport { type HttpTransportApp } from \"../mcp/http-transport.js\";\nimport { McpResourceRegistry } from \"../mcp/resource-registry.js\";\nimport { McpToolRegistry } from \"../mcp/tool-registry.js\";\nimport { loadModel } from \"../mcp/tools/shared.js\";\nimport { startActiveSpan } from \"../telemetry/index.js\";\nimport { CLIError, getErrorMessage } from \"../utils/errors.js\";\nimport { getCliVersion } from \"../utils/spec-version.js\";\n\nconst cliVersion = getCliVersion();\n\n/**\n * Prompt the user (via stdout/stdin) for where to store a newly generated API key.\n * Only invoked when stdin is a TTY \u2014 non-interactive launches (e.g. an MCP client\n * spawning `dr mcp`) fall back to the default path without prompting.\n */\nconst promptForKeyPath: ApiKeyStoragePrompt = async (defaultPath) => {\n  const result = await text({\n    message: \"Where should the MCP API key be stored?\",\n    placeholder: defaultPath,\n    defaultValue: defaultPath,\n  });\n\n  if (isCancel(result) || typeof result !== \"string\" || result.trim().length === 0) {\n    return defaultPath;\n  }\n  return result.trim();\n};\n\nasync function createConfiguredServer() {\n  const { McpServer } = await import(\"@modelcontextprotocol/sdk/server/mcp.js\");\n\n  const server = new McpServer({\n    name: \"documentation-robotics\",\n    version: cliVersion,\n  });\n\n  // Register the model tool surface and spec/model manifest resources.\n  new McpToolRegistry().registerAll(server);\n  await new McpResourceRegistry().registerAll(server);\n\n  // Warm the model cache at startup so it's loaded once and held in memory\n  // for the session, rather than waiting for the first tool call. Best-effort:\n  // a server started outside a DR project still starts up cleanly \u2014 the first\n  // tool call that supplies a valid rootPath will load and cache it.\n  try {\n    await loadModel();\n  } catch (error) {\n    process.stderr.write(`[mcp] model warmup skipped: ${getErrorMessage(error)}\\n`);\n  }\n\n  return server;\n}\n\n/**\n * DR_MCP_ROOT_PATH lets a launcher (DrBot under `dr visualize`) point the server at a model\n * without relying on the cwd its MCP client spawns it in: tools default to this directory. A bad\n * path is reported and the server keeps its own cwd.\n */\nexport function serveFromRootPathEnv(env: NodeJS.ProcessEnv = process.env): void {\n  const rootPath = env.DR_MCP_ROOT_PATH;\n  if (!rootPath) return;\n  try {\n    process.chdir(rootPath);\n  } catch (error) {\n    process.stderr.write(\n      `[mcp] DR_MCP_ROOT_PATH=${rootPath} is not usable (${getErrorMessage(error)}); serving from ${process.cwd()}\\n`\n    );\n  }\n}\n\nexport interface McpCommandOptions {\n  /** Force-generate a new API key, overwrite it at the configured storage path, and exit. */\n  regenerateKey?: boolean;\n  /** Transport type: \"stdio\" or \"http\" (default: \"stdio\") */\n  transport?: \"stdio\" | \"http\";\n  /** Port for HTTP transport (default: 3100, only meaningful with --transport http) */\n  port?: number;\n  /** Host address for HTTP transport (default: \"127.0.0.1\", only meaningful with --transport http) */\n  host?: string;\n}\n\nexport async function mcpCommand(options: McpCommandOptions = {}): Promise<void> {\n  const keyManager = new ApiKeyManager();\n\n  try {\n    if (options.regenerateKey) {\n      await startActiveSpan(\"mcp.key.rotate\", async (span) => {\n        const isInteractive = process.stdin.isTTY === true;\n        const { key, path } = await keyManager.rotate(\n          isInteractive ? promptForKeyPath : undefined\n        );\n        span.setAttribute(\"mcp.key.path\", path);\n        process.stderr.write(`Generated new MCP API key, stored at ${path}\\n`);\n        process.stderr.write(`MCP API key: ${key}\\n`);\n      });\n      return;\n    }\n\n    serveFromRootPathEnv();\n\n    const isInteractive = process.stdin.isTTY === true;\n    const { key, path, isNew } = await keyManager.ensureKey(\n      isInteractive ? promptForKeyPath : undefined\n    );\n\n    if (isNew) {\n      process.stderr.write(`Generated new MCP API key, stored at ${path}\\n`);\n    }\n    // Print the API key to stderr (never stdout, which carries the MCP protocol) so a user can\n    // configure a client. A launcher that already passes the key (DrBot) sets DR_MCP_QUIET_KEY so\n    // the key does not end up in the client's MCP server logs on every start.\n    const quietKey = /^(1|true|yes|on)$/i.test(process.env.DR_MCP_QUIET_KEY ?? \"\");\n    if (!quietKey) {\n      process.stderr.write(`MCP API key: ${key}\\n`);\n    }\n\n    if (options.transport === \"http\") {\n      // HTTP transport implementation\n      const { createMcpHttpApp, startHttpServer, closeHttpServer } = await import(\n        \"../mcp/http-transport.js\"\n      );\n\n      const host = options.host || \"127.0.0.1\";\n      const port = options.port || 3100;\n\n      let httpApp: HttpTransportApp;\n      const httpServer = await startActiveSpan(\n        \"mcp.server.start\",\n        async (span) => {\n          httpApp = await createMcpHttpApp(keyManager, key, createConfiguredServer, host);\n          const server = await startHttpServer(httpApp, host, port);\n\n          span.setAttribute(\"mcp.server.name\", \"documentation-robotics\");\n          span.setAttribute(\"mcp.server.version\", cliVersion);\n          span.setAttribute(\"mcp.server.transport\", \"http\");\n          span.setAttribute(\"mcp.server.host\", host);\n          span.setAttribute(\"mcp.server.port\", port);\n\n          return server;\n        }\n      );\n\n      // Warn if bound to 0.0.0.0\n      if (host === \"0.0.0.0\") {\n        process.stderr.write(\n          \"Warning: Server bound to 0.0.0.0. DNS rebinding protection is not automatic; ensure your MCP client validates the Host header.\\n\"\n        );\n      }\n\n      process.stderr.write(`Documentation Robotics MCP server ready (http://${host}:${port}/mcp)\\n`);\n\n      // Set up graceful shutdown on SIGINT/SIGTERM\n      let resolveKeepAlive: () => void;\n      const keepAlive = new Promise<void>((resolve) => {\n        resolveKeepAlive = resolve;\n      });\n\n      let isShuttingDown = false;\n      const shutdown = async () => {\n        if (isShuttingDown) {\n          return;\n        }\n        isShuttingDown = true;\n\n        try {\n          process.stderr.write(\"\\nShutting down HTTP server...\\n\");\n          await closeHttpServer(httpServer, httpApp);\n        } catch (error) {\n          const errorMsg = error instanceof Error ? error.message : \"unknown error\";\n          process.stderr.write(`[mcp] Error closing server: ${errorMsg}\\n`);\n        } finally {\n          resolveKeepAlive!();\n        }\n      };\n\n      process.on(\"SIGINT\", shutdown);\n      process.on(\"SIGTERM\", shutdown);\n\n      // Keep the process alive until explicitly terminated\n      await keepAlive;\n    } else {\n      // Stdio transport (default)\n      // Validate that the environment has the correct API key configured\n      if (!keyManager.validate(process.env.DR_MCP_API_KEY, key)) {\n        // A quiet launch passed a key that doesn't match: show the right one so it can be fixed.\n        if (quietKey) process.stderr.write(`MCP API key: ${key}\\n`);\n        process.stderr.write(\n          \"Error: DR_MCP_API_KEY environment variable is missing or does not match the stored API key.\\n\"\n        );\n        process.stderr.write(\n          \"Set DR_MCP_API_KEY to the key above when configuring the MCP client, e.g.:\\n\"\n        );\n        process.stderr.write(\n          '  { \"command\": \"dr\", \"args\": [\"mcp\"], \"env\": { \"DR_MCP_API_KEY\": \"<key>\" } }\\n'\n        );\n        throw new CLIError(\"MCP authentication failed: invalid or missing DR_MCP_API_KEY\", 1);\n      }\n\n      const transport = await startActiveSpan(\n        \"mcp.server.start\",\n        async (span) => {\n          const { StdioServerTransport } = await import(\"@modelcontextprotocol/sdk/server/stdio.js\");\n\n          const server = await createConfiguredServer();\n\n          span.setAttribute(\"mcp.server.name\", \"documentation-robotics\");\n          span.setAttribute(\"mcp.server.version\", cliVersion);\n          span.setAttribute(\"mcp.server.transport\", \"stdio\");\n\n          const serverTransport = new StdioServerTransport();\n          await server.connect(serverTransport);\n          return serverTransport;\n        }\n      );\n\n      process.stderr.write(\"Documentation Robotics MCP server ready (stdio)\\n\");\n\n      // Keep the process alive for the lifetime of the stdio session; the\n      // transport closes (and the process exits) when stdin closes. Shutdown\n      // is implicit in the `mcp.server.start` span's end, so no separate\n      // zero-duration \"stop\" span is recorded here.\n      await new Promise<void>((resolve) => {\n        transport.onclose = () => resolve();\n      });\n    }\n  } catch (error) {\n    if (error instanceof CLIError) {\n      throw error;\n    }\n\n    let message = getErrorMessage(error);\n\n    // Handle port-in-use errors with actionable message\n    if (error instanceof Error && (error as NodeJS.ErrnoException).code === \"EADDRINUSE\") {\n      const port = options.port || 3100;\n      message = `Port ${port} is already in use. Try a different port with --port or kill the process using that port.`;\n    }\n\n    process.stderr.write(`Error: ${message}\\n`);\n    throw new CLIError(message, 1);\n  }\n}\n"],
-  "mappings": "AAUA,SAAS,UAAU,YAAY;AAC/B,SAAS,qBAA+C;AAExD,SAAS,2BAA2B;AACpC,SAAS,uBAAuB;AAChC,SAAS,iBAAiB;AAC1B,SAAS,uBAAuB;AAChC,SAAS,UAAU,uBAAuB;AAC1C,SAAS,qBAAqB;AAE9B,MAAM,aAAa,cAAa,GAO1B,mBAAwC,OAAO,gBAAe;AAClE,QAAM,SAAS,MAAM,KAAK;IACxB,SAAS;IACT,aAAa;IACb,cAAc;GACf;AAED,SAAI,SAAS,MAAM,KAAK,OAAO,UAAW,YAAY,OAAO,KAAI,EAAG,WAAW,IACtE,cAEF,OAAO,KAAI;AACpB;AAEA,eAAe,yBAAsB;AACnC,QAAM,EAAE,UAAS,IAAK,MAAM,OAAO,yCAAyC,GAEtE,SAAS,IAAI,UAAU;IAC3B,MAAM;IACN,SAAS;GACV;AAGD,MAAI,gBAAe,EAAG,YAAY,MAAM,GACxC,MAAM,IAAI,oBAAmB,EAAG,YAAY,MAAM;AAMlD,MAAI;AACF,UAAM,UAAS;EACjB,SAAS,OAAO;AACd,YAAQ,OAAO,MAAM,+BAA+B,gBAAgB,KAAK,CAAC;CAAI;EAChF;AAEA,SAAO;AACT;AAOM,SAAU,qBAAqB,MAAyB,QAAQ,KAAG;AACvE,QAAM,WAAW,IAAI;AACrB,MAAK;AACL,QAAI;AACF,cAAQ,MAAM,QAAQ;IACxB,SAAS,OAAO;AACd,cAAQ,OAAO,MACb,0BAA0B,QAAQ,mBAAmB,gBAAgB,KAAK,CAAC,mBAAmB,QAAQ,IAAG,CAAE;CAAI;IAEnH;AACF;AAaA,eAAsB,WAAW,UAA6B,CAAA,GAAE;AAC9D,QAAM,aAAa,IAAI,cAAa;AAEpC,MAAI;AACF,QAAI,QAAQ,eAAe;AACzB,YAAM,gBAAgB,kBAAkB,OAAO,SAAQ;AACrD,cAAMA,iBAAgB,QAAQ,MAAM,UAAU,IACxC,EAAE,KAAAC,MAAK,MAAAC,MAAI,IAAK,MAAM,WAAW,OACrCF,iBAAgB,mBAAmB,MAAS;AAE9C,aAAK,aAAa,gBAAgBE,KAAI,GACtC,QAAQ,OAAO,MAAM,wCAAwCA,KAAI;CAAI,GACrE,QAAQ,OAAO,MAAM,gBAAgBD,IAAG;CAAI;MAC9C,CAAC;AACD;IACF;AAEA,yBAAoB;AAEpB,UAAM,gBAAgB,QAAQ,MAAM,UAAU,IACxC,EAAE,KAAK,MAAM,MAAK,IAAK,MAAM,WAAW,UAC5C,gBAAgB,mBAAmB,MAAS;AAG9C,IAAI,SACF,QAAQ,OAAO,MAAM,wCAAwC,IAAI;CAAI;AAKvE,UAAM,WAAW,qBAAqB,KAAK,QAAQ,IAAI,oBAAoB,EAAE;AAK7E,QAJK,YACH,QAAQ,OAAO,MAAM,gBAAgB,GAAG;CAAI,GAG1C,QAAQ,cAAc,QAAQ;AAEhC,YAAM,EAAE,kBAAkB,iBAAiB,gBAAe,IAAK,MAAM,OACnE,0BAA0B,GAGtB,OAAO,QAAQ,QAAQ,aACvB,OAAO,QAAQ,QAAQ;AAE7B,UAAI;AACJ,YAAM,aAAa,MAAM,gBACvB,oBACA,OAAO,SAAQ;AACb,kBAAU,MAAM,iBAAiB,YAAY,KAAK,wBAAwB,IAAI;AAC9E,cAAM,SAAS,MAAM,gBAAgB,SAAS,MAAM,IAAI;AAExD,oBAAK,aAAa,mBAAmB,wBAAwB,GAC7D,KAAK,aAAa,sBAAsB,UAAU,GAClD,KAAK,aAAa,wBAAwB,MAAM,GAChD,KAAK,aAAa,mBAAmB,IAAI,GACzC,KAAK,aAAa,mBAAmB,IAAI,GAElC;MACT,CAAC;AAIH,MAAI,SAAS,aACX,QAAQ,OAAO,MACb;CAAkI,GAItI,QAAQ,OAAO,MAAM,mDAAmD,IAAI,IAAI,IAAI;CAAS;AAG7F,UAAI;AACJ,YAAM,YAAY,IAAI,QAAc,CAAC,YAAW;AAC9C,2BAAmB;MACrB,CAAC;AAED,UAAI,iBAAiB;AACrB,YAAM,WAAW,YAAW;AAC1B,YAAI,iBAGJ;2BAAiB;AAEjB,cAAI;AACF,oBAAQ,OAAO,MAAM;;CAAkC,GACvD,MAAM,gBAAgB,YAAY,OAAO;UAC3C,SAAS,OAAO;AACd,kBAAM,WAAW,iBAAiB,QAAQ,MAAM,UAAU;AAC1D,oBAAQ,OAAO,MAAM,+BAA+B,QAAQ;CAAI;UAClE;AACE,6BAAiB;UACnB;;MACF;AAEA,cAAQ,GAAG,UAAU,QAAQ,GAC7B,QAAQ,GAAG,WAAW,QAAQ,GAG9B,MAAM;IACR,OAAO;AAGL,UAAI,CAAC,WAAW,SAAS,QAAQ,IAAI,gBAAgB,GAAG;AAEtD,cAAI,YAAU,QAAQ,OAAO,MAAM,gBAAgB,GAAG;CAAI,GAC1D,QAAQ,OAAO,MACb;CAA+F,GAEjG,QAAQ,OAAO,MACb;CAA8E,GAEhF,QAAQ,OAAO,MACb;CAAgF,GAE5E,IAAI,SAAS,gEAAgE,CAAC;AAGtF,YAAM,YAAY,MAAM,gBACtB,oBACA,OAAO,SAAQ;AACb,cAAM,EAAE,qBAAoB,IAAK,MAAM,OAAO,2CAA2C,GAEnF,SAAS,MAAM,uBAAsB;AAE3C,aAAK,aAAa,mBAAmB,wBAAwB,GAC7D,KAAK,aAAa,sBAAsB,UAAU,GAClD,KAAK,aAAa,wBAAwB,OAAO;AAEjD,cAAM,kBAAkB,IAAI,qBAAoB;AAChD,qBAAM,OAAO,QAAQ,eAAe,GAC7B;MACT,CAAC;AAGH,cAAQ,OAAO,MAAM;CAAmD,GAMxE,MAAM,IAAI,QAAc,CAAC,YAAW;AAClC,kBAAU,UAAU,MAAM,QAAO;MACnC,CAAC;IACH;EACF,SAAS,OAAO;AACd,QAAI,iBAAiB;AACnB,YAAM;AAGR,QAAI,UAAU,gBAAgB,KAAK;AAGnC,UAAI,iBAAiB,SAAU,MAAgC,SAAS,iBAEtE,UAAU,QADG,QAAQ,QAAQ,IACP,8FAGxB,QAAQ,OAAO,MAAM,UAAU,OAAO;CAAI,GACpC,IAAI,SAAS,SAAS,CAAC;EAC/B;AACF;",
-  "names": ["isInteractive", "key", "path"]
+  "sourcesContent": ["/**\n * MCP server command - Starts an MCP server for AI assistant integration\n *\n * Supports two transport modes:\n * - stdio (default): In-process over stdio; stdout carries the MCP JSON-RPC protocol. The client\n *   spawns the server as the same user, so there is no API key: anything that can start\n *   `dr mcp` can already read the model files.\n * - http: HTTP server with Express, exposing MCP at /mcp endpoint with bearer token auth\n *\n * All non-protocol output goes to stderr.\n */\n\nimport { isCancel, text } from \"@clack/prompts\";\nimport type { McpServer } from \"@modelcontextprotocol/sdk/server/mcp.js\";\nimport { RootsListChangedNotificationSchema } from \"@modelcontextprotocol/sdk/types.js\";\nimport { ApiKeyManager, type ApiKeyStoragePrompt } from \"../mcp/api-key-manager.js\";\nimport { type HttpTransportApp } from \"../mcp/http-transport.js\";\nimport { followClientRoots, markEnvProjectRoot } from \"../mcp/project-root.js\";\nimport { McpResourceRegistry } from \"../mcp/resource-registry.js\";\nimport { SERVER_INSTRUCTIONS } from \"../mcp/server-instructions.js\";\nimport { McpToolRegistry } from \"../mcp/tool-registry.js\";\nimport { clearModelCache, loadModel } from \"../mcp/tools/shared.js\";\nimport { startActiveSpan } from \"../telemetry/index.js\";\nimport { CLIError, getErrorMessage } from \"../utils/errors.js\";\nimport { getCliVersion } from \"../utils/spec-version.js\";\n\nconst cliVersion = getCliVersion();\n\n/**\n * Prompt the user (via stdout/stdin) for where to store a newly generated API key.\n * Only invoked when stdin is a TTY \u2014 non-interactive launches (e.g. an MCP client\n * spawning `dr mcp`) fall back to the default path without prompting.\n */\nconst promptForKeyPath: ApiKeyStoragePrompt = async (defaultPath) => {\n  const result = await text({\n    message: \"Where should the MCP API key be stored?\",\n    placeholder: defaultPath,\n    defaultValue: defaultPath,\n  });\n\n  if (isCancel(result) || typeof result !== \"string\" || result.trim().length === 0) {\n    return defaultPath;\n  }\n  return result.trim();\n};\n\nasync function createConfiguredServer(): Promise<McpServer> {\n  const { McpServer } = await import(\"@modelcontextprotocol/sdk/server/mcp.js\");\n\n  const server = new McpServer(\n    { name: \"documentation-robotics\", version: cliVersion },\n    { instructions: SERVER_INSTRUCTIONS }\n  );\n\n  // Register the model tool surface and spec/model manifest resources.\n  new McpToolRegistry().registerAll(server);\n  await new McpResourceRegistry().registerAll(server);\n  return server;\n}\n\n/**\n * Loads the default project's model into the cache so the first tool call doesn't wait for it.\n * Best-effort: a server started outside a DR project still starts cleanly, and the first tool\n * call that supplies a valid rootPath loads and caches that model.\n */\nasync function warmModel(): Promise<void> {\n  try {\n    await loadModel();\n  } catch (error) {\n    process.stderr.write(`[mcp] model warmup skipped: ${getErrorMessage(error)}\\n`);\n  }\n}\n\n/**\n * Serve the project the client has open: once the client has initialized, ask for its roots\n * (mcp/project-root.ts), and ask again whenever it says they changed. A client without roots\n * leaves the server in its working directory.\n */\nfunction serveClientRoots(server: McpServer): void {\n  const inner = server.server;\n  const follow = () =>\n    followClientRoots(() => inner.listRoots(), { onChange: () => clearModelCache() }).then(warmModel);\n\n  inner.setNotificationHandler(RootsListChangedNotificationSchema, async () => {\n    await follow();\n  });\n  inner.oninitialized = () => {\n    if (inner.getClientCapabilities()?.roots) {\n      void follow();\n    } else {\n      void warmModel();\n    }\n  };\n}\n\n/**\n * DR_MCP_ROOT_PATH lets a launcher (DrBot under `dr visualize`) point the server at a model\n * without relying on the cwd its MCP client spawns it in: tools default to this directory, and\n * the client's roots are not consulted. A bad path is reported and the server keeps its own cwd.\n * Returns whether the server now serves from DR_MCP_ROOT_PATH.\n */\nexport function serveFromRootPathEnv(env: NodeJS.ProcessEnv = process.env): boolean {\n  const rootPath = env.DR_MCP_ROOT_PATH;\n  if (!rootPath) return false;\n  try {\n    process.chdir(rootPath);\n  } catch (error) {\n    process.stderr.write(\n      `[mcp] DR_MCP_ROOT_PATH=${rootPath} is not usable (${getErrorMessage(error)}); serving from ${process.cwd()}\\n`\n    );\n    return false;\n  }\n  markEnvProjectRoot(process.cwd());\n  return true;\n}\n\nexport interface McpCommandOptions {\n  /** Force-generate a new API key, overwrite it at the configured storage path, and exit. */\n  regenerateKey?: boolean;\n  /** Transport type: \"stdio\" or \"http\" (default: \"stdio\") */\n  transport?: \"stdio\" | \"http\";\n  /** Port for HTTP transport (default: 3100, only meaningful with --transport http) */\n  port?: number;\n  /** Host address for HTTP transport (default: \"127.0.0.1\", only meaningful with --transport http) */\n  host?: string;\n}\n\nexport async function mcpCommand(options: McpCommandOptions = {}): Promise<void> {\n  const keyManager = new ApiKeyManager();\n\n  try {\n    if (options.regenerateKey) {\n      await startActiveSpan(\"mcp.key.rotate\", async (span) => {\n        const isInteractive = process.stdin.isTTY === true;\n        const { key, path } = await keyManager.rotate(\n          isInteractive ? promptForKeyPath : undefined\n        );\n        span.setAttribute(\"mcp.key.path\", path);\n        process.stderr.write(`Generated new MCP API key, stored at ${path}\\n`);\n        process.stderr.write(`MCP API key: ${key}\\n`);\n      });\n      return;\n    }\n\n    const servesEnvRoot = serveFromRootPathEnv();\n\n    if (options.transport === \"http\") {\n      // HTTP clients authenticate with the API key as a bearer token.\n      const isInteractive = process.stdin.isTTY === true;\n      const { key, path, isNew } = await keyManager.ensureKey(\n        isInteractive ? promptForKeyPath : undefined\n      );\n      if (isNew) {\n        process.stderr.write(`Generated new MCP API key, stored at ${path}\\n`);\n      }\n      // Print the key to stderr so a user can configure a client. DR_MCP_QUIET_KEY keeps it out\n      // of logs for a launcher that already has it.\n      if (!/^(1|true|yes|on)$/i.test(process.env.DR_MCP_QUIET_KEY ?? \"\")) {\n        process.stderr.write(`MCP API key: ${key}\\n`);\n      }\n\n      // One server for every HTTP session, so the project is the env root or the cwd (client\n      // roots are per session and can't move a shared working directory).\n      await warmModel();\n      const { createMcpHttpApp, startHttpServer, closeHttpServer } = await import(\n        \"../mcp/http-transport.js\"\n      );\n\n      const host = options.host || \"127.0.0.1\";\n      const port = options.port || 3100;\n\n      let httpApp: HttpTransportApp;\n      const httpServer = await startActiveSpan(\n        \"mcp.server.start\",\n        async (span) => {\n          httpApp = await createMcpHttpApp(keyManager, key, createConfiguredServer, host);\n          const server = await startHttpServer(httpApp, host, port);\n\n          span.setAttribute(\"mcp.server.name\", \"documentation-robotics\");\n          span.setAttribute(\"mcp.server.version\", cliVersion);\n          span.setAttribute(\"mcp.server.transport\", \"http\");\n          span.setAttribute(\"mcp.server.host\", host);\n          span.setAttribute(\"mcp.server.port\", port);\n\n          return server;\n        }\n      );\n\n      // Warn if bound to 0.0.0.0\n      if (host === \"0.0.0.0\") {\n        process.stderr.write(\n          \"Warning: Server bound to 0.0.0.0. DNS rebinding protection is not automatic; ensure your MCP client validates the Host header.\\n\"\n        );\n      }\n\n      process.stderr.write(`Documentation Robotics MCP server ready (http://${host}:${port}/mcp)\\n`);\n\n      // Set up graceful shutdown on SIGINT/SIGTERM\n      let resolveKeepAlive: () => void;\n      const keepAlive = new Promise<void>((resolve) => {\n        resolveKeepAlive = resolve;\n      });\n\n      let isShuttingDown = false;\n      const shutdown = async () => {\n        if (isShuttingDown) {\n          return;\n        }\n        isShuttingDown = true;\n\n        try {\n          process.stderr.write(\"\\nShutting down HTTP server...\\n\");\n          await closeHttpServer(httpServer, httpApp);\n        } catch (error) {\n          const errorMsg = error instanceof Error ? error.message : \"unknown error\";\n          process.stderr.write(`[mcp] Error closing server: ${errorMsg}\\n`);\n        } finally {\n          resolveKeepAlive!();\n        }\n      };\n\n      process.on(\"SIGINT\", shutdown);\n      process.on(\"SIGTERM\", shutdown);\n\n      // Keep the process alive until explicitly terminated\n      await keepAlive;\n    } else {\n      // Stdio transport (default). No API key: the client spawns this process as its own user.\n      const transport = await startActiveSpan(\n        \"mcp.server.start\",\n        async (span) => {\n          const { StdioServerTransport } = await import(\"@modelcontextprotocol/sdk/server/stdio.js\");\n\n          const server = await createConfiguredServer();\n          if (servesEnvRoot) {\n            await warmModel();\n          } else {\n            serveClientRoots(server);\n          }\n\n          span.setAttribute(\"mcp.server.name\", \"documentation-robotics\");\n          span.setAttribute(\"mcp.server.version\", cliVersion);\n          span.setAttribute(\"mcp.server.transport\", \"stdio\");\n\n          const serverTransport = new StdioServerTransport();\n          await server.connect(serverTransport);\n          return serverTransport;\n        }\n      );\n\n      process.stderr.write(\"Documentation Robotics MCP server ready (stdio)\\n\");\n\n      // Keep the process alive for the lifetime of the stdio session; the\n      // transport closes (and the process exits) when stdin closes. Shutdown\n      // is implicit in the `mcp.server.start` span's end, so no separate\n      // zero-duration \"stop\" span is recorded here.\n      await new Promise<void>((resolve) => {\n        transport.onclose = () => resolve();\n      });\n    }\n  } catch (error) {\n    if (error instanceof CLIError) {\n      throw error;\n    }\n\n    let message = getErrorMessage(error);\n\n    // Handle port-in-use errors with actionable message\n    if (error instanceof Error && (error as NodeJS.ErrnoException).code === \"EADDRINUSE\") {\n      const port = options.port || 3100;\n      message = `Port ${port} is already in use. Try a different port with --port or kill the process using that port.`;\n    }\n\n    process.stderr.write(`Error: ${message}\\n`);\n    throw new CLIError(message, 1);\n  }\n}\n"],
+  "mappings": "AAYA,SAAS,UAAU,YAAY;AAE/B,SAAS,0CAA0C;AACnD,SAAS,qBAA+C;AAExD,SAAS,mBAAmB,0BAA0B;AACtD,SAAS,2BAA2B;AACpC,SAAS,2BAA2B;AACpC,SAAS,uBAAuB;AAChC,SAAS,iBAAiB,iBAAiB;AAC3C,SAAS,uBAAuB;AAChC,SAAS,UAAU,uBAAuB;AAC1C,SAAS,qBAAqB;AAE9B,MAAM,aAAa,cAAa,GAO1B,mBAAwC,OAAO,gBAAe;AAClE,QAAM,SAAS,MAAM,KAAK;IACxB,SAAS;IACT,aAAa;IACb,cAAc;GACf;AAED,SAAI,SAAS,MAAM,KAAK,OAAO,UAAW,YAAY,OAAO,KAAI,EAAG,WAAW,IACtE,cAEF,OAAO,KAAI;AACpB;AAEA,eAAe,yBAAsB;AACnC,QAAM,EAAE,UAAS,IAAK,MAAM,OAAO,yCAAyC,GAEtE,SAAS,IAAI,UACjB,EAAE,MAAM,0BAA0B,SAAS,WAAU,GACrD,EAAE,cAAc,oBAAmB,CAAE;AAIvC,aAAI,gBAAe,EAAG,YAAY,MAAM,GACxC,MAAM,IAAI,oBAAmB,EAAG,YAAY,MAAM,GAC3C;AACT;AAOA,eAAe,YAAS;AACtB,MAAI;AACF,UAAM,UAAS;EACjB,SAAS,OAAO;AACd,YAAQ,OAAO,MAAM,+BAA+B,gBAAgB,KAAK,CAAC;CAAI;EAChF;AACF;AAOA,SAAS,iBAAiB,QAAiB;AACzC,QAAM,QAAQ,OAAO,QACf,SAAS,MACb,kBAAkB,MAAM,MAAM,UAAS,GAAI,EAAE,UAAU,MAAM,gBAAe,EAAE,CAAE,EAAE,KAAK,SAAS;AAElG,QAAM,uBAAuB,oCAAoC,YAAW;AAC1E,UAAM,OAAM;EACd,CAAC,GACD,MAAM,gBAAgB,MAAK;AACzB,IAAI,MAAM,sBAAqB,GAAI,QAC5B,OAAM,IAEN,UAAS;EAElB;AACF;AAQM,SAAU,qBAAqB,MAAyB,QAAQ,KAAG;AACvE,QAAM,WAAW,IAAI;AACrB,MAAI,CAAC;AAAU,WAAO;AACtB,MAAI;AACF,YAAQ,MAAM,QAAQ;EACxB,SAAS,OAAO;AACd,mBAAQ,OAAO,MACb,0BAA0B,QAAQ,mBAAmB,gBAAgB,KAAK,CAAC,mBAAmB,QAAQ,IAAG,CAAE;CAAI,GAE1G;EACT;AACA,4BAAmB,QAAQ,IAAG,CAAE,GACzB;AACT;AAaA,eAAsB,WAAW,UAA6B,CAAA,GAAE;AAC9D,QAAM,aAAa,IAAI,cAAa;AAEpC,MAAI;AACF,QAAI,QAAQ,eAAe;AACzB,YAAM,gBAAgB,kBAAkB,OAAO,SAAQ;AACrD,cAAM,gBAAgB,QAAQ,MAAM,UAAU,IACxC,EAAE,KAAK,KAAI,IAAK,MAAM,WAAW,OACrC,gBAAgB,mBAAmB,MAAS;AAE9C,aAAK,aAAa,gBAAgB,IAAI,GACtC,QAAQ,OAAO,MAAM,wCAAwC,IAAI;CAAI,GACrE,QAAQ,OAAO,MAAM,gBAAgB,GAAG;CAAI;MAC9C,CAAC;AACD;IACF;AAEA,UAAM,gBAAgB,qBAAoB;AAE1C,QAAI,QAAQ,cAAc,QAAQ;AAEhC,YAAM,gBAAgB,QAAQ,MAAM,UAAU,IACxC,EAAE,KAAK,MAAM,MAAK,IAAK,MAAM,WAAW,UAC5C,gBAAgB,mBAAmB,MAAS;AAE9C,MAAI,SACF,QAAQ,OAAO,MAAM,wCAAwC,IAAI;CAAI,GAIlE,qBAAqB,KAAK,QAAQ,IAAI,oBAAoB,EAAE,KAC/D,QAAQ,OAAO,MAAM,gBAAgB,GAAG;CAAI,GAK9C,MAAM,UAAS;AACf,YAAM,EAAE,kBAAkB,iBAAiB,gBAAe,IAAK,MAAM,OACnE,0BAA0B,GAGtB,OAAO,QAAQ,QAAQ,aACvB,OAAO,QAAQ,QAAQ;AAE7B,UAAI;AACJ,YAAM,aAAa,MAAM,gBACvB,oBACA,OAAO,SAAQ;AACb,kBAAU,MAAM,iBAAiB,YAAY,KAAK,wBAAwB,IAAI;AAC9E,cAAM,SAAS,MAAM,gBAAgB,SAAS,MAAM,IAAI;AAExD,oBAAK,aAAa,mBAAmB,wBAAwB,GAC7D,KAAK,aAAa,sBAAsB,UAAU,GAClD,KAAK,aAAa,wBAAwB,MAAM,GAChD,KAAK,aAAa,mBAAmB,IAAI,GACzC,KAAK,aAAa,mBAAmB,IAAI,GAElC;MACT,CAAC;AAIH,MAAI,SAAS,aACX,QAAQ,OAAO,MACb;CAAkI,GAItI,QAAQ,OAAO,MAAM,mDAAmD,IAAI,IAAI,IAAI;CAAS;AAG7F,UAAI;AACJ,YAAM,YAAY,IAAI,QAAc,CAAC,YAAW;AAC9C,2BAAmB;MACrB,CAAC;AAED,UAAI,iBAAiB;AACrB,YAAM,WAAW,YAAW;AAC1B,YAAI,iBAGJ;2BAAiB;AAEjB,cAAI;AACF,oBAAQ,OAAO,MAAM;;CAAkC,GACvD,MAAM,gBAAgB,YAAY,OAAO;UAC3C,SAAS,OAAO;AACd,kBAAM,WAAW,iBAAiB,QAAQ,MAAM,UAAU;AAC1D,oBAAQ,OAAO,MAAM,+BAA+B,QAAQ;CAAI;UAClE;AACE,6BAAiB;UACnB;;MACF;AAEA,cAAQ,GAAG,UAAU,QAAQ,GAC7B,QAAQ,GAAG,WAAW,QAAQ,GAG9B,MAAM;IACR,OAAO;AAEL,YAAM,YAAY,MAAM,gBACtB,oBACA,OAAO,SAAQ;AACb,cAAM,EAAE,qBAAoB,IAAK,MAAM,OAAO,2CAA2C,GAEnF,SAAS,MAAM,uBAAsB;AAC3C,QAAI,gBACF,MAAM,UAAS,IAEf,iBAAiB,MAAM,GAGzB,KAAK,aAAa,mBAAmB,wBAAwB,GAC7D,KAAK,aAAa,sBAAsB,UAAU,GAClD,KAAK,aAAa,wBAAwB,OAAO;AAEjD,cAAM,kBAAkB,IAAI,qBAAoB;AAChD,qBAAM,OAAO,QAAQ,eAAe,GAC7B;MACT,CAAC;AAGH,cAAQ,OAAO,MAAM;CAAmD,GAMxE,MAAM,IAAI,QAAc,CAAC,YAAW;AAClC,kBAAU,UAAU,MAAM,QAAO;MACnC,CAAC;IACH;EACF,SAAS,OAAO;AACd,QAAI,iBAAiB;AACnB,YAAM;AAGR,QAAI,UAAU,gBAAgB,KAAK;AAGnC,UAAI,iBAAiB,SAAU,MAAgC,SAAS,iBAEtE,UAAU,QADG,QAAQ,QAAQ,IACP,8FAGxB,QAAQ,OAAO,MAAM,UAAU,OAAO;CAAI,GACpC,IAAI,SAAS,SAAS,CAAC;EAC/B;AACF;",
+  "names": []
 }

package/dist/commands/relationship.js

@@ -4,7 +4,7 @@ import { StagingAreaManager } from "../core/staging-area.js";
 import { findElementLayer } from "../utils/element-utils.js";
 import { isValidLayerName } from "../core/layers.js";
 import { CLIError, ErrorCategory, handleError, handleSuccess, handleInfo, getErrorMessage } from "../utils/errors.js";
-import { getValidRelationships, getValidPredicatesForSource, getValidDestinationsForSourceAndPredicate } from "../generated/relationship-index.js";
+import { getValidRelationships } from "../generated/relationship-index.js";
 import { normalizeNodeType } from "../generated/node-types.js";
 import { ModelReportOrchestrator } from "../reports/model-report-orchestrator.js";
 import { emitLog, SeverityNumber } from "../telemetry/index.js";
@@ -13,7 +13,7 @@ import { RelationshipCatalog } from "../core/relationship-catalog.js";
 import { collectKnownIdentifiers, findDanglingRelationships } from "../core/relationship-prune.js";
 import { isJson } from "../utils/globals.js";
 import { ChangesetGraph } from "../core/changeset-graph.js";
-import { liveRelationshipsBetween, planRelationshipAdd, stageRelationshipDeletes } from "../core/relationship-staging.js";
+import { liveRelationshipsBetween, planRelationshipAdd, stageRelationshipDeletes, validateRelationshipCombination } from "../core/relationship-staging.js";
 function constructSpecNodeId(element, elementId) {
   const layerId = element.layer_id, type = element.type;
   if (typeof layerId != "string" || !layerId)
@@ -23,16 +23,6 @@ function constructSpecNodeId(element, elementId) {
   const normalizedType = normalizeNodeType(layerId, type);
   return `${layerId}.${normalizedType}`;
 }
-function validateRelationshipCombination(sourceSpecNodeId, predicate, destSpecNodeId) {
-  if (getValidRelationships(sourceSpecNodeId, predicate, destSpecNodeId).length > 0)
-    return { valid: !0 };
-  const suggestions = [], validPredicates = getValidPredicatesForSource(sourceSpecNodeId);
-  if (validPredicates.length > 0 ? suggestions.push(`Valid predicates for ${sourceSpecNodeId}: ${validPredicates.join(", ")}`) : suggestions.push(`No valid relationships defined for ${sourceSpecNodeId}`), predicate) {
-    const validDests = getValidDestinationsForSourceAndPredicate(sourceSpecNodeId, predicate);
-    validDests.length > 0 ? suggestions.push(`Valid destinations for --[${predicate}]-->: ${validDests.join(", ")}`) : suggestions.push(`No valid destinations for ${sourceSpecNodeId} --[${predicate}]-->`);
-  }
-  return { valid: !1, suggestions };
-}
 function getRelationshipConstraints(sourceSpecNodeId, predicate, destSpecNodeId) {
   const rels = getValidRelationships(sourceSpecNodeId, predicate, destSpecNodeId);
   if (rels.length === 0)