claude-flow/.claude/worktrees/adr-120-impl/v3/@claude-flow/shared/src/mcp/index.ts

Ruflo - Enterprise AI agent orchestration for Claude Code. Deploy 60+ specialized agents in coordinated swarms with self-learning, fault-tolerant consensus, vector memory, and MCP integration

/**
 * V3 MCP Module
 *
 * Optimized MCP (Model Context Protocol) implementation for Claude-Flow V3
 *
 * Features:
 * - High-performance server with <400ms startup
 * - Connection pooling with max 10 connections
 * - Multiple transport support (stdio, http, websocket, in-process)
 * - Fast tool registry with <10ms registration
 * - Session management with timeout handling
 * - Comprehensive metrics and monitoring
 *
 * Performance Targets:
 * - Server startup: <400ms
 * - Tool registration: <10ms
 * - Tool execution: <50ms overhead
 * - Connection acquire: <5ms
 *
 * @module @claude-flow/mcp
 * @version 3.0.0
 */

// Core types - using 'export type' for TypeScript isolatedModules compatibility
export type {
  // Protocol types
  JsonRpcVersion,
  RequestId,
  MCPMessage,
  MCPRequest,
  MCPResponse,
  MCPNotification,
  MCPError,

  // Server configuration
  TransportType,
  AuthMethod,
  AuthConfig,
  LoadBalancerConfig,
  ConnectionPoolConfig,
  MCPServerConfig,

  // Session types
  SessionState,
  MCPSession,
  MCPClientInfo,

  // Capability types
  MCPCapabilities,
  MCPProtocolVersion,
  MCPInitializeParams,
  MCPInitializeResult,

  // Tool types
  JSONSchema,
  ToolContext,
  ToolHandler,
  MCPTool,
  ToolCallResult,
  ToolRegistrationOptions,

  // Transport types
  RequestHandler,
  NotificationHandler,
  TransportHealthStatus,
  ITransport,

  // Connection pool types
  ConnectionState,
  PooledConnection,
  ConnectionPoolStats,
  IConnectionPool,

  // Metrics types
  ToolCallMetrics,
  MCPServerMetrics,
  SessionMetrics,

  // Event types
  MCPEventType,
  MCPEvent,
  EventHandler,

  // Logger
  LogLevel,
  ILogger,
} from './types.js';

// Import types for local use in quickStart function
import type { MCPServerConfig, ILogger } from './types.js';

// Error handling - values (not types)
export { ErrorCodes, MCPServerError } from './types.js';

// Server - class and interface exports
import { MCPServer, createMCPServer } from './server.js';
export { MCPServer, createMCPServer };
export type { IMCPServer } from './server.js';

// Tool Registry
export { ToolRegistry, createToolRegistry, defineTool } from './tool-registry.js';

// Session Manager - class and factory exports
import { SessionManager, createSessionManager } from './session-manager.js';
export { SessionManager, createSessionManager };
export type { SessionConfig } from './session-manager.js';

// Connection Pool
export { ConnectionPool, createConnectionPool } from './connection-pool.js';

// Transport layer - values
export {
  // Factory
  createTransport,
  createInProcessTransport,
  TransportManager,
  createTransportManager,
  DEFAULT_TRANSPORT_CONFIGS,

  // Specific transports
  StdioTransport,
  HttpTransport,
  WebSocketTransport,
} from './transport/index.js';

// Transport layer - types
export type {
  TransportConfig,
  StdioTransportConfig,
  HttpTransportConfig,
  WebSocketTransportConfig,
} from './transport/index.js';

/**
 * Quick start function to create and configure an MCP server
 *
 * @example
 * ```typescript
 * import { quickStart } from '@claude-flow/mcp';
 *
 * const server = await quickStart({
 *   transport: 'stdio',
 *   name: 'My MCP Server',
 * });
 *
 * // Register custom tools
 * server.registerTool({
 *   name: 'my-tool',
 *   description: 'My custom tool',
 *   inputSchema: { type: 'object', properties: {} },
 *   handler: async () => ({ result: 'success' }),
 * });
 *
 * // Start server
 * await server.start();
 * ```
 */
export async function quickStart(
  config: Partial<MCPServerConfig>,
  logger?: ILogger
): Promise<MCPServer> {
  // Create default logger if not provided
  const defaultLogger: ILogger = logger || {
    debug: (msg, data) => console.debug(`[DEBUG] ${msg}`, data || ''),
    info: (msg, data) => console.info(`[INFO] ${msg}`, data || ''),
    warn: (msg, data) => console.warn(`[WARN] ${msg}`, data || ''),
    error: (msg, data) => console.error(`[ERROR] ${msg}`, data || ''),
  };

  const server = createMCPServer(config, defaultLogger);

  return server;
}

/**
 * Module version
 */
export const VERSION = '3.0.0';

/**
 * Module name
 */
export const MODULE_NAME = '@claude-flow/mcp';