claude-brain 0.30.2 → 0.30.3

package/src/server/handlers/tools/schemas.ts

@@ -1,179 +1,179 @@
-/**
- * Zod Validation Schemas for Tool Handlers
- * Provides input validation for all MCP tool handlers
- */
-
-import { z } from 'zod'
-
-// ============================================================================
-// Project Context & Management
-// ============================================================================
-
-export const GetProjectContextSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  include_memories: z.boolean().optional().default(true),
-  memory_limit: z.number().int().min(1).max(100).optional().default(5)
-})
-
-export const ListProjectsSchema = z.object({
-  status_filter: z.enum(['active', 'archived', 'planning', 'all']).optional().default('active')
-})
-
-export const CreateProjectSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required')
-    .regex(/^[a-z0-9-]+$/, 'project_name must be lowercase with hyphens only'),
-  description: z.string().optional(),
-  tech_stack: z.array(z.string()).optional().default([]),
-  status: z.enum(['planning', 'active', 'on-hold', 'completed']).optional().default('planning'),
-  tags: z.array(z.string()).optional().default([])
-})
-
-export const InitProjectSchema = z.object({
-  project_path: z.string().optional(),
-  project_name: z.string().optional(),
-  save_to_memory: z.boolean().optional().default(true)
-})
-
-// ============================================================================
-// Decision & Memory
-// ============================================================================
-
-export const RememberDecisionSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  context: z.string().min(1, 'context is required'),
-  decision: z.string().min(1, 'decision is required'),
-  reasoning: z.string().min(1, 'reasoning is required'),
-  alternatives_considered: z.string().optional(),
-  tags: z.array(z.string()).optional().default([])
-})
-
-export const RecallSimilarSchema = z.object({
-  query: z.string().min(1, 'query is required'),
-  project_filter: z.string().optional(),
-  limit: z.number().int().min(1).max(50).optional().default(5),
-  min_similarity: z.number().min(0).max(1).optional().default(0.3)
-})
-
-// ============================================================================
-// Smart Context & Automation
-// ============================================================================
-
-export const SmartContextSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  current_task: z.string().min(1, 'current_task is required'),
-  min_similarity: z.number().min(0).max(1).optional().default(0.3)
-})
-
-export const AutoRememberSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  text: z.string().min(1, 'text is required'),
-  confidence_threshold: z.number().min(0).max(1).optional().default(0.7)
-})
-
-// ============================================================================
-// Code Standards
-// ============================================================================
-
-export const GetCodeStandardsSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  language: z.string().optional()
-})
-
-// ============================================================================
-// Progress Tracking
-// ============================================================================
-
-export const UpdateProgressSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  completed_task: z.string().min(1, 'completed_task is required'),
-  next_steps: z.string().min(1, 'next_steps is required'),
-  notes: z.string().optional()
-})
-
-// ============================================================================
-// Patterns & Corrections
-// ============================================================================
-
-export const RecognizePatternSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  pattern_type: z.enum(['solution', 'anti-pattern', 'best-practice', 'common-issue']),
-  description: z.string().min(1, 'description is required'),
-  example: z.string().optional(),
-  confidence: z.number().min(0).max(1).optional().default(0.8),
-  context: z.string().optional()
-})
-
-export const RecordCorrectionSchema = z.object({
-  project_name: z.string().min(1, 'project_name is required'),
-  original: z.string().min(1, 'original is required'),
-  correction: z.string().min(1, 'correction is required'),
-  reasoning: z.string().min(1, 'reasoning is required'),
-  context: z.string().optional(),
-  confidence: z.number().min(0).max(1).optional().default(0.9)
-})
-
-export const GetPatternsSchema = z.object({
-  project_name: z.string().optional(),
-  pattern_type: z.enum(['solution', 'anti-pattern', 'best-practice', 'common-issue']).optional(),
-  limit: z.number().int().min(1).max(100).optional().default(10),
-  query: z.string().optional()
-})
-
-export const GetCorrectionsSchema = z.object({
-  project_name: z.string().optional(),
-  limit: z.number().int().min(1).max(100).optional().default(10),
-  query: z.string().optional()
-})
-
-// ============================================================================
-// Phase 12 Status
-// ============================================================================
-
-export const GetPhase12StatusSchema = z.object({})
-
-// ============================================================================
-// Phase 16 - Unified Brain Tool
-// ============================================================================
-
-export const BrainSchema = z.object({
-  message: z.string().min(1, 'message is required'),
-  project: z.string().optional(),
-  action: z.enum(['auto', 'store', 'recall', 'update', 'delete']).optional()
-})
-
-// ============================================================================
-// Code Intelligence
-// ============================================================================
-
-export const SearchCodeSchema = z.object({
-  query: z.string().min(1, 'query is required'),
-  project: z.string().min(1, 'project is required'),
-  type: z.enum(['symbols', 'files', 'dependencies']).optional().default('symbols'),
-  file_path: z.string().optional(),
-  limit: z.number().int().min(1).max(100).optional().default(20)
-})
-
-// ============================================================================
-// Type exports for validated inputs
-// ============================================================================
-
-export type ValidatedGetProjectContext = z.infer<typeof GetProjectContextSchema>
-export type ValidatedListProjects = z.infer<typeof ListProjectsSchema>
-export type ValidatedCreateProject = z.infer<typeof CreateProjectSchema>
-export type ValidatedInitProject = z.infer<typeof InitProjectSchema>
-export type ValidatedRememberDecision = z.infer<typeof RememberDecisionSchema>
-export type ValidatedRecallSimilar = z.infer<typeof RecallSimilarSchema>
-export type ValidatedSmartContext = z.infer<typeof SmartContextSchema>
-export type ValidatedAutoRemember = z.infer<typeof AutoRememberSchema>
-export type ValidatedGetCodeStandards = z.infer<typeof GetCodeStandardsSchema>
-export type ValidatedUpdateProgress = z.infer<typeof UpdateProgressSchema>
-export type ValidatedRecognizePattern = z.infer<typeof RecognizePatternSchema>
-export type ValidatedRecordCorrection = z.infer<typeof RecordCorrectionSchema>
-export type ValidatedGetPatterns = z.infer<typeof GetPatternsSchema>
-export type ValidatedGetCorrections = z.infer<typeof GetCorrectionsSchema>
-
-// Phase 16
-export type ValidatedBrain = z.infer<typeof BrainSchema>
-
-// Code Intelligence
-export type ValidatedSearchCode = z.infer<typeof SearchCodeSchema>
+/**
+ * Zod Validation Schemas for Tool Handlers
+ * Provides input validation for all MCP tool handlers
+ */
+
+import { z } from 'zod'
+
+// ============================================================================
+// Project Context & Management
+// ============================================================================
+
+export const GetProjectContextSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  include_memories: z.boolean().optional().default(true),
+  memory_limit: z.number().int().min(1).max(100).optional().default(5)
+})
+
+export const ListProjectsSchema = z.object({
+  status_filter: z.enum(['active', 'archived', 'planning', 'all']).optional().default('active')
+})
+
+export const CreateProjectSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required')
+    .regex(/^[a-z0-9-]+$/, 'project_name must be lowercase with hyphens only'),
+  description: z.string().optional(),
+  tech_stack: z.array(z.string()).optional().default([]),
+  status: z.enum(['planning', 'active', 'on-hold', 'completed']).optional().default('planning'),
+  tags: z.array(z.string()).optional().default([])
+})
+
+export const InitProjectSchema = z.object({
+  project_path: z.string().optional(),
+  project_name: z.string().optional(),
+  save_to_memory: z.boolean().optional().default(true)
+})
+
+// ============================================================================
+// Decision & Memory
+// ============================================================================
+
+export const RememberDecisionSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  context: z.string().min(1, 'context is required'),
+  decision: z.string().min(1, 'decision is required'),
+  reasoning: z.string().min(1, 'reasoning is required'),
+  alternatives_considered: z.string().optional(),
+  tags: z.array(z.string()).optional().default([])
+})
+
+export const RecallSimilarSchema = z.object({
+  query: z.string().min(1, 'query is required'),
+  project_filter: z.string().optional(),
+  limit: z.number().int().min(1).max(50).optional().default(5),
+  min_similarity: z.number().min(0).max(1).optional().default(0.3)
+})
+
+// ============================================================================
+// Smart Context & Automation
+// ============================================================================
+
+export const SmartContextSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  current_task: z.string().min(1, 'current_task is required'),
+  min_similarity: z.number().min(0).max(1).optional().default(0.3)
+})
+
+export const AutoRememberSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  text: z.string().min(1, 'text is required'),
+  confidence_threshold: z.number().min(0).max(1).optional().default(0.7)
+})
+
+// ============================================================================
+// Code Standards
+// ============================================================================
+
+export const GetCodeStandardsSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  language: z.string().optional()
+})
+
+// ============================================================================
+// Progress Tracking
+// ============================================================================
+
+export const UpdateProgressSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  completed_task: z.string().min(1, 'completed_task is required'),
+  next_steps: z.string().min(1, 'next_steps is required'),
+  notes: z.string().optional()
+})
+
+// ============================================================================
+// Patterns & Corrections
+// ============================================================================
+
+export const RecognizePatternSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  pattern_type: z.enum(['solution', 'anti-pattern', 'best-practice', 'common-issue']),
+  description: z.string().min(1, 'description is required'),
+  example: z.string().optional(),
+  confidence: z.number().min(0).max(1).optional().default(0.8),
+  context: z.string().optional()
+})
+
+export const RecordCorrectionSchema = z.object({
+  project_name: z.string().min(1, 'project_name is required'),
+  original: z.string().min(1, 'original is required'),
+  correction: z.string().min(1, 'correction is required'),
+  reasoning: z.string().min(1, 'reasoning is required'),
+  context: z.string().optional(),
+  confidence: z.number().min(0).max(1).optional().default(0.9)
+})
+
+export const GetPatternsSchema = z.object({
+  project_name: z.string().optional(),
+  pattern_type: z.enum(['solution', 'anti-pattern', 'best-practice', 'common-issue']).optional(),
+  limit: z.number().int().min(1).max(100).optional().default(10),
+  query: z.string().optional()
+})
+
+export const GetCorrectionsSchema = z.object({
+  project_name: z.string().optional(),
+  limit: z.number().int().min(1).max(100).optional().default(10),
+  query: z.string().optional()
+})
+
+// ============================================================================
+// Phase 12 Status
+// ============================================================================
+
+export const GetPhase12StatusSchema = z.object({})
+
+// ============================================================================
+// Phase 16 - Unified Brain Tool
+// ============================================================================
+
+export const BrainSchema = z.object({
+  message: z.string().min(1, 'message is required'),
+  project: z.string().optional(),
+  action: z.enum(['auto', 'store', 'recall', 'update', 'delete']).optional()
+})
+
+// ============================================================================
+// Code Intelligence
+// ============================================================================
+
+export const SearchCodeSchema = z.object({
+  query: z.string().min(1, 'query is required'),
+  project: z.string().min(1, 'project is required'),
+  type: z.enum(['symbols', 'files', 'dependencies']).optional().default('symbols'),
+  file_path: z.string().optional(),
+  limit: z.number().int().min(1).max(100).optional().default(20)
+})
+
+// ============================================================================
+// Type exports for validated inputs
+// ============================================================================
+
+export type ValidatedGetProjectContext = z.infer<typeof GetProjectContextSchema>
+export type ValidatedListProjects = z.infer<typeof ListProjectsSchema>
+export type ValidatedCreateProject = z.infer<typeof CreateProjectSchema>
+export type ValidatedInitProject = z.infer<typeof InitProjectSchema>
+export type ValidatedRememberDecision = z.infer<typeof RememberDecisionSchema>
+export type ValidatedRecallSimilar = z.infer<typeof RecallSimilarSchema>
+export type ValidatedSmartContext = z.infer<typeof SmartContextSchema>
+export type ValidatedAutoRemember = z.infer<typeof AutoRememberSchema>
+export type ValidatedGetCodeStandards = z.infer<typeof GetCodeStandardsSchema>
+export type ValidatedUpdateProgress = z.infer<typeof UpdateProgressSchema>
+export type ValidatedRecognizePattern = z.infer<typeof RecognizePatternSchema>
+export type ValidatedRecordCorrection = z.infer<typeof RecordCorrectionSchema>
+export type ValidatedGetPatterns = z.infer<typeof GetPatternsSchema>
+export type ValidatedGetCorrections = z.infer<typeof GetCorrectionsSchema>
+
+// Phase 16
+export type ValidatedBrain = z.infer<typeof BrainSchema>
+
+// Code Intelligence
+export type ValidatedSearchCode = z.infer<typeof SearchCodeSchema>

package/src/server/handlers/tools/search-code.ts

@@ -1,122 +1,122 @@
-/**
- * Search Code Handler
- * Exposes code intelligence (symbol/file/dependency search) as a direct MCP tool
- */
-
-import type { Logger } from 'pino'
-import type { ToolResponse } from '@/tools/types'
-import { getCodeQuery, isServicesInitialized } from '@/server/services'
-import { ToolValidator } from '@/server/utils/validators'
-import { ResponseFormatter } from '@/server/utils/response-formatter'
-import { ErrorHandler } from '@/server/utils/error-handler'
-import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
-import { SearchCodeSchema } from './schemas'
-
-export async function handleSearchCode(
-  args: unknown,
-  logger: Logger
-): Promise<ToolResponse> {
-  try {
-    const input = ToolValidator.validate(args, SearchCodeSchema)
-    const { query, project, type, file_path, limit } = input
-
-    logger.debug({ query, project, type, limit }, 'Searching code index')
-
-    if (!isServicesInitialized()) {
-      throw new McpError(
-        ErrorCode.InternalError,
-        'Services not initialized. The server may still be starting up.'
-      )
-    }
-
-    const codeQuery = getCodeQuery()
-    if (!codeQuery) {
-      return ResponseFormatter.text(
-        'Code intelligence is not available. Run `claude-brain reindex` to index your project first.'
-      )
-    }
-
-    if (type === 'files') {
-      const results = codeQuery.findFiles(query, project)
-      if (results.length === 0) {
-        return ResponseFormatter.text(
-          `No files found matching "${query}" in project "${project}". Try \`claude-brain reindex\` if the project hasn't been indexed.`
-        )
-      }
-
-      const lines = results.map(r => {
-        const parts = [r.filePath]
-        if (r.language) parts.push(`(${r.language})`)
-        if (r.summary) parts.push(`— ${r.summary}`)
-        parts.push(`[${r.symbolCount} symbols, ${r.lineCount} lines]`)
-        return parts.join(' ')
-      })
-
-      return ResponseFormatter.text(
-        `Found ${results.length} file${results.length === 1 ? '' : 's'} matching "${query}":\n\n${lines.join('\n')}`
-      )
-    }
-
-    if (type === 'dependencies') {
-      const targetPath = file_path || query
-      const deps = codeQuery.getDependencies(targetPath, project)
-
-      const parts: string[] = [`Dependencies for ${deps.file}:\n`]
-
-      if (deps.imports.length > 0) {
-        parts.push('**Imports:**')
-        for (const imp of deps.imports) {
-          const names = imp.names.length > 0 ? ` { ${imp.names.join(', ')} }` : ''
-          parts.push(`  → ${imp.file}${names}`)
-        }
-      } else {
-        parts.push('**Imports:** none')
-      }
-
-      parts.push('')
-
-      if (deps.importedBy.length > 0) {
-        parts.push('**Imported by:**')
-        for (const dep of deps.importedBy) {
-          const names = dep.names.length > 0 ? ` { ${dep.names.join(', ')} }` : ''
-          parts.push(`  ← ${dep.file}${names}`)
-        }
-      } else {
-        parts.push('**Imported by:** none')
-      }
-
-      return ResponseFormatter.text(parts.join('\n'))
-    }
-
-    // Default: symbols
-    const results = codeQuery.findSymbols(query, project, limit)
-    if (results.length === 0) {
-      return ResponseFormatter.text(
-        `No symbols found matching "${query}" in project "${project}". Try \`claude-brain reindex\` if the project hasn't been indexed.`
-      )
-    }
-
-    const lines = results.map(r => {
-      const sig = r.signature ? ` ${r.signature}` : ''
-      const lineRange = r.lineEnd ? `${r.lineStart}-${r.lineEnd}` : `${r.lineStart}`
-      const conf = r.confidence < 1.0 ? ` (${(r.confidence * 100).toFixed(0)}%)` : ''
-      return `${r.symbol} (${r.type})${sig} → ${r.filePath}:${lineRange}${conf}`
-    })
-
-    return ResponseFormatter.text(
-      `Found ${results.length} symbol${results.length === 1 ? '' : 's'} matching "${query}":\n\n${lines.join('\n')}`
-    )
-
-  } catch (error) {
-    ErrorHandler.logError(logger, error, { tool: 'search_code', args })
-
-    if (error instanceof McpError) {
-      throw error
-    }
-
-    throw new McpError(
-      ErrorCode.InternalError,
-      `Failed to search code: ${error instanceof Error ? error.message : 'Unknown error'}`
-    )
-  }
-}
+/**
+ * Search Code Handler
+ * Exposes code intelligence (symbol/file/dependency search) as a direct MCP tool
+ */
+
+import type { Logger } from 'pino'
+import type { ToolResponse } from '@/tools/types'
+import { getCodeQuery, isServicesInitialized } from '@/server/services'
+import { ToolValidator } from '@/server/utils/validators'
+import { ResponseFormatter } from '@/server/utils/response-formatter'
+import { ErrorHandler } from '@/server/utils/error-handler'
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js'
+import { SearchCodeSchema } from './schemas'
+
+export async function handleSearchCode(
+  args: unknown,
+  logger: Logger
+): Promise<ToolResponse> {
+  try {
+    const input = ToolValidator.validate(args, SearchCodeSchema)
+    const { query, project, type, file_path, limit } = input
+
+    logger.debug({ query, project, type, limit }, 'Searching code index')
+
+    if (!isServicesInitialized()) {
+      throw new McpError(
+        ErrorCode.InternalError,
+        'Services not initialized. The server may still be starting up.'
+      )
+    }
+
+    const codeQuery = getCodeQuery()
+    if (!codeQuery) {
+      return ResponseFormatter.text(
+        'Code intelligence is not available. Run `claude-brain reindex` to index your project first.'
+      )
+    }
+
+    if (type === 'files') {
+      const results = codeQuery.findFiles(query, project)
+      if (results.length === 0) {
+        return ResponseFormatter.text(
+          `No files found matching "${query}" in project "${project}". Try \`claude-brain reindex\` if the project hasn't been indexed.`
+        )
+      }
+
+      const lines = results.map(r => {
+        const parts = [r.filePath]
+        if (r.language) parts.push(`(${r.language})`)
+        if (r.summary) parts.push(`— ${r.summary}`)
+        parts.push(`[${r.symbolCount} symbols, ${r.lineCount} lines]`)
+        return parts.join(' ')
+      })
+
+      return ResponseFormatter.text(
+        `Found ${results.length} file${results.length === 1 ? '' : 's'} matching "${query}":\n\n${lines.join('\n')}`
+      )
+    }
+
+    if (type === 'dependencies') {
+      const targetPath = file_path || query
+      const deps = codeQuery.getDependencies(targetPath, project)
+
+      const parts: string[] = [`Dependencies for ${deps.file}:\n`]
+
+      if (deps.imports.length > 0) {
+        parts.push('**Imports:**')
+        for (const imp of deps.imports) {
+          const names = imp.names.length > 0 ? ` { ${imp.names.join(', ')} }` : ''
+          parts.push(`  → ${imp.file}${names}`)
+        }
+      } else {
+        parts.push('**Imports:** none')
+      }
+
+      parts.push('')
+
+      if (deps.importedBy.length > 0) {
+        parts.push('**Imported by:**')
+        for (const dep of deps.importedBy) {
+          const names = dep.names.length > 0 ? ` { ${dep.names.join(', ')} }` : ''
+          parts.push(`  ← ${dep.file}${names}`)
+        }
+      } else {
+        parts.push('**Imported by:** none')
+      }
+
+      return ResponseFormatter.text(parts.join('\n'))
+    }
+
+    // Default: symbols
+    const results = codeQuery.findSymbols(query, project, limit)
+    if (results.length === 0) {
+      return ResponseFormatter.text(
+        `No symbols found matching "${query}" in project "${project}". Try \`claude-brain reindex\` if the project hasn't been indexed.`
+      )
+    }
+
+    const lines = results.map(r => {
+      const sig = r.signature ? ` ${r.signature}` : ''
+      const lineRange = r.lineEnd ? `${r.lineStart}-${r.lineEnd}` : `${r.lineStart}`
+      const conf = r.confidence < 1.0 ? ` (${(r.confidence * 100).toFixed(0)}%)` : ''
+      return `${r.symbol} (${r.type})${sig} → ${r.filePath}:${lineRange}${conf}`
+    })
+
+    return ResponseFormatter.text(
+      `Found ${results.length} symbol${results.length === 1 ? '' : 's'} matching "${query}":\n\n${lines.join('\n')}`
+    )
+
+  } catch (error) {
+    ErrorHandler.logError(logger, error, { tool: 'search_code', args })
+
+    if (error instanceof McpError) {
+      throw error
+    }
+
+    throw new McpError(
+      ErrorCode.InternalError,
+      `Failed to search code: ${error instanceof Error ? error.message : 'Unknown error'}`
+    )
+  }
+}