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

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,