@juspay/neurolink 1.10.0 → 1.11.0

package/dist/lib/mcp/registry.d.ts

@@ -1,196 +1,47 @@
 /**
- * NeuroLink MCP Tool Registry System
- * Central registry for managing MCP servers and tools with execution capabilities
- * Supports tool discovery, registration, and orchestrated execution
+ * MCP Registry - Plugin Registration and Management
  */
-import type { NeuroLinkMCPServer, NeuroLinkMCPTool, NeuroLinkExecutionContext, ToolResult } from "./factory.js";
-import { ContextManager } from "./context-manager.js";
+import type { DiscoveredMCP } from "./contracts/mcp-contract.js";
 /**
- * Tool registration information
+ * Simple MCP registry for plugin management
  */
-export interface ToolRegistration {
-    tool: NeuroLinkMCPTool;
-    serverId: string;
-    serverTitle: string;
-    serverCategory?: string;
-    qualifiedName: string;
-    simpleName: string;
-    registeredAt: number;
-}
-/**
- * Tool execution options
- */
-export interface ToolExecutionOptions {
-    validateInput?: boolean;
-    validatePermissions?: boolean;
-    trackMetrics?: boolean;
-    timeoutMs?: number;
-}
-/**
- * Tool search criteria
- */
-export interface ToolSearchCriteria {
-    name?: string;
-    category?: string;
-    serverId?: string;
-    serverCategory?: string;
-    permissions?: string[];
-    implemented?: boolean;
-}
-/**
- * Registry statistics
- */
-export interface RegistryStats {
-    totalServers: number;
-    totalTools: number;
-    toolsByCategory: Record<string, number>;
-    serversByCategory: Record<string, number>;
-    executionCount: number;
-    averageExecutionTime: number;
-    errorRate: number;
-}
-/**
- * Central MCP Tool Registry
- * Manages all MCP servers and their tools with advanced execution capabilities
- */
-export declare class MCPToolRegistry {
-    private servers;
-    private tools;
-    private contextManager;
-    private executionCount;
-    private totalExecutionTime;
-    private errorCount;
-    constructor(contextManager?: ContextManager);
-    /**
-     * Register an MCP server and all its tools
-     *
-     * @param server MCP server to register
-     * @throws Error if server ID already exists
-     */
-    registerServer(server: NeuroLinkMCPServer): Promise<void>;
+export declare class MCPRegistry {
+    private plugins;
     /**
-     * Register a single tool from a server
-     *
-     * @param server Source server
-     * @param toolName Tool name
-     * @param tool Tool implementation
+     * Register a plugin
      */
-    private registerToolFromServer;
+    register(plugin: DiscoveredMCP): void;
     /**
-     * Execute a tool with comprehensive error handling and context tracking
-     *
-     * @param toolName Tool name (simple or qualified)
-     * @param params Tool parameters
-     * @param context Execution context
-     * @param options Execution options
-     * @returns Tool execution result
+     * Unregister a plugin
      */
-    executeTool(toolName: string, params: any, context: NeuroLinkExecutionContext, options?: ToolExecutionOptions): Promise<ToolResult>;
+    unregister(name: string): boolean;
     /**
-     * List all available tools with optional filtering
-     *
-     * @param criteria Search criteria for filtering tools
-     * @returns Array of tool information
+     * Get a plugin
      */
-    listTools(criteria?: ToolSearchCriteria): {
-        name: string;
-        qualifiedName: string;
-        description: string;
-        server: string;
-        serverTitle: string;
-        category?: string;
-        serverCategory?: string;
-        permissions?: string[];
-        isImplemented?: boolean;
-    }[];
+    get(name: string): DiscoveredMCP | undefined;
     /**
-     * Get detailed information about a specific tool
-     *
-     * @param toolName Tool name (simple or qualified)
-     * @returns Detailed tool information or undefined if not found
+     * List all plugins
      */
-    getToolInfo(toolName: string): {
-        tool: NeuroLinkMCPTool;
-        server: NeuroLinkMCPServer;
-        registration: ToolRegistration;
-    } | undefined;
+    list(): DiscoveredMCP[];
     /**
-     * Get registry statistics
-     *
-     * @returns Comprehensive registry statistics
+     * Check if plugin exists
      */
-    getStats(): RegistryStats;
+    has(name: string): boolean;
     /**
-     * Unregister a server and all its tools
-     *
-     * @param serverId Server ID to unregister
-     * @returns Whether server was found and removed
-     */
-    unregisterServer(serverId: string): boolean;
-    /**
-     * Clear all servers and tools
+     * Clear all plugins
      */
     clear(): void;
-    /**
-     * Create timeout promise wrapper
-     *
-     * @param promise Promise to wrap
-     * @param timeoutMs Timeout in milliseconds
-     * @param timeoutMessage Error message for timeout
-     * @returns Promise that rejects on timeout
-     */
-    private createTimeoutPromise;
-    /**
-     * Update execution metrics
-     *
-     * @param executionTime Execution time in milliseconds
-     * @param success Whether execution was successful
-     */
-    private updateExecutionMetrics;
 }
 /**
- * Get the default singleton instance of the MCPToolRegistry.
- * This function ensures that the registry is only instantiated once.
- * Using a function getter helps prevent circular dependency issues during module initialization.
- *
- * @returns The singleton MCPToolRegistry instance.
- */
-export declare function getDefaultToolRegistry(): MCPToolRegistry;
-/**
- * @deprecated The direct export `defaultToolRegistry` is deprecated and will be removed.
- * Please use `getDefaultToolRegistry()` instead to avoid circular dependency issues.
- */
-export declare const defaultToolRegistry: MCPToolRegistry;
-/**
- * Utility function to register server with default registry
- *
- * @param server MCP server to register
- */
-export declare function registerServer(server: NeuroLinkMCPServer): Promise<void>;
-/**
- * Utility function to execute tool with default registry
- *
- * @param toolName Tool name to execute
- * @param params Tool parameters
- * @param context Execution context
- * @param options Execution options
- * @returns Tool execution result
- */
-export declare function executeTool(toolName: string, params: any, context: NeuroLinkExecutionContext, options?: ToolExecutionOptions): Promise<ToolResult>;
-/**
- * Utility function to list tools with default registry
- *
- * @param criteria Search criteria
- * @returns Array of tool information
+ * Default registry instance
  */
-export declare function listTools(criteria?: ToolSearchCriteria): {
-    name: string;
-    qualifiedName: string;
-    description: string;
-    server: string;
-    serverTitle: string;
-    category?: string;
-    serverCategory?: string;
-    permissions?: string[];
-    isImplemented?: boolean;
-}[];
+export declare const mcpRegistry: MCPRegistry;
+export type MCPToolRegistry = MCPRegistry;
+export declare const defaultToolRegistry: MCPRegistry;
+export { mcpRegistry as defaultMCPRegistry };
+export interface ToolExecutionOptions {
+    preferredSource?: string;
+    fallbackEnabled?: boolean;
+    validateBeforeExecution?: boolean;
+    timeoutMs?: number;
+}

package/dist/lib/mcp/registry.js

@@ -1,400 +1,59 @@
 /**
- * NeuroLink MCP Tool Registry System
- * Central registry for managing MCP servers and tools with execution capabilities
- * Supports tool discovery, registration, and orchestrated execution
+ * MCP Registry - Plugin Registration and Management
  */
-import { ContextManager, ContextValidator } from "./context-manager.js";
-import { registryLogger, mcpLogger } from "./logging.js";
+import { registryLogger } from "./logging.js";
 /**
- * Central MCP Tool Registry
- * Manages all MCP servers and their tools with advanced execution capabilities
+ * Simple MCP registry for plugin management
  */
-export class MCPToolRegistry {
-    servers = new Map();
-    tools = new Map();
-    contextManager;
-    // Execution tracking
-    executionCount = 0;
-    totalExecutionTime = 0;
-    errorCount = 0;
-    constructor(contextManager) {
-        this.contextManager = contextManager || new ContextManager();
-    }
-    /**
-     * Register an MCP server and all its tools
-     *
-     * @param server MCP server to register
-     * @throws Error if server ID already exists
-     */
-    async registerServer(server) {
-        // Check for duplicate server ID - if already registered, skip silently
-        if (this.servers.has(server.id)) {
-            mcpLogger.debug(`Server with ID '${server.id}' is already registered, skipping registration`);
-            return;
-        }
-        // Register the server
-        this.servers.set(server.id, server);
-        // Register all tools from the server
-        for (const [toolName, tool] of Object.entries(server.tools)) {
-            await this.registerToolFromServer(server, toolName, tool);
-        }
-        registryLogger.debug(`Registered server '${server.id}' with ${Object.keys(server.tools).length} tools`);
-    }
-    /**
-     * Register a single tool from a server
-     *
-     * @param server Source server
-     * @param toolName Tool name
-     * @param tool Tool implementation
-     */
-    async registerToolFromServer(server, toolName, tool) {
-        const qualifiedName = `${server.id}.${toolName}`;
-        const simpleName = toolName;
-        const registration = {
-            tool,
-            serverId: server.id,
-            serverTitle: server.title,
-            serverCategory: server.category,
-            qualifiedName,
-            simpleName,
-            registeredAt: Date.now(),
-        };
-        // Register with both qualified and simple names
-        this.tools.set(qualifiedName, registration);
-        // Only register simple name if it doesn't conflict
-        if (!this.tools.has(simpleName)) {
-            this.tools.set(simpleName, registration);
-        }
-        else {
-            registryLogger.warn(`Tool name conflict: '${simpleName}' already exists, use qualified name '${qualifiedName}'`);
-        }
-    }
+export class MCPRegistry {
+    plugins = new Map();
     /**
-     * Execute a tool with comprehensive error handling and context tracking
-     *
-     * @param toolName Tool name (simple or qualified)
-     * @param params Tool parameters
-     * @param context Execution context
-     * @param options Execution options
-     * @returns Tool execution result
+     * Register a plugin
      */
-    async executeTool(toolName, params, context, options = {}) {
-        const startTime = Date.now();
-        const { validateInput = true, validatePermissions = true, trackMetrics = true, timeoutMs = 30000, } = options;
-        try {
-            // Find the tool
-            const registration = this.tools.get(toolName);
-            if (!registration) {
-                throw new Error(`Tool not found: ${toolName}`);
-            }
-            const { tool, serverId, serverTitle } = registration;
-            // Validate context
-            if (validatePermissions) {
-                const contextValidation = ContextValidator.validateContext(context);
-                if (!contextValidation.isValid) {
-                    throw new Error(`Context validation failed: ${contextValidation.errors.join(", ")}`);
-                }
-                // Check tool permissions
-                if (tool.permissions &&
-                    !ContextValidator.hasPermissions(context, tool.permissions)) {
-                    throw new Error(`Insufficient permissions for tool '${toolName}'. Required: ${tool.permissions.join(", ")}`);
-                }
-            }
-            // Validate input parameters if schema provided
-            if (validateInput && tool.inputSchema) {
-                try {
-                    tool.inputSchema.parse(params);
-                }
-                catch (error) {
-                    const errorMessage = error instanceof Error ? error.message : String(error);
-                    throw new Error(`Input validation failed for tool '${toolName}': ${errorMessage}`);
-                }
-            }
-            // Add tool to execution chain
-            this.contextManager.addToToolChain(context, toolName);
-            // Execute tool with timeout
-            const executeWithTimeout = this.createTimeoutPromise(tool.execute(params, context), timeoutMs, `Tool '${toolName}' execution timeout`);
-            const result = await executeWithTimeout;
-            // Add execution metadata
-            const executionTime = Date.now() - startTime;
-            const enhancedResult = {
-                ...result,
-                metadata: {
-                    ...result.metadata,
-                    toolName,
-                    serverId,
-                    serverTitle,
-                    sessionId: context.sessionId,
-                    timestamp: Date.now(),
-                    executionTime,
-                },
-            };
-            // Track metrics
-            if (trackMetrics) {
-                this.updateExecutionMetrics(executionTime, result.success);
-            }
-            // Validate output if schema provided
-            if (tool.outputSchema && result.success) {
-                try {
-                    tool.outputSchema.parse(result.data);
-                }
-                catch (error) {
-                    const errorMessage = error instanceof Error ? error.message : String(error);
-                    registryLogger.warn(`Output validation warning for tool '${toolName}': ${errorMessage}`);
-                }
-            }
-            registryLogger.debug(`Executed tool '${toolName}' in ${executionTime}ms - ${result.success ? "SUCCESS" : "FAILED"}`);
-            return enhancedResult;
-        }
-        catch (error) {
-            const executionTime = Date.now() - startTime;
-            // Track error metrics
-            if (trackMetrics) {
-                this.updateExecutionMetrics(executionTime, false);
-            }
-            const errorMessage = error instanceof Error ? error.message : String(error);
-            const errorResult = {
-                success: false,
-                error: errorMessage,
-                metadata: {
-                    toolName,
-                    sessionId: context.sessionId,
-                    timestamp: Date.now(),
-                    executionTime,
-                },
-            };
-            registryLogger.error(`Tool execution failed '${toolName}': ${errorMessage}`);
-            return errorResult;
-        }
+    register(plugin) {
+        this.plugins.set(plugin.metadata.name, plugin);
+        registryLogger.info(`Registered plugin: ${plugin.metadata.name}`);
     }
     /**
-     * List all available tools with optional filtering
-     *
-     * @param criteria Search criteria for filtering tools
-     * @returns Array of tool information
+     * Unregister a plugin
      */
-    listTools(criteria = {}) {
-        const tools = [];
-        // Get unique tools (prefer qualified names over simple names)
-        const uniqueTools = new Map();
-        for (const [name, registration] of this.tools) {
-            if (name.includes(".")) {
-                // Qualified name
-                uniqueTools.set(registration.qualifiedName, registration);
-            }
-            else if (!uniqueTools.has(registration.qualifiedName)) {
-                uniqueTools.set(registration.qualifiedName, registration);
-            }
-        }
-        for (const registration of uniqueTools.values()) {
-            const { tool, serverId, serverTitle, serverCategory, qualifiedName, simpleName, } = registration;
-            // Apply filters
-            if (criteria.name &&
-                !simpleName.toLowerCase().includes(criteria.name.toLowerCase())) {
-                continue;
-            }
-            if (criteria.category && tool.category !== criteria.category) {
-                continue;
-            }
-            if (criteria.serverId && serverId !== criteria.serverId) {
-                continue;
-            }
-            if (criteria.serverCategory &&
-                serverCategory !== criteria.serverCategory) {
-                continue;
-            }
-            if (criteria.implemented !== undefined &&
-                tool.isImplemented !== criteria.implemented) {
-                continue;
-            }
-            if (criteria.permissions && criteria.permissions.length > 0) {
-                const toolPermissions = tool.permissions || [];
-                const hasAllPermissions = criteria.permissions.every((p) => toolPermissions.includes(p));
-                if (!hasAllPermissions) {
-                    continue;
-                }
-            }
-            tools.push({
-                name: simpleName,
-                qualifiedName,
-                description: tool.description,
-                server: serverId,
-                serverTitle,
-                category: tool.category,
-                serverCategory,
-                permissions: tool.permissions,
-                isImplemented: tool.isImplemented,
-            });
+    unregister(name) {
+        const removed = this.plugins.delete(name);
+        if (removed) {
+            registryLogger.info(`Unregistered plugin: ${name}`);
         }
-        return tools.sort((a, b) => a.qualifiedName.localeCompare(b.qualifiedName));
+        return removed;
     }
     /**
-     * Get detailed information about a specific tool
-     *
-     * @param toolName Tool name (simple or qualified)
-     * @returns Detailed tool information or undefined if not found
+     * Get a plugin
      */
-    getToolInfo(toolName) {
-        const registration = this.tools.get(toolName);
-        if (!registration) {
-            return undefined;
-        }
-        const server = this.servers.get(registration.serverId);
-        if (!server) {
-            return undefined;
-        }
-        return {
-            tool: registration.tool,
-            server,
-            registration,
-        };
+    get(name) {
+        return this.plugins.get(name);
     }
     /**
-     * Get registry statistics
-     *
-     * @returns Comprehensive registry statistics
+     * List all plugins
      */
-    getStats() {
-        const toolsByCategory = {};
-        const serversByCategory = {};
-        // Count tools by category
-        for (const registration of this.tools.values()) {
-            const category = registration.tool.category || "uncategorized";
-            toolsByCategory[category] = (toolsByCategory[category] || 0) + 1;
-        }
-        // Count servers by category
-        for (const server of this.servers.values()) {
-            const category = server.category || "uncategorized";
-            serversByCategory[category] = (serversByCategory[category] || 0) + 1;
-        }
-        return {
-            totalServers: this.servers.size,
-            totalTools: new Set(Array.from(this.tools.values()).map((r) => r.qualifiedName)).size,
-            toolsByCategory,
-            serversByCategory,
-            executionCount: this.executionCount,
-            averageExecutionTime: this.executionCount > 0
-                ? this.totalExecutionTime / this.executionCount
-                : 0,
-            errorRate: this.executionCount > 0 ? this.errorCount / this.executionCount : 0,
-        };
+    list() {
+        return Array.from(this.plugins.values());
     }
     /**
-     * Unregister a server and all its tools
-     *
-     * @param serverId Server ID to unregister
-     * @returns Whether server was found and removed
+     * Check if plugin exists
      */
-    unregisterServer(serverId) {
-        const server = this.servers.get(serverId);
-        if (!server) {
-            return false;
-        }
-        // Remove all tools from this server
-        const toolsToRemove = [];
-        for (const [name, registration] of this.tools) {
-            if (registration.serverId === serverId) {
-                toolsToRemove.push(name);
-            }
-        }
-        for (const toolName of toolsToRemove) {
-            this.tools.delete(toolName);
-        }
-        // Remove the server
-        this.servers.delete(serverId);
-        registryLogger.debug(`Unregistered server '${serverId}' and ${toolsToRemove.length} tools`);
-        return true;
+    has(name) {
+        return this.plugins.has(name);
     }
     /**
-     * Clear all servers and tools
+     * Clear all plugins
      */
     clear() {
-        this.servers.clear();
-        this.tools.clear();
-        this.executionCount = 0;
-        this.totalExecutionTime = 0;
-        this.errorCount = 0;
-        registryLogger.debug("Cleared all servers and tools");
-    }
-    /**
-     * Create timeout promise wrapper
-     *
-     * @param promise Promise to wrap
-     * @param timeoutMs Timeout in milliseconds
-     * @param timeoutMessage Error message for timeout
-     * @returns Promise that rejects on timeout
-     */
-    createTimeoutPromise(promise, timeoutMs, timeoutMessage) {
-        return Promise.race([
-            promise,
-            new Promise((_, reject) => {
-                setTimeout(() => reject(new Error(timeoutMessage)), timeoutMs);
-            }),
-        ]);
-    }
-    /**
-     * Update execution metrics
-     *
-     * @param executionTime Execution time in milliseconds
-     * @param success Whether execution was successful
-     */
-    updateExecutionMetrics(executionTime, success) {
-        this.executionCount++;
-        this.totalExecutionTime += executionTime;
-        if (!success) {
-            this.errorCount++;
-        }
+        this.plugins.clear();
+        registryLogger.info("Registry cleared");
     }
 }
 /**
  * Default registry instance
- * Can be used across the application for consistent tool management
  */
-let defaultRegistryInstance = null;
-/**
- * Get the default singleton instance of the MCPToolRegistry.
- * This function ensures that the registry is only instantiated once.
- * Using a function getter helps prevent circular dependency issues during module initialization.
- *
- * @returns The singleton MCPToolRegistry instance.
- */
-export function getDefaultToolRegistry() {
-    if (!defaultRegistryInstance) {
-        defaultRegistryInstance = new MCPToolRegistry();
-    }
-    return defaultRegistryInstance;
-}
-/**
- * @deprecated The direct export `defaultToolRegistry` is deprecated and will be removed.
- * Please use `getDefaultToolRegistry()` instead to avoid circular dependency issues.
- */
-export const defaultToolRegistry = getDefaultToolRegistry();
-/**
- * Utility function to register server with default registry
- *
- * @param server MCP server to register
- */
-export async function registerServer(server) {
-    return getDefaultToolRegistry().registerServer(server);
-}
-/**
- * Utility function to execute tool with default registry
- *
- * @param toolName Tool name to execute
- * @param params Tool parameters
- * @param context Execution context
- * @param options Execution options
- * @returns Tool execution result
- */
-export async function executeTool(toolName, params, context, options) {
-    return getDefaultToolRegistry().executeTool(toolName, params, context, options);
-}
-/**
- * Utility function to list tools with default registry
- *
- * @param criteria Search criteria
- * @returns Array of tool information
- */
-export function listTools(criteria) {
-    return getDefaultToolRegistry().listTools(criteria);
-}
+export const mcpRegistry = new MCPRegistry();
+export const defaultToolRegistry = mcpRegistry;
+// Additional exports for compatibility
+export { mcpRegistry as defaultMCPRegistry };