aiyou-dev 3.5.15

package/v3/@aiyou-dev/shared/dist/mcp/session-manager.d.ts

@@ -0,0 +1,136 @@
+/**
+ * V3 MCP Session Manager
+ *
+ * Manages MCP sessions with:
+ * - Session lifecycle management
+ * - Authentication integration
+ * - Session timeout handling
+ * - Concurrent session support
+ * - Session metrics and monitoring
+ *
+ * Performance Targets:
+ * - Session creation: <5ms
+ * - Session lookup: <1ms
+ * - Session cleanup: <10ms
+ */
+import { EventEmitter } from 'events';
+import { MCPSession, SessionState, SessionMetrics, MCPInitializeParams, TransportType, ILogger } from './types.js';
+/**
+ * Session configuration
+ */
+export interface SessionConfig {
+    maxSessions?: number;
+    sessionTimeout?: number;
+    cleanupInterval?: number;
+    enableMetrics?: boolean;
+}
+/**
+ * Session Manager
+ *
+ * Handles creation, tracking, and cleanup of MCP sessions
+ */
+export declare class SessionManager extends EventEmitter {
+    private readonly logger;
+    private readonly sessions;
+    private readonly config;
+    private cleanupTimer?;
+    private sessionCounter;
+    private totalCreated;
+    private totalClosed;
+    private totalExpired;
+    constructor(logger: ILogger, config?: SessionConfig);
+    /**
+     * Create a new session
+     */
+    createSession(transport: TransportType): MCPSession;
+    /**
+     * Initialize a session with client information
+     */
+    initializeSession(sessionId: string, params: MCPInitializeParams): MCPSession | undefined;
+    /**
+     * Authenticate a session
+     */
+    authenticateSession(sessionId: string): boolean;
+    /**
+     * Get a session by ID
+     */
+    getSession(sessionId: string): MCPSession | undefined;
+    /**
+     * Check if session exists
+     */
+    hasSession(sessionId: string): boolean;
+    /**
+     * Get all active sessions
+     */
+    getActiveSessions(): MCPSession[];
+    /**
+     * Update session activity timestamp
+     */
+    updateActivity(sessionId: string): boolean;
+    /**
+     * Set session state
+     */
+    setState(sessionId: string, state: SessionState): boolean;
+    /**
+     * Set session metadata
+     */
+    setMetadata(sessionId: string, key: string, value: unknown): boolean;
+    /**
+     * Get session metadata
+     */
+    getMetadata(sessionId: string, key: string): unknown;
+    /**
+     * Close a session
+     */
+    closeSession(sessionId: string, reason?: string): boolean;
+    /**
+     * Remove a session (alias for closeSession)
+     */
+    removeSession(sessionId: string): boolean;
+    /**
+     * Get session metrics
+     */
+    getSessionMetrics(): SessionMetrics;
+    /**
+     * Get detailed statistics
+     */
+    getStats(): {
+        total: number;
+        byState: Record<SessionState, number>;
+        byTransport: Record<TransportType, number>;
+        totalCreated: number;
+        totalClosed: number;
+        totalExpired: number;
+        oldestSession?: Date;
+        newestSession?: Date;
+    };
+    /**
+     * Clean up expired sessions
+     */
+    cleanupExpiredSessions(): number;
+    /**
+     * Start cleanup timer
+     */
+    private startCleanupTimer;
+    /**
+     * Stop cleanup timer
+     */
+    private stopCleanupTimer;
+    /**
+     * Generate a unique session ID
+     */
+    private generateSessionId;
+    /**
+     * Clear all sessions
+     */
+    clearAll(): void;
+    /**
+     * Destroy the session manager
+     */
+    destroy(): void;
+}
+/**
+ * Create a session manager
+ */
+export declare function createSessionManager(logger: ILogger, config?: SessionConfig): SessionManager;
+//# sourceMappingURL=session-manager.d.ts.map

package/v3/@aiyou-dev/shared/dist/mcp/session-manager.js

@@ -0,0 +1,335 @@
+/**
+ * V3 MCP Session Manager
+ *
+ * Manages MCP sessions with:
+ * - Session lifecycle management
+ * - Authentication integration
+ * - Session timeout handling
+ * - Concurrent session support
+ * - Session metrics and monitoring
+ *
+ * Performance Targets:
+ * - Session creation: <5ms
+ * - Session lookup: <1ms
+ * - Session cleanup: <10ms
+ */
+import { EventEmitter } from 'events';
+/**
+ * Default session configuration
+ */
+const DEFAULT_SESSION_CONFIG = {
+    maxSessions: 100,
+    sessionTimeout: 30 * 60 * 1000, // 30 minutes
+    cleanupInterval: 60 * 1000, // 1 minute
+    enableMetrics: true,
+};
+/**
+ * Session Manager
+ *
+ * Handles creation, tracking, and cleanup of MCP sessions
+ */
+export class SessionManager extends EventEmitter {
+    logger;
+    sessions = new Map();
+    config;
+    cleanupTimer;
+    sessionCounter = 0;
+    // Statistics
+    totalCreated = 0;
+    totalClosed = 0;
+    totalExpired = 0;
+    constructor(logger, config = {}) {
+        super();
+        this.logger = logger;
+        this.config = { ...DEFAULT_SESSION_CONFIG, ...config };
+        this.startCleanupTimer();
+    }
+    /**
+     * Create a new session
+     */
+    createSession(transport) {
+        // Check max sessions
+        if (this.sessions.size >= this.config.maxSessions) {
+            throw new Error(`Maximum sessions (${this.config.maxSessions}) reached`);
+        }
+        const id = this.generateSessionId();
+        const now = new Date();
+        const session = {
+            id,
+            state: 'created',
+            transport,
+            createdAt: now,
+            lastActivityAt: now,
+            isInitialized: false,
+            isAuthenticated: false,
+        };
+        this.sessions.set(id, session);
+        this.totalCreated++;
+        this.logger.debug('Session created', { id, transport });
+        this.emit('session:created', session);
+        return session;
+    }
+    /**
+     * Initialize a session with client information
+     */
+    initializeSession(sessionId, params) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            this.logger.warn('Session not found for initialization', { sessionId });
+            return undefined;
+        }
+        session.state = 'ready';
+        session.isInitialized = true;
+        session.clientInfo = params.clientInfo;
+        session.protocolVersion = params.protocolVersion;
+        session.capabilities = params.capabilities;
+        session.lastActivityAt = new Date();
+        this.logger.info('Session initialized', {
+            sessionId,
+            clientInfo: params.clientInfo,
+            protocolVersion: params.protocolVersion,
+        });
+        this.emit('session:initialized', session);
+        return session;
+    }
+    /**
+     * Authenticate a session
+     */
+    authenticateSession(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return false;
+        }
+        session.isAuthenticated = true;
+        session.lastActivityAt = new Date();
+        this.logger.debug('Session authenticated', { sessionId });
+        this.emit('session:authenticated', session);
+        return true;
+    }
+    /**
+     * Get a session by ID
+     */
+    getSession(sessionId) {
+        return this.sessions.get(sessionId);
+    }
+    /**
+     * Check if session exists
+     */
+    hasSession(sessionId) {
+        return this.sessions.has(sessionId);
+    }
+    /**
+     * Get all active sessions
+     */
+    getActiveSessions() {
+        return Array.from(this.sessions.values()).filter((s) => s.state === 'ready' || s.state === 'created' || s.state === 'initializing');
+    }
+    /**
+     * Update session activity timestamp
+     */
+    updateActivity(sessionId) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return false;
+        }
+        session.lastActivityAt = new Date();
+        return true;
+    }
+    /**
+     * Set session state
+     */
+    setState(sessionId, state) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return false;
+        }
+        const oldState = session.state;
+        session.state = state;
+        session.lastActivityAt = new Date();
+        this.logger.debug('Session state changed', {
+            sessionId,
+            oldState,
+            newState: state,
+        });
+        this.emit('session:stateChanged', { session, oldState, newState: state });
+        return true;
+    }
+    /**
+     * Set session metadata
+     */
+    setMetadata(sessionId, key, value) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return false;
+        }
+        if (!session.metadata) {
+            session.metadata = {};
+        }
+        session.metadata[key] = value;
+        session.lastActivityAt = new Date();
+        return true;
+    }
+    /**
+     * Get session metadata
+     */
+    getMetadata(sessionId, key) {
+        const session = this.sessions.get(sessionId);
+        return session?.metadata?.[key];
+    }
+    /**
+     * Close a session
+     */
+    closeSession(sessionId, reason) {
+        const session = this.sessions.get(sessionId);
+        if (!session) {
+            return false;
+        }
+        session.state = 'closed';
+        this.sessions.delete(sessionId);
+        this.totalClosed++;
+        this.logger.info('Session closed', { sessionId, reason });
+        this.emit('session:closed', { session, reason });
+        return true;
+    }
+    /**
+     * Remove a session (alias for closeSession)
+     */
+    removeSession(sessionId) {
+        return this.closeSession(sessionId);
+    }
+    /**
+     * Get session metrics
+     */
+    getSessionMetrics() {
+        let authenticated = 0;
+        let active = 0;
+        for (const session of this.sessions.values()) {
+            if (session.isAuthenticated)
+                authenticated++;
+            if (session.state === 'ready')
+                active++;
+        }
+        return {
+            total: this.sessions.size,
+            active,
+            authenticated,
+            expired: this.totalExpired,
+        };
+    }
+    /**
+     * Get detailed statistics
+     */
+    getStats() {
+        const byState = {
+            created: 0,
+            initializing: 0,
+            ready: 0,
+            closing: 0,
+            closed: 0,
+            error: 0,
+        };
+        const byTransport = {
+            stdio: 0,
+            http: 0,
+            websocket: 0,
+            'in-process': 0,
+        };
+        let oldest;
+        let newest;
+        for (const session of this.sessions.values()) {
+            byState[session.state] = (byState[session.state] || 0) + 1;
+            byTransport[session.transport] = (byTransport[session.transport] || 0) + 1;
+            if (!oldest || session.createdAt < oldest) {
+                oldest = session.createdAt;
+            }
+            if (!newest || session.createdAt > newest) {
+                newest = session.createdAt;
+            }
+        }
+        return {
+            total: this.sessions.size,
+            byState: byState,
+            byTransport: byTransport,
+            totalCreated: this.totalCreated,
+            totalClosed: this.totalClosed,
+            totalExpired: this.totalExpired,
+            oldestSession: oldest,
+            newestSession: newest,
+        };
+    }
+    /**
+     * Clean up expired sessions
+     */
+    cleanupExpiredSessions() {
+        const now = Date.now();
+        const expired = [];
+        for (const [id, session] of this.sessions) {
+            const inactiveTime = now - session.lastActivityAt.getTime();
+            if (inactiveTime > this.config.sessionTimeout) {
+                expired.push(id);
+            }
+        }
+        for (const id of expired) {
+            const session = this.sessions.get(id);
+            if (session) {
+                session.state = 'closed';
+                this.sessions.delete(id);
+                this.totalExpired++;
+                this.logger.info('Session expired', { sessionId: id });
+                this.emit('session:expired', session);
+            }
+        }
+        if (expired.length > 0) {
+            this.logger.info('Cleaned up expired sessions', { count: expired.length });
+        }
+        return expired.length;
+    }
+    /**
+     * Start cleanup timer
+     */
+    startCleanupTimer() {
+        this.cleanupTimer = setInterval(() => {
+            this.cleanupExpiredSessions();
+        }, this.config.cleanupInterval);
+    }
+    /**
+     * Stop cleanup timer
+     */
+    stopCleanupTimer() {
+        if (this.cleanupTimer) {
+            clearInterval(this.cleanupTimer);
+            this.cleanupTimer = undefined;
+        }
+    }
+    /**
+     * Generate a unique session ID
+     */
+    generateSessionId() {
+        return `session-${++this.sessionCounter}-${Date.now()}-${Math.random().toString(36).substring(2, 8)}`;
+    }
+    /**
+     * Clear all sessions
+     */
+    clearAll() {
+        for (const id of this.sessions.keys()) {
+            this.closeSession(id, 'Session manager cleared');
+        }
+        this.logger.info('All sessions cleared');
+    }
+    /**
+     * Destroy the session manager
+     */
+    destroy() {
+        this.stopCleanupTimer();
+        this.clearAll();
+        this.removeAllListeners();
+        this.logger.info('Session manager destroyed');
+    }
+}
+/**
+ * Create a session manager
+ */
+export function createSessionManager(logger, config) {
+    return new SessionManager(logger, config);
+}
+//# sourceMappingURL=session-manager.js.map

package/v3/@aiyou-dev/shared/dist/mcp/tool-registry.d.ts

@@ -0,0 +1,178 @@
+/**
+ * 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';
+import { MCPTool, JSONSchema, ToolHandler, ToolContext, ToolCallResult, ToolRegistrationOptions, ILogger } from './types.js';
+/**
+ * Tool metadata for enhanced lookup
+ */
+interface ToolMetadata {
+    tool: MCPTool;
+    registeredAt: Date;
+    callCount: number;
+    lastCalled?: Date;
+    avgExecutionTime: number;
+    errorCount: number;
+}
+/**
+ * Tool search options
+ */
+interface ToolSearchOptions {
+    category?: string;
+    tags?: string[];
+    deprecated?: boolean;
+    cacheable?: boolean;
+}
+/**
+ * Tool Registry
+ *
+ * Manages registration, lookup, and execution of MCP tools
+ */
+export declare class ToolRegistry extends EventEmitter {
+    private readonly logger;
+    private readonly tools;
+    private readonly categoryIndex;
+    private readonly tagIndex;
+    private defaultContext?;
+    private totalRegistrations;
+    private totalLookups;
+    private totalExecutions;
+    constructor(logger: ILogger);
+    /**
+     * Register a tool
+     */
+    register(tool: MCPTool, options?: ToolRegistrationOptions): boolean;
+    /**
+     * Register multiple tools at once
+     */
+    registerBatch(tools: MCPTool[], options?: ToolRegistrationOptions): {
+        registered: number;
+        failed: string[];
+    };
+    /**
+     * Unregister a tool
+     */
+    unregister(name: string): boolean;
+    /**
+     * Get a tool by name
+     */
+    getTool(name: string): MCPTool | undefined;
+    /**
+     * Check if a tool exists
+     */
+    hasTool(name: string): boolean;
+    /**
+     * Get tool count
+     */
+    getToolCount(): number;
+    /**
+     * Get all tool names
+     */
+    getToolNames(): string[];
+    /**
+     * List all tools with metadata
+     */
+    listTools(): Array<{
+        name: string;
+        description: string;
+        category?: string;
+        tags?: string[];
+        deprecated?: boolean;
+    }>;
+    /**
+     * Search tools by criteria
+     */
+    search(options: ToolSearchOptions): MCPTool[];
+    /**
+     * Get tools by category
+     */
+    getByCategory(category: string): MCPTool[];
+    /**
+     * Get tools by tag
+     */
+    getByTag(tag: string): MCPTool[];
+    /**
+     * Get all categories
+     */
+    getCategories(): string[];
+    /**
+     * Get all tags
+     */
+    getTags(): string[];
+    /**
+     * Execute a tool
+     */
+    execute(name: string, input: Record<string, unknown>, context?: ToolContext): Promise<ToolCallResult>;
+    /**
+     * Set default execution context
+     */
+    setDefaultContext(context: ToolContext): void;
+    /**
+     * Get tool metadata
+     */
+    getMetadata(name: string): ToolMetadata | undefined;
+    /**
+     * Get registry statistics
+     */
+    getStats(): {
+        totalTools: number;
+        totalCategories: number;
+        totalTags: number;
+        totalRegistrations: number;
+        totalLookups: number;
+        totalExecutions: number;
+        topTools: Array<{
+            name: string;
+            calls: number;
+        }>;
+    };
+    /**
+     * Validate a tool definition
+     */
+    validateTool(tool: MCPTool): {
+        valid: boolean;
+        errors: string[];
+    };
+    /**
+     * Validate JSON Schema
+     */
+    private validateSchema;
+    /**
+     * Update average execution time
+     */
+    private updateAverageExecutionTime;
+    /**
+     * Clear all tools
+     */
+    clear(): void;
+}
+/**
+ * Create a tool registry
+ */
+export declare function createToolRegistry(logger: ILogger): ToolRegistry;
+/**
+ * Helper to create a tool definition
+ */
+export declare function defineTool<TInput = Record<string, unknown>, TOutput = unknown>(name: string, description: string, inputSchema: JSONSchema, handler: ToolHandler<TInput, TOutput>, options?: {
+    category?: string;
+    tags?: string[];
+    version?: string;
+    deprecated?: boolean;
+    cacheable?: boolean;
+    cacheTTL?: number;
+    timeout?: number;
+}): MCPTool<TInput, TOutput>;
+export {};
+//# sourceMappingURL=tool-registry.d.ts.map