@optave/codegraph 3.0.3 → 3.1.0

package/src/cfg.js

@@ -1046,15 +1046,29 @@ export function buildFunctionCFG(functionNode, langId) {
 export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
   // Lazily init WASM parsers if needed
   let parsers = null;
-  let extToLang = null;
   let needsFallback = false;
 
+  // Always build ext→langId map so native-only builds (where _langId is unset)
+  // can still derive the language from the file extension.
+  const extToLang = new Map();
+  for (const entry of LANGUAGE_REGISTRY) {
+    for (const ext of entry.extensions) {
+      extToLang.set(ext, entry.id);
+    }
+  }
+
   for (const [relPath, symbols] of fileSymbols) {
     if (!symbols._tree) {
       const ext = path.extname(relPath).toLowerCase();
       if (CFG_EXTENSIONS.has(ext)) {
-        needsFallback = true;
-        break;
+        // Check if all function/method defs already have native CFG data
+        const hasNativeCfg = symbols.definitions
+          .filter((d) => (d.kind === 'function' || d.kind === 'method') && d.line)
+          .every((d) => d.cfg === null || d.cfg?.blocks?.length);
+        if (!hasNativeCfg) {
+          needsFallback = true;
+          break;
+        }
       }
     }
   }
@@ -1062,12 +1076,6 @@ export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
   if (needsFallback) {
     const { createParsers } = await import('./parser.js');
     parsers = await createParsers();
-    extToLang = new Map();
-    for (const entry of LANGUAGE_REGISTRY) {
-      for (const ext of entry.extensions) {
-        extToLang.set(ext, entry.id);
-      }
-    }
   }
 
   let getParserFn = null;
@@ -1102,9 +1110,14 @@ export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
       let tree = symbols._tree;
       let langId = symbols._langId;
 
-      // WASM fallback if no cached tree
-      if (!tree) {
-        if (!extToLang || !getParserFn) continue;
+      // Check if all defs already have native CFG — skip WASM parse if so
+      const allNative = symbols.definitions
+        .filter((d) => (d.kind === 'function' || d.kind === 'method') && d.line)
+        .every((d) => d.cfg === null || d.cfg?.blocks?.length);
+
+      // WASM fallback if no cached tree and not all native
+      if (!tree && !allNative) {
+        if (!getParserFn) continue;
         langId = extToLang.get(ext);
         if (!langId || !CFG_LANG_IDS.has(langId)) continue;
 
@@ -1127,7 +1140,7 @@ export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
       }
 
       if (!langId) {
-        langId = extToLang ? extToLang.get(ext) : null;
+        langId = extToLang.get(ext);
         if (!langId) continue;
       }
 
@@ -1135,7 +1148,7 @@ export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
       if (!cfgRules) continue;
 
       const complexityRules = COMPLEXITY_RULES.get(langId);
-      if (!complexityRules) continue;
+      // complexityRules only needed for WASM fallback path
 
       for (const def of symbols.definitions) {
         if (def.kind !== 'function' && def.kind !== 'method') continue;
@@ -1144,11 +1157,19 @@ export async function buildCFGData(db, fileSymbols, rootDir, _engineOpts) {
         const row = getNodeId.get(def.name, relPath, def.line);
         if (!row) continue;
 
-        const funcNode = findFunctionNode(tree.rootNode, def.line, def.endLine, complexityRules);
-        if (!funcNode) continue;
+        // Native path: use pre-computed CFG from Rust engine
+        let cfg = null;
+        if (def.cfg?.blocks?.length) {
+          cfg = def.cfg;
+        } else {
+          // WASM fallback: compute CFG from tree-sitter AST
+          if (!tree || !complexityRules) continue;
+          const funcNode = findFunctionNode(tree.rootNode, def.line, def.endLine, complexityRules);
+          if (!funcNode) continue;
+          cfg = buildFunctionCFG(funcNode, langId);
+        }
 
-        const cfg = buildFunctionCFG(funcNode, langId);
-        if (cfg.blocks.length === 0) continue;
+        if (!cfg || cfg.blocks.length === 0) continue;
 
         // Clear old CFG data for this function
         deleteEdges.run(row.id);

package/src/cli.js

@@ -277,6 +277,7 @@ program
   .option('--limit <number>', 'Max results to return')
   .option('--offset <number>', 'Skip N results (default: 0)')
   .option('--ndjson', 'Newline-delimited JSON output')
+  .option('--unused', 'Show only exports with zero consumers')
   .action((file, opts) => {
     fileExports(file, opts.db, {
       noTests: resolveNoTests(opts),
@@ -284,6 +285,7 @@ program
       limit: opts.limit ? parseInt(opts.limit, 10) : undefined,
       offset: opts.offset ? parseInt(opts.offset, 10) : undefined,
       ndjson: opts.ndjson,
+      unused: opts.unused,
     });
   });
 
@@ -1137,6 +1139,39 @@ program
     });
   });
 
+program
+  .command('sequence <name>')
+  .description('Generate a Mermaid sequence diagram from call graph edges (participants = files)')
+  .option('--depth <n>', 'Max forward traversal depth', '10')
+  .option('--dataflow', 'Annotate with parameter names and return arrows from dataflow table')
+  .option('-d, --db <path>', 'Path to graph.db')
+  .option('-f, --file <path>', 'Scope to a specific file (partial match)')
+  .option('-k, --kind <kind>', 'Filter by symbol kind')
+  .option('-T, --no-tests', 'Exclude test/spec files from results')
+  .option('--include-tests', 'Include test/spec files (overrides excludeTests config)')
+  .option('-j, --json', 'Output as JSON')
+  .option('--limit <number>', 'Max results to return')
+  .option('--offset <number>', 'Skip N results (default: 0)')
+  .option('--ndjson', 'Newline-delimited JSON output')
+  .action(async (name, opts) => {
+    if (opts.kind && !EVERY_SYMBOL_KIND.includes(opts.kind)) {
+      console.error(`Invalid kind "${opts.kind}". Valid: ${EVERY_SYMBOL_KIND.join(', ')}`);
+      process.exit(1);
+    }
+    const { sequence } = await import('./sequence.js');
+    sequence(name, opts.db, {
+      depth: parseInt(opts.depth, 10),
+      file: opts.file,
+      kind: opts.kind,
+      noTests: resolveNoTests(opts),
+      json: opts.json,
+      dataflow: opts.dataflow,
+      limit: opts.limit ? parseInt(opts.limit, 10) : undefined,
+      offset: opts.offset ? parseInt(opts.offset, 10) : undefined,
+      ndjson: opts.ndjson,
+    });
+  });
+
 program
   .command('dataflow <name>')
   .description('Show data flow for a function: parameters, return consumers, mutations')

package/src/dataflow.js

@@ -1005,11 +1005,19 @@ function collectIdentifiers(node, out, rules) {
 export async function buildDataflowEdges(db, fileSymbols, rootDir, _engineOpts) {
   // Lazily init WASM parsers if needed
   let parsers = null;
-  let extToLang = null;
   let needsFallback = false;
 
+  // Always build ext→langId map so native-only builds (where _langId is unset)
+  // can still derive the language from the file extension.
+  const extToLang = new Map();
+  for (const entry of LANGUAGE_REGISTRY) {
+    for (const ext of entry.extensions) {
+      extToLang.set(ext, entry.id);
+    }
+  }
+
   for (const [relPath, symbols] of fileSymbols) {
-    if (!symbols._tree) {
+    if (!symbols._tree && !symbols.dataflow) {
       const ext = path.extname(relPath).toLowerCase();
       if (DATAFLOW_EXTENSIONS.has(ext)) {
         needsFallback = true;
@@ -1021,12 +1029,6 @@ export async function buildDataflowEdges(db, fileSymbols, rootDir, _engineOpts)
   if (needsFallback) {
     const { createParsers } = await import('./parser.js');
     parsers = await createParsers();
-    extToLang = new Map();
-    for (const entry of LANGUAGE_REGISTRY) {
-      for (const ext of entry.extensions) {
-        extToLang.set(ext, entry.id);
-      }
-    }
   }
 
   let getParserFn = null;
@@ -1061,41 +1063,45 @@ export async function buildDataflowEdges(db, fileSymbols, rootDir, _engineOpts)
       const ext = path.extname(relPath).toLowerCase();
       if (!DATAFLOW_EXTENSIONS.has(ext)) continue;
 
-      let tree = symbols._tree;
-      let langId = symbols._langId;
+      // Use native dataflow data if available — skip WASM extraction
+      let data = symbols.dataflow;
+      if (!data) {
+        let tree = symbols._tree;
+        let langId = symbols._langId;
+
+        // WASM fallback if no cached tree
+        if (!tree) {
+          if (!getParserFn) continue;
+          langId = extToLang.get(ext);
+          if (!langId || !DATAFLOW_LANG_IDS.has(langId)) continue;
+
+          const absPath = path.join(rootDir, relPath);
+          let code;
+          try {
+            code = fs.readFileSync(absPath, 'utf-8');
+          } catch {
+            continue;
+          }
 
-      // WASM fallback if no cached tree
-      if (!tree) {
-        if (!extToLang || !getParserFn) continue;
-        langId = extToLang.get(ext);
-        if (!langId || !DATAFLOW_LANG_IDS.has(langId)) continue;
+          const parser = getParserFn(parsers, absPath);
+          if (!parser) continue;
 
-        const absPath = path.join(rootDir, relPath);
-        let code;
-        try {
-          code = fs.readFileSync(absPath, 'utf-8');
-        } catch {
-          continue;
+          try {
+            tree = parser.parse(code);
+          } catch {
+            continue;
+          }
         }
 
-        const parser = getParserFn(parsers, absPath);
-        if (!parser) continue;
-
-        try {
-          tree = parser.parse(code);
-        } catch {
-          continue;
+        if (!langId) {
+          langId = extToLang.get(ext);
+          if (!langId) continue;
         }
-      }
 
-      if (!langId) {
-        langId = extToLang ? extToLang.get(ext) : null;
-        if (!langId) continue;
-      }
-
-      if (!DATAFLOW_RULES.has(langId)) continue;
+        if (!DATAFLOW_RULES.has(langId)) continue;
 
-      const data = extractDataflow(tree, relPath, symbols.definitions, langId);
+        data = extractDataflow(tree, relPath, symbols.definitions, langId);
+      }
 
       // Resolve function names to node IDs in this file first, then globally
       function resolveNode(funcName) {

package/src/db.js

@@ -225,6 +225,13 @@ export const MIGRATIONS = [
       CREATE INDEX IF NOT EXISTS idx_ast_kind_name ON ast_nodes(kind, name);
     `,
   },
+  {
+    version: 14,
+    up: `
+      ALTER TABLE nodes ADD COLUMN exported INTEGER DEFAULT 0;
+      CREATE INDEX IF NOT EXISTS idx_nodes_exported ON nodes(exported);
+    `,
+  },
 ];
 
 export function getBuildMeta(db, key) {

package/src/flow.js

@@ -7,7 +7,7 @@
 
 import { openReadonlyOrFail } from './db.js';
 import { paginateResult, printNdjson } from './paginate.js';
-import { isTestFile, kindIcon } from './queries.js';
+import { findMatchingNodes, isTestFile, kindIcon } from './queries.js';
 import { FRAMEWORK_ENTRY_PREFIXES } from './structure.js';
 
 /**
@@ -95,12 +95,12 @@ export function flowData(name, dbPath, opts = {}) {
   const noTests = opts.noTests || false;
 
   // Phase 1: Direct LIKE match on full name
-  let matchNode = findBestMatch(db, name, opts);
+  let matchNode = findMatchingNodes(db, name, opts)[0] ?? null;
 
   // Phase 2: Prefix-stripped matching — try adding framework prefixes
   if (!matchNode) {
     for (const prefix of FRAMEWORK_ENTRY_PREFIXES) {
-      matchNode = findBestMatch(db, `${prefix}${name}`, opts);
+      matchNode = findMatchingNodes(db, `${prefix}${name}`, opts)[0] ?? null;
       if (matchNode) break;
     }
   }
@@ -219,73 +219,6 @@ export function flowData(name, dbPath, opts = {}) {
   return paginateResult(base, 'steps', { limit: opts.limit, offset: opts.offset });
 }
 
-/**
- * Find the best matching node using the same relevance scoring as queries.js findMatchingNodes.
- */
-function findBestMatch(db, name, opts = {}) {
-  const kinds = opts.kind
-    ? [opts.kind]
-    : [
-        'function',
-        'method',
-        'class',
-        'interface',
-        'type',
-        'struct',
-        'enum',
-        'trait',
-        'record',
-        'module',
-      ];
-  const placeholders = kinds.map(() => '?').join(', ');
-  const params = [`%${name}%`, ...kinds];
-
-  let fileCondition = '';
-  if (opts.file) {
-    fileCondition = ' AND n.file LIKE ?';
-    params.push(`%${opts.file}%`);
-  }
-
-  const rows = db
-    .prepare(
-      `SELECT n.*, COALESCE(fi.cnt, 0) AS fan_in
-       FROM nodes n
-       LEFT JOIN (
-         SELECT target_id, COUNT(*) AS cnt FROM edges WHERE kind = 'calls' GROUP BY target_id
-       ) fi ON fi.target_id = n.id
-       WHERE n.name LIKE ? AND n.kind IN (${placeholders})${fileCondition}`,
-    )
-    .all(...params);
-
-  const noTests = opts.noTests || false;
-  const nodes = noTests ? rows.filter((n) => !isTestFile(n.file)) : rows;
-
-  if (nodes.length === 0) return null;
-
-  const lowerQuery = name.toLowerCase();
-  for (const node of nodes) {
-    const lowerName = node.name.toLowerCase();
-    const bareName = lowerName.includes('.') ? lowerName.split('.').pop() : lowerName;
-
-    let matchScore;
-    if (lowerName === lowerQuery || bareName === lowerQuery) {
-      matchScore = 100;
-    } else if (lowerName.startsWith(lowerQuery) || bareName.startsWith(lowerQuery)) {
-      matchScore = 60;
-    } else if (lowerName.includes(`.${lowerQuery}`) || lowerName.includes(`${lowerQuery}.`)) {
-      matchScore = 40;
-    } else {
-      matchScore = 10;
-    }
-
-    const fanInBonus = Math.min(Math.log2(node.fan_in + 1) * 5, 25);
-    node._relevance = matchScore + fanInBonus;
-  }
-
-  nodes.sort((a, b) => b._relevance - a._relevance);
-  return nodes[0];
-}
-
 /**
  * CLI formatter — text or JSON output.
  */

package/src/index.js

@@ -121,7 +121,6 @@ export { isNativeAvailable } from './native.js';
 export { matchOwners, owners, ownersData, ownersForFiles, parseCodeowners } from './owners.js';
 // Pagination utilities
 export { MCP_DEFAULTS, MCP_MAX_LIMIT, paginate, paginateResult, printNdjson } from './paginate.js';
-
 // Unified parser API
 export { getActiveEngine, isWasmAvailable, parseFileAuto, parseFilesAuto } from './parser.js';
 // Query functions (data-returning)
@@ -170,6 +169,8 @@ export {
   saveRegistry,
   unregisterRepo,
 } from './registry.js';
+// Sequence diagram generation
+export { sequence, sequenceData, sequenceToMermaid } from './sequence.js';
 // Snapshot management
 export {
   snapshotDelete,

package/src/mcp.js

@@ -113,6 +113,11 @@ const BASE_TOOLS = [
       properties: {
         file: { type: 'string', description: 'File path (partial match supported)' },
         no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+        unused: {
+          type: 'boolean',
+          description: 'Show only exports with zero consumers',
+          default: false,
+        },
         ...PAGINATION_PROPS,
       },
       required: ['file'],
@@ -418,6 +423,43 @@ const BASE_TOOLS = [
       },
     },
   },
+  {
+    name: 'sequence',
+    description:
+      'Generate a Mermaid sequence diagram from call graph edges. Participants are files, messages are function calls between them.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: {
+          type: 'string',
+          description: 'Entry point or function name to trace from (partial match)',
+        },
+        depth: { type: 'number', description: 'Max forward traversal depth', default: 10 },
+        format: {
+          type: 'string',
+          enum: ['mermaid', 'json'],
+          description: 'Output format (default: mermaid)',
+        },
+        dataflow: {
+          type: 'boolean',
+          description: 'Annotate with parameter names and return arrows',
+          default: false,
+        },
+        file: {
+          type: 'string',
+          description: 'Scope search to functions in this file (partial match)',
+        },
+        kind: {
+          type: 'string',
+          enum: EVERY_SYMBOL_KIND,
+          description: 'Filter to a specific symbol kind',
+        },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+        ...PAGINATION_PROPS,
+      },
+      required: ['name'],
+    },
+  },
   {
     name: 'complexity',
     description:
@@ -902,6 +944,7 @@ export async function startMCPServer(customDbPath, options = {}) {
         case 'file_exports':
           result = exportsData(args.file, dbPath, {
             noTests: args.no_tests,
+            unused: args.unused,
             limit: Math.min(args.limit ?? MCP_DEFAULTS.file_exports, MCP_MAX_LIMIT),
             offset: args.offset ?? 0,
           });
@@ -1165,6 +1208,23 @@ export async function startMCPServer(customDbPath, options = {}) {
           }
           break;
         }
+        case 'sequence': {
+          const { sequenceData, sequenceToMermaid } = await import('./sequence.js');
+          const seqResult = sequenceData(args.name, dbPath, {
+            depth: args.depth,
+            file: args.file,
+            kind: args.kind,
+            dataflow: args.dataflow,
+            noTests: args.no_tests,
+            limit: Math.min(args.limit ?? MCP_DEFAULTS.execution_flow, MCP_MAX_LIMIT),
+            offset: args.offset ?? 0,
+          });
+          result =
+            args.format === 'json'
+              ? seqResult
+              : { text: sequenceToMermaid(seqResult), ...seqResult };
+          break;
+        }
         case 'complexity': {
           const { complexityData } = await import('./complexity.js');
           result = complexityData(dbPath, {

package/src/native.js

@@ -12,9 +12,27 @@ import os from 'node:os';
 let _cached; // undefined = not yet tried, null = failed, object = module
 let _loadError = null;
 
-/** Map of (platform-arch) → npm package name. */
+/**
+ * Detect whether the current Linux environment uses glibc or musl.
+ * Returns 'gnu' for glibc, 'musl' for musl, 'gnu' as fallback.
+ */
+function detectLibc() {
+  try {
+    const { readdirSync } = require('node:fs');
+    const files = readdirSync('/lib');
+    if (files.some((f) => f.startsWith('ld-musl-') && f.endsWith('.so.1'))) {
+      return 'musl';
+    }
+  } catch {}
+  return 'gnu';
+}
+
+/** Map of (platform-arch[-libc]) → npm package name. */
 const PLATFORM_PACKAGES = {
-  'linux-x64': '@optave/codegraph-linux-x64-gnu',
+  'linux-x64-gnu': '@optave/codegraph-linux-x64-gnu',
+  'linux-x64-musl': '@optave/codegraph-linux-x64-musl',
+  'linux-arm64-gnu': '@optave/codegraph-linux-arm64-gnu',
+  'linux-arm64-musl': '@optave/codegraph-linux-arm64-musl', // not yet published — placeholder for future CI target
   'darwin-arm64': '@optave/codegraph-darwin-arm64',
   'darwin-x64': '@optave/codegraph-darwin-x64',
   'win32-x64': '@optave/codegraph-win32-x64-msvc',
@@ -29,7 +47,9 @@ export function loadNative() {
 
   const require = createRequire(import.meta.url);
 
-  const key = `${os.platform()}-${os.arch()}`;
+  const platform = os.platform();
+  const arch = os.arch();
+  const key = platform === 'linux' ? `${platform}-${arch}-${detectLibc()}` : `${platform}-${arch}`;
   const pkg = PLATFORM_PACKAGES[key];
   if (pkg) {
     try {