@documentation-robotics/cli 0.3.0 → 0.4.0

package/dist/analyzers/cbm-analyzer.js

@@ -1,12 +1,13 @@
 import { spawnSync } from "child_process";
 import * as path from "path";
-import { readFile } from "fs/promises";
+import { readFile, writeFile } from "fs/promises";
 import { CLIError, ErrorCategory, handleWarning } from "../utils/errors.js";
 import { VALID_HTTP_METHODS } from "./types.js";
 import { StdioClient } from "./stdio-client.js";
 import { readIndexMeta, writeIndexMeta } from "./session-state.js";
 import { getCliVersion } from "./version.js";
 import { StagingAreaManager } from "../core/staging-area.js";
+import { DR_OWN_DIRS } from "./code-changes.js";
 import { VerifyEngine } from "./verify-engine.js";
 import { clampDepth, shapeCallGraphNode, determineEdgeType } from "./call-graph-utils.js";
 import { evaluateHeuristic, capConfidence, matchPattern, getKnownHeuristicNames } from "./heuristic-utils.js";
@@ -16,7 +17,8 @@ import { cbmCardGraph } from "./cards/graph.js";
 import { classifyTier0, defaultLabels } from "./cards/classify.js";
 import { CARD_FIELDS } from "./cards/build.js";
 import { tier0RulesFrom } from "./tier0-rules.js";
-const SOURCE_CODE_EXT_RE = /\.(ts|tsx|js|jsx|mjs|cjs|vue|svelte|py|go|rb|java|rs|php|cs|cpp|c|h|kt|swift|sh|bash)$/i;
+import { collectExtraction, extractionConfigFrom } from "./cards/extract.js";
+const EXTRACT_CALL_TIMEOUT_MS = 12e4, SOURCE_CODE_EXT_RE = /\.(ts|tsx|js|jsx|mjs|cjs|vue|svelte|py|go|rb|java|rs|php|cs|cpp|c|h|kt|swift|sh|bash)$/i;
 function isLikelyHttpRoute(qualifiedName) {
   const httpPath = qualifiedName.replace(/^__route__[A-Z]+__/, "");
   return !SOURCE_CODE_EXT_RE.test(httpPath) && !httpPath.startsWith("/.dr/") && !httpPath.startsWith("/schemas/");
@@ -293,7 +295,14 @@ class CbmAnalyzer {
         const errorMsg = error instanceof Error ? error.message : String(error);
         handleWarning("Failed to read active changeset", [`Error: ${errorMsg}`, "Continuing with indexing"]);
       }
-      const indexResponse = await tools.indexRepository(effectiveCodebaseRoot), timestamp = (/* @__PURE__ */ new Date()).toISOString(), nodeCount = typeof indexResponse.nodes == "number" ? indexResponse.nodes : 0, edgeCount = typeof indexResponse.edges == "number" ? indexResponse.edges : 0;
+      try {
+        const ignore = await ensureCbmIgnore(effectiveCodebaseRoot, projectRoot);
+        ignore.added.length > 0 && console.log(`Added ${ignore.added.join(", ")} to ${ignore.path} so the index leaves DR's own files out`);
+      } catch (error) {
+        const errorMsg = error instanceof Error ? error.message : String(error);
+        handleWarning("Could not update .cbmignore; DR's own files may be indexed", [`Error: ${errorMsg}`]);
+      }
+      const timestamp = (/* @__PURE__ */ new Date()).toISOString(), indexResponse = await tools.indexRepository(effectiveCodebaseRoot, options?.timeoutMs ? { timeout: options.timeoutMs } : void 0), nodeCount = typeof indexResponse.nodes == "number" ? indexResponse.nodes : 0, edgeCount = typeof indexResponse.edges == "number" ? indexResponse.edges : 0;
       typeof indexResponse.nodes != "number" && handleWarning("Invalid node count from index_repository", [
         `Expected number, got ${typeof indexResponse.nodes}: ${String(indexResponse.nodes)}`,
         "Defaulting to 0 - index may be incomplete or corrupted"
@@ -1183,6 +1192,41 @@ class CbmAnalyzer {
       client.close();
     }
   }
+  /** The Tier-0 rules, edge mappings and node-mapping decisions extraction runs with. */
+  extractionConfig() {
+    const nodeMappings = this.mapper.getNodeLabels().map((label) => this.mapper.getNodeMapping(label)).filter((m) => !!m), edgeMappings = this.mapper.getEdgeTypes().map((type) => this.mapper.getEdgeMapping(type)).filter((m) => !!m);
+    return extractionConfigFrom(tier0RulesFrom(this.mapper.getHeuristics()), edgeMappings, nodeMappings);
+  }
+  /**
+   * What `codebase_extract` reads from the index (src/analyzers/cards/extract.ts): cards with
+   * their Tier-0 outcomes, the mapped edges and the symbols of the files asked about. Read-only.
+   * Graph reads get two minutes each, past the client's 30s default, for large graphs.
+   *
+   * @throws CLIError if the project is not indexed or the mapping has no Tier-0 rules
+   */
+  async extract(projectRoot, options = {}) {
+    const effectiveCodebaseRoot = options.codebaseRoot ?? projectRoot, config = this.extractionConfig();
+    if (config.rules.length === 0)
+      throw new CLIError(`The ${this.name} analyzer mapping has no Tier-0 rules`, ErrorCategory.USER);
+    const status = await this.status(projectRoot, { codebaseRoot: effectiveCodebaseRoot });
+    if (!status.indexed)
+      throw new CLIError(`Project not indexed: ${projectRoot}`, ErrorCategory.NOT_FOUND, ["Run `dr analyzer index` to index the project"]);
+    if (!status.detected.installed)
+      throw new CLIError("CBM analyzer not installed", ErrorCategory.NOT_FOUND, [
+        "Install codebase-memory-mcp: npm install -g codebase-memory-mcp"
+      ]);
+    const client = new StdioClient();
+    try {
+      client.spawn(status.detected.binary_path), await client.initialize({ name: "dr-cli", version: await getCliVersion() });
+      const tools = new CbmTools(client, { timeout: EXTRACT_CALL_TIMEOUT_MS }), projectName = await this.resolveProjectName(tools, effectiveCodebaseRoot);
+      return await collectExtraction(cbmCardGraph(tools, projectName), config, this.name, {
+        sourceRoot: effectiveCodebaseRoot,
+        ...options
+      });
+    } finally {
+      client.close();
+    }
+  }
   /**
    * Verify that graph-discovered routes align with model endpoints
    *
@@ -1222,8 +1266,36 @@ class CbmAnalyzer {
     }
   }
 }
+const CBM_IGNORE_HEADING = "# Added by Documentation Robotics: its model, changesets and analyzer state are not code.";
+async function ensureCbmIgnore(codebaseRoot, projectRoot) {
+  const file = path.join(codebaseRoot, ".cbmignore");
+  let text = "", created = !0;
+  try {
+    text = await readFile(file, "utf-8"), created = !1;
+  } catch (error) {
+    if (error.code !== "ENOENT")
+      throw error;
+  }
+  const lines = text.split(/\r?\n/).map((line) => line.trim()).filter((line) => line && !line.startsWith("#")), folder = (line) => line.replace(/^\/+/, "").replace(/\/+$/, ""), listed = new Set(lines.filter((line) => !line.startsWith("!")).map(folder)), negated = new Set(lines.filter((line) => line.startsWith("!")).map((line) => folder(line.slice(1)))), added = [];
+  for (const dir of DR_OWN_DIRS) {
+    const relative = path.relative(codebaseRoot, path.join(projectRoot, dir)).split(path.sep).join("/");
+    !relative || relative.startsWith("..") || path.isAbsolute(relative) || !listed.has(relative) && !negated.has(relative) && added.push(`/${relative}/`);
+  }
+  if (added.length === 0)
+    return { path: file, added, created: !1 };
+  const separator = text.length === 0 ? "" : text.endsWith(`
+`) ? `
+` : `
+
+`, heading = text.includes(CBM_IGNORE_HEADING) ? "" : `${CBM_IGNORE_HEADING}
+`;
+  return await writeFile(file, `${text}${separator}${heading}${added.join(`
+`)}
+`), { path: file, added, created };
+}
 export {
   CbmAnalyzer,
-  SOURCE_CODE_EXT_RE
+  SOURCE_CODE_EXT_RE,
+  ensureCbmIgnore
 };
 //# sourceMappingURL=cbm-analyzer.js.map