@optave/codegraph 2.1.0 → 2.1.1-dev.00f091c

package/src/extractors/javascript.js

@@ -313,6 +313,18 @@ function extractImplementsFromNode(node) {
   return result;
 }
 
+function extractReceiverName(objNode) {
+  if (!objNode) return undefined;
+  if (objNode.type === 'identifier') return objNode.text;
+  if (objNode.type === 'this') return 'this';
+  if (objNode.type === 'super') return 'super';
+  if (objNode.type === 'member_expression') {
+    const prop = objNode.childForFieldName('property');
+    if (prop) return objNode.text;
+  }
+  return objNode.text;
+}
+
 function extractCallInfo(fn, callNode) {
   if (fn.type === 'identifier') {
     return { name: fn.text, line: callNode.startPosition.row + 1 };
@@ -335,19 +347,25 @@ function extractCallInfo(fn, callNode) {
 
     if (prop.type === 'string' || prop.type === 'string_fragment') {
       const methodName = prop.text.replace(/['"]/g, '');
-      if (methodName)
-        return { name: methodName, line: callNode.startPosition.row + 1, dynamic: true };
+      if (methodName) {
+        const receiver = extractReceiverName(obj);
+        return { name: methodName, line: callNode.startPosition.row + 1, dynamic: true, receiver };
+      }
     }
 
-    return { name: prop.text, line: callNode.startPosition.row + 1 };
+    const receiver = extractReceiverName(obj);
+    return { name: prop.text, line: callNode.startPosition.row + 1, receiver };
   }
 
   if (fn.type === 'subscript_expression') {
+    const obj = fn.childForFieldName('object');
     const index = fn.childForFieldName('index');
     if (index && (index.type === 'string' || index.type === 'template_string')) {
       const methodName = index.text.replace(/['"`]/g, '');
-      if (methodName && !methodName.includes('$'))
-        return { name: methodName, line: callNode.startPosition.row + 1, dynamic: true };
+      if (methodName && !methodName.includes('$')) {
+        const receiver = extractReceiverName(obj);
+        return { name: methodName, line: callNode.startPosition.row + 1, dynamic: true, receiver };
+      }
     }
   }
 

package/src/extractors/php.js

@@ -206,7 +206,10 @@ export function extractPHPSymbols(tree, _filePath) {
       case 'member_call_expression': {
         const name = node.childForFieldName('name');
         if (name) {
-          calls.push({ name: name.text, line: node.startPosition.row + 1 });
+          const obj = node.childForFieldName('object');
+          const call = { name: name.text, line: node.startPosition.row + 1 };
+          if (obj) call.receiver = obj.text;
+          calls.push(call);
         }
         break;
       }
@@ -214,7 +217,10 @@ export function extractPHPSymbols(tree, _filePath) {
       case 'scoped_call_expression': {
         const name = node.childForFieldName('name');
         if (name) {
-          calls.push({ name: name.text, line: node.startPosition.row + 1 });
+          const scope = node.childForFieldName('scope');
+          const call = { name: name.text, line: node.startPosition.row + 1 };
+          if (scope) call.receiver = scope.text;
+          calls.push(call);
         }
         break;
       }

package/src/extractors/python.js

@@ -69,12 +69,19 @@ export function extractPythonSymbols(tree, _filePath) {
         const fn = node.childForFieldName('function');
         if (fn) {
           let callName = null;
+          let receiver;
           if (fn.type === 'identifier') callName = fn.text;
           else if (fn.type === 'attribute') {
             const attr = fn.childForFieldName('attribute');
             if (attr) callName = attr.text;
+            const obj = fn.childForFieldName('object');
+            if (obj) receiver = obj.text;
+          }
+          if (callName) {
+            const call = { name: callName, line: node.startPosition.row + 1 };
+            if (receiver) call.receiver = receiver;
+            calls.push(call);
           }
-          if (callName) calls.push({ name: callName, line: node.startPosition.row + 1 });
         }
         break;
       }

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,6 +38,7 @@ export { isNativeAvailable } from './native.js';
 export { getActiveEngine, parseFileAuto, parseFilesAuto } from './parser.js';
 // Query functions (data-returning)
 export {
+  contextData,
   diffImpactData,
   fileDepsData,
   fnDepsData,

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

@@ -100,6 +100,34 @@ const BASE_TOOLS = [
       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,
+        },
+        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: 'diff_impact',
     description: 'Analyze git diff to find which functions changed and their transitive callers',
@@ -245,12 +273,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 +298,7 @@ export async function startMCPServer(customDbPath, options = {}) {
     fileDepsData,
     fnDepsData,
     fnImpactData,
+    contextData,
     diffImpactData,
     listFunctionsData,
   } = await import('./queries.js');
@@ -279,9 +311,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 {
@@ -343,6 +377,14 @@ export async function startMCPServer(customDbPath, options = {}) {
             noTests: args.no_tests,
           });
           break;
+        case 'context':
+          result = contextData(args.name, dbPath, {
+            depth: args.depth,
+            noSource: args.no_source,
+            noTests: args.no_tests,
+            includeTests: args.include_tests,
+          });
+          break;
         case 'diff_impact':
           result = diffImpactData(dbPath, {
             staged: args.staged,

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,