@optave/codegraph 3.1.5 → 3.3.0

package/src/domain/analysis/context.js

@@ -5,14 +5,18 @@ import {
   findCrossFileCallTargets,
   findDbPath,
   findFileNodes,
+  findImplementors,
   findImportSources,
   findImportTargets,
+  findInterfaces,
   findIntraFileCallEdges,
   findNodeChildren,
   findNodesByFile,
   getComplexityForNode,
   openReadonlyOrFail,
 } from '../../db/index.js';
+import { loadConfig } from '../../infrastructure/config.js';
+import { debug } from '../../infrastructure/logger.js';
 import { isTestFile } from '../../infrastructure/test-filter.js';
 import {
   createFileLinesReader,
@@ -26,7 +30,175 @@ import { normalizeSymbol } from '../../shared/normalize.js';
 import { paginateResult } from '../../shared/paginate.js';
 import { findMatchingNodes } from './symbol-lookup.js';
 
-function explainFileImpl(db, target, getFileLines) {
+function buildCallees(db, node, repoRoot, getFileLines, opts) {
+  const { noTests, depth, displayOpts } = opts;
+  const calleeRows = findCallees(db, node.id);
+  const filteredCallees = noTests ? calleeRows.filter((c) => !isTestFile(c.file)) : calleeRows;
+
+  const callees = filteredCallees.map((c) => {
+    const cLines = getFileLines(c.file);
+    const summary = cLines ? extractSummary(cLines, c.line, displayOpts) : null;
+    let calleeSource = null;
+    if (depth >= 1) {
+      calleeSource = readSourceRange(repoRoot, c.file, c.line, c.end_line, displayOpts);
+    }
+    return {
+      name: c.name,
+      kind: c.kind,
+      file: c.file,
+      line: c.line,
+      endLine: c.end_line || null,
+      summary,
+      source: calleeSource,
+    };
+  });
+
+  if (depth > 1) {
+    const visited = new Set(filteredCallees.map((c) => c.id));
+    visited.add(node.id);
+    let frontier = filteredCallees.map((c) => c.id);
+    const maxDepth = Math.min(depth, 5);
+    for (let d = 2; d <= maxDepth; d++) {
+      const nextFrontier = [];
+      for (const fid of frontier) {
+        const deeper = findCallees(db, fid);
+        for (const c of deeper) {
+          if (!visited.has(c.id) && (!noTests || !isTestFile(c.file))) {
+            visited.add(c.id);
+            nextFrontier.push(c.id);
+            const cLines = getFileLines(c.file);
+            callees.push({
+              name: c.name,
+              kind: c.kind,
+              file: c.file,
+              line: c.line,
+              endLine: c.end_line || null,
+              summary: cLines ? extractSummary(cLines, c.line, displayOpts) : null,
+              source: readSourceRange(repoRoot, c.file, c.line, c.end_line, displayOpts),
+            });
+          }
+        }
+      }
+      frontier = nextFrontier;
+      if (frontier.length === 0) break;
+    }
+  }
+
+  return callees;
+}
+
+function buildCallers(db, node, noTests) {
+  let callerRows = findCallers(db, node.id);
+
+  if (node.kind === 'method' && node.name.includes('.')) {
+    const methodName = node.name.split('.').pop();
+    const relatedMethods = resolveMethodViaHierarchy(db, methodName);
+    for (const rm of relatedMethods) {
+      if (rm.id === node.id) continue;
+      const extraCallers = findCallers(db, rm.id);
+      callerRows.push(...extraCallers.map((c) => ({ ...c, viaHierarchy: rm.name })));
+    }
+  }
+  if (noTests) callerRows = callerRows.filter((c) => !isTestFile(c.file));
+
+  return callerRows.map((c) => ({
+    name: c.name,
+    kind: c.kind,
+    file: c.file,
+    line: c.line,
+    viaHierarchy: c.viaHierarchy || undefined,
+  }));
+}
+
+const INTERFACE_LIKE_KINDS = new Set(['interface', 'trait']);
+const IMPLEMENTOR_KINDS = new Set(['class', 'struct', 'record', 'enum']);
+
+function buildImplementationInfo(db, node, noTests) {
+  // For interfaces/traits: show who implements them
+  if (INTERFACE_LIKE_KINDS.has(node.kind)) {
+    let impls = findImplementors(db, node.id);
+    if (noTests) impls = impls.filter((n) => !isTestFile(n.file));
+    return {
+      implementors: impls.map((n) => ({ name: n.name, kind: n.kind, file: n.file, line: n.line })),
+    };
+  }
+  // For classes/structs: show what they implement
+  if (IMPLEMENTOR_KINDS.has(node.kind)) {
+    let ifaces = findInterfaces(db, node.id);
+    if (noTests) ifaces = ifaces.filter((n) => !isTestFile(n.file));
+    if (ifaces.length > 0) {
+      return {
+        implements: ifaces.map((n) => ({ name: n.name, kind: n.kind, file: n.file, line: n.line })),
+      };
+    }
+  }
+  return {};
+}
+
+function buildRelatedTests(db, node, getFileLines, includeTests) {
+  const testCallerRows = findCallers(db, node.id);
+  const testCallers = testCallerRows.filter((c) => isTestFile(c.file));
+
+  const testsByFile = new Map();
+  for (const tc of testCallers) {
+    if (!testsByFile.has(tc.file)) testsByFile.set(tc.file, []);
+    testsByFile.get(tc.file).push(tc);
+  }
+
+  const relatedTests = [];
+  for (const [file] of testsByFile) {
+    const tLines = getFileLines(file);
+    const testNames = [];
+    if (tLines) {
+      for (const tl of tLines) {
+        const tm = tl.match(/(?:it|test|describe)\s*\(\s*['"`]([^'"`]+)['"`]/);
+        if (tm) testNames.push(tm[1]);
+      }
+    }
+    const testSource = includeTests && tLines ? tLines.join('\n') : undefined;
+    relatedTests.push({
+      file,
+      testCount: testNames.length,
+      testNames,
+      source: testSource,
+    });
+  }
+
+  return relatedTests;
+}
+
+function getComplexityMetrics(db, nodeId) {
+  try {
+    const cRow = getComplexityForNode(db, nodeId);
+    if (!cRow) return null;
+    return {
+      cognitive: cRow.cognitive,
+      cyclomatic: cRow.cyclomatic,
+      maxNesting: cRow.max_nesting,
+      maintainabilityIndex: cRow.maintainability_index || 0,
+      halsteadVolume: cRow.halstead_volume || 0,
+    };
+  } catch (e) {
+    debug(`complexity lookup failed for node ${nodeId}: ${e.message}`);
+    return null;
+  }
+}
+
+function getNodeChildrenSafe(db, nodeId) {
+  try {
+    return findNodeChildren(db, nodeId).map((c) => ({
+      name: c.name,
+      kind: c.kind,
+      line: c.line,
+      endLine: c.end_line || null,
+    }));
+  } catch (e) {
+    debug(`findNodeChildren failed for node ${nodeId}: ${e.message}`);
+    return [];
+  }
+}
+
+function explainFileImpl(db, target, getFileLines, displayOpts) {
   const fileNodes = findFileNodes(db, `%${target}%`);
   if (fileNodes.length === 0) return [];
 
@@ -42,21 +214,17 @@ function explainFileImpl(db, target, getFileLines) {
       kind: s.kind,
       line: s.line,
       role: s.role || null,
-      summary: fileLines ? extractSummary(fileLines, s.line) : null,
-      signature: fileLines ? extractSignature(fileLines, s.line) : null,
+      summary: fileLines ? extractSummary(fileLines, s.line, displayOpts) : null,
+      signature: fileLines ? extractSignature(fileLines, s.line, displayOpts) : null,
     });
 
     const publicApi = symbols.filter((s) => publicIds.has(s.id)).map(mapSymbol);
     const internal = symbols.filter((s) => !publicIds.has(s.id)).map(mapSymbol);
 
-    // Imports / importedBy
     const imports = findImportTargets(db, fn.id).map((r) => ({ file: r.file }));
-
     const importedBy = findImportSources(db, fn.id).map((r) => ({ file: r.file }));
 
-    // Intra-file data flow
     const intraEdges = findIntraFileCallEdges(db, fn.file);
-
     const dataFlowMap = new Map();
     for (const edge of intraEdges) {
       if (!dataFlowMap.has(edge.caller_name)) dataFlowMap.set(edge.caller_name, []);
@@ -67,7 +235,6 @@ function explainFileImpl(db, target, getFileLines) {
       callees,
     }));
 
-    // Line count: prefer node_metrics (actual), fall back to MAX(end_line)
     const metric = db
       .prepare(`SELECT nm.line_count FROM node_metrics nm WHERE nm.node_id = ?`)
       .get(fn.id);
@@ -92,10 +259,10 @@ function explainFileImpl(db, target, getFileLines) {
   });
 }
 
-function explainFunctionImpl(db, target, noTests, getFileLines) {
+function explainFunctionImpl(db, target, noTests, getFileLines, displayOpts) {
   let nodes = db
     .prepare(
-      `SELECT * FROM nodes WHERE name LIKE ? AND kind IN ('function','method','class','interface','type','struct','enum','trait','record','module') ORDER BY file, line`,
+      `SELECT * FROM nodes WHERE name LIKE ? AND kind IN ('function','method','class','interface','type','struct','enum','trait','record','module','constant') ORDER BY file, line`,
     )
     .all(`%${target}%`);
   if (noTests) nodes = nodes.filter((n) => !isTestFile(n.file));
@@ -105,8 +272,8 @@ function explainFunctionImpl(db, target, noTests, getFileLines) {
   return nodes.slice(0, 10).map((node) => {
     const fileLines = getFileLines(node.file);
     const lineCount = node.end_line ? node.end_line - node.line + 1 : null;
-    const summary = fileLines ? extractSummary(fileLines, node.line) : null;
-    const signature = fileLines ? extractSignature(fileLines, node.line) : null;
+    const summary = fileLines ? extractSummary(fileLines, node.line, displayOpts) : null;
+    const signature = fileLines ? extractSignature(fileLines, node.line, displayOpts) : null;
 
     const callees = findCallees(db, node.id).map((c) => ({
       name: c.name,
@@ -129,29 +296,12 @@ function explainFunctionImpl(db, target, noTests, getFileLines) {
       .filter((r) => isTestFile(r.file) && !seenFiles.has(r.file) && seenFiles.add(r.file))
       .map((r) => ({ file: r.file }));
 
-    // Complexity metrics
-    let complexityMetrics = null;
-    try {
-      const cRow = getComplexityForNode(db, node.id);
-      if (cRow) {
-        complexityMetrics = {
-          cognitive: cRow.cognitive,
-          cyclomatic: cRow.cyclomatic,
-          maxNesting: cRow.max_nesting,
-          maintainabilityIndex: cRow.maintainability_index || 0,
-          halsteadVolume: cRow.halstead_volume || 0,
-        };
-      }
-    } catch {
-      /* table may not exist */
-    }
-
     return {
       ...normalizeSymbol(node, db, hc),
       lineCount,
       summary,
       signature,
-      complexity: complexityMetrics,
+      complexity: getComplexityMetrics(db, node.id),
       callees,
       callers,
       relatedTests,
@@ -159,6 +309,42 @@ function explainFunctionImpl(db, target, noTests, getFileLines) {
   });
 }
 
+function explainCallees(
+  parentResults,
+  currentDepth,
+  visited,
+  db,
+  noTests,
+  getFileLines,
+  displayOpts,
+) {
+  if (currentDepth <= 0) return;
+  for (const r of parentResults) {
+    const newCallees = [];
+    for (const callee of r.callees) {
+      const key = `${callee.name}:${callee.file}:${callee.line}`;
+      if (visited.has(key)) continue;
+      visited.add(key);
+      const calleeResults = explainFunctionImpl(
+        db,
+        callee.name,
+        noTests,
+        getFileLines,
+        displayOpts,
+      );
+      const exact = calleeResults.find((cr) => cr.file === callee.file && cr.line === callee.line);
+      if (exact) {
+        exact._depth = (r._depth || 0) + 1;
+        newCallees.push(exact);
+      }
+    }
+    if (newCallees.length > 0) {
+      r.depDetails = newCallees;
+      explainCallees(newCallees, currentDepth - 1, visited, db, noTests, getFileLines, displayOpts);
+    }
+  }
+}
+
 // ─── Exported functions ──────────────────────────────────────────────────
 
 export function contextData(name, customDbPath, opts = {}) {
@@ -169,6 +355,9 @@ export function contextData(name, customDbPath, opts = {}) {
     const noTests = opts.noTests || false;
     const includeTests = opts.includeTests || false;
 
+    const config = opts.config || loadConfig();
+    const displayOpts = config.display || {};
+
     const dbPath = findDbPath(customDbPath);
     const repoRoot = path.resolve(path.dirname(dbPath), '..');
 
@@ -177,156 +366,27 @@ export function contextData(name, customDbPath, opts = {}) {
       return { name, results: [] };
     }
 
-    // No hardcoded slice — pagination handles bounding via limit/offset
-
     const getFileLines = createFileLinesReader(repoRoot);
 
     const results = nodes.map((node) => {
       const fileLines = getFileLines(node.file);
 
-      // Source
       const source = noSource
         ? null
-        : readSourceRange(repoRoot, node.file, node.line, node.end_line);
-
-      // Signature
-      const signature = fileLines ? extractSignature(fileLines, node.line) : null;
+        : readSourceRange(repoRoot, node.file, node.line, node.end_line, displayOpts);
 
-      // Callees
-      const calleeRows = findCallees(db, node.id);
-      const filteredCallees = noTests ? calleeRows.filter((c) => !isTestFile(c.file)) : calleeRows;
+      const signature = fileLines ? extractSignature(fileLines, node.line, displayOpts) : null;
 
-      const callees = filteredCallees.map((c) => {
-        const cLines = getFileLines(c.file);
-        const summary = cLines ? extractSummary(cLines, c.line) : null;
-        let calleeSource = null;
-        if (depth >= 1) {
-          calleeSource = readSourceRange(repoRoot, c.file, c.line, c.end_line);
-        }
-        return {
-          name: c.name,
-          kind: c.kind,
-          file: c.file,
-          line: c.line,
-          endLine: c.end_line || null,
-          summary,
-          source: calleeSource,
-        };
+      const callees = buildCallees(db, node, repoRoot, getFileLines, {
+        noTests,
+        depth,
+        displayOpts,
       });
-
-      // Deep callee expansion via BFS (depth > 1, capped at 5)
-      if (depth > 1) {
-        const visited = new Set(filteredCallees.map((c) => c.id));
-        visited.add(node.id);
-        let frontier = filteredCallees.map((c) => c.id);
-        const maxDepth = Math.min(depth, 5);
-        for (let d = 2; d <= maxDepth; d++) {
-          const nextFrontier = [];
-          for (const fid of frontier) {
-            const deeper = findCallees(db, fid);
-            for (const c of deeper) {
-              if (!visited.has(c.id) && (!noTests || !isTestFile(c.file))) {
-                visited.add(c.id);
-                nextFrontier.push(c.id);
-                const cLines = getFileLines(c.file);
-                callees.push({
-                  name: c.name,
-                  kind: c.kind,
-                  file: c.file,
-                  line: c.line,
-                  endLine: c.end_line || null,
-                  summary: cLines ? extractSummary(cLines, c.line) : null,
-                  source: readSourceRange(repoRoot, c.file, c.line, c.end_line),
-                });
-              }
-            }
-          }
-          frontier = nextFrontier;
-          if (frontier.length === 0) break;
-        }
-      }
-
-      // Callers
-      let callerRows = findCallers(db, node.id);
-
-      // Method hierarchy resolution
-      if (node.kind === 'method' && node.name.includes('.')) {
-        const methodName = node.name.split('.').pop();
-        const relatedMethods = resolveMethodViaHierarchy(db, methodName);
-        for (const rm of relatedMethods) {
-          if (rm.id === node.id) continue;
-          const extraCallers = findCallers(db, rm.id);
-          callerRows.push(...extraCallers.map((c) => ({ ...c, viaHierarchy: rm.name })));
-        }
-      }
-      if (noTests) callerRows = callerRows.filter((c) => !isTestFile(c.file));
-
-      const callers = callerRows.map((c) => ({
-        name: c.name,
-        kind: c.kind,
-        file: c.file,
-        line: c.line,
-        viaHierarchy: c.viaHierarchy || undefined,
-      }));
-
-      // Related tests: callers that live in test files
-      const testCallerRows = findCallers(db, node.id);
-      const testCallers = testCallerRows.filter((c) => isTestFile(c.file));
-
-      const testsByFile = new Map();
-      for (const tc of testCallers) {
-        if (!testsByFile.has(tc.file)) testsByFile.set(tc.file, []);
-        testsByFile.get(tc.file).push(tc);
-      }
-
-      const relatedTests = [];
-      for (const [file] of testsByFile) {
-        const tLines = getFileLines(file);
-        const testNames = [];
-        if (tLines) {
-          for (const tl of tLines) {
-            const tm = tl.match(/(?:it|test|describe)\s*\(\s*['"`]([^'"`]+)['"`]/);
-            if (tm) testNames.push(tm[1]);
-          }
-        }
-        const testSource = includeTests && tLines ? tLines.join('\n') : undefined;
-        relatedTests.push({
-          file,
-          testCount: testNames.length,
-          testNames,
-          source: testSource,
-        });
-      }
-
-      // Complexity metrics
-      let complexityMetrics = null;
-      try {
-        const cRow = getComplexityForNode(db, node.id);
-        if (cRow) {
-          complexityMetrics = {
-            cognitive: cRow.cognitive,
-            cyclomatic: cRow.cyclomatic,
-            maxNesting: cRow.max_nesting,
-            maintainabilityIndex: cRow.maintainability_index || 0,
-            halsteadVolume: cRow.halstead_volume || 0,
-          };
-        }
-      } catch {
-        /* table may not exist */
-      }
-
-      // Children (parameters, properties, constants)
-      let nodeChildren = [];
-      try {
-        nodeChildren = findNodeChildren(db, node.id).map((c) => ({
-          name: c.name,
-          kind: c.kind,
-          line: c.line,
-          endLine: c.end_line || null,
-        }));
-      } catch {
-        /* parent_id column may not exist */
-      }
+      const callers = buildCallers(db, node, noTests);
+      const relatedTests = buildRelatedTests(db, node, getFileLines, includeTests);
+      const complexityMetrics = getComplexityMetrics(db, node.id);
+      const nodeChildren = getNodeChildrenSafe(db, node.id);
+      const implInfo = buildImplementationInfo(db, node, noTests);
 
       return {
         name: node.name,
@@ -342,6 +402,7 @@ export function contextData(name, customDbPath, opts = {}) {
         callees,
         callers,
         relatedTests,
+        ...implInfo,
       };
     });
 
@@ -359,6 +420,9 @@ export function explainData(target, customDbPath, opts = {}) {
     const depth = opts.depth || 0;
     const kind = isFileLikeTarget(target) ? 'file' : 'function';
 
+    const config = opts.config || loadConfig();
+    const displayOpts = config.display || {};
+
     const dbPath = findDbPath(customDbPath);
     const repoRoot = path.resolve(path.dirname(dbPath), '..');
 
@@ -366,38 +430,12 @@ export function explainData(target, customDbPath, opts = {}) {
 
     const results =
       kind === 'file'
-        ? explainFileImpl(db, target, getFileLines)
-        : explainFunctionImpl(db, target, noTests, getFileLines);
+        ? explainFileImpl(db, target, getFileLines, displayOpts)
+        : explainFunctionImpl(db, target, noTests, getFileLines, displayOpts);
 
-    // Recursive dependency explanation for function targets
     if (kind === 'function' && depth > 0 && results.length > 0) {
       const visited = new Set(results.map((r) => `${r.name}:${r.file}:${r.line}`));
-
-      function explainCallees(parentResults, currentDepth) {
-        if (currentDepth <= 0) return;
-        for (const r of parentResults) {
-          const newCallees = [];
-          for (const callee of r.callees) {
-            const key = `${callee.name}:${callee.file}:${callee.line}`;
-            if (visited.has(key)) continue;
-            visited.add(key);
-            const calleeResults = explainFunctionImpl(db, callee.name, noTests, getFileLines);
-            const exact = calleeResults.find(
-              (cr) => cr.file === callee.file && cr.line === callee.line,
-            );
-            if (exact) {
-              exact._depth = (r._depth || 0) + 1;
-              newCallees.push(exact);
-            }
-          }
-          if (newCallees.length > 0) {
-            r.depDetails = newCallees;
-            explainCallees(newCallees, currentDepth - 1);
-          }
-        }
-      }
-
-      explainCallees(results, depth);
+      explainCallees(results, depth, visited, db, noTests, getFileLines, displayOpts);
     }
 
     const base = { target, kind, results };