code-auditor-mcp 1.19.0 → 1.21.0

package/dist/mcp-tools-shared.js

@@ -0,0 +1,1161 @@
+/**
+ * Shared MCP Tools Logic
+ *
+ * This contains all the tool definitions and handlers that are shared
+ * between stdio and HTTP/UI MCP server implementations.
+ */
+import { createAuditRunner } from './auditRunner.js';
+import { searchFunctions, findDefinition, syncFileIndex } from './codeIndexService.js';
+import { CodeMapGenerator } from './services/CodeMapGenerator.js';
+import { analyzeDocumentation } from './analyzers/documentationAnalyzer.js';
+import { CodeIndexDB } from './codeIndexDB.js';
+import path from 'node:path';
+import fs from 'node:fs/promises';
+import chalk from 'chalk';
+export 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',
+            },
+        ],
+    },
+    // Database Schema Management Tools
+    {
+        name: 'generate_schema_discovery_sql',
+        description: 'Generate SQL queries for LLMs to extract database schema information automatically',
+        parameters: [
+            {
+                name: 'databaseType',
+                type: 'string',
+                required: true,
+                enum: ['postgresql', 'mysql', 'sqlite', 'sqlserver', 'oracle'],
+                description: 'Type of database to generate queries for',
+            },
+            {
+                name: 'includeIndexes',
+                type: 'boolean',
+                required: false,
+                default: true,
+                description: 'Include queries to discover indexes',
+            },
+            {
+                name: 'includeConstraints',
+                type: 'boolean',
+                required: false,
+                default: true,
+                description: 'Include queries to discover foreign key constraints',
+            },
+            {
+                name: 'specificTables',
+                type: 'array',
+                required: false,
+                description: 'Limit discovery to specific table names (optional)',
+            },
+        ],
+    },
+    {
+        name: 'get_schemas',
+        description: 'List all loaded database schemas with their metadata',
+        parameters: [],
+    },
+    {
+        name: 'search_schema',
+        description: 'Search for tables, columns, or relationships in loaded schemas',
+        parameters: [
+            {
+                name: 'query',
+                type: 'string',
+                required: true,
+                description: 'Search query for table/column names or descriptions',
+            },
+            {
+                name: 'schemaId',
+                type: 'string',
+                required: false,
+                description: 'Limit search to specific schema ID',
+            },
+            {
+                name: 'searchType',
+                type: 'string',
+                required: false,
+                enum: ['tables', 'columns', 'relationships', 'all'],
+                default: 'all',
+                description: 'Type of schema elements to search',
+            },
+        ],
+    },
+    {
+        name: 'analyze_schema_usage',
+        description: 'Analyze how database tables are used in the codebase',
+        parameters: [
+            {
+                name: 'path',
+                type: 'string',
+                required: false,
+                default: '.',
+                description: 'Path to analyze for schema usage patterns',
+            },
+            {
+                name: 'schemaId',
+                type: 'string',
+                required: false,
+                description: 'Schema ID to analyze against',
+            },
+            {
+                name: 'includeUsagePatterns',
+                type: 'boolean',
+                required: false,
+                default: true,
+                description: 'Whether to include detailed usage patterns',
+            },
+        ],
+    },
+    {
+        name: 'find_table_usage',
+        description: 'Find all functions that interact with a specific database table',
+        parameters: [
+            {
+                name: 'tableName',
+                type: 'string',
+                required: true,
+                description: 'Name of the table to find usage for',
+            },
+            {
+                name: 'usageType',
+                type: 'string',
+                required: false,
+                enum: ['query', 'insert', 'update', 'delete', 'reference', 'all'],
+                default: 'all',
+                description: 'Type of table usage to find',
+            },
+        ],
+    },
+    {
+        name: 'validate_schema_consistency',
+        description: 'Validate schema consistency and find potential issues',
+        parameters: [
+            {
+                name: 'schemaId',
+                type: 'string',
+                required: false,
+                description: 'Schema ID to validate (validates all if not specified)',
+            },
+            {
+                name: 'checkCircularDeps',
+                type: 'boolean',
+                required: false,
+                default: true,
+                description: 'Check for circular dependencies',
+            },
+            {
+                name: 'checkNamingConventions',
+                type: 'boolean',
+                required: false,
+                default: true,
+                description: 'Check naming convention compliance',
+            },
+        ],
+    },
+];
+// UI-specific tools (only available in UI mode)
+export const uiTools = [
+    {
+        name: 'audit_dashboard',
+        description: 'Generates an interactive dashboard with detailed audit findings, code maps, and remediation options.',
+        parameters: [
+            {
+                name: 'path',
+                type: 'string',
+                required: false,
+                description: 'Path to audit',
+                default: '.',
+            },
+            {
+                name: 'analyzers',
+                type: 'array',
+                required: false,
+                description: 'Analyzers to run',
+                default: ['solid', 'dry', 'documentation', 'react', 'data-access'],
+            },
+            {
+                name: 'minSeverity',
+                type: 'string',
+                required: false,
+                description: 'Minimum severity level',
+                default: 'warning',
+                enum: ['info', 'warning', 'critical'],
+            },
+        ],
+    },
+    {
+        name: 'code_map_viewer',
+        description: 'Generates an interactive, navigable code map with file structure, complexity analysis, and documentation coverage.',
+        parameters: [
+            {
+                name: 'path',
+                type: 'string',
+                required: false,
+                description: 'Path to analyze',
+                default: '.',
+            },
+        ],
+    },
+];
+/**
+ * Shared tool handler implementations
+ */
+export class ToolHandlers {
+    static async handleAudit(args) {
+        const auditPath = path.resolve(args.path || process.cwd());
+        const indexFunctions = args.indexFunctions !== false; // Default true
+        const generateCodeMap = args.generateCodeMap !== false; // Default true
+        // 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 }),
+        };
+        const runner = createAuditRunner(options);
+        const auditResult = await runner.run();
+        // 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 {
+                    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
+        return {
+            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: ToolHandlers.calculateHealthScore(auditResult),
+            },
+            violations: ToolHandlers.getAllViolations(auditResult).slice(0, 100), // Limit to first 100
+            recommendations: auditResult.recommendations,
+            ...(indexingResult && { functionIndexing: indexingResult }),
+            ...(codeMapResult && { codeMap: codeMapResult }),
+        };
+    }
+    static async handleAuditHealth(args) {
+        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,
+            ...(Object.keys(analyzerConfigs).length > 0 && { analyzerConfigs }),
+        });
+        const auditResult = await runner.run();
+        const healthScore = ToolHandlers.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 {
+                    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'
+                };
+            }
+        }
+        return {
+            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: ToolHandlers.getHealthRecommendation(healthScore, auditResult),
+            ...(indexingResult && { functionIndexing: indexingResult }),
+            ...(codeMapResult && { codeMap: codeMapResult }),
+        };
+    }
+    // Add all other tool handlers here following the same pattern...
+    // (I'll include key ones for brevity)
+    static async handleSearchCode(args) {
+        const query = args.query;
+        const filters = args.filters;
+        const limit = args.limit || 50;
+        const offset = args.offset || 0;
+        if (query !== undefined && typeof query !== 'string') {
+            throw new Error('query must be a string');
+        }
+        return await searchFunctions({
+            query,
+            filters,
+            limit,
+            offset
+        });
+    }
+    static async handleFindDefinition(args) {
+        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);
+        return definition || { error: 'Function not found' };
+    }
+    // Schema Management Tool Handlers
+    static async handleGenerateSchemaDiscoverySQL(args) {
+        const databaseType = args.databaseType;
+        const includeIndexes = args.includeIndexes !== false;
+        const includeConstraints = args.includeConstraints !== false;
+        const specificTables = args.specificTables;
+        if (!databaseType || typeof databaseType !== 'string') {
+            throw new Error('databaseType must be a non-empty string');
+        }
+        const sqlQueries = ToolHandlers.generateSchemaDiscoveryQueries(databaseType, includeIndexes, includeConstraints, specificTables);
+        return {
+            databaseType,
+            queries: sqlQueries,
+            instructions: [
+                "Execute these SQL queries against your database",
+                "Copy the results and use create_schema_from_sql_result to import them",
+                "Each query discovers different aspects of your database schema",
+                "Run them in order and collect all results"
+            ],
+            nextStep: "Use create_schema_from_sql_result with the query results"
+        };
+    }
+    static async handleCreateSchemaFromSqlResult(args) {
+        const schemaName = args.schemaName;
+        const databaseType = args.databaseType;
+        const tablesData = args.tablesData;
+        const columnsData = args.columnsData;
+        const constraintsData = args.constraintsData;
+        if (!schemaName || !databaseType || !tablesData || !columnsData) {
+            throw new Error('schemaName, databaseType, tablesData, and columnsData are required');
+        }
+        try {
+            const schema = ToolHandlers.buildSchemaFromSqlData(schemaName, databaseType, tablesData, columnsData, constraintsData);
+            const db = CodeIndexDB.getInstance();
+            await db.initialize();
+            const schemaId = await db.storeSchema(schema);
+            return {
+                success: true,
+                schemaId,
+                schemaName: schema.name,
+                stats: {
+                    databaseCount: schema.databases.length,
+                    tableCount: schema.databases.reduce((acc, db) => acc + db.tables.length, 0),
+                    columnCount: schema.databases.reduce((acc, db) => acc + db.tables.reduce((tAcc, table) => tAcc + table.columns.length, 0), 0)
+                },
+                message: `Schema '${schema.name}' created from SQL results`
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to create schema from SQL results: ${error instanceof Error ? error.message : 'Unknown error'}`
+            };
+        }
+    }
+    static async handleAddTableManually(args) {
+        const schemaName = args.schemaName;
+        const tableName = args.tableName;
+        const columns = args.columns;
+        const databaseType = args.databaseType || 'postgresql';
+        if (!schemaName || !tableName || !columns || !Array.isArray(columns)) {
+            throw new Error('schemaName, tableName, and columns array are required');
+        }
+        try {
+            const db = CodeIndexDB.getInstance();
+            await db.initialize();
+            // Get existing schema or create new one
+            const existingSchemas = await db.getAllSchemas();
+            let schema = existingSchemas.find(s => s.schema.name === schemaName)?.schema;
+            if (!schema) {
+                // Create new schema
+                schema = {
+                    version: "1.0.0",
+                    name: schemaName,
+                    description: `Manually created schema: ${schemaName}`,
+                    databases: [{
+                            name: 'default',
+                            type: databaseType,
+                            tables: []
+                        }]
+                };
+            }
+            // Add the table
+            const newTable = {
+                name: tableName,
+                type: 'table',
+                columns: columns.map(col => ({
+                    name: col.name,
+                    type: col.type,
+                    nullable: col.nullable !== false,
+                    primaryKey: col.primaryKey || false,
+                    unique: col.unique || false,
+                    indexed: col.indexed || false,
+                    description: col.description
+                })),
+                references: [],
+                indexes: []
+            };
+            schema.databases[0].tables.push(newTable);
+            // Store updated schema
+            const schemaId = await db.storeSchema(schema);
+            return {
+                success: true,
+                schemaId,
+                tableName,
+                columnCount: columns.length,
+                message: `Table '${tableName}' added to schema '${schemaName}'`
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to add table: ${error instanceof Error ? error.message : 'Unknown error'}`
+            };
+        }
+    }
+    static async handleListSchemas(args) {
+        try {
+            const db = CodeIndexDB.getInstance();
+            await db.initialize();
+            const schemas = await db.getAllSchemas();
+            const stats = await db.getSchemaStats();
+            return {
+                schemas: schemas.map(s => ({
+                    schemaId: s.schemaId,
+                    name: s.schema.name,
+                    description: s.schema.description,
+                    databaseCount: s.schema.databases.length,
+                    tableCount: s.metadata.tableCount,
+                    relationshipCount: s.metadata.relationshipCount,
+                    indexedAt: s.metadata.indexedAt
+                })),
+                totalStats: stats
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to list schemas: ${error instanceof Error ? error.message : 'Unknown error'}`
+            };
+        }
+    }
+    static async handleSearchSchemaElements(args) {
+        const query = args.query;
+        const elementType = args.elementType || 'all';
+        if (!query || typeof query !== 'string') {
+            throw new Error('query must be a non-empty string');
+        }
+        try {
+            const db = CodeIndexDB.getInstance();
+            await db.initialize();
+            const schemas = (await db.getAllSchemas()).map(s => s.schema);
+            const results = [];
+            const queryLower = query.toLowerCase();
+            for (const schema of schemas) {
+                for (const database of schema.databases) {
+                    for (const table of database.tables) {
+                        // Search tables
+                        if ((elementType === 'all' || elementType === 'tables') &&
+                            (table.name.toLowerCase().includes(queryLower) ||
+                                table.description?.toLowerCase().includes(queryLower))) {
+                            results.push({
+                                type: 'table',
+                                tableName: table.name,
+                                databaseName: database.name,
+                                schemaName: schema.name,
+                                description: table.description,
+                                columnCount: table.columns.length,
+                                tags: table.tags
+                            });
+                        }
+                        // Search columns
+                        if (elementType === 'all' || elementType === 'columns') {
+                            for (const column of table.columns) {
+                                if (column.name.toLowerCase().includes(queryLower) ||
+                                    column.description?.toLowerCase().includes(queryLower) ||
+                                    column.type.toLowerCase().includes(queryLower)) {
+                                    results.push({
+                                        type: 'column',
+                                        columnName: column.name,
+                                        tableName: table.name,
+                                        databaseName: database.name,
+                                        schemaName: schema.name,
+                                        columnType: column.type,
+                                        description: column.description,
+                                        nullable: column.nullable,
+                                        primaryKey: column.primaryKey
+                                    });
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+            return {
+                query,
+                elementType,
+                resultCount: results.length,
+                results
+            };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: `Failed to search schema elements: ${error instanceof Error ? error.message : 'Unknown error'}`
+            };
+        }
+    }
+    // Schema Helper Functions
+    static generateSchemaDiscoveryQueries(databaseType, includeIndexes, includeConstraints, specificTables) {
+        const queries = [];
+        const tableFilter = specificTables && specificTables.length > 0
+            ? `WHERE table_name IN (${specificTables.map(t => `'${t}'`).join(', ')})`
+            : '';
+        switch (databaseType.toLowerCase()) {
+            case 'postgresql':
+                queries.push({
+                    name: 'tables',
+                    sql: `SELECT table_name, table_type, table_schema 
+                FROM information_schema.tables 
+                WHERE table_schema NOT IN ('information_schema', 'pg_catalog') ${tableFilter}
+                ORDER BY table_schema, table_name;`,
+                    description: 'Get all tables and views'
+                });
+                queries.push({
+                    name: 'columns',
+                    sql: `SELECT table_name, column_name, data_type, is_nullable, column_default, 
+                       character_maximum_length, numeric_precision, numeric_scale
+                FROM information_schema.columns 
+                WHERE table_schema NOT IN ('information_schema', 'pg_catalog') ${tableFilter}
+                ORDER BY table_name, ordinal_position;`,
+                    description: 'Get all columns with types and constraints'
+                });
+                if (includeConstraints) {
+                    queries.push({
+                        name: 'foreign_keys',
+                        sql: `SELECT tc.table_name, kcu.column_name, ccu.table_name AS foreign_table_name,
+                         ccu.column_name AS foreign_column_name, rc.delete_rule, rc.update_rule
+                  FROM information_schema.table_constraints AS tc 
+                  JOIN information_schema.key_column_usage AS kcu ON tc.constraint_name = kcu.constraint_name
+                  JOIN information_schema.constraint_column_usage AS ccu ON ccu.constraint_name = tc.constraint_name
+                  JOIN information_schema.referential_constraints AS rc ON tc.constraint_name = rc.constraint_name
+                  WHERE tc.constraint_type = 'FOREIGN KEY' ${tableFilter.replace('table_name', 'tc.table_name')}
+                  ORDER BY tc.table_name, kcu.column_name;`,
+                        description: 'Get foreign key relationships'
+                    });
+                }
+                if (includeIndexes) {
+                    queries.push({
+                        name: 'indexes',
+                        sql: `SELECT tablename, indexname, indexdef 
+                  FROM pg_indexes 
+                  WHERE schemaname NOT IN ('information_schema', 'pg_catalog') ${tableFilter.replace('table_name', 'tablename')}
+                  ORDER BY tablename, indexname;`,
+                        description: 'Get all indexes'
+                    });
+                }
+                break;
+            case 'mysql':
+                queries.push({
+                    name: 'tables',
+                    sql: `SELECT table_name, table_type, table_schema 
+                FROM information_schema.tables 
+                WHERE table_schema = DATABASE() ${tableFilter}
+                ORDER BY table_name;`,
+                    description: 'Get all tables and views'
+                });
+                queries.push({
+                    name: 'columns',
+                    sql: `SELECT table_name, column_name, data_type, is_nullable, column_default,
+                       character_maximum_length, numeric_precision, numeric_scale,
+                       column_key, extra
+                FROM information_schema.columns 
+                WHERE table_schema = DATABASE() ${tableFilter}
+                ORDER BY table_name, ordinal_position;`,
+                    description: 'Get all columns with types and constraints'
+                });
+                if (includeConstraints) {
+                    queries.push({
+                        name: 'foreign_keys',
+                        sql: `SELECT table_name, column_name, referenced_table_name, referenced_column_name,
+                         delete_rule, update_rule
+                  FROM information_schema.key_column_usage 
+                  WHERE table_schema = DATABASE() AND referenced_table_name IS NOT NULL ${tableFilter}
+                  ORDER BY table_name, column_name;`,
+                        description: 'Get foreign key relationships'
+                    });
+                }
+                break;
+            case 'sqlite':
+                queries.push({
+                    name: 'tables',
+                    sql: `SELECT name as table_name, type as table_type 
+                FROM sqlite_master 
+                WHERE type IN ('table', 'view') AND name NOT LIKE 'sqlite_%'
+                ORDER BY name;`,
+                    description: 'Get all tables and views'
+                });
+                queries.push({
+                    name: 'table_info',
+                    sql: `-- Run this for each table: PRAGMA table_info(table_name);
+                -- This will give you column information for each table`,
+                    description: 'Get column information (run PRAGMA table_info for each table)'
+                });
+                break;
+        }
+        return queries;
+    }
+    static buildSchemaFromSqlData(schemaName, databaseType, tablesData, columnsData, constraintsData) {
+        // Build schema from SQL results
+        const tables = [];
+        const tableMap = new Map();
+        // Process tables
+        const tableRows = Array.isArray(tablesData) ? tablesData : tablesData.rows || [];
+        for (const row of tableRows) {
+            const tableName = row.table_name || row.TABLE_NAME;
+            if (!tableMap.has(tableName)) {
+                tableMap.set(tableName, {
+                    name: tableName,
+                    type: 'table',
+                    columns: [],
+                    references: [],
+                    indexes: []
+                });
+            }
+        }
+        // Process columns
+        const columnRows = Array.isArray(columnsData) ? columnsData : columnsData.rows || [];
+        for (const row of columnRows) {
+            const tableName = row.table_name || row.TABLE_NAME;
+            const table = tableMap.get(tableName);
+            if (table) {
+                table.columns.push({
+                    name: row.column_name || row.COLUMN_NAME,
+                    type: row.data_type || row.DATA_TYPE,
+                    nullable: (row.is_nullable || row.IS_NULLABLE) === 'YES',
+                    primaryKey: (row.column_key || row.COLUMN_KEY) === 'PRI',
+                    defaultValue: row.column_default || row.COLUMN_DEFAULT,
+                    length: row.character_maximum_length || row.CHARACTER_MAXIMUM_LENGTH,
+                    precision: row.numeric_precision || row.NUMERIC_PRECISION,
+                    scale: row.numeric_scale || row.NUMERIC_SCALE
+                });
+            }
+        }
+        // Process constraints if provided
+        if (constraintsData) {
+            const constraintRows = Array.isArray(constraintsData) ? constraintsData : constraintsData.rows || [];
+            for (const row of constraintRows) {
+                const tableName = row.table_name || row.TABLE_NAME;
+                const table = tableMap.get(tableName);
+                if (table) {
+                    table.references.push({
+                        foreignKey: row.column_name || row.COLUMN_NAME,
+                        referencedTable: row.foreign_table_name || row.REFERENCED_TABLE_NAME,
+                        referencedColumn: row.foreign_column_name || row.REFERENCED_COLUMN_NAME,
+                        onDelete: row.delete_rule || row.DELETE_RULE,
+                        onUpdate: row.update_rule || row.UPDATE_RULE
+                    });
+                }
+            }
+        }
+        return {
+            version: "1.0.0",
+            name: schemaName,
+            description: `Schema discovered from ${databaseType} database`,
+            databases: [{
+                    name: 'main',
+                    type: databaseType,
+                    tables: Array.from(tableMap.values())
+                }],
+            metadata: {
+                createdAt: new Date().toISOString(),
+                source: 'sql-discovery'
+            }
+        };
+    }
+    // Helper functions
+    static getAllViolations(result) {
+        const violations = [];
+        for (const [analyzerName, analyzerResult] of Object.entries(result.analyzerResults)) {
+            for (const violation of analyzerResult.violations) {
+                violations.push({
+                    ...violation,
+                    analyzer: analyzerName,
+                });
+            }
+        }
+        return violations;
+    }
+    static calculateHealthScore(result) {
+        const filesAnalyzed = result.metadata?.filesAnalyzed || 1;
+        const critical = result.summary.criticalIssues || 0;
+        const warnings = result.summary.warnings || 0;
+        const suggestions = result.summary.suggestions || 0;
+        const weights = {
+            critical: 10,
+            warning: 3,
+            suggestion: 0.5
+        };
+        const weightedViolations = (critical * weights.critical) +
+            (warnings * weights.warning) +
+            (suggestions * weights.suggestion);
+        const violationsPerFile = weightedViolations / filesAnalyzed;
+        let score = 100 - (violationsPerFile * 2);
+        return Math.max(0, Math.round(Math.min(100, score)));
+    }
+    static getHealthRecommendation(score, result) {
+        if (score >= 90)
+            return 'Excellent code health!';
+        if (score >= 70)
+            return 'Good code health with room for improvement';
+        if (result.summary.criticalIssues > 0) {
+            return `Fix ${result.summary.criticalIssues} critical violations first`;
+        }
+        return 'Code health needs attention - run detailed audit';
+    }
+}
+//# sourceMappingURL=mcp-tools-shared.js.map