@optave/codegraph 1.1.0 → 1.4.1

package/src/mcp.js

@@ -1,139 +1,311 @@
-/**
- * MCP (Model Context Protocol) server for codegraph.
- * Exposes codegraph queries as tools that AI coding assistants can call.
- *
- * Requires: npm install @modelcontextprotocol/sdk
- */
-
-import { createRequire } from 'node:module';
-import { findDbPath } from './db.js';
-import { findCycles } from './cycles.js';
-
-const TOOLS = [
-  {
-    name: 'query_function',
-    description: 'Find callers and callees of a function by name',
-    inputSchema: {
-      type: 'object',
-      properties: {
-        name: { type: 'string', description: 'Function name to query (supports partial match)' },
-        depth: { type: 'number', description: 'Traversal depth for transitive callers', default: 2 }
-      },
-      required: ['name']
-    }
-  },
-  {
-    name: 'file_deps',
-    description: 'Show what a file imports and what imports it',
-    inputSchema: {
-      type: 'object',
-      properties: {
-        file: { type: 'string', description: 'File path (partial match supported)' }
-      },
-      required: ['file']
-    }
-  },
-  {
-    name: 'impact_analysis',
-    description: 'Show files affected by changes to a given file (transitive)',
-    inputSchema: {
-      type: 'object',
-      properties: {
-        file: { type: 'string', description: 'File path to analyze' }
-      },
-      required: ['file']
-    }
-  },
-  {
-    name: 'find_cycles',
-    description: 'Detect circular dependencies in the codebase',
-    inputSchema: {
-      type: 'object',
-      properties: {}
-    }
-  },
-  {
-    name: 'module_map',
-    description: 'Get high-level overview of most-connected files',
-    inputSchema: {
-      type: 'object',
-      properties: {
-        limit: { type: 'number', description: 'Number of top files to show', default: 20 }
-      }
-    }
-  }
-];
-
-export { TOOLS };
-
-/**
- * Start the MCP server.
- * This function requires @modelcontextprotocol/sdk to be installed.
- */
-export async function startMCPServer(customDbPath) {
-  let Server, StdioServerTransport;
-  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;
-  } catch {
-    console.error(
-      'MCP server requires @modelcontextprotocol/sdk.\n' +
-      'Install it with: npm install @modelcontextprotocol/sdk'
-    );
-    process.exit(1);
-  }
-
-  // Lazy import query functions to avoid circular deps at module load
-  const { queryNameData, impactAnalysisData, moduleMapData, fileDepsData } = await import('./queries.js');
-
-  const require = createRequire(import.meta.url);
-  const Database = require('better-sqlite3');
-
-  const server = new Server(
-    { name: 'codegraph', version: '1.0.0' },
-    { capabilities: { tools: {} } }
-  );
-
-  server.setRequestHandler('tools/list', async () => ({ tools: TOOLS }));
-
-  server.setRequestHandler('tools/call', async (request) => {
-    const { name, arguments: args } = request.params;
-    const dbPath = customDbPath || undefined;
-
-    try {
-      let result;
-      switch (name) {
-        case 'query_function':
-          result = queryNameData(args.name, dbPath);
-          break;
-        case 'file_deps':
-          result = fileDepsData(args.file, dbPath);
-          break;
-        case 'impact_analysis':
-          result = impactAnalysisData(args.file, dbPath);
-          break;
-        case 'find_cycles': {
-          const db = new Database(findDbPath(dbPath), { readonly: true });
-          const cycles = findCycles(db);
-          db.close();
-          result = { cycles, count: cycles.length };
-          break;
-        }
-        case 'module_map':
-          result = moduleMapData(dbPath, args.limit || 20);
-          break;
-        default:
-          return { content: [{ type: 'text', text: `Unknown tool: ${name}` }], isError: true };
-      }
-
-      return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
-    } catch (err) {
-      return { content: [{ type: 'text', text: `Error: ${err.message}` }], isError: true };
-    }
-  });
-
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
-}
+/**
+ * MCP (Model Context Protocol) server for codegraph.
+ * Exposes codegraph queries as tools that AI coding assistants can call.
+ *
+ * Requires: npm install @modelcontextprotocol/sdk
+ */
+
+import { createRequire } from 'node:module';
+import { findCycles } from './cycles.js';
+import { findDbPath } from './db.js';
+
+const TOOLS = [
+  {
+    name: 'query_function',
+    description: 'Find callers and callees of a function by name',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Function name to query (supports partial match)' },
+        depth: {
+          type: 'number',
+          description: 'Traversal depth for transitive callers',
+          default: 2,
+        },
+      },
+      required: ['name'],
+    },
+  },
+  {
+    name: 'file_deps',
+    description: 'Show what a file imports and what imports it',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        file: { type: 'string', description: 'File path (partial match supported)' },
+      },
+      required: ['file'],
+    },
+  },
+  {
+    name: 'impact_analysis',
+    description: 'Show files affected by changes to a given file (transitive)',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        file: { type: 'string', description: 'File path to analyze' },
+      },
+      required: ['file'],
+    },
+  },
+  {
+    name: 'find_cycles',
+    description: 'Detect circular dependencies in the codebase',
+    inputSchema: {
+      type: 'object',
+      properties: {},
+    },
+  },
+  {
+    name: 'module_map',
+    description: 'Get high-level overview of most-connected files',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        limit: { type: 'number', description: 'Number of top files to show', default: 20 },
+      },
+    },
+  },
+  {
+    name: 'fn_deps',
+    description: 'Show function-level dependency chain: what a function calls and what calls it',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Function/method/class name (partial match)' },
+        depth: { type: 'number', description: 'Transitive caller depth', default: 3 },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+      required: ['name'],
+    },
+  },
+  {
+    name: 'fn_impact',
+    description:
+      'Show function-level blast radius: all functions transitively affected by changes to a function',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Function/method/class name (partial match)' },
+        depth: { type: 'number', description: 'Max traversal depth', default: 5 },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+      required: ['name'],
+    },
+  },
+  {
+    name: 'diff_impact',
+    description: 'Analyze git diff to find which functions changed and their transitive callers',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        staged: { type: 'boolean', description: 'Analyze staged changes only', default: false },
+        ref: { type: 'string', description: 'Git ref to diff against (default: HEAD)' },
+        depth: { type: 'number', description: 'Transitive caller depth', default: 3 },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+    },
+  },
+  {
+    name: 'semantic_search',
+    description:
+      'Search code symbols by meaning using embeddings (requires prior `codegraph embed`)',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        query: { type: 'string', description: 'Natural language search query' },
+        limit: { type: 'number', description: 'Max results to return', default: 15 },
+        min_score: { type: 'number', description: 'Minimum similarity score (0-1)', default: 0.2 },
+      },
+      required: ['query'],
+    },
+  },
+  {
+    name: 'export_graph',
+    description: 'Export the dependency graph in DOT (Graphviz), Mermaid, or JSON format',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        format: {
+          type: 'string',
+          enum: ['dot', 'mermaid', 'json'],
+          description: 'Export format',
+        },
+        file_level: {
+          type: 'boolean',
+          description: 'File-level graph (true) or function-level (false)',
+          default: true,
+        },
+      },
+      required: ['format'],
+    },
+  },
+  {
+    name: 'list_functions',
+    description:
+      'List functions, methods, and classes in the codebase, optionally filtered by file or name pattern',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        file: { type: 'string', description: 'Filter by file path (partial match)' },
+        pattern: { type: 'string', description: 'Filter by function name (partial match)' },
+        no_tests: { type: 'boolean', description: 'Exclude test files', default: false },
+      },
+    },
+  },
+];
+
+export { TOOLS };
+
+/**
+ * Start the MCP server.
+ * This function requires @modelcontextprotocol/sdk to be installed.
+ */
+export async function startMCPServer(customDbPath) {
+  let Server, StdioServerTransport;
+  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;
+  } catch {
+    console.error(
+      'MCP server requires @modelcontextprotocol/sdk.\n' +
+        'Install it with: npm install @modelcontextprotocol/sdk',
+    );
+    process.exit(1);
+  }
+
+  // Lazy import query functions to avoid circular deps at module load
+  const {
+    queryNameData,
+    impactAnalysisData,
+    moduleMapData,
+    fileDepsData,
+    fnDepsData,
+    fnImpactData,
+    diffImpactData,
+    listFunctionsData,
+  } = await import('./queries.js');
+
+  const require = createRequire(import.meta.url);
+  const Database = require('better-sqlite3');
+
+  const server = new Server(
+    { name: 'codegraph', version: '1.0.0' },
+    { capabilities: { tools: {} } },
+  );
+
+  server.setRequestHandler('tools/list', async () => ({ tools: TOOLS }));
+
+  server.setRequestHandler('tools/call', async (request) => {
+    const { name, arguments: args } = request.params;
+    const dbPath = customDbPath || undefined;
+
+    try {
+      let result;
+      switch (name) {
+        case 'query_function':
+          result = queryNameData(args.name, dbPath);
+          break;
+        case 'file_deps':
+          result = fileDepsData(args.file, dbPath);
+          break;
+        case 'impact_analysis':
+          result = impactAnalysisData(args.file, dbPath);
+          break;
+        case 'find_cycles': {
+          const db = new Database(findDbPath(dbPath), { readonly: true });
+          const cycles = findCycles(db);
+          db.close();
+          result = { cycles, count: cycles.length };
+          break;
+        }
+        case 'module_map':
+          result = moduleMapData(dbPath, args.limit || 20);
+          break;
+        case 'fn_deps':
+          result = fnDepsData(args.name, dbPath, {
+            depth: args.depth,
+            noTests: args.no_tests,
+          });
+          break;
+        case 'fn_impact':
+          result = fnImpactData(args.name, dbPath, {
+            depth: args.depth,
+            noTests: args.no_tests,
+          });
+          break;
+        case 'diff_impact':
+          result = diffImpactData(dbPath, {
+            staged: args.staged,
+            ref: args.ref,
+            depth: args.depth,
+            noTests: args.no_tests,
+          });
+          break;
+        case 'semantic_search': {
+          const { searchData } = await import('./embedder.js');
+          result = await searchData(args.query, dbPath, {
+            limit: args.limit,
+            minScore: args.min_score,
+          });
+          if (result === null) {
+            return {
+              content: [
+                { type: 'text', text: 'Semantic search unavailable. Run `codegraph embed` first.' },
+              ],
+              isError: true,
+            };
+          }
+          break;
+        }
+        case 'export_graph': {
+          const { exportDOT, exportMermaid, exportJSON } = await import('./export.js');
+          const db = new Database(findDbPath(dbPath), { readonly: true });
+          const fileLevel = args.file_level !== false;
+          switch (args.format) {
+            case 'dot':
+              result = exportDOT(db, { fileLevel });
+              break;
+            case 'mermaid':
+              result = exportMermaid(db, { fileLevel });
+              break;
+            case 'json':
+              result = exportJSON(db);
+              break;
+            default:
+              db.close();
+              return {
+                content: [
+                  {
+                    type: 'text',
+                    text: `Unknown format: ${args.format}. Use dot, mermaid, or json.`,
+                  },
+                ],
+                isError: true,
+              };
+          }
+          db.close();
+          break;
+        }
+        case 'list_functions':
+          result = listFunctionsData(dbPath, {
+            file: args.file,
+            pattern: args.pattern,
+            noTests: args.no_tests,
+          });
+          break;
+        default:
+          return { content: [{ type: 'text', text: `Unknown tool: ${name}` }], isError: true };
+      }
+
+      return { content: [{ type: 'text', text: JSON.stringify(result, null, 2) }] };
+    } catch (err) {
+      return { content: [{ type: 'text', text: `Error: ${err.message}` }], isError: true };
+    }
+  });
+
+  const transport = new StdioServerTransport();
+  await server.connect(transport);
+}

package/src/native.js

@@ -0,0 +1,68 @@
+/**
+ * Native addon loader with graceful fallback to WASM.
+ *
+ * Tries to load the platform-specific napi-rs binary built from
+ * crates/codegraph-core.  If unavailable the caller should fall back
+ * to the existing WASM pipeline.
+ */
+
+import { createRequire } from 'node:module';
+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. */
+const PLATFORM_PACKAGES = {
+  'linux-x64': '@optave/codegraph-linux-x64-gnu',
+  'darwin-arm64': '@optave/codegraph-darwin-arm64',
+  'darwin-x64': '@optave/codegraph-darwin-x64',
+  'win32-x64': '@optave/codegraph-win32-x64-msvc',
+};
+
+/**
+ * Try to load the native napi addon.
+ * Returns the module on success, null on failure.
+ */
+export function loadNative() {
+  if (_cached !== undefined) return _cached;
+
+  const require = createRequire(import.meta.url);
+
+  const key = `${os.platform()}-${os.arch()}`;
+  const pkg = PLATFORM_PACKAGES[key];
+  if (pkg) {
+    try {
+      _cached = require(pkg);
+      return _cached;
+    } catch (err) {
+      _loadError = err;
+    }
+  } else {
+    _loadError = new Error(`Unsupported platform: ${key}`);
+  }
+
+  _cached = null;
+  return null;
+}
+
+/**
+ * Check whether the native engine is available on this platform.
+ */
+export function isNativeAvailable() {
+  return loadNative() !== null;
+}
+
+/**
+ * Return the native module or throw if not available.
+ */
+export function getNative() {
+  const mod = loadNative();
+  if (!mod) {
+    throw new Error(
+      `Native codegraph-core not available: ${_loadError?.message || 'unknown error'}. ` +
+        'Install the platform package or use --engine wasm.',
+    );
+  }
+  return mod;
+}