@optave/codegraph 2.1.1-dev.3c12b64 → 2.2.0

package/src/extractors/ruby.js

@@ -170,7 +170,10 @@ export function extractRubySymbols(tree, _filePath) {
               }
             }
           } else {
-            calls.push({ name: methodNode.text, line: node.startPosition.row + 1 });
+            const recv = node.childForFieldName('receiver');
+            const call = { name: methodNode.text, line: node.startPosition.row + 1 };
+            if (recv) call.receiver = recv.text;
+            calls.push(call);
           }
         }
         break;

package/src/extractors/rust.js

@@ -135,10 +135,20 @@ export function extractRustSymbols(tree, _filePath) {
             calls.push({ name: fn.text, line: node.startPosition.row + 1 });
           } else if (fn.type === 'field_expression') {
             const field = fn.childForFieldName('field');
-            if (field) calls.push({ name: field.text, line: node.startPosition.row + 1 });
+            if (field) {
+              const value = fn.childForFieldName('value');
+              const call = { name: field.text, line: node.startPosition.row + 1 };
+              if (value) call.receiver = value.text;
+              calls.push(call);
+            }
           } else if (fn.type === 'scoped_identifier') {
             const name = fn.childForFieldName('name');
-            if (name) calls.push({ name: name.text, line: node.startPosition.row + 1 });
+            if (name) {
+              const path = fn.childForFieldName('path');
+              const call = { name: name.text, line: node.startPosition.row + 1 };
+              if (path) call.receiver = path.text;
+              calls.push(call);
+            }
           }
         }
         break;

package/src/index.js

@@ -38,7 +38,12 @@ export { isNativeAvailable } from './native.js';
 export { getActiveEngine, parseFileAuto, parseFilesAuto } from './parser.js';
 // Query functions (data-returning)
 export {
+  ALL_SYMBOL_KINDS,
+  contextData,
   diffImpactData,
+  explainData,
+  FALSE_POSITIVE_CALLER_THRESHOLD,
+  FALSE_POSITIVE_NAMES,
   fileDepsData,
   fnDepsData,
   fnImpactData,
@@ -46,6 +51,7 @@ export {
   moduleMapData,
   queryNameData,
   statsData,
+  whereData,
 } from './queries.js';
 // Registry (multi-repo)
 export {

package/src/journal.js

@@ -0,0 +1,109 @@
+import fs from 'node:fs';
+import path from 'node:path';
+import { debug, warn } from './logger.js';
+
+export const JOURNAL_FILENAME = 'changes.journal';
+const HEADER_PREFIX = '# codegraph-journal v1 ';
+
+/**
+ * Read and validate the change journal.
+ * Returns { valid, timestamp, changed[], removed[] } or { valid: false }.
+ */
+export function readJournal(rootDir) {
+  const journalPath = path.join(rootDir, '.codegraph', JOURNAL_FILENAME);
+  let content;
+  try {
+    content = fs.readFileSync(journalPath, 'utf-8');
+  } catch {
+    return { valid: false };
+  }
+
+  const lines = content.split('\n');
+  if (lines.length === 0 || !lines[0].startsWith(HEADER_PREFIX)) {
+    debug('Journal has malformed or missing header');
+    return { valid: false };
+  }
+
+  const timestamp = Number(lines[0].slice(HEADER_PREFIX.length).trim());
+  if (!Number.isFinite(timestamp) || timestamp <= 0) {
+    debug('Journal has invalid timestamp');
+    return { valid: false };
+  }
+
+  const changed = [];
+  const removed = [];
+  const seenChanged = new Set();
+  const seenRemoved = new Set();
+
+  for (let i = 1; i < lines.length; i++) {
+    const line = lines[i].trim();
+    if (!line || line.startsWith('#')) continue;
+
+    if (line.startsWith('DELETED ')) {
+      const filePath = line.slice(8);
+      if (filePath && !seenRemoved.has(filePath)) {
+        seenRemoved.add(filePath);
+        removed.push(filePath);
+      }
+    } else {
+      if (!seenChanged.has(line)) {
+        seenChanged.add(line);
+        changed.push(line);
+      }
+    }
+  }
+
+  return { valid: true, timestamp, changed, removed };
+}
+
+/**
+ * Append changed/deleted paths to the journal.
+ * Creates the journal with a header if it doesn't exist.
+ */
+export function appendJournalEntries(rootDir, entries) {
+  const dir = path.join(rootDir, '.codegraph');
+  const journalPath = path.join(dir, JOURNAL_FILENAME);
+
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+
+  // If journal doesn't exist, create with a placeholder header
+  if (!fs.existsSync(journalPath)) {
+    fs.writeFileSync(journalPath, `${HEADER_PREFIX}0\n`);
+  }
+
+  const lines = entries.map((e) => {
+    if (e.deleted) return `DELETED ${e.file}`;
+    return e.file;
+  });
+
+  fs.appendFileSync(journalPath, `${lines.join('\n')}\n`);
+}
+
+/**
+ * Write a fresh journal header after a successful build.
+ * Atomic: write to temp file then rename.
+ */
+export function writeJournalHeader(rootDir, timestamp) {
+  const dir = path.join(rootDir, '.codegraph');
+  const journalPath = path.join(dir, JOURNAL_FILENAME);
+  const tmpPath = `${journalPath}.tmp`;
+
+  if (!fs.existsSync(dir)) {
+    fs.mkdirSync(dir, { recursive: true });
+  }
+
+  try {
+    fs.writeFileSync(tmpPath, `${HEADER_PREFIX}${timestamp}\n`);
+    fs.renameSync(tmpPath, journalPath);
+  } catch (err) {
+    warn(`Failed to write journal header: ${err.message}`);
+    // Clean up temp file if rename failed
+    try {
+      fs.unlinkSync(tmpPath);
+    } catch {
+      /* ignore */
+    }
+  }
+}

package/src/mcp.js

@@ -8,6 +8,7 @@
 import { createRequire } from 'node:module';
 import { findCycles } from './cycles.js';
 import { findDbPath } from './db.js';
+import { ALL_SYMBOL_KINDS } from './queries.js';
 
 const REPO_PROP = {
   repo: {
@@ -29,6 +30,7 @@ const BASE_TOOLS = [
           description: 'Traversal depth for transitive callers',
           default: 2,
         },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
       required: ['name'],
     },
@@ -40,6 +42,7 @@ const BASE_TOOLS = [
       type: 'object',
       properties: {
         file: { type: 'string', description: 'File path (partial match supported)' },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
       required: ['file'],
     },
@@ -51,6 +54,7 @@ const BASE_TOOLS = [
       type: 'object',
       properties: {
         file: { type: 'string', description: 'File path to analyze' },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
       required: ['file'],
     },
@@ -70,6 +74,7 @@ const BASE_TOOLS = [
       type: 'object',
       properties: {
         limit: { type: 'number', description: 'Number of top files to show', default: 20 },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
     },
   },
@@ -81,6 +86,15 @@ const BASE_TOOLS = [
       properties: {
         name: { type: 'string', description: 'Function/method/class name (partial match)' },
         depth: { type: 'number', description: 'Transitive caller depth', default: 3 },
+        file: {
+          type: 'string',
+          description: 'Scope search to functions in this file (partial match)',
+        },
+        kind: {
+          type: 'string',
+          enum: ALL_SYMBOL_KINDS,
+          description: 'Filter to a specific symbol kind',
+        },
         no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
       required: ['name'],
@@ -95,11 +109,88 @@ const BASE_TOOLS = [
       properties: {
         name: { type: 'string', description: 'Function/method/class name (partial match)' },
         depth: { type: 'number', description: 'Max traversal depth', default: 5 },
+        file: {
+          type: 'string',
+          description: 'Scope search to functions in this file (partial match)',
+        },
+        kind: {
+          type: 'string',
+          enum: ALL_SYMBOL_KINDS,
+          description: 'Filter to a specific symbol kind',
+        },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+      required: ['name'],
+    },
+  },
+  {
+    name: 'context',
+    description:
+      'Full context for a function: source code, dependencies with summaries, callers, signature, and related tests — everything needed to understand or modify a function in one call',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Function/method/class name (partial match)' },
+        depth: {
+          type: 'number',
+          description: 'Include callee source up to N levels deep (0=no source, 1=direct)',
+          default: 0,
+        },
+        file: {
+          type: 'string',
+          description: 'Scope search to functions in this file (partial match)',
+        },
+        kind: {
+          type: 'string',
+          enum: ALL_SYMBOL_KINDS,
+          description: 'Filter to a specific symbol kind',
+        },
+        no_source: {
+          type: 'boolean',
+          description: 'Skip source extraction (metadata only)',
+          default: false,
+        },
         no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+        include_tests: {
+          type: 'boolean',
+          description: 'Include test file source code',
+          default: false,
+        },
       },
       required: ['name'],
     },
   },
+  {
+    name: 'explain',
+    description:
+      'Structural summary of a file or function: public/internal API, data flow, dependencies. No LLM needed.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        target: { type: 'string', description: 'File path or function name' },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+      required: ['target'],
+    },
+  },
+  {
+    name: 'where',
+    description:
+      'Find where a symbol is defined and used, or list symbols/imports/exports for a file. Minimal, fast lookup.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        target: { type: 'string', description: 'Symbol name or file path' },
+        file_mode: {
+          type: 'boolean',
+          description: 'Treat target as file path (list symbols/imports/exports)',
+          default: false,
+        },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+      required: ['target'],
+    },
+  },
   {
     name: 'diff_impact',
     description: 'Analyze git diff to find which functions changed and their transitive callers',
@@ -195,6 +286,7 @@ const BASE_TOOLS = [
           description: 'Rank files or directories',
         },
         limit: { type: 'number', description: 'Number of results to return', default: 10 },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
       },
     },
   },
@@ -245,12 +337,15 @@ export { TOOLS, buildToolList };
 export async function startMCPServer(customDbPath, options = {}) {
   const { allowedRepos } = options;
   const multiRepo = options.multiRepo || !!allowedRepos;
-  let Server, StdioServerTransport;
+  let Server, StdioServerTransport, ListToolsRequestSchema, CallToolRequestSchema;
   try {
     const sdk = await import('@modelcontextprotocol/sdk/server/index.js');
     Server = sdk.Server;
     const transport = await import('@modelcontextprotocol/sdk/server/stdio.js');
     StdioServerTransport = transport.StdioServerTransport;
+    const types = await import('@modelcontextprotocol/sdk/types.js');
+    ListToolsRequestSchema = types.ListToolsRequestSchema;
+    CallToolRequestSchema = types.CallToolRequestSchema;
   } catch {
     console.error(
       'MCP server requires @modelcontextprotocol/sdk.\n' +
@@ -267,6 +362,9 @@ export async function startMCPServer(customDbPath, options = {}) {
     fileDepsData,
     fnDepsData,
     fnImpactData,
+    contextData,
+    explainData,
+    whereData,
     diffImpactData,
     listFunctionsData,
   } = await import('./queries.js');
@@ -279,9 +377,11 @@ export async function startMCPServer(customDbPath, options = {}) {
     { capabilities: { tools: {} } },
   );
 
-  server.setRequestHandler('tools/list', async () => ({ tools: buildToolList(multiRepo) }));
+  server.setRequestHandler(ListToolsRequestSchema, async () => ({
+    tools: buildToolList(multiRepo),
+  }));
 
-  server.setRequestHandler('tools/call', async (request) => {
+  server.setRequestHandler(CallToolRequestSchema, async (request) => {
     const { name, arguments: args } = request.params;
 
     try {
@@ -313,13 +413,13 @@ export async function startMCPServer(customDbPath, options = {}) {
       let result;
       switch (name) {
         case 'query_function':
-          result = queryNameData(args.name, dbPath);
+          result = queryNameData(args.name, dbPath, { noTests: args.no_tests });
           break;
         case 'file_deps':
-          result = fileDepsData(args.file, dbPath);
+          result = fileDepsData(args.file, dbPath, { noTests: args.no_tests });
           break;
         case 'impact_analysis':
-          result = impactAnalysisData(args.file, dbPath);
+          result = impactAnalysisData(args.file, dbPath, { noTests: args.no_tests });
           break;
         case 'find_cycles': {
           const db = new Database(findDbPath(dbPath), { readonly: true });
@@ -329,17 +429,40 @@ export async function startMCPServer(customDbPath, options = {}) {
           break;
         }
         case 'module_map':
-          result = moduleMapData(dbPath, args.limit || 20);
+          result = moduleMapData(dbPath, args.limit || 20, { noTests: args.no_tests });
           break;
         case 'fn_deps':
           result = fnDepsData(args.name, dbPath, {
             depth: args.depth,
+            file: args.file,
+            kind: args.kind,
             noTests: args.no_tests,
           });
           break;
         case 'fn_impact':
           result = fnImpactData(args.name, dbPath, {
             depth: args.depth,
+            file: args.file,
+            kind: args.kind,
+            noTests: args.no_tests,
+          });
+          break;
+        case 'context':
+          result = contextData(args.name, dbPath, {
+            depth: args.depth,
+            file: args.file,
+            kind: args.kind,
+            noSource: args.no_source,
+            noTests: args.no_tests,
+            includeTests: args.include_tests,
+          });
+          break;
+        case 'explain':
+          result = explainData(args.target, dbPath, { noTests: args.no_tests });
+          break;
+        case 'where':
+          result = whereData(args.target, dbPath, {
+            file: args.file_mode,
             noTests: args.no_tests,
           });
           break;
@@ -418,6 +541,7 @@ export async function startMCPServer(customDbPath, options = {}) {
             metric: args.metric,
             level: args.level,
             limit: args.limit,
+            noTests: args.no_tests,
           });
           break;
         }

package/src/parser.js

@@ -101,6 +101,7 @@ function normalizeNativeSymbols(result) {
       name: c.name,
       line: c.line,
       dynamic: c.dynamic,
+      receiver: c.receiver,
     })),
     imports: (result.imports || []).map((i) => ({
       source: i.source,