claude-brain 0.15.2 → 0.16.0

package/src/server/index.ts

@@ -1,40 +1,40 @@
-/**
- * Server Module Exports
- */
-
-export { ClaudeBrainMCPServer } from './mcp-server'
-export type {
-  ConnectionState,
-  ServerStatus,
-  ServerEvents,
-  ServerOptions
-} from './types'
-
-// Re-export utilities
-export {
-  ResponseFormatter,
-  ToolValidator,
-  ErrorHandler,
-  RequestContext,
-  Timer,
-  createTimer,
-  generateShortId,
-  type ErrorInfo
-} from './utils'
-
-// Re-export handlers
-export { handleListTools, handleCallTool } from './handlers'
-
-// Re-export services
-export {
-  initializeServices,
-  shutdownServices,
-  getServices,
-  getMemoryService,
-  getVaultService,
-  isServicesInitialized,
-  type Services
-} from './services'
-
-// Phase 6.5: Re-export providers
-export { PromptProvider, ResourceProvider } from './providers'
+/**
+ * Server Module Exports
+ */
+
+export { ClaudeBrainMCPServer } from './mcp-server'
+export type {
+  ConnectionState,
+  ServerStatus,
+  ServerEvents,
+  ServerOptions
+} from './types'
+
+// Re-export utilities
+export {
+  ResponseFormatter,
+  ToolValidator,
+  ErrorHandler,
+  RequestContext,
+  Timer,
+  createTimer,
+  generateShortId,
+  type ErrorInfo
+} from './utils'
+
+// Re-export handlers
+export { handleListTools, handleCallTool } from './handlers'
+
+// Re-export services
+export {
+  initializeServices,
+  shutdownServices,
+  getServices,
+  getMemoryService,
+  getVaultService,
+  isServicesInitialized,
+  type Services
+} from './services'
+
+// Phase 6.5: Re-export providers
+export { PromptProvider, ResourceProvider } from './providers'

package/src/server/mcp-server.ts

@@ -1,283 +1,283 @@
-/**
- * Claude Brain MCP Server
- * Main MCP server implementation using the Model Context Protocol SDK
- * Phase 6.5: Enhanced with system prompts and resources
- */
-
-import { Server } from '@modelcontextprotocol/sdk/server/index.js'
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-  ListPromptsRequestSchema,
-  GetPromptRequestSchema,
-  ListResourcesRequestSchema,
-  ReadResourceRequestSchema
-} from '@modelcontextprotocol/sdk/types.js'
-import type { Logger } from 'pino'
-import type { Config } from '@/config'
-import type { ConnectionState, ServerStatus, ServerOptions, ServerEvents } from './types'
-import { PromptProvider } from './providers/prompts'
-import { ResourceProvider } from './providers/resources'
-
-export class ClaudeBrainMCPServer {
-  private server: Server
-  private logger: Logger
-  private config: Config
-  private isRunning: boolean = false
-  private connectionState: ConnectionState = 'disconnected'
-  private startedAt?: Date
-  private requestCount: number = 0
-  private events: ServerEvents
-
-  // Phase 6.5: Prompt and Resource providers
-  private promptProvider: PromptProvider
-  private resourceProvider: ResourceProvider
-
-  constructor(config: Config, logger: Logger, options: ServerOptions = {}) {
-    this.config = config
-    this.logger = logger.child({ component: 'mcp-server' })
-    this.events = options.events || {}
-
-    // Phase 6.5: Initialize providers
-    this.promptProvider = new PromptProvider(logger)
-    this.resourceProvider = new ResourceProvider(logger)
-
-    // Initialize MCP Server with metadata and enhanced capabilities
-    this.server = new Server(
-      {
-        name: config.serverName,
-        version: config.serverVersion
-      },
-      {
-        capabilities: {
-          tools: {},
-          prompts: {},    // Phase 6.5: Enable prompts
-          resources: {}   // Phase 6.5: Enable resources
-        }
-      }
-    )
-
-    // Setup request handlers
-    this.setupHandlers()
-
-    // Setup error handling
-    this.setupErrorHandling()
-
-    this.logger.debug('MCP server instance created with Phase 6.5 enhancements')
-  }
-
-  /**
-   * Setup MCP request handlers for tools, prompts, and resources
-   */
-  private setupHandlers(): void {
-    // Handle tools/list request
-    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
-      this.logger.debug('Handling tools/list request')
-
-      // Import dynamically to avoid circular dependencies
-      const { handleListTools } = await import('./handlers/list-tools')
-      return await handleListTools()
-    })
-
-    // Handle tools/call request
-    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
-      const toolName = request.params.name
-      const args = request.params.arguments || {}
-
-      this.requestCount++
-      const requestId = `req_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 9)}`
-
-      this.logger.debug({
-        tool: toolName,
-        args,
-        requestId
-      }, 'Handling tools/call request')
-
-      // Emit request event
-      this.events.onRequest?.(toolName, requestId)
-
-      const startTime = Date.now()
-
-      try {
-        // Import dynamically to avoid circular dependencies
-        const { handleCallTool } = await import('./handlers/call-tool')
-        const result = await handleCallTool(toolName, args, this.logger)
-
-        const duration = Date.now() - startTime
-        this.logger.debug({
-          tool: toolName,
-          requestId,
-          durationMs: duration
-        }, 'Tool call completed')
-
-        // Emit response event
-        this.events.onResponse?.(toolName, requestId, duration)
-
-        return result
-      } catch (error) {
-        const duration = Date.now() - startTime
-        this.logger.error({
-          tool: toolName,
-          requestId,
-          durationMs: duration,
-          error: error instanceof Error ? error.message : String(error)
-        }, 'Tool call failed')
-
-        throw error
-      }
-    })
-
-    // Phase 6.5: Handle prompts/list request
-    this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
-      this.logger.debug('Handling prompts/list request')
-      return await this.promptProvider.listPrompts()
-    })
-
-    // Phase 6.5: Handle prompts/get request
-    this.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
-      this.logger.debug({ name: request.params.name }, 'Handling prompts/get request')
-      return await this.promptProvider.getPrompt(request.params.name)
-    })
-
-    // Phase 6.5: Handle resources/list request
-    this.server.setRequestHandler(ListResourcesRequestSchema, async () => {
-      this.logger.debug('Handling resources/list request')
-      return await this.resourceProvider.listResources()
-    })
-
-    // Phase 6.5: Handle resources/read request
-    this.server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
-      this.logger.debug({ uri: request.params.uri }, 'Handling resources/read request')
-      return await this.resourceProvider.readResource(request.params.uri)
-    })
-
-    this.logger.debug('Request handlers configured (tools, prompts, resources)')
-  }
-
-  /**
-   * Setup error handling for the server
-   */
-  private setupErrorHandling(): void {
-    this.server.onerror = (error) => {
-      this.logger.error({
-        error: error instanceof Error ? {
-          message: error.message,
-          stack: error.stack,
-          name: error.name
-        } : error
-      }, 'MCP server error')
-
-      this.events.onError?.(error instanceof Error ? error : new Error(String(error)))
-    }
-  }
-
-  /**
-   * Update connection state and emit events
-   */
-  private setConnectionState(state: ConnectionState): void {
-    const previousState = this.connectionState
-    this.connectionState = state
-
-    if (previousState !== state) {
-      this.logger.info({
-        previousState,
-        newState: state
-      }, 'Connection state changed')
-
-      this.events.onConnectionStateChange?.(state)
-    }
-  }
-
-  /**
-   * Start the MCP server and connect via stdio transport
-   */
-  async start(): Promise<void> {
-    if (this.isRunning) {
-      this.logger.warn('Server already running')
-      return
-    }
-
-    try {
-      this.logger.info('Starting MCP server...')
-      this.setConnectionState('connecting')
-
-      const transport = new StdioServerTransport()
-      await this.server.connect(transport)
-
-      this.isRunning = true
-      this.startedAt = new Date()
-      this.setConnectionState('connected')
-
-      this.logger.info({
-        name: this.config.serverName,
-        version: this.config.serverVersion,
-        startedAt: this.startedAt.toISOString()
-      }, 'MCP server started successfully')
-
-    } catch (error) {
-      this.setConnectionState('disconnected')
-      this.logger.error({
-        error: error instanceof Error ? {
-          message: error.message,
-          stack: error.stack
-        } : error
-      }, 'Failed to start MCP server')
-      throw error
-    }
-  }
-
-  /**
-   * Stop the MCP server gracefully
-   */
-  async stop(): Promise<void> {
-    if (!this.isRunning) {
-      this.logger.debug('Server not running, skip stop')
-      return
-    }
-
-    try {
-      this.logger.info('Stopping MCP server...')
-      await this.server.close()
-
-      this.isRunning = false
-      this.setConnectionState('disconnected')
-
-      this.logger.info({
-        totalRequests: this.requestCount,
-        uptimeMs: this.startedAt ? Date.now() - this.startedAt.getTime() : 0
-      }, 'MCP server stopped')
-
-    } catch (error) {
-      this.logger.error({
-        error: error instanceof Error ? {
-          message: error.message,
-          stack: error.stack
-        } : error
-      }, 'Error stopping MCP server')
-      throw error
-    }
-  }
-
-  /**
-   * Get current server status
-   */
-  getStatus(): ServerStatus {
-    return {
-      running: this.isRunning,
-      name: this.config.serverName,
-      version: this.config.serverVersion,
-      connectionState: this.connectionState,
-      startedAt: this.startedAt,
-      requestCount: this.requestCount,
-      capabilities: ['tools', 'prompts', 'resources']
-    }
-  }
-
-  /**
-   * Health check method
-   */
-  isHealthy(): boolean {
-    return this.isRunning && this.connectionState === 'connected'
-  }
-}
+/**
+ * Claude Brain MCP Server
+ * Main MCP server implementation using the Model Context Protocol SDK
+ * Phase 6.5: Enhanced with system prompts and resources
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js'
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+  ListPromptsRequestSchema,
+  GetPromptRequestSchema,
+  ListResourcesRequestSchema,
+  ReadResourceRequestSchema
+} from '@modelcontextprotocol/sdk/types.js'
+import type { Logger } from 'pino'
+import type { Config } from '@/config'
+import type { ConnectionState, ServerStatus, ServerOptions, ServerEvents } from './types'
+import { PromptProvider } from './providers/prompts'
+import { ResourceProvider } from './providers/resources'
+
+export class ClaudeBrainMCPServer {
+  private server: Server
+  private logger: Logger
+  private config: Config
+  private isRunning: boolean = false
+  private connectionState: ConnectionState = 'disconnected'
+  private startedAt?: Date
+  private requestCount: number = 0
+  private events: ServerEvents
+
+  // Phase 6.5: Prompt and Resource providers
+  private promptProvider: PromptProvider
+  private resourceProvider: ResourceProvider
+
+  constructor(config: Config, logger: Logger, options: ServerOptions = {}) {
+    this.config = config
+    this.logger = logger.child({ component: 'mcp-server' })
+    this.events = options.events || {}
+
+    // Phase 6.5: Initialize providers
+    this.promptProvider = new PromptProvider(logger)
+    this.resourceProvider = new ResourceProvider(logger)
+
+    // Initialize MCP Server with metadata and enhanced capabilities
+    this.server = new Server(
+      {
+        name: config.serverName,
+        version: config.serverVersion
+      },
+      {
+        capabilities: {
+          tools: {},
+          prompts: {},    // Phase 6.5: Enable prompts
+          resources: {}   // Phase 6.5: Enable resources
+        }
+      }
+    )
+
+    // Setup request handlers
+    this.setupHandlers()
+
+    // Setup error handling
+    this.setupErrorHandling()
+
+    this.logger.debug('MCP server instance created with Phase 6.5 enhancements')
+  }
+
+  /**
+   * Setup MCP request handlers for tools, prompts, and resources
+   */
+  private setupHandlers(): void {
+    // Handle tools/list request
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+      this.logger.debug('Handling tools/list request')
+
+      // Import dynamically to avoid circular dependencies
+      const { handleListTools } = await import('./handlers/list-tools')
+      return await handleListTools()
+    })
+
+    // Handle tools/call request
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      const toolName = request.params.name
+      const args = request.params.arguments || {}
+
+      this.requestCount++
+      const requestId = `req_${Date.now().toString(36)}_${Math.random().toString(36).slice(2, 9)}`
+
+      this.logger.debug({
+        tool: toolName,
+        args,
+        requestId
+      }, 'Handling tools/call request')
+
+      // Emit request event
+      this.events.onRequest?.(toolName, requestId)
+
+      const startTime = Date.now()
+
+      try {
+        // Import dynamically to avoid circular dependencies
+        const { handleCallTool } = await import('./handlers/call-tool')
+        const result = await handleCallTool(toolName, args, this.logger)
+
+        const duration = Date.now() - startTime
+        this.logger.debug({
+          tool: toolName,
+          requestId,
+          durationMs: duration
+        }, 'Tool call completed')
+
+        // Emit response event
+        this.events.onResponse?.(toolName, requestId, duration)
+
+        return result
+      } catch (error) {
+        const duration = Date.now() - startTime
+        this.logger.error({
+          tool: toolName,
+          requestId,
+          durationMs: duration,
+          error: error instanceof Error ? error.message : String(error)
+        }, 'Tool call failed')
+
+        throw error
+      }
+    })
+
+    // Phase 6.5: Handle prompts/list request
+    this.server.setRequestHandler(ListPromptsRequestSchema, async () => {
+      this.logger.debug('Handling prompts/list request')
+      return await this.promptProvider.listPrompts()
+    })
+
+    // Phase 6.5: Handle prompts/get request
+    this.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+      this.logger.debug({ name: request.params.name }, 'Handling prompts/get request')
+      return await this.promptProvider.getPrompt(request.params.name)
+    })
+
+    // Phase 6.5: Handle resources/list request
+    this.server.setRequestHandler(ListResourcesRequestSchema, async () => {
+      this.logger.debug('Handling resources/list request')
+      return await this.resourceProvider.listResources()
+    })
+
+    // Phase 6.5: Handle resources/read request
+    this.server.setRequestHandler(ReadResourceRequestSchema, async (request) => {
+      this.logger.debug({ uri: request.params.uri }, 'Handling resources/read request')
+      return await this.resourceProvider.readResource(request.params.uri)
+    })
+
+    this.logger.debug('Request handlers configured (tools, prompts, resources)')
+  }
+
+  /**
+   * Setup error handling for the server
+   */
+  private setupErrorHandling(): void {
+    this.server.onerror = (error) => {
+      this.logger.error({
+        error: error instanceof Error ? {
+          message: error.message,
+          stack: error.stack,
+          name: error.name
+        } : error
+      }, 'MCP server error')
+
+      this.events.onError?.(error instanceof Error ? error : new Error(String(error)))
+    }
+  }
+
+  /**
+   * Update connection state and emit events
+   */
+  private setConnectionState(state: ConnectionState): void {
+    const previousState = this.connectionState
+    this.connectionState = state
+
+    if (previousState !== state) {
+      this.logger.info({
+        previousState,
+        newState: state
+      }, 'Connection state changed')
+
+      this.events.onConnectionStateChange?.(state)
+    }
+  }
+
+  /**
+   * Start the MCP server and connect via stdio transport
+   */
+  async start(): Promise<void> {
+    if (this.isRunning) {
+      this.logger.warn('Server already running')
+      return
+    }
+
+    try {
+      this.logger.info('Starting MCP server...')
+      this.setConnectionState('connecting')
+
+      const transport = new StdioServerTransport()
+      await this.server.connect(transport)
+
+      this.isRunning = true
+      this.startedAt = new Date()
+      this.setConnectionState('connected')
+
+      this.logger.info({
+        name: this.config.serverName,
+        version: this.config.serverVersion,
+        startedAt: this.startedAt.toISOString()
+      }, 'MCP server started successfully')
+
+    } catch (error) {
+      this.setConnectionState('disconnected')
+      this.logger.error({
+        error: error instanceof Error ? {
+          message: error.message,
+          stack: error.stack
+        } : error
+      }, 'Failed to start MCP server')
+      throw error
+    }
+  }
+
+  /**
+   * Stop the MCP server gracefully
+   */
+  async stop(): Promise<void> {
+    if (!this.isRunning) {
+      this.logger.debug('Server not running, skip stop')
+      return
+    }
+
+    try {
+      this.logger.info('Stopping MCP server...')
+      await this.server.close()
+
+      this.isRunning = false
+      this.setConnectionState('disconnected')
+
+      this.logger.info({
+        totalRequests: this.requestCount,
+        uptimeMs: this.startedAt ? Date.now() - this.startedAt.getTime() : 0
+      }, 'MCP server stopped')
+
+    } catch (error) {
+      this.logger.error({
+        error: error instanceof Error ? {
+          message: error.message,
+          stack: error.stack
+        } : error
+      }, 'Error stopping MCP server')
+      throw error
+    }
+  }
+
+  /**
+   * Get current server status
+   */
+  getStatus(): ServerStatus {
+    return {
+      running: this.isRunning,
+      name: this.config.serverName,
+      version: this.config.serverVersion,
+      connectionState: this.connectionState,
+      startedAt: this.startedAt,
+      requestCount: this.requestCount,
+      capabilities: ['tools', 'prompts', 'resources']
+    }
+  }
+
+  /**
+   * Health check method
+   */
+  isHealthy(): boolean {
+    return this.isRunning && this.connectionState === 'connected'
+  }
+}

package/src/server/providers/index.ts

@@ -1,7 +1,7 @@
-/**
- * Server Providers Exports
- * Phase 6.5: System prompts and resources
- */
-
-export { PromptProvider } from './prompts'
-export { ResourceProvider } from './resources'
+/**
+ * Server Providers Exports
+ * Phase 6.5: System prompts and resources
+ */
+
+export { PromptProvider } from './prompts'
+export { ResourceProvider } from './resources'