code-auditor-mcp
Version:
Multi-language code quality auditor with MCP server - Analyze TypeScript, JavaScript, and Go code for SOLID principles, DRY violations, security patterns, and more
1,024 lines (1,023 loc) • 54.5 kB
JavaScript
#!/usr/bin/env node
// MCP servers use stdio: stdout for protocol messages, stderr for logging
// All log messages must go to stderr to avoid interfering with MCP protocol
import chalk from 'chalk';
console.error(chalk.blue('[INFO]'), 'Loading modules...');
import { Server } from '@modelcontextprotocol/sdk/server/index.js';
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
import { CallToolRequestSchema, ListToolsRequestSchema, InitializeRequestSchema } from '@modelcontextprotocol/sdk/types.js';
console.error(chalk.blue('[INFO]'), 'MCP SDK loaded');
import { createAuditRunner } from './auditRunner.js';
console.error(chalk.blue('[INFO]'), 'Audit runner loaded');
import { searchFunctions, findDefinition, syncFileIndex, getDatabase } from './codeIndexService.js';
import { CodeMapGenerator } from './services/CodeMapGenerator.js';
import { analyzeDocumentation } from './analyzers/documentationAnalyzer.js';
import { ConfigGeneratorFactory } from './generators/ConfigGeneratorFactory.js';
import { DEFAULT_SERVER_URL, IS_DEV_MODE } from './constants.js';
console.error(chalk.blue('[INFO]'), 'Code index service loaded');
import path from 'node:path';
import fs from 'node:fs/promises';
import { createWriteStream } from 'node:fs';
import { CodeIndexDB } from './codeIndexDB.js';
console.error(chalk.blue('[INFO]'), 'All modules loaded successfully');
// Set up file logging (only in development mode)
let logStream = null;
if (IS_DEV_MODE) {
const logFilePath = path.join(process.cwd(), 'mcp-server.log');
logStream = createWriteStream(logFilePath, { flags: 'a' });
console.error(chalk.blue('[INFO]'), `Development mode: Logging to file: ${logFilePath}`);
}
// Save original console methods
const originalConsoleLog = console.log;
const originalConsoleError = console.error;
// Override console.log (with conditional file logging)
console.log = (...args) => {
if (logStream) {
const message = args.map(arg => typeof arg === 'object' ? JSON.stringify(arg) : String(arg)).join(' ');
const timestamp = new Date().toISOString();
logStream.write(`[${timestamp}] [LOG] ${message}\n`);
}
originalConsoleLog.apply(console, args);
};
// Override console.error (with conditional file logging)
console.error = (...args) => {
if (logStream) {
const message = args.map(arg => typeof arg === 'object' ? JSON.stringify(arg) : String(arg)).join(' ');
const timestamp = new Date().toISOString();
logStream.write(`[${timestamp}] [ERROR] ${message}\n`);
}
originalConsoleError.apply(console, args);
};
const tools = [
// Core Audit Tools
{
name: 'audit',
description: 'Run a comprehensive code audit on files or directories, including React component analysis',
parameters: [
{
name: 'path',
type: 'string',
required: false,
description: 'The file or directory path to audit (defaults to current directory)',
default: process.cwd(),
},
{
name: 'analyzers',
type: 'array',
required: false,
description: 'List of analyzers to run (solid, dry, documentation, react, data-access)',
default: ['solid', 'dry', 'documentation', 'react', 'data-access'],
},
{
name: 'minSeverity',
type: 'string',
required: false,
description: 'Minimum severity level to report',
default: 'warning',
enum: ['info', 'warning', 'critical'],
},
{
name: 'indexFunctions',
type: 'boolean',
required: false,
description: 'Automatically index functions during audit',
default: true,
},
{
name: 'analyzerConfigs',
type: 'object',
required: false,
description: 'Analyzer-specific configuration overrides (e.g., SOLID thresholds, DRY settings)',
},
{
name: 'generateCodeMap',
type: 'boolean',
required: false,
description: 'Generate and return a human-readable code map as part of the audit results',
default: true,
},
],
},
{
name: 'audit_health',
description: 'Quick health check of a codebase with key metrics',
parameters: [
{
name: 'path',
type: 'string',
required: false,
description: 'The directory path to check',
default: process.cwd(),
},
{
name: 'threshold',
type: 'number',
required: false,
description: 'Health score threshold (0-100) for pass/fail',
default: 70,
},
{
name: 'indexFunctions',
type: 'boolean',
required: false,
description: 'Automatically index functions during health check',
default: true,
},
{
name: 'analyzerConfigs',
type: 'object',
required: false,
description: 'Analyzer-specific configuration overrides (e.g., SOLID thresholds, DRY settings)',
},
{
name: 'generateCodeMap',
type: 'boolean',
required: false,
description: 'Generate and return a human-readable code map as part of the health check results',
default: true,
},
],
},
// Code Index Tools
{
name: 'search_code',
description: 'Search indexed functions and React components with natural language queries. Supports operators: entity:component, component:functional|class|memo|forwardRef, hook:useState|useEffect|etc, prop:propName, dep:packageName, dependency:lodash, uses:express, calls:functionName, calledby:functionName, dependents-of:functionName, used-by:functionName, depends-on:module, imports-from:file, unused-imports, dead-imports, type:fileType, file:path, lang:language, complexity:1-10, jsdoc:true|false',
parameters: [
{
name: 'query',
type: 'string',
required: true,
description: 'Search query with natural language and/or operators. Examples: "Button component:functional", "entity:component hook:useState", "render prop:onClick", "dep:lodash", "calls:validateUser", "unused-imports", "dependents-of:authenticate"',
},
{
name: 'filters',
type: 'object',
required: false,
description: 'Optional filters (language, filePath, dependencies, componentType, entityType, searchMode). Set searchMode to "content" to search within function bodies, "metadata" for names/signatures only, or "both" for combined search',
},
{
name: 'limit',
type: 'number',
required: false,
description: 'Maximum results to return',
default: 50,
},
{
name: 'offset',
type: 'number',
required: false,
description: 'Offset for pagination',
default: 0,
},
],
},
{
name: 'find_definition',
description: 'Find the exact definition of a specific function or React component',
parameters: [
{
name: 'name',
type: 'string',
required: true,
description: 'Function or component name to find',
},
{
name: 'filePath',
type: 'string',
required: false,
description: 'Optional file path to narrow search',
},
],
},
{
name: 'sync_index',
description: 'Synchronize, cleanup, or reset the code index',
parameters: [
{
name: 'mode',
type: 'string',
required: false,
description: 'Operation mode: sync (update), cleanup (remove stale), or reset (clear all)',
default: 'sync',
enum: ['sync', 'cleanup', 'reset'],
},
{
name: 'path',
type: 'string',
required: false,
description: 'Optional specific path to sync',
},
],
},
// AI Configuration Tool
{
name: 'generate_ai_config',
description: 'Generate configuration files for AI coding assistants',
parameters: [
{
name: 'tools',
type: 'array',
required: true,
description: 'AI tools to configure (cursor, continue, copilot, claude, zed, windsurf, cody, aider, cline, pearai)',
},
{
name: 'outputDir',
type: 'string',
required: false,
description: 'Output directory for configuration files',
default: '.',
},
],
},
// Workflow Guide Tool
{
name: 'get_workflow_guide',
description: 'Get recommended workflows and best practices for using code auditor tools effectively',
parameters: [
{
name: 'scenario',
type: 'string',
required: false,
description: 'Specific scenario: initial-setup, react-development, code-review, find-patterns, maintenance. Leave empty to see all.',
},
],
},
// Analyzer Configuration Tools
{
name: 'set_analyzer_config',
description: 'Set or update analyzer configuration that persists across audit runs',
parameters: [
{
name: 'analyzerName',
type: 'string',
required: true,
description: 'The analyzer to configure (solid, dry, security, etc.)',
},
{
name: 'config',
type: 'object',
required: true,
description: 'Configuration object for the analyzer (e.g., thresholds, rules)',
},
{
name: 'projectPath',
type: 'string',
required: false,
description: 'Optional project path for project-specific config (defaults to global)',
},
],
},
{
name: 'get_analyzer_config',
description: 'Get current configuration for an analyzer',
parameters: [
{
name: 'analyzerName',
type: 'string',
required: false,
description: 'Specific analyzer name, or omit to get all configs',
},
{
name: 'projectPath',
type: 'string',
required: false,
description: 'Optional project path to get project-specific config',
},
],
},
{
name: 'reset_analyzer_config',
description: 'Reset analyzer configuration to defaults',
parameters: [
{
name: 'analyzerName',
type: 'string',
required: false,
description: 'Specific analyzer to reset, or omit to reset all',
},
{
name: 'projectPath',
type: 'string',
required: false,
description: 'Optional project path to reset only project-specific config',
},
],
},
{
name: 'get_code_map_section',
description: 'Retrieve a specific section of a previously generated code map',
parameters: [
{
name: 'mapId',
type: 'string',
required: true,
description: 'The map ID returned from a previous audit with code map generation',
},
{
name: 'sectionType',
type: 'string',
required: true,
description: 'The section type to retrieve (e.g., overview, files, dependencies, documentation)',
},
],
},
{
name: 'list_code_map_sections',
description: 'List all available sections for a code map',
parameters: [
{
name: 'mapId',
type: 'string',
required: true,
description: 'The map ID returned from a previous audit',
},
],
},
];
async function startMcpServer() {
console.error(chalk.blue('[INFO]'), 'Starting Code Auditor MCP Server...');
console.error(chalk.blue('[DEBUG]'), `Process PID: ${process.pid}`);
console.error(chalk.blue('[DEBUG]'), `Node version: ${process.version}`);
console.error(chalk.blue('[DEBUG]'), `Working directory: ${process.cwd()}`);
const server = new Server({
name: 'code-auditor',
}, {
capabilities: {
tools: {},
},
});
console.error(chalk.blue('[INFO]'), 'Server instance created');
console.error(chalk.blue('[DEBUG]'), 'Server capabilities:', JSON.stringify({ tools: {} }));
// Add error handler
server.onerror = (error) => {
console.error(chalk.red('[ERROR]'), 'Server error occurred:', error);
console.error(chalk.red('[ERROR]'), 'Error stack:', error.stack);
};
// Handle tool listing
console.error(chalk.blue('[INFO]'), 'Setting up request handlers...');
server.setRequestHandler(ListToolsRequestSchema, async (request) => {
console.error(chalk.blue('[DEBUG]'), 'Received ListTools request');
console.error(chalk.blue('[DEBUG]'), 'Request:', JSON.stringify(request, null, 2));
console.error(chalk.blue('[DEBUG]'), 'Handling ListTools request');
const response = {
tools: tools.map(tool => ({
name: tool.name,
description: tool.description,
inputSchema: {
type: 'object',
properties: tool.parameters.reduce((acc, param) => {
acc[param.name] = {
type: param.type,
description: param.description,
...(param.default !== undefined && { default: param.default }),
...(param.enum && { enum: param.enum }),
};
return acc;
}, {}),
required: tool.parameters.filter(p => p.required).map(p => p.name),
},
})),
};
console.error(chalk.blue('[DEBUG]'), `Returning ${response.tools.length} tools`);
console.error(chalk.blue('[DEBUG]'), 'ListTools response:', JSON.stringify(response, null, 2));
return response;
});
console.error(chalk.blue('[INFO]'), `Registered ${tools.length} tools`);
// Add handler for initialize request
server.setRequestHandler(InitializeRequestSchema, async (request) => {
console.error(chalk.blue('[DEBUG]'), 'Received initialize request');
console.error(chalk.blue('[DEBUG]'), 'Request:', JSON.stringify(request, null, 2));
const response = {
protocolVersion: '2024-11-05',
capabilities: {
tools: {},
},
serverInfo: {
name: 'code-auditor',
version: '1.0.0',
},
};
console.error(chalk.blue('[DEBUG]'), 'Initialize response:', JSON.stringify(response, null, 2));
return response;
});
// Handle tool execution
server.setRequestHandler(CallToolRequestSchema, async (request) => {
console.error(chalk.blue('[DEBUG]'), 'Received CallTool request');
console.error(chalk.blue('[DEBUG]'), 'Request:', JSON.stringify(request, null, 2));
const { name, arguments: args } = request.params;
console.error(chalk.blue('[DEBUG]'), `Tool name: ${name}`);
console.error(chalk.blue('[DEBUG]'), `Tool arguments:`, JSON.stringify(args, null, 2));
console.error(chalk.blue('[DEBUG]'), `Handling CallTool request for: ${name}`);
try {
console.error(chalk.blue('[DEBUG]'), `Starting tool execution for: ${name}`);
let result;
switch (name) {
case 'audit': {
const auditPath = path.resolve(args.path || process.cwd());
const indexFunctions = args.indexFunctions !== false; // Default true
const generateCodeMap = args.generateCodeMap || false;
// Check if path is a file or directory
const stats = await fs.stat(auditPath).catch(() => null);
const isFile = stats?.isFile() || false;
// Get stored analyzer configs from database
const db = CodeIndexDB.getInstance();
await db.initialize();
const storedConfigs = await db.getAllAnalyzerConfigs(auditPath);
// Merge stored configs with any provided configs
const analyzerConfigs = {
...storedConfigs,
...(args.analyzerConfigs || {})
};
const options = {
projectRoot: isFile ? path.dirname(auditPath) : auditPath,
enabledAnalyzers: args.analyzers || ['solid', 'dry', 'documentation', 'react', 'data-access'],
minSeverity: (args.minSeverity || 'warning'),
verbose: false,
indexFunctions,
...(isFile && { includePaths: [auditPath] }),
...(Object.keys(analyzerConfigs).length > 0 && { analyzerConfigs }),
};
console.error(chalk.blue('[DEBUG]'), 'Creating audit runner with options:', JSON.stringify(options, null, 2));
const runner = createAuditRunner(options);
console.error(chalk.blue('[DEBUG]'), 'About to run audit...');
const auditResult = await runner.run();
console.error(chalk.blue('[DEBUG]'), 'Audit completed, result summary:', auditResult.summary);
// Handle function indexing if enabled and functions were collected
let indexingResult = null;
if (indexFunctions && auditResult.metadata.fileToFunctionsMap) {
try {
const syncStats = { added: 0, updated: 0, removed: 0 };
// Sync each file's functions to handle additions, updates, and removals
for (const [filePath, functions] of Object.entries(auditResult.metadata.fileToFunctionsMap)) {
const fileStats = await syncFileIndex(filePath, functions);
syncStats.added += fileStats.added;
syncStats.updated += fileStats.updated;
syncStats.removed += fileStats.removed;
}
indexingResult = {
success: true,
registered: syncStats.added + syncStats.updated,
failed: 0,
syncStats
};
console.error(chalk.blue('[INFO]'), `Synced functions: ${syncStats.added} added, ${syncStats.updated} updated, ${syncStats.removed} removed`);
}
catch (error) {
console.error(chalk.yellow('[WARN]'), 'Failed to sync functions:', error);
}
}
// Generate code map if requested and functions were indexed
let codeMapResult = null;
if (generateCodeMap && indexingResult && indexingResult.success) {
try {
const mapGenerator = new CodeMapGenerator();
const mapOptions = {
includeComplexity: true,
includeDocumentation: true,
includeDependencies: true,
includeUsage: false,
groupByDirectory: true,
maxDepth: 10,
showUnusedImports: true,
minComplexity: 7,
};
// Generate documentation metrics
let documentation = undefined;
try {
// Use existing file discovery from audit result
const files = Object.keys(auditResult.metadata.fileToFunctionsMap || {});
if (files.length > 0) {
const docResult = await analyzeDocumentation(files);
documentation = docResult.metrics;
}
}
catch (docError) {
console.error(chalk.yellow('[WARN]'), 'Failed to analyze documentation:', docError);
}
// Use paginated code map generation
const paginatedResult = await mapGenerator.generatePaginatedCodeMap(auditPath, {
...mapOptions,
includeDocumentation: !!documentation
});
codeMapResult = {
success: true,
mapId: paginatedResult.mapId,
summary: paginatedResult.summary,
quickPreview: paginatedResult.quickPreview,
sections: paginatedResult.summary.sectionsAvailable,
documentationCoverage: documentation?.coverageScore
};
console.error(chalk.blue('[INFO]'), `Generated paginated code map: ${paginatedResult.summary.stats.totalFiles} files, ${paginatedResult.summary.totalSections} sections`);
}
catch (error) {
console.error(chalk.yellow('[WARN]'), 'Failed to generate code map:', error);
codeMapResult = {
success: false,
error: error instanceof Error ? error.message : 'Failed to generate code map'
};
}
}
// Format for MCP
result = {
summary: {
totalViolations: auditResult.summary.totalViolations,
criticalIssues: auditResult.summary.criticalIssues,
warnings: auditResult.summary.warnings,
suggestions: auditResult.summary.suggestions,
filesAnalyzed: auditResult.metadata.filesAnalyzed,
executionTime: auditResult.metadata.auditDuration,
healthScore: calculateHealthScore(auditResult),
},
violations: getAllViolations(auditResult).slice(0, 100), // Limit to first 100
recommendations: auditResult.recommendations,
...(indexingResult && { functionIndexing: indexingResult }),
...(codeMapResult && { codeMap: codeMapResult }),
};
break;
}
case 'audit_health': {
const auditPath = path.resolve(args.path || process.cwd());
const threshold = args.threshold || 70;
const indexFunctions = args.indexFunctions !== false; // Default true
const generateCodeMap = args.generateCodeMap !== false; // Default true
// Get stored analyzer configs from database
const db = CodeIndexDB.getInstance();
await db.initialize();
const storedConfigs = await db.getAllAnalyzerConfigs(auditPath);
// Merge stored configs with any provided configs
const analyzerConfigs = {
...storedConfigs,
...(args.analyzerConfigs || {})
};
const runner = createAuditRunner({
projectRoot: auditPath,
enabledAnalyzers: ['solid', 'dry', 'documentation', 'react', 'data-access'],
minSeverity: 'warning',
verbose: false,
indexFunctions, // Pass the flag to the runner
...(Object.keys(analyzerConfigs).length > 0 && { analyzerConfigs }),
});
const auditResult = await runner.run();
const healthScore = calculateHealthScore(auditResult);
// Handle function indexing if enabled and functions were collected
let indexingResult = null;
if (indexFunctions && auditResult.metadata.fileToFunctionsMap) {
try {
const syncStats = { added: 0, updated: 0, removed: 0 };
// Sync each file's functions to handle additions, updates, and removals
for (const [filePath, functions] of Object.entries(auditResult.metadata.fileToFunctionsMap)) {
const fileStats = await syncFileIndex(filePath, functions);
syncStats.added += fileStats.added;
syncStats.updated += fileStats.updated;
syncStats.removed += fileStats.removed;
}
indexingResult = {
success: true,
registered: syncStats.added + syncStats.updated,
failed: 0,
syncStats
};
console.error(chalk.blue('[INFO]'), `Synced functions: ${syncStats.added} added, ${syncStats.updated} updated, ${syncStats.removed} removed`);
}
catch (error) {
console.error(chalk.yellow('[WARN]'), 'Failed to sync functions:', error);
}
}
// Generate code map if requested and functions were indexed
let codeMapResult = null;
if (generateCodeMap && indexingResult && indexingResult.success) {
try {
const mapGenerator = new CodeMapGenerator();
const mapOptions = {
includeComplexity: true,
includeDocumentation: true,
includeDependencies: true,
includeUsage: false,
groupByDirectory: true,
maxDepth: 8, // Slightly smaller for health check
showUnusedImports: true,
minComplexity: 7,
};
// Generate documentation metrics
let documentation = undefined;
try {
// Use existing file discovery from audit result
const files = Object.keys(auditResult.metadata.fileToFunctionsMap || {});
if (files.length > 0) {
const docResult = await analyzeDocumentation(files);
documentation = docResult.metrics;
}
}
catch (docError) {
console.error(chalk.yellow('[WARN]'), 'Failed to analyze documentation:', docError);
}
// Use paginated code map generation for health check too
const paginatedResult = await mapGenerator.generatePaginatedCodeMap(auditPath, {
...mapOptions,
includeDocumentation: !!documentation
});
codeMapResult = {
success: true,
mapId: paginatedResult.mapId,
summary: paginatedResult.summary,
quickPreview: paginatedResult.quickPreview,
sections: paginatedResult.summary.sectionsAvailable,
documentationCoverage: documentation?.coverageScore
};
console.error(chalk.blue('[INFO]'), `Generated paginated code map: ${paginatedResult.summary.stats.totalFiles} files, ${paginatedResult.summary.totalSections} sections`);
}
catch (error) {
console.error(chalk.yellow('[WARN]'), 'Failed to generate code map:', error);
codeMapResult = {
success: false,
error: error instanceof Error ? error.message : 'Failed to generate code map'
};
}
}
result = {
healthScore,
threshold,
passed: healthScore >= threshold,
status: healthScore >= threshold ? 'healthy' : 'needs-attention',
metrics: {
filesAnalyzed: auditResult.metadata.filesAnalyzed,
totalViolations: auditResult.summary.totalViolations,
criticalViolations: auditResult.summary.criticalIssues,
warningViolations: auditResult.summary.warnings,
},
recommendation: getHealthRecommendation(healthScore, auditResult),
...(indexingResult && { functionIndexing: indexingResult }),
...(codeMapResult && { codeMap: codeMapResult }),
};
break;
}
case 'search_code': {
const query = args.query;
const filters = args.filters;
const limit = args.limit || 50;
const offset = args.offset || 0;
// Allow empty query to return all results
if (query !== undefined && typeof query !== 'string') {
throw new Error('query must be a string');
}
result = await searchFunctions({
query,
filters,
limit,
offset
});
break;
}
case 'find_definition': {
const name = args.name;
const filePath = args.filePath;
if (!name || typeof name !== 'string') {
throw new Error('name must be a non-empty string');
}
const definition = await findDefinition(name, filePath);
result = definition || { error: 'Function not found' };
break;
}
case 'generate_ai_config': {
const tools = args.tools;
const serverUrl = args.serverUrl || DEFAULT_SERVER_URL;
const outputDir = args.outputDir || '.';
const overwrite = args.overwrite || false;
if (!Array.isArray(tools) || tools.length === 0) {
throw new Error('tools parameter must be a non-empty array');
}
const factory = new ConfigGeneratorFactory(serverUrl);
const generatedFiles = [];
const errors = [];
for (const tool of tools) {
try {
const generator = factory.createGenerator(tool);
if (!generator) {
errors.push(`Unknown tool: ${tool}`);
continue;
}
const config = generator.generateConfig();
const outputPath = path.resolve(outputDir, config.filename);
// Check if file exists
let fileExists = false;
try {
await fs.access(outputPath);
fileExists = true;
}
catch {
// File doesn't exist
}
if (fileExists && !overwrite) {
errors.push(`File already exists: ${config.filename} (use overwrite: true to replace)`);
continue;
}
// Ensure directory exists
await fs.mkdir(path.dirname(outputPath), { recursive: true });
// Write main config file
await fs.writeFile(outputPath, config.content);
generatedFiles.push(config.filename);
// Write additional files if any
if (config.additionalFiles) {
for (const additionalFile of config.additionalFiles) {
const additionalPath = path.resolve(outputDir, additionalFile.filename);
await fs.mkdir(path.dirname(additionalPath), { recursive: true });
await fs.writeFile(additionalPath, additionalFile.content);
generatedFiles.push(additionalFile.filename);
}
}
}
catch (error) {
errors.push(`Failed to generate config for ${tool}: ${error instanceof Error ? error.message : 'Unknown error'}`);
}
}
result = {
success: errors.length === 0,
generatedFiles,
errors: errors.length > 0 ? errors : undefined,
totalRequested: tools.length,
totalGenerated: generatedFiles.length
};
break;
}
case 'get_workflow_guide': {
const scenario = args.scenario;
const { getWorkflowGuide, getWorkflowTips } = await import('./mcp-tools/workflowGuide.js');
try {
const workflows = getWorkflowGuide(scenario);
const tips = getWorkflowTips();
result = {
success: true,
...(scenario ? { workflow: workflows } : { workflows }),
tips
};
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Unknown error',
availableScenarios: ['initial-setup', 'react-development', 'code-review', 'find-patterns', 'maintenance', 'analyzer-configuration']
};
}
break;
}
case 'sync_index': {
const mode = args.mode || 'sync';
const targetPath = args.path;
const db = CodeIndexDB.getInstance();
await db.initialize();
switch (mode) {
case 'cleanup': {
const cleanupResult = await db.bulkCleanup();
result = {
mode: 'cleanup',
success: true,
scannedFiles: cleanupResult.scannedCount,
removedEntries: cleanupResult.removedCount,
removedFiles: cleanupResult.removedFiles,
errors: cleanupResult.errors,
message: `Cleaned up ${cleanupResult.removedCount} entries from ${cleanupResult.removedFiles.length} deleted files`
};
break;
}
case 'reset': {
await db.clearIndex();
result = {
mode: 'reset',
success: true,
message: 'Index cleared successfully'
};
break;
}
case 'sync':
default: {
if (targetPath) {
// Sync specific file
const syncResult = await db.synchronizeFile(path.resolve(targetPath));
result = {
mode: 'sync',
success: true,
path: targetPath,
...(syncResult || { message: 'File not found' })
};
}
else {
// Deep sync all files
const syncResult = await db.deepSync();
result = {
mode: 'sync',
success: true,
syncedFiles: syncResult.syncedFiles,
addedFunctions: syncResult.addedFunctions,
updatedFunctions: syncResult.updatedFunctions,
removedFunctions: syncResult.removedFunctions,
errors: syncResult.errors,
message: `Synced ${syncResult.syncedFiles} files: ${syncResult.addedFunctions} added, ${syncResult.updatedFunctions} updated, ${syncResult.removedFunctions} removed`
};
}
break;
}
}
break;
}
case 'set_analyzer_config': {
const analyzerName = args.analyzerName;
const config = args.config;
const projectPath = args.projectPath;
const db = CodeIndexDB.getInstance();
await db.initialize();
try {
await db.storeAnalyzerConfig(analyzerName, config, {
projectPath,
isGlobal: !projectPath
});
result = {
success: true,
message: `Configuration for ${analyzerName} analyzer has been saved${projectPath ? ` for project ${projectPath}` : ' globally'}`,
analyzer: analyzerName,
scope: projectPath ? 'project' : 'global',
config
};
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Failed to save configuration'
};
}
break;
}
case 'get_analyzer_config': {
const analyzerName = args.analyzerName;
const projectPath = args.projectPath;
const db = CodeIndexDB.getInstance();
await db.initialize();
try {
if (analyzerName) {
const config = await db.getAnalyzerConfig(analyzerName, projectPath);
result = {
success: true,
analyzer: analyzerName,
config: config || null,
scope: projectPath ? 'project' : 'global',
message: config ? 'Configuration found' : 'No custom configuration found, using defaults'
};
}
else {
const configs = await db.getAllAnalyzerConfigs(projectPath);
result = {
success: true,
configs,
scope: projectPath ? 'project' : 'global',
message: `Found ${Object.keys(configs).length} analyzer configurations`
};
}
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Failed to get configuration'
};
}
break;
}
case 'reset_analyzer_config': {
const analyzerName = args.analyzerName;
const projectPath = args.projectPath;
const db = CodeIndexDB.getInstance();
await db.initialize();
try {
if (analyzerName) {
const deleted = await db.deleteAnalyzerConfig(analyzerName, {
projectPath,
isGlobal: !projectPath
});
result = {
success: deleted,
message: deleted
? `Configuration for ${analyzerName} analyzer has been reset${projectPath ? ` for project ${projectPath}` : ' globally'}`
: `No configuration found for ${analyzerName} analyzer`,
analyzer: analyzerName,
scope: projectPath ? 'project' : 'global'
};
}
else {
await db.resetAnalyzerConfigs(projectPath);
result = {
success: true,
message: projectPath
? `All project-specific configurations for ${projectPath} have been reset`
: 'All analyzer configurations have been reset to defaults',
scope: projectPath ? 'project' : 'global'
};
}
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Failed to reset configuration'
};
}
break;
}
case 'get_code_map_section': {
const mapId = args.mapId;
const sectionType = args.sectionType;
if (!mapId || !sectionType) {
result = {
success: false,
error: 'Both mapId and sectionType are required'
};
break;
}
try {
const db = await getDatabase();
const section = await db.getCodeMapSection(mapId, sectionType);
if (section) {
result = {
success: true,
mapId,
sectionType,
content: section.content,
metadata: section.metadata
};
}
else {
result = {
success: false,
error: `Section '${sectionType}' not found for map '${mapId}'`
};
}
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Failed to retrieve code map section'
};
}
break;
}
case 'list_code_map_sections': {
const mapId = args.mapId;
if (!mapId) {
result = {
success: false,
error: 'mapId is required'
};
break;
}
try {
const db = await getDatabase();
const sections = await db.listCodeMapSections(mapId);
result = {
success: true,
mapId,
sections,
totalSections: sections.length
};
}
catch (error) {
result = {
success: false,
error: error instanceof Error ? error.message : 'Failed to list code map sections'
};
}
break;
}
default:
throw new Error(`Unknown tool: ${name}`);
}
const response = {
content: [
{
type: 'text',
text: JSON.stringify(result, null, 2),
},
],
};
console.error(chalk.blue('[DEBUG]'), `Tool ${name} executed successfully`);
console.error(chalk.blue('[DEBUG]'), 'CallTool response:', JSON.stringify(response, null, 2).substring(0, 500) + '...');
return response;
}
catch (error) {
console.error(chalk.red('[ERROR]'), `Tool ${name} execution failed:`, error);
console.error(chalk.red('[ERROR]'), 'Error stack:', error instanceof Error ? error.stack : 'No stack trace');
return {
content: [
{
type: 'text',
text: JSON.stringify({
error: error instanceof Error ? error.message : 'Unknown error',
tool: name,
}),
},
],
isError: true,
};
}
});
console.error(chalk.blue('[INFO]'), 'Request handlers configured');
const transport = new StdioServerTransport();
console.error(chalk.blue('[INFO]'), 'Creating stdio transport...');
// Add transport event handlers if available
if (transport.onclose) {
transport.onclose = () => {