@optave/codegraph 3.1.3 → 3.1.5

package/src/{mcp.js → mcp/tool-registry.js}

@@ -1,16 +1,11 @@
 /**
- * MCP (Model Context Protocol) server for codegraph.
- * Exposes codegraph queries as tools that AI coding assistants can call.
+ * MCP tool schema registry.
  *
- * Requires: npm install @modelcontextprotocol/sdk
+ * Owns BASE_TOOLS, LIST_REPOS_TOOL, buildToolList(), and the backward-compatible TOOLS export.
  */
 
-import { createRequire } from 'node:module';
-import { AST_NODE_KINDS } from './ast.js';
-import { findCycles } from './cycles.js';
-import { findDbPath } from './db.js';
-import { MCP_DEFAULTS, MCP_MAX_LIMIT } from './paginate.js';
-import { diffImpactMermaid, EVERY_EDGE_KIND, EVERY_SYMBOL_KIND, VALID_ROLES } from './queries.js';
+import { EVERY_EDGE_KIND, EVERY_SYMBOL_KIND, VALID_ROLES } from '../domain/queries.js';
+import { AST_NODE_KINDS } from '../features/ast.js';
 
 const REPO_PROP = {
   repo: {
@@ -788,7 +783,7 @@ const LIST_REPOS_TOOL = {
  * @param {boolean} multiRepo - If true, inject `repo` prop into each tool and append `list_repos`
  * @returns {object[]}
  */
-function buildToolList(multiRepo) {
+export function buildToolList(multiRepo) {
   if (!multiRepo) return BASE_TOOLS;
   return [
     ...BASE_TOOLS.map((tool) => ({
@@ -803,668 +798,4 @@ function buildToolList(multiRepo) {
 }
 
 // Backward-compatible export: full multi-repo tool list
-const TOOLS = buildToolList(true);
-
-export { TOOLS, buildToolList };
-
-/**
- * Start the MCP server.
- * This function requires @modelcontextprotocol/sdk to be installed.
- *
- * @param {string} [customDbPath] - Path to a specific graph.db
- * @param {object} [options]
- * @param {boolean} [options.multiRepo] - Enable multi-repo access (default: false)
- * @param {string[]} [options.allowedRepos] - Restrict access to these repo names only
- */
-export async function startMCPServer(customDbPath, options = {}) {
-  const { allowedRepos } = options;
-  const multiRepo = options.multiRepo || !!allowedRepos;
-  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' +
-        'Install it with: npm install @modelcontextprotocol/sdk',
-    );
-    process.exit(1);
-  }
-
-  // Connect transport FIRST so the server can receive the client's
-  // `initialize` request while heavy modules (queries, better-sqlite3)
-  // are still loading.  These are lazy-loaded on the first tool call
-  // and cached for subsequent calls.
-  let _queries;
-  let _Database;
-
-  async function getQueries() {
-    if (!_queries) {
-      _queries = await import('./queries.js');
-    }
-    return _queries;
-  }
-
-  function getDatabase() {
-    if (!_Database) {
-      const require = createRequire(import.meta.url);
-      _Database = require('better-sqlite3');
-    }
-    return _Database;
-  }
-
-  const server = new Server(
-    { name: 'codegraph', version: '1.0.0' },
-    { capabilities: { tools: {} } },
-  );
-
-  server.setRequestHandler(ListToolsRequestSchema, async () => ({
-    tools: buildToolList(multiRepo),
-  }));
-
-  server.setRequestHandler(CallToolRequestSchema, async (request) => {
-    const { name, arguments: args } = request.params;
-    try {
-      const {
-        impactAnalysisData,
-        moduleMapData,
-        fileDepsData,
-        exportsData,
-        fnDepsData,
-        fnImpactData,
-        pathData,
-        contextData,
-        childrenData,
-        explainData,
-        whereData,
-        diffImpactData,
-        listFunctionsData,
-        rolesData,
-      } = await getQueries();
-      const Database = getDatabase();
-      if (!multiRepo && args.repo) {
-        throw new Error(
-          'Multi-repo access is disabled. Restart with `codegraph mcp --multi-repo` to access other repositories.',
-        );
-      }
-      if (!multiRepo && name === 'list_repos') {
-        throw new Error(
-          'Multi-repo access is disabled. Restart with `codegraph mcp --multi-repo` to list repositories.',
-        );
-      }
-
-      let dbPath = customDbPath || undefined;
-      if (args.repo) {
-        if (allowedRepos && !allowedRepos.includes(args.repo)) {
-          throw new Error(`Repository "${args.repo}" is not in the allowed repos list.`);
-        }
-        const { resolveRepoDbPath } = await import('./registry.js');
-        const resolved = resolveRepoDbPath(args.repo);
-        if (!resolved)
-          throw new Error(
-            `Repository "${args.repo}" not found in registry or its database is missing.`,
-          );
-        dbPath = resolved;
-      }
-
-      let result;
-      switch (name) {
-        case 'query': {
-          const qMode = args.mode || 'deps';
-          if (qMode === 'path') {
-            if (!args.to) {
-              result = { error: 'path mode requires a "to" argument' };
-              break;
-            }
-            result = pathData(args.name, args.to, dbPath, {
-              maxDepth: args.depth ?? 10,
-              edgeKinds: args.edge_kinds,
-              reverse: args.reverse,
-              fromFile: args.from_file,
-              toFile: args.to_file,
-              kind: args.kind,
-              noTests: args.no_tests,
-            });
-          } else {
-            result = fnDepsData(args.name, dbPath, {
-              depth: args.depth,
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.query, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          }
-          break;
-        }
-        case 'path':
-          result = pathData(args.from, args.to, dbPath, {
-            maxDepth: args.depth ?? 10,
-            edgeKinds: args.edge_kinds,
-            fromFile: args.from_file,
-            toFile: args.to_file,
-            noTests: args.no_tests,
-          });
-          break;
-        case 'file_deps':
-          result = fileDepsData(args.file, dbPath, {
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.file_deps, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        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,
-          });
-          break;
-        case 'impact_analysis':
-          result = impactAnalysisData(args.file, dbPath, {
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.impact_analysis, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          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, { 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,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.fn_impact, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          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,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.context, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        case 'symbol_children':
-          result = childrenData(args.name, dbPath, {
-            file: args.file,
-            kind: args.kind,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.context, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        case 'where':
-          result = whereData(args.target, dbPath, {
-            file: args.file_mode,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.where, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        case 'diff_impact':
-          if (args.format === 'mermaid') {
-            result = diffImpactMermaid(dbPath, {
-              staged: args.staged,
-              ref: args.ref,
-              depth: args.depth,
-              noTests: args.no_tests,
-            });
-          } else {
-            result = diffImpactData(dbPath, {
-              staged: args.staged,
-              ref: args.ref,
-              depth: args.depth,
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.diff_impact, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          }
-          break;
-        case 'semantic_search': {
-          const mode = args.mode || 'hybrid';
-          const searchOpts = {
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.semantic_search, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-            minScore: args.min_score,
-          };
-
-          if (mode === 'keyword') {
-            const { ftsSearchData } = await import('./embedder.js');
-            result = ftsSearchData(args.query, dbPath, searchOpts);
-            if (result === null) {
-              return {
-                content: [
-                  {
-                    type: 'text',
-                    text: 'No FTS5 index found. Run `codegraph embed` to build the keyword index.',
-                  },
-                ],
-                isError: true,
-              };
-            }
-          } else if (mode === 'semantic') {
-            const { searchData } = await import('./embedder.js');
-            result = await searchData(args.query, dbPath, searchOpts);
-            if (result === null) {
-              return {
-                content: [
-                  {
-                    type: 'text',
-                    text: 'Semantic search unavailable. Run `codegraph embed` first.',
-                  },
-                ],
-                isError: true,
-              };
-            }
-          } else {
-            // hybrid (default) — falls back to semantic if no FTS5
-            const { hybridSearchData, searchData } = await import('./embedder.js');
-            result = await hybridSearchData(args.query, dbPath, searchOpts);
-            if (result === null) {
-              result = await searchData(args.query, dbPath, searchOpts);
-              if (result === null) {
-                return {
-                  content: [
-                    {
-                      type: 'text',
-                      text: 'Semantic search unavailable. Run `codegraph embed` first.',
-                    },
-                  ],
-                  isError: true,
-                };
-              }
-            }
-          }
-          break;
-        }
-        case 'export_graph': {
-          const {
-            exportDOT,
-            exportGraphML,
-            exportGraphSON,
-            exportJSON,
-            exportMermaid,
-            exportNeo4jCSV,
-          } = await import('./export.js');
-          const db = new Database(findDbPath(dbPath), { readonly: true });
-          const fileLevel = args.file_level !== false;
-          const exportLimit = args.limit
-            ? Math.min(args.limit, MCP_MAX_LIMIT)
-            : MCP_DEFAULTS.export_graph;
-          switch (args.format) {
-            case 'dot':
-              result = exportDOT(db, { fileLevel, limit: exportLimit });
-              break;
-            case 'mermaid':
-              result = exportMermaid(db, { fileLevel, limit: exportLimit });
-              break;
-            case 'json':
-              result = exportJSON(db, {
-                limit: exportLimit,
-                offset: args.offset ?? 0,
-              });
-              break;
-            case 'graphml':
-              result = exportGraphML(db, { fileLevel, limit: exportLimit });
-              break;
-            case 'graphson':
-              result = exportGraphSON(db, {
-                fileLevel,
-                limit: exportLimit,
-                offset: args.offset ?? 0,
-              });
-              break;
-            case 'neo4j':
-              result = exportNeo4jCSV(db, { fileLevel, limit: exportLimit });
-              break;
-            default:
-              db.close();
-              return {
-                content: [
-                  {
-                    type: 'text',
-                    text: `Unknown format: ${args.format}. Use dot, mermaid, json, graphml, graphson, or neo4j.`,
-                  },
-                ],
-                isError: true,
-              };
-          }
-          db.close();
-          break;
-        }
-        case 'list_functions':
-          result = listFunctionsData(dbPath, {
-            file: args.file,
-            pattern: args.pattern,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.list_functions, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        case 'node_roles':
-          result = rolesData(dbPath, {
-            role: args.role,
-            file: args.file,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.node_roles, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        case 'structure': {
-          const { structureData } = await import('./structure.js');
-          result = structureData(dbPath, {
-            directory: args.directory,
-            depth: args.depth,
-            sort: args.sort,
-            full: args.full,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.structure, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        }
-        case 'co_changes': {
-          const { coChangeData, coChangeTopData } = await import('./cochange.js');
-          result = args.file
-            ? coChangeData(args.file, dbPath, {
-                limit: Math.min(args.limit ?? MCP_DEFAULTS.co_changes, MCP_MAX_LIMIT),
-                offset: args.offset ?? 0,
-                minJaccard: args.min_jaccard,
-                noTests: args.no_tests,
-              })
-            : coChangeTopData(dbPath, {
-                limit: Math.min(args.limit ?? MCP_DEFAULTS.co_changes, MCP_MAX_LIMIT),
-                offset: args.offset ?? 0,
-                minJaccard: args.min_jaccard,
-                noTests: args.no_tests,
-              });
-          break;
-        }
-        case 'execution_flow': {
-          if (args.list) {
-            const { listEntryPointsData } = await import('./flow.js');
-            result = listEntryPointsData(dbPath, {
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.execution_flow, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          } else {
-            if (!args.name) {
-              result = { error: 'Provide a name or set list=true' };
-              break;
-            }
-            const { flowData } = await import('./flow.js');
-            result = flowData(args.name, dbPath, {
-              depth: args.depth,
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.execution_flow, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          }
-          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, {
-            target: args.name,
-            file: args.file,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.complexity, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-            sort: args.sort,
-            aboveThreshold: args.above_threshold,
-            health: args.health,
-            noTests: args.no_tests,
-            kind: args.kind,
-          });
-          break;
-        }
-        case 'communities': {
-          const { communitiesData } = await import('./communities.js');
-          result = communitiesData(dbPath, {
-            functions: args.functions,
-            resolution: args.resolution,
-            drift: args.drift,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.communities, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        }
-        case 'code_owners': {
-          const { ownersData } = await import('./owners.js');
-          result = ownersData(dbPath, {
-            file: args.file,
-            owner: args.owner,
-            boundary: args.boundary,
-            kind: args.kind,
-            noTests: args.no_tests,
-          });
-          break;
-        }
-        case 'audit': {
-          if (args.quick) {
-            result = explainData(args.target, dbPath, {
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.explain, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          } else {
-            const { auditData } = await import('./audit.js');
-            result = auditData(args.target, dbPath, {
-              depth: args.depth,
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-            });
-          }
-          break;
-        }
-        case 'batch_query': {
-          const { batchData } = await import('./batch.js');
-          result = batchData(args.command, args.targets, dbPath, {
-            depth: args.depth,
-            file: args.file,
-            kind: args.kind,
-            noTests: args.no_tests,
-          });
-          break;
-        }
-        case 'triage': {
-          if (args.level === 'file' || args.level === 'directory') {
-            const { hotspotsData } = await import('./structure.js');
-            const TRIAGE_TO_HOTSPOT = {
-              risk: 'fan-in',
-              complexity: 'density',
-              churn: 'coupling',
-              mi: 'fan-in',
-            };
-            const metric = TRIAGE_TO_HOTSPOT[args.sort] ?? args.sort;
-            result = hotspotsData(dbPath, {
-              metric,
-              level: args.level,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.hotspots, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-              noTests: args.no_tests,
-            });
-          } else {
-            const { triageData } = await import('./triage.js');
-            result = triageData(dbPath, {
-              sort: args.sort,
-              minScore: args.min_score,
-              role: args.role,
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-              weights: args.weights,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.triage, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          }
-          break;
-        }
-        case 'branch_compare': {
-          const { branchCompareData, branchCompareMermaid } = await import('./branch-compare.js');
-          const bcData = await branchCompareData(args.base, args.target, {
-            depth: args.depth,
-            noTests: args.no_tests,
-          });
-          result = args.format === 'mermaid' ? branchCompareMermaid(bcData) : bcData;
-          break;
-        }
-        case 'cfg': {
-          const { cfgData, cfgToDOT, cfgToMermaid } = await import('./cfg.js');
-          const cfgResult = cfgData(args.name, dbPath, {
-            file: args.file,
-            kind: args.kind,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.query, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          if (args.format === 'dot') {
-            result = { text: cfgToDOT(cfgResult) };
-          } else if (args.format === 'mermaid') {
-            result = { text: cfgToMermaid(cfgResult) };
-          } else {
-            result = cfgResult;
-          }
-          break;
-        }
-        case 'dataflow': {
-          const dfMode = args.mode || 'edges';
-          if (dfMode === 'impact') {
-            const { dataflowImpactData } = await import('./dataflow.js');
-            result = dataflowImpactData(args.name, dbPath, {
-              depth: args.depth,
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.fn_impact, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          } else {
-            const { dataflowData } = await import('./dataflow.js');
-            result = dataflowData(args.name, dbPath, {
-              file: args.file,
-              kind: args.kind,
-              noTests: args.no_tests,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.query, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          }
-          break;
-        }
-        case 'check': {
-          const isDiffMode = args.ref || args.staged;
-
-          if (!isDiffMode && !args.rules) {
-            // No ref, no staged → run manifesto rules on whole codebase
-            const { manifestoData } = await import('./manifesto.js');
-            result = manifestoData(dbPath, {
-              file: args.file,
-              noTests: args.no_tests,
-              kind: args.kind,
-              limit: Math.min(args.limit ?? MCP_DEFAULTS.manifesto, MCP_MAX_LIMIT),
-              offset: args.offset ?? 0,
-            });
-          } else {
-            const { checkData } = await import('./check.js');
-            const checkResult = checkData(dbPath, {
-              ref: args.ref,
-              staged: args.staged,
-              cycles: args.cycles,
-              blastRadius: args.blast_radius,
-              signatures: args.signatures,
-              boundaries: args.boundaries,
-              depth: args.depth,
-              noTests: args.no_tests,
-            });
-
-            if (args.rules) {
-              const { manifestoData } = await import('./manifesto.js');
-              const manifestoResult = manifestoData(dbPath, {
-                file: args.file,
-                noTests: args.no_tests,
-                kind: args.kind,
-                limit: Math.min(args.limit ?? MCP_DEFAULTS.manifesto, MCP_MAX_LIMIT),
-                offset: args.offset ?? 0,
-              });
-              result = { check: checkResult, manifesto: manifestoResult };
-            } else {
-              result = checkResult;
-            }
-          }
-          break;
-        }
-        case 'ast_query': {
-          const { astQueryData } = await import('./ast.js');
-          result = astQueryData(args.pattern, dbPath, {
-            kind: args.kind,
-            file: args.file,
-            noTests: args.no_tests,
-            limit: Math.min(args.limit ?? MCP_DEFAULTS.ast_query, MCP_MAX_LIMIT),
-            offset: args.offset ?? 0,
-          });
-          break;
-        }
-        case 'list_repos': {
-          const { listRepos, pruneRegistry } = await import('./registry.js');
-          pruneRegistry();
-          let repos = listRepos();
-          if (allowedRepos) {
-            repos = repos.filter((r) => allowedRepos.includes(r.name));
-          }
-          result = { repos };
-          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);
-}
+export const TOOLS = buildToolList(true);