moflo 4.8.19 → 4.8.20

package/src/@claude-flow/shared/dist/mcp/tool-registry.js

@@ -0,0 +1,439 @@
+/**
+ * V3 MCP Tool Registry
+ *
+ * High-performance tool management with:
+ * - Fast O(1) lookup using Map
+ * - Category-based organization
+ * - Tool validation on registration
+ * - Dynamic registration/unregistration
+ * - Caching for frequently used tools
+ *
+ * Performance Targets:
+ * - Tool registration: <10ms
+ * - Tool lookup: <1ms
+ * - Tool validation: <5ms
+ */
+import { EventEmitter } from 'events';
+/**
+ * Tool Registry
+ *
+ * Manages registration, lookup, and execution of MCP tools
+ */
+export class ToolRegistry extends EventEmitter {
+    logger;
+    tools = new Map();
+    categoryIndex = new Map();
+    tagIndex = new Map();
+    defaultContext;
+    // Performance tracking
+    totalRegistrations = 0;
+    totalLookups = 0;
+    totalExecutions = 0;
+    constructor(logger) {
+        super();
+        this.logger = logger;
+    }
+    /**
+     * Register a tool
+     */
+    register(tool, options = {}) {
+        const startTime = performance.now();
+        // Check for existing tool
+        if (this.tools.has(tool.name) && !options.override) {
+            this.logger.warn('Tool already registered', { name: tool.name });
+            return false;
+        }
+        // Validate tool if requested
+        if (options.validate !== false) {
+            const validation = this.validateTool(tool);
+            if (!validation.valid) {
+                this.logger.error('Tool validation failed', {
+                    name: tool.name,
+                    errors: validation.errors,
+                });
+                return false;
+            }
+        }
+        // Create metadata
+        const metadata = {
+            tool,
+            registeredAt: new Date(),
+            callCount: 0,
+            avgExecutionTime: 0,
+            errorCount: 0,
+        };
+        // Register tool
+        this.tools.set(tool.name, metadata);
+        this.totalRegistrations++;
+        // Update category index
+        if (tool.category) {
+            if (!this.categoryIndex.has(tool.category)) {
+                this.categoryIndex.set(tool.category, new Set());
+            }
+            this.categoryIndex.get(tool.category).add(tool.name);
+        }
+        // Update tag index
+        if (tool.tags) {
+            for (const tag of tool.tags) {
+                if (!this.tagIndex.has(tag)) {
+                    this.tagIndex.set(tag, new Set());
+                }
+                this.tagIndex.get(tag).add(tool.name);
+            }
+        }
+        const duration = performance.now() - startTime;
+        this.logger.debug('Tool registered', {
+            name: tool.name,
+            category: tool.category,
+            duration: `${duration.toFixed(2)}ms`,
+        });
+        this.emit('tool:registered', tool.name);
+        return true;
+    }
+    /**
+     * Register multiple tools at once
+     */
+    registerBatch(tools, options = {}) {
+        const startTime = performance.now();
+        const failed = [];
+        let registered = 0;
+        for (const tool of tools) {
+            if (this.register(tool, options)) {
+                registered++;
+            }
+            else {
+                failed.push(tool.name);
+            }
+        }
+        const duration = performance.now() - startTime;
+        this.logger.info('Batch registration complete', {
+            total: tools.length,
+            registered,
+            failed: failed.length,
+            duration: `${duration.toFixed(2)}ms`,
+        });
+        return { registered, failed };
+    }
+    /**
+     * Unregister a tool
+     */
+    unregister(name) {
+        const metadata = this.tools.get(name);
+        if (!metadata) {
+            return false;
+        }
+        // Remove from category index
+        if (metadata.tool.category) {
+            const categoryTools = this.categoryIndex.get(metadata.tool.category);
+            categoryTools?.delete(name);
+            if (categoryTools?.size === 0) {
+                this.categoryIndex.delete(metadata.tool.category);
+            }
+        }
+        // Remove from tag index
+        if (metadata.tool.tags) {
+            for (const tag of metadata.tool.tags) {
+                const tagTools = this.tagIndex.get(tag);
+                tagTools?.delete(name);
+                if (tagTools?.size === 0) {
+                    this.tagIndex.delete(tag);
+                }
+            }
+        }
+        this.tools.delete(name);
+        this.logger.debug('Tool unregistered', { name });
+        this.emit('tool:unregistered', name);
+        return true;
+    }
+    /**
+     * Get a tool by name
+     */
+    getTool(name) {
+        this.totalLookups++;
+        return this.tools.get(name)?.tool;
+    }
+    /**
+     * Check if a tool exists
+     */
+    hasTool(name) {
+        return this.tools.has(name);
+    }
+    /**
+     * Get tool count
+     */
+    getToolCount() {
+        return this.tools.size;
+    }
+    /**
+     * Get all tool names
+     */
+    getToolNames() {
+        return Array.from(this.tools.keys());
+    }
+    /**
+     * List all tools with metadata
+     */
+    listTools() {
+        return Array.from(this.tools.values()).map(({ tool }) => ({
+            name: tool.name,
+            description: tool.description,
+            category: tool.category,
+            tags: tool.tags,
+            deprecated: tool.deprecated,
+        }));
+    }
+    /**
+     * Search tools by criteria
+     */
+    search(options) {
+        let results;
+        // Filter by category
+        if (options.category) {
+            const categoryTools = this.categoryIndex.get(options.category);
+            if (!categoryTools)
+                return [];
+            results = new Set(categoryTools);
+        }
+        // Filter by tags (intersection)
+        if (options.tags && options.tags.length > 0) {
+            for (const tag of options.tags) {
+                const tagTools = this.tagIndex.get(tag);
+                if (!tagTools)
+                    return [];
+                if (results) {
+                    results = new Set([...results].filter((name) => tagTools.has(name)));
+                }
+                else {
+                    results = new Set(tagTools);
+                }
+            }
+        }
+        // Get all tools if no filters applied
+        if (!results) {
+            results = new Set(this.tools.keys());
+        }
+        // Convert to tools and apply additional filters
+        const tools = [];
+        for (const name of results) {
+            const metadata = this.tools.get(name);
+            if (!metadata)
+                continue;
+            const tool = metadata.tool;
+            // Filter by deprecated status
+            if (options.deprecated !== undefined && tool.deprecated !== options.deprecated) {
+                continue;
+            }
+            // Filter by cacheable status
+            if (options.cacheable !== undefined && tool.cacheable !== options.cacheable) {
+                continue;
+            }
+            tools.push(tool);
+        }
+        return tools;
+    }
+    /**
+     * Get tools by category
+     */
+    getByCategory(category) {
+        const toolNames = this.categoryIndex.get(category);
+        if (!toolNames)
+            return [];
+        return Array.from(toolNames)
+            .map((name) => this.tools.get(name)?.tool)
+            .filter((tool) => tool !== undefined);
+    }
+    /**
+     * Get tools by tag
+     */
+    getByTag(tag) {
+        const toolNames = this.tagIndex.get(tag);
+        if (!toolNames)
+            return [];
+        return Array.from(toolNames)
+            .map((name) => this.tools.get(name)?.tool)
+            .filter((tool) => tool !== undefined);
+    }
+    /**
+     * Get all categories
+     */
+    getCategories() {
+        return Array.from(this.categoryIndex.keys());
+    }
+    /**
+     * Get all tags
+     */
+    getTags() {
+        return Array.from(this.tagIndex.keys());
+    }
+    /**
+     * Execute a tool
+     */
+    async execute(name, input, context) {
+        const startTime = performance.now();
+        const metadata = this.tools.get(name);
+        if (!metadata) {
+            return {
+                content: [{ type: 'text', text: `Tool not found: ${name}` }],
+                isError: true,
+            };
+        }
+        // Build execution context with required sessionId
+        const execContext = {
+            sessionId: context?.sessionId || this.defaultContext?.sessionId || 'default-session',
+            ...this.defaultContext,
+            ...context,
+        };
+        this.totalExecutions++;
+        metadata.callCount++;
+        metadata.lastCalled = new Date();
+        try {
+            this.emit('tool:called', { name, input });
+            const result = await metadata.tool.handler(input, execContext);
+            const duration = performance.now() - startTime;
+            this.updateAverageExecutionTime(metadata, duration);
+            this.logger.debug('Tool executed', {
+                name,
+                duration: `${duration.toFixed(2)}ms`,
+                success: true,
+            });
+            this.emit('tool:completed', { name, duration, success: true });
+            // Format result
+            return {
+                content: [{
+                        type: 'text',
+                        text: typeof result === 'string' ? result : JSON.stringify(result, null, 2),
+                    }],
+                isError: false,
+            };
+        }
+        catch (error) {
+            const duration = performance.now() - startTime;
+            metadata.errorCount++;
+            this.logger.error('Tool execution failed', { name, error });
+            this.emit('tool:error', { name, error, duration });
+            return {
+                content: [{
+                        type: 'text',
+                        text: `Error: ${error instanceof Error ? error.message : String(error)}`,
+                    }],
+                isError: true,
+            };
+        }
+    }
+    /**
+     * Set default execution context
+     */
+    setDefaultContext(context) {
+        this.defaultContext = context;
+    }
+    /**
+     * Get tool metadata
+     */
+    getMetadata(name) {
+        return this.tools.get(name);
+    }
+    /**
+     * Get registry statistics
+     */
+    getStats() {
+        // Get top 10 most used tools
+        const topTools = Array.from(this.tools.entries())
+            .map(([name, metadata]) => ({ name, calls: metadata.callCount }))
+            .sort((a, b) => b.calls - a.calls)
+            .slice(0, 10);
+        return {
+            totalTools: this.tools.size,
+            totalCategories: this.categoryIndex.size,
+            totalTags: this.tagIndex.size,
+            totalRegistrations: this.totalRegistrations,
+            totalLookups: this.totalLookups,
+            totalExecutions: this.totalExecutions,
+            topTools,
+        };
+    }
+    /**
+     * Validate a tool definition
+     */
+    validateTool(tool) {
+        const errors = [];
+        if (!tool.name || typeof tool.name !== 'string') {
+            errors.push('Tool name is required and must be a string');
+        }
+        else if (!/^[a-zA-Z][a-zA-Z0-9_/:-]*$/.test(tool.name)) {
+            errors.push('Tool name must start with a letter and contain only alphanumeric characters, underscores, slashes, colons, and hyphens');
+        }
+        if (!tool.description || typeof tool.description !== 'string') {
+            errors.push('Tool description is required and must be a string');
+        }
+        if (!tool.inputSchema || typeof tool.inputSchema !== 'object') {
+            errors.push('Tool inputSchema is required and must be an object');
+        }
+        else {
+            const schemaErrors = this.validateSchema(tool.inputSchema);
+            errors.push(...schemaErrors);
+        }
+        if (typeof tool.handler !== 'function') {
+            errors.push('Tool handler is required and must be a function');
+        }
+        return {
+            valid: errors.length === 0,
+            errors,
+        };
+    }
+    /**
+     * Validate JSON Schema
+     */
+    validateSchema(schema, path = '') {
+        const errors = [];
+        if (!schema.type) {
+            errors.push(`${path || 'schema'}: type is required`);
+        }
+        if (schema.type === 'object' && schema.properties) {
+            for (const [key, propSchema] of Object.entries(schema.properties)) {
+                const propPath = path ? `${path}.${key}` : key;
+                errors.push(...this.validateSchema(propSchema, propPath));
+            }
+        }
+        if (schema.type === 'array' && schema.items) {
+            errors.push(...this.validateSchema(schema.items, `${path}[]`));
+        }
+        return errors;
+    }
+    /**
+     * Update average execution time
+     */
+    updateAverageExecutionTime(metadata, duration) {
+        const n = metadata.callCount;
+        metadata.avgExecutionTime =
+            ((metadata.avgExecutionTime * (n - 1)) + duration) / n;
+    }
+    /**
+     * Clear all tools
+     */
+    clear() {
+        this.tools.clear();
+        this.categoryIndex.clear();
+        this.tagIndex.clear();
+        this.logger.info('Tool registry cleared');
+        this.emit('registry:cleared');
+    }
+}
+/**
+ * Create a tool registry
+ */
+export function createToolRegistry(logger) {
+    return new ToolRegistry(logger);
+}
+/**
+ * Helper to create a tool definition
+ */
+export function defineTool(name, description, inputSchema, handler, options) {
+    return {
+        name,
+        description,
+        inputSchema,
+        handler,
+        ...options,
+    };
+}
+//# sourceMappingURL=tool-registry.js.map

package/src/@claude-flow/shared/dist/mcp/transport/http.d.ts

@@ -0,0 +1,104 @@
+/**
+ * V3 MCP HTTP Transport
+ *
+ * HTTP/REST transport for MCP communication:
+ * - Express-based with optimized middleware
+ * - WebSocket support for real-time notifications
+ * - Connection pooling for client connections
+ * - Security headers with helmet
+ *
+ * Performance Targets:
+ * - Request handling: <20ms overhead
+ * - WebSocket message: <5ms
+ */
+import { EventEmitter } from 'events';
+import { ITransport, TransportType, MCPNotification, RequestHandler, NotificationHandler, TransportHealthStatus, ILogger, AuthConfig } from '../types.js';
+/**
+ * HTTP Transport Configuration
+ */
+export interface HttpTransportConfig {
+    host: string;
+    port: number;
+    tlsEnabled?: boolean;
+    tlsCert?: string;
+    tlsKey?: string;
+    corsEnabled?: boolean;
+    corsOrigins?: string[];
+    auth?: AuthConfig;
+    maxRequestSize?: string;
+    requestTimeout?: number;
+}
+/**
+ * HTTP Transport Implementation
+ */
+export declare class HttpTransport extends EventEmitter implements ITransport {
+    private readonly logger;
+    private readonly config;
+    readonly type: TransportType;
+    private requestHandler?;
+    private notificationHandler?;
+    private app;
+    private server?;
+    private wss?;
+    private running;
+    private activeConnections;
+    private messagesReceived;
+    private messagesSent;
+    private errors;
+    private httpRequests;
+    private wsMessages;
+    constructor(logger: ILogger, config: HttpTransportConfig);
+    /**
+     * Start the transport
+     */
+    start(): Promise<void>;
+    /**
+     * Stop the transport
+     */
+    stop(): Promise<void>;
+    /**
+     * Register request handler
+     */
+    onRequest(handler: RequestHandler): void;
+    /**
+     * Register notification handler
+     */
+    onNotification(handler: NotificationHandler): void;
+    /**
+     * Get health status
+     */
+    getHealthStatus(): Promise<TransportHealthStatus>;
+    /**
+     * Send notification to all connected WebSocket clients
+     */
+    sendNotification(notification: MCPNotification): Promise<void>;
+    /**
+     * Setup Express middleware
+     */
+    private setupMiddleware;
+    /**
+     * Setup Express routes
+     */
+    private setupRoutes;
+    /**
+     * Setup WebSocket handlers
+     */
+    private setupWebSocketHandlers;
+    /**
+     * Handle HTTP request
+     */
+    private handleHttpRequest;
+    /**
+     * Handle WebSocket message
+     */
+    private handleWebSocketMessage;
+    /**
+     * Validate authentication
+     */
+    private validateAuth;
+}
+/**
+ * Create HTTP transport
+ */
+export declare function createHttpTransport(logger: ILogger, config: HttpTransportConfig): HttpTransport;
+//# sourceMappingURL=http.d.ts.map