@umsai/ums-code 0.1.4-v2 → 0.3.0-v2

package/dist/src/nonInteractive/control/ControlService.js

@@ -0,0 +1,154 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Control Service
+ *
+ * Facade layer providing domain-grouped APIs for control plane operations.
+ * Shares controller instances with ControlDispatcher to ensure single source
+ * of truth and state consistency.
+ */
+export class ControlService {
+    dispatcher;
+    /**
+     * Construct ControlService
+     *
+     * @param context - Control context (unused directly, passed to dispatcher)
+     * @param dispatcher - Control dispatcher that owns the controller instances
+     */
+    constructor(context, dispatcher) {
+        this.dispatcher = dispatcher;
+    }
+    /**
+     * Permission Domain API
+     *
+     * Handles tool execution permissions, approval checks, and callbacks.
+     * Delegates to the shared PermissionController instance.
+     */
+    // get permission(): PermissionServiceAPI {
+    //   const controller = this.dispatcher.permissionController;
+    //   return {
+    //     /**
+    //      * Check if a tool should be allowed based on current permission settings
+    //      *
+    //      * Evaluates permission mode and tool registry to determine if execution
+    //      * should proceed. Can optionally modify tool arguments based on confirmation details.
+    //      *
+    //      * @param toolRequest - Tool call request information
+    //      * @param confirmationDetails - Optional confirmation details for UI
+    //      * @returns Permission decision with optional updated arguments
+    //      */
+    //     shouldAllowTool: controller.shouldAllowTool.bind(controller),
+    //
+    //     /**
+    //      * Build UI suggestions for tool confirmation dialogs
+    //      *
+    //      * Creates actionable permission suggestions based on tool confirmation details.
+    //      *
+    //      * @param confirmationDetails - Tool confirmation details
+    //      * @returns Array of permission suggestions or null
+    //      */
+    //     buildPermissionSuggestions:
+    //       controller.buildPermissionSuggestions.bind(controller),
+    //
+    //     /**
+    //      * Get callback for monitoring tool call status updates
+    //      *
+    //      * Returns callback function for integration with CoreToolScheduler.
+    //      *
+    //      * @returns Callback function for tool call updates
+    //      */
+    //     getToolCallUpdateCallback:
+    //       controller.getToolCallUpdateCallback.bind(controller),
+    //   };
+    // }
+    /**
+     * System Domain API
+     *
+     * Handles system-level operations and session management.
+     * Delegates to the shared SystemController instance.
+     */
+    get system() {
+        const controller = this.dispatcher.systemController;
+        return {
+            /**
+             * Get control capabilities
+             *
+             * Returns the control capabilities object indicating what control
+             * features are available. Used exclusively for the initialize
+             * control response. System messages do not include capabilities.
+             *
+             * @returns Control capabilities object
+             */
+            getControlCapabilities: () => controller.buildControlCapabilities(),
+        };
+    }
+    /**
+     * MCP Domain API
+     *
+     * Handles Model Context Protocol server interactions.
+     * Delegates to the shared MCPController instance.
+     */
+    // get mcp(): McpServiceAPI {
+    //   return {
+    //     /**
+    //      * Get or create MCP client for a server (lazy initialization)
+    //      *
+    //      * Returns existing client or creates new connection.
+    //      *
+    //      * @param serverName - Name of the MCP server
+    //      * @returns Promise with client and config
+    //      */
+    //     getMcpClient: async (serverName: string) => {
+    //       // MCPController has a private method getOrCreateMcpClient
+    //       // We need to expose it via the API
+    //       // For now, throw error as placeholder
+    //       // The actual implementation will be added when we update MCPController
+    //       throw new Error(
+    //         `getMcpClient not yet implemented in ControlService. Server: ${serverName}`,
+    //       );
+    //     },
+    //
+    //     /**
+    //      * List all available MCP servers
+    //      *
+    //      * Returns names of configured/connected MCP servers.
+    //      *
+    //      * @returns Array of server names
+    //      */
+    //     listServers: () => {
+    //       // Get servers from context
+    //       const sdkServers = Array.from(
+    //         this.dispatcher.mcpController['context'].sdkMcpServers,
+    //       );
+    //       const cliServers = Array.from(
+    //         this.dispatcher.mcpController['context'].mcpClients.keys(),
+    //       );
+    //       return [...new Set([...sdkServers, ...cliServers])];
+    //     },
+    //   };
+    // }
+    /**
+     * Hook Domain API
+     *
+     * Handles hook callback processing (placeholder for future expansion).
+     * Delegates to the shared HookController instance.
+     */
+    // get hook(): HookServiceAPI {
+    //   // HookController has no public methods yet - controller access reserved for future use
+    //   return {};
+    // }
+    /**
+     * Cleanup all controllers
+     *
+     * Should be called on session shutdown. Delegates to dispatcher's shutdown
+     * method to ensure all controllers are properly cleaned up.
+     */
+    cleanup() {
+        // Delegate to dispatcher which manages controller cleanup
+        this.dispatcher.shutdown();
+    }
+}
+//# sourceMappingURL=ControlService.js.map

package/dist/src/nonInteractive/control/ControlService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ControlService.js","sourceRoot":"","sources":["../../../../src/nonInteractive/control/ControlService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAiCH;;;;;;GAMG;AACH,MAAM,OAAO,cAAc;IACjB,UAAU,CAAoB;IAEtC;;;;;OAKG;IACH,YAAY,OAAwB,EAAE,UAA6B;QACjE,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED;;;;;OAKG;IACH,2CAA2C;IAC3C,6DAA6D;IAC7D,aAAa;IACb,UAAU;IACV,gFAAgF;IAChF,SAAS;IACT,+EAA+E;IAC/E,6FAA6F;IAC7F,SAAS;IACT,4DAA4D;IAC5D,2EAA2E;IAC3E,sEAAsE;IACtE,UAAU;IACV,oEAAoE;IACpE,EAAE;IACF,UAAU;IACV,4DAA4D;IAC5D,SAAS;IACT,uFAAuF;IACvF,SAAS;IACT,gEAAgE;IAChE,0DAA0D;IAC1D,UAAU;IACV,kCAAkC;IAClC,gEAAgE;IAChE,EAAE;IACF,UAAU;IACV,8DAA8D;IAC9D,SAAS;IACT,2EAA2E;IAC3E,SAAS;IACT,0DAA0D;IAC1D,UAAU;IACV,iCAAiC;IACjC,+DAA+D;IAC/D,OAAO;IACP,IAAI;IAEJ;;;;;OAKG;IACH,IAAI,MAAM;QACR,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC;QACpD,OAAO;YACL;;;;;;;;eAQG;YACH,sBAAsB,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,wBAAwB,EAAE;SACpE,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,6BAA6B;IAC7B,aAAa;IACb,UAAU;IACV,qEAAqE;IACrE,SAAS;IACT,4DAA4D;IAC5D,SAAS;IACT,oDAAoD;IACpD,iDAAiD;IACjD,UAAU;IACV,oDAAoD;IACpD,mEAAmE;IACnE,4CAA4C;IAC5C,+CAA+C;IAC/C,gFAAgF;IAChF,yBAAyB;IACzB,uFAAuF;IACvF,WAAW;IACX,SAAS;IACT,EAAE;IACF,UAAU;IACV,wCAAwC;IACxC,SAAS;IACT,4DAA4D;IAC5D,SAAS;IACT,wCAAwC;IACxC,UAAU;IACV,2BAA2B;IAC3B,oCAAoC;IACpC,uCAAuC;IACvC,kEAAkE;IAClE,WAAW;IACX,uCAAuC;IACvC,sEAAsE;IACtE,WAAW;IACX,6DAA6D;IAC7D,SAAS;IACT,OAAO;IACP,IAAI;IAEJ;;;;;OAKG;IACH,+BAA+B;IAC/B,4FAA4F;IAC5F,eAAe;IACf,IAAI;IAEJ;;;;;OAKG;IACH,OAAO;QACL,0DAA0D;QAC1D,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC;CACF"}

package/dist/src/nonInteractive/control/controllers/baseController.d.ts

@@ -0,0 +1,50 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { IControlContext } from '../ControlContext.js';
+import type { ControlRequestPayload, ControlResponse } from '../../types.js';
+/**
+ * Registry interface for controllers to register/deregister pending requests
+ */
+export interface IPendingRequestRegistry {
+    registerIncomingRequest(requestId: string, controller: string, abortController: AbortController, timeoutId: NodeJS.Timeout): void;
+    deregisterIncomingRequest(requestId: string): void;
+    registerOutgoingRequest(requestId: string, controller: string, resolve: (response: ControlResponse) => void, reject: (error: Error) => void, timeoutId: NodeJS.Timeout): void;
+    deregisterOutgoingRequest(requestId: string): void;
+}
+/**
+ * Abstract base controller class
+ *
+ * Subclasses should implement handleRequestPayload() to process specific
+ * control request types.
+ */
+export declare abstract class BaseController {
+    protected context: IControlContext;
+    protected registry: IPendingRequestRegistry;
+    protected controllerName: string;
+    constructor(context: IControlContext, registry: IPendingRequestRegistry, controllerName: string);
+    /**
+     * Handle an incoming control request
+     *
+     * Manages lifecycle: register -> process -> deregister
+     */
+    handleRequest(payload: ControlRequestPayload, requestId: string): Promise<Record<string, unknown>>;
+    /**
+     * Send an outgoing control request to SDK
+     *
+     * Manages lifecycle: register -> send -> wait for response -> deregister
+     */
+    sendControlRequest(payload: ControlRequestPayload, timeoutMs?: number): Promise<ControlResponse>;
+    /**
+     * Abstract method: Handle specific request payload
+     *
+     * Subclasses must implement this to process their domain-specific requests.
+     */
+    protected abstract handleRequestPayload(payload: ControlRequestPayload, signal: AbortSignal): Promise<Record<string, unknown>>;
+    /**
+     * Cleanup resources
+     */
+    cleanup(): void;
+}

package/dist/src/nonInteractive/control/controllers/baseController.js

@@ -0,0 +1,102 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Base Controller
+ *
+ * Abstract base class for domain-specific control plane controllers.
+ * Provides common functionality for:
+ * - Handling incoming control requests (SDK -> CLI)
+ * - Sending outgoing control requests (CLI -> SDK)
+ * - Request lifecycle management with timeout and cancellation
+ * - Integration with central pending request registry
+ */
+import { randomUUID } from 'node:crypto';
+const DEFAULT_REQUEST_TIMEOUT_MS = 30000; // 30 seconds
+/**
+ * Abstract base controller class
+ *
+ * Subclasses should implement handleRequestPayload() to process specific
+ * control request types.
+ */
+export class BaseController {
+    context;
+    registry;
+    controllerName;
+    constructor(context, registry, controllerName) {
+        this.context = context;
+        this.registry = registry;
+        this.controllerName = controllerName;
+    }
+    /**
+     * Handle an incoming control request
+     *
+     * Manages lifecycle: register -> process -> deregister
+     */
+    async handleRequest(payload, requestId) {
+        const requestAbortController = new AbortController();
+        // Setup timeout
+        const timeoutId = setTimeout(() => {
+            requestAbortController.abort();
+            this.registry.deregisterIncomingRequest(requestId);
+            if (this.context.debugMode) {
+                console.error(`[${this.controllerName}] Request timeout: ${requestId}`);
+            }
+        }, DEFAULT_REQUEST_TIMEOUT_MS);
+        // Register with central registry
+        this.registry.registerIncomingRequest(requestId, this.controllerName, requestAbortController, timeoutId);
+        try {
+            const response = await this.handleRequestPayload(payload, requestAbortController.signal);
+            // Success - deregister
+            this.registry.deregisterIncomingRequest(requestId);
+            return response;
+        }
+        catch (error) {
+            // Error - deregister
+            this.registry.deregisterIncomingRequest(requestId);
+            throw error;
+        }
+    }
+    /**
+     * Send an outgoing control request to SDK
+     *
+     * Manages lifecycle: register -> send -> wait for response -> deregister
+     */
+    async sendControlRequest(payload, timeoutMs = DEFAULT_REQUEST_TIMEOUT_MS) {
+        const requestId = randomUUID();
+        return new Promise((resolve, reject) => {
+            // Setup timeout
+            const timeoutId = setTimeout(() => {
+                this.registry.deregisterOutgoingRequest(requestId);
+                reject(new Error('Control request timeout'));
+                if (this.context.debugMode) {
+                    console.error(`[${this.controllerName}] Outgoing request timeout: ${requestId}`);
+                }
+            }, timeoutMs);
+            // Register with central registry
+            this.registry.registerOutgoingRequest(requestId, this.controllerName, resolve, reject, timeoutId);
+            // Send control request
+            const request = {
+                type: 'control_request',
+                request_id: requestId,
+                request: payload,
+            };
+            try {
+                this.context.streamJson.send(request);
+            }
+            catch (error) {
+                this.registry.deregisterOutgoingRequest(requestId);
+                reject(error);
+            }
+        });
+    }
+    /**
+     * Cleanup resources
+     */
+    cleanup() {
+        // Subclasses can override to add cleanup logic
+    }
+}
+//# sourceMappingURL=baseController.js.map

package/dist/src/nonInteractive/control/controllers/baseController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"baseController.js","sourceRoot":"","sources":["../../../../../src/nonInteractive/control/controllers/baseController.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;;;GASG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAQzC,MAAM,0BAA0B,GAAG,KAAK,CAAC,CAAC,aAAa;AAwBvD;;;;;GAKG;AACH,MAAM,OAAgB,cAAc;IACxB,OAAO,CAAkB;IACzB,QAAQ,CAA0B;IAClC,cAAc,CAAS;IAEjC,YACE,OAAwB,EACxB,QAAiC,EACjC,cAAsB;QAEtB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,aAAa,CACjB,OAA8B,EAC9B,SAAiB;QAEjB,MAAM,sBAAsB,GAAG,IAAI,eAAe,EAAE,CAAC;QAErD,gBAAgB;QAChB,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,sBAAsB,CAAC,KAAK,EAAE,CAAC;YAC/B,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;YACnD,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,cAAc,sBAAsB,SAAS,EAAE,CAAC,CAAC;YAC1E,CAAC;QACH,CAAC,EAAE,0BAA0B,CAAC,CAAC;QAE/B,iCAAiC;QACjC,IAAI,CAAC,QAAQ,CAAC,uBAAuB,CACnC,SAAS,EACT,IAAI,CAAC,cAAc,EACnB,sBAAsB,EACtB,SAAS,CACV,CAAC;QAEF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAC9C,OAAO,EACP,sBAAsB,CAAC,MAAM,CAC9B,CAAC;YAEF,uBAAuB;YACvB,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;YAEnD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,qBAAqB;YACrB,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;YACnD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,kBAAkB,CACtB,OAA8B,EAC9B,YAAoB,0BAA0B;QAE9C,MAAM,SAAS,GAAG,UAAU,EAAE,CAAC;QAE/B,OAAO,IAAI,OAAO,CAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtD,gBAAgB;YAChB,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;gBAChC,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;gBACnD,MAAM,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC,CAAC;gBAC7C,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;oBAC3B,OAAO,CAAC,KAAK,CACX,IAAI,IAAI,CAAC,cAAc,+BAA+B,SAAS,EAAE,CAClE,CAAC;gBACJ,CAAC;YACH,CAAC,EAAE,SAAS,CAAC,CAAC;YAEd,iCAAiC;YACjC,IAAI,CAAC,QAAQ,CAAC,uBAAuB,CACnC,SAAS,EACT,IAAI,CAAC,cAAc,EACnB,OAAO,EACP,MAAM,EACN,SAAS,CACV,CAAC;YAEF,uBAAuB;YACvB,MAAM,OAAO,GAAsB;gBACjC,IAAI,EAAE,iBAAiB;gBACvB,UAAU,EAAE,SAAS;gBACrB,OAAO,EAAE,OAAO;aACjB,CAAC;YAEF,IAAI,CAAC;gBACH,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,QAAQ,CAAC,yBAAyB,CAAC,SAAS,CAAC,CAAC;gBACnD,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAYD;;OAEG;IACH,OAAO;QACL,+CAA+C;IACjD,CAAC;CACF"}

package/dist/src/nonInteractive/control/controllers/hookController.d.ts

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Hook Controller
+ *
+ * Handles hook-related control requests:
+ * - hook_callback: Process hook callbacks (placeholder for future)
+ */
+import { BaseController } from './baseController.js';
+import type { ControlRequestPayload } from '../../types.js';
+export declare class HookController extends BaseController {
+    /**
+     * Handle hook control requests
+     */
+    protected handleRequestPayload(payload: ControlRequestPayload, _signal: AbortSignal): Promise<Record<string, unknown>>;
+    /**
+     * Handle hook_callback request
+     *
+     * Processes hook callbacks (placeholder implementation)
+     */
+    private handleHookCallback;
+}

package/dist/src/nonInteractive/control/controllers/hookController.js

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Hook Controller
+ *
+ * Handles hook-related control requests:
+ * - hook_callback: Process hook callbacks (placeholder for future)
+ */
+import { BaseController } from './baseController.js';
+export class HookController extends BaseController {
+    /**
+     * Handle hook control requests
+     */
+    async handleRequestPayload(payload, _signal) {
+        switch (payload.subtype) {
+            case 'hook_callback':
+                return this.handleHookCallback(payload);
+            default:
+                throw new Error(`Unsupported request subtype in HookController`);
+        }
+    }
+    /**
+     * Handle hook_callback request
+     *
+     * Processes hook callbacks (placeholder implementation)
+     */
+    async handleHookCallback(payload) {
+        if (this.context.debugMode) {
+            console.error(`[HookController] Hook callback: ${payload.callback_id}`);
+        }
+        // Hook callback processing not yet implemented
+        return {
+            result: 'Hook callback processing not yet implemented',
+            callback_id: payload.callback_id,
+            tool_use_id: payload.tool_use_id,
+        };
+    }
+}
+//# sourceMappingURL=hookController.js.map

package/dist/src/nonInteractive/control/controllers/hookController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hookController.js","sourceRoot":"","sources":["../../../../../src/nonInteractive/control/controllers/hookController.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;GAKG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAMrD,MAAM,OAAO,cAAe,SAAQ,cAAc;IAChD;;OAEG;IACO,KAAK,CAAC,oBAAoB,CAClC,OAA8B,EAC9B,OAAoB;QAEpB,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,KAAK,eAAe;gBAClB,OAAO,IAAI,CAAC,kBAAkB,CAAC,OAAiC,CAAC,CAAC;YAEpE;gBACE,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,kBAAkB,CAC9B,OAA+B;QAE/B,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CAAC,mCAAmC,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;QAC1E,CAAC;QAED,+CAA+C;QAC/C,OAAO;YACL,MAAM,EAAE,8CAA8C;YACtD,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,WAAW,EAAE,OAAO,CAAC,WAAW;SACjC,CAAC;IACJ,CAAC;CACF"}

package/dist/src/nonInteractive/control/controllers/mcpController.d.ts

@@ -0,0 +1,42 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * MCP Controller
+ *
+ * Handles MCP-related control requests:
+ * - mcp_message: Route MCP messages
+ * - mcp_server_status: Return MCP server status
+ */
+import { BaseController } from './baseController.js';
+import type { ControlRequestPayload } from '../../types.js';
+export declare class MCPController extends BaseController {
+    /**
+     * Handle MCP control requests
+     */
+    protected handleRequestPayload(payload: ControlRequestPayload, _signal: AbortSignal): Promise<Record<string, unknown>>;
+    /**
+     * Handle mcp_message request
+     *
+     * Routes JSON-RPC messages to MCP servers
+     */
+    private handleMcpMessage;
+    /**
+     * Handle mcp_server_status request
+     *
+     * Returns status of registered MCP servers
+     */
+    private handleMcpStatus;
+    /**
+     * Get or create MCP client for a server
+     *
+     * Implements lazy connection and caching
+     */
+    private getOrCreateMcpClient;
+    /**
+     * Cleanup MCP clients
+     */
+    cleanup(): void;
+}

package/dist/src/nonInteractive/control/controllers/mcpController.js

@@ -0,0 +1,205 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * MCP Controller
+ *
+ * Handles MCP-related control requests:
+ * - mcp_message: Route MCP messages
+ * - mcp_server_status: Return MCP server status
+ */
+import { BaseController } from './baseController.js';
+import { ResultSchema } from '@modelcontextprotocol/sdk/types.js';
+import { connectToMcpServer, MCP_DEFAULT_TIMEOUT_MSEC, } from '@umsai/ums-code-core';
+export class MCPController extends BaseController {
+    /**
+     * Handle MCP control requests
+     */
+    async handleRequestPayload(payload, _signal) {
+        switch (payload.subtype) {
+            case 'mcp_message':
+                return this.handleMcpMessage(payload);
+            case 'mcp_server_status':
+                return this.handleMcpStatus();
+            default:
+                throw new Error(`Unsupported request subtype in MCPController`);
+        }
+    }
+    /**
+     * Handle mcp_message request
+     *
+     * Routes JSON-RPC messages to MCP servers
+     */
+    async handleMcpMessage(payload) {
+        const serverNameRaw = payload.server_name;
+        if (typeof serverNameRaw !== 'string' ||
+            serverNameRaw.trim().length === 0) {
+            throw new Error('Missing server_name in mcp_message request');
+        }
+        const message = payload.message;
+        if (!message || typeof message !== 'object') {
+            throw new Error('Missing or invalid message payload for mcp_message request');
+        }
+        // Get or create MCP client
+        let clientEntry;
+        try {
+            clientEntry = await this.getOrCreateMcpClient(serverNameRaw.trim());
+        }
+        catch (error) {
+            throw new Error(error instanceof Error
+                ? error.message
+                : 'Failed to connect to MCP server');
+        }
+        const method = message.method;
+        if (typeof method !== 'string' || method.trim().length === 0) {
+            throw new Error('Invalid MCP message: missing method');
+        }
+        const jsonrpcVersion = typeof message.jsonrpc === 'string' ? message.jsonrpc : '2.0';
+        const messageId = message.id;
+        const params = message.params;
+        const timeout = typeof clientEntry.config.timeout === 'number'
+            ? clientEntry.config.timeout
+            : MCP_DEFAULT_TIMEOUT_MSEC;
+        try {
+            // Handle notification (no id)
+            if (messageId === undefined) {
+                await clientEntry.client.notification({
+                    method,
+                    params,
+                });
+                return {
+                    subtype: 'mcp_message',
+                    mcp_response: {
+                        jsonrpc: jsonrpcVersion,
+                        id: null,
+                        result: { success: true, acknowledged: true },
+                    },
+                };
+            }
+            // Handle request (with id)
+            const result = await clientEntry.client.request({
+                method,
+                params,
+            }, ResultSchema, { timeout });
+            return {
+                subtype: 'mcp_message',
+                mcp_response: {
+                    jsonrpc: jsonrpcVersion,
+                    id: messageId,
+                    result,
+                },
+            };
+        }
+        catch (error) {
+            // If connection closed, remove from cache
+            if (error instanceof Error && /closed/i.test(error.message)) {
+                this.context.mcpClients.delete(serverNameRaw.trim());
+            }
+            const errorCode = typeof error?.code === 'number'
+                ? error.code
+                : -32603;
+            const errorMessage = error instanceof Error
+                ? error.message
+                : 'Failed to execute MCP request';
+            const errorData = error?.data;
+            const errorBody = {
+                code: errorCode,
+                message: errorMessage,
+            };
+            if (errorData !== undefined) {
+                errorBody['data'] = errorData;
+            }
+            return {
+                subtype: 'mcp_message',
+                mcp_response: {
+                    jsonrpc: jsonrpcVersion,
+                    id: messageId ?? null,
+                    error: errorBody,
+                },
+            };
+        }
+    }
+    /**
+     * Handle mcp_server_status request
+     *
+     * Returns status of registered MCP servers
+     */
+    async handleMcpStatus() {
+        const status = {};
+        // Include SDK MCP servers
+        for (const serverName of this.context.sdkMcpServers) {
+            status[serverName] = 'connected';
+        }
+        // Include CLI-managed MCP clients
+        for (const serverName of this.context.mcpClients.keys()) {
+            status[serverName] = 'connected';
+        }
+        if (this.context.debugMode) {
+            console.error(`[MCPController] MCP status: ${Object.keys(status).length} servers`);
+        }
+        return status;
+    }
+    /**
+     * Get or create MCP client for a server
+     *
+     * Implements lazy connection and caching
+     */
+    async getOrCreateMcpClient(serverName) {
+        // Check cache first
+        const cached = this.context.mcpClients.get(serverName);
+        if (cached) {
+            return cached;
+        }
+        // Get server configuration
+        const provider = this.context.config;
+        if (typeof provider.getMcpServers !== 'function') {
+            throw new Error(`MCP server "${serverName}" is not configured`);
+        }
+        const servers = provider.getMcpServers() ?? {};
+        const serverConfig = servers[serverName];
+        if (!serverConfig) {
+            throw new Error(`MCP server "${serverName}" is not configured`);
+        }
+        const debugMode = typeof provider.getDebugMode === 'function'
+            ? provider.getDebugMode()
+            : false;
+        const workspaceContext = typeof provider.getWorkspaceContext === 'function'
+            ? provider.getWorkspaceContext()
+            : undefined;
+        if (!workspaceContext) {
+            throw new Error('Workspace context is not available for MCP connection');
+        }
+        // Connect to MCP server
+        const client = await connectToMcpServer(serverName, serverConfig, debugMode, workspaceContext);
+        // Cache the client
+        const entry = { client, config: serverConfig };
+        this.context.mcpClients.set(serverName, entry);
+        if (this.context.debugMode) {
+            console.error(`[MCPController] Connected to MCP server: ${serverName}`);
+        }
+        return entry;
+    }
+    /**
+     * Cleanup MCP clients
+     */
+    cleanup() {
+        if (this.context.debugMode) {
+            console.error(`[MCPController] Cleaning up ${this.context.mcpClients.size} MCP clients`);
+        }
+        // Close all MCP clients
+        for (const [serverName, { client }] of this.context.mcpClients.entries()) {
+            try {
+                client.close();
+            }
+            catch (error) {
+                if (this.context.debugMode) {
+                    console.error(`[MCPController] Failed to close MCP client ${serverName}:`, error);
+                }
+            }
+        }
+        this.context.mcpClients.clear();
+    }
+}
+//# sourceMappingURL=mcpController.js.map

package/dist/src/nonInteractive/control/controllers/mcpController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcpController.js","sourceRoot":"","sources":["../../../../../src/nonInteractive/control/controllers/mcpController.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,OAAO,EAAE,YAAY,EAAE,MAAM,oCAAoC,CAAC;AASlE,OAAO,EACL,kBAAkB,EAClB,wBAAwB,GACzB,MAAM,sBAAsB,CAAC;AAE9B,MAAM,OAAO,aAAc,SAAQ,cAAc;IAC/C;;OAEG;IACO,KAAK,CAAC,oBAAoB,CAClC,OAA8B,EAC9B,OAAoB;QAEpB,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAsC,CAAC,CAAC;YAEvE,KAAK,mBAAmB;gBACtB,OAAO,IAAI,CAAC,eAAe,EAAE,CAAC;YAEhC;gBACE,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,gBAAgB,CAC5B,OAAoC;QAEpC,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,CAAC;QAC1C,IACE,OAAO,aAAa,KAAK,QAAQ;YACjC,aAAa,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EACjC,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAChC,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;QACJ,CAAC;QAED,2BAA2B;QAC3B,IAAI,WAAwD,CAAC;QAC7D,IAAI,CAAC;YACH,WAAW,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,KAAK,YAAY,KAAK;gBACpB,CAAC,CAAC,KAAK,CAAC,OAAO;gBACf,CAAC,CAAC,iCAAiC,CACtC,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QAED,MAAM,cAAc,GAClB,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC;QAChE,MAAM,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,MAAM,OAAO,GACX,OAAO,WAAW,CAAC,MAAM,CAAC,OAAO,KAAK,QAAQ;YAC5C,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO;YAC5B,CAAC,CAAC,wBAAwB,CAAC;QAE/B,IAAI,CAAC;YACH,8BAA8B;YAC9B,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC;oBACpC,MAAM;oBACN,MAAM;iBACP,CAAC,CAAC;gBACH,OAAO;oBACL,OAAO,EAAE,aAAa;oBACtB,YAAY,EAAE;wBACZ,OAAO,EAAE,cAAc;wBACvB,EAAE,EAAE,IAAI;wBACR,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE;qBAC9C;iBACF,CAAC;YACJ,CAAC;YAED,2BAA2B;YAC3B,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,OAAO,CAC7C;gBACE,MAAM;gBACN,MAAM;aACP,EACD,YAAY,EACZ,EAAE,OAAO,EAAE,CACZ,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE,aAAa;gBACtB,YAAY,EAAE;oBACZ,OAAO,EAAE,cAAc;oBACvB,EAAE,EAAE,SAAS;oBACb,MAAM;iBACP;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,0CAA0C;YAC1C,IAAI,KAAK,YAAY,KAAK,IAAI,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC5D,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,SAAS,GACb,OAAQ,KAA4B,EAAE,IAAI,KAAK,QAAQ;gBACrD,CAAC,CAAG,KAA0B,CAAC,IAAe;gBAC9C,CAAC,CAAC,CAAC,KAAK,CAAC;YACb,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK;gBACpB,CAAC,CAAC,KAAK,CAAC,OAAO;gBACf,CAAC,CAAC,+BAA+B,CAAC;YACtC,MAAM,SAAS,GAAI,KAA4B,EAAE,IAAI,CAAC;YAEtD,MAAM,SAAS,GAA4B;gBACzC,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,YAAY;aACtB,CAAC;YACF,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;gBAC5B,SAAS,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;YAChC,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,aAAa;gBACtB,YAAY,EAAE;oBACZ,OAAO,EAAE,cAAc;oBACvB,EAAE,EAAE,SAAS,IAAI,IAAI;oBACrB,KAAK,EAAE,SAAS;iBACjB;aACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,eAAe;QAC3B,MAAM,MAAM,GAA2B,EAAE,CAAC;QAE1C,0BAA0B;QAC1B,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;YACpD,MAAM,CAAC,UAAU,CAAC,GAAG,WAAW,CAAC;QACnC,CAAC;QAED,kCAAkC;QAClC,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC;YACxD,MAAM,CAAC,UAAU,CAAC,GAAG,WAAW,CAAC;QACnC,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CACX,+BAA+B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,UAAU,CACpE,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,oBAAoB,CAChC,UAAkB;QAElB,oBAAoB;QACpB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACvD,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,2BAA2B;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,MAI7B,CAAC;QAEF,IAAI,OAAO,QAAQ,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YACjD,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,qBAAqB,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,OAAO,GAAG,QAAQ,CAAC,aAAa,EAAE,IAAI,EAAE,CAAC;QAC/C,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,KAAK,CAAC,eAAe,UAAU,qBAAqB,CAAC,CAAC;QAClE,CAAC;QAED,MAAM,SAAS,GACb,OAAO,QAAQ,CAAC,YAAY,KAAK,UAAU;YACzC,CAAC,CAAC,QAAQ,CAAC,YAAY,EAAE;YACzB,CAAC,CAAC,KAAK,CAAC;QAEZ,MAAM,gBAAgB,GACpB,OAAO,QAAQ,CAAC,mBAAmB,KAAK,UAAU;YAChD,CAAC,CAAC,QAAQ,CAAC,mBAAmB,EAAE;YAChC,CAAC,CAAC,SAAS,CAAC;QAEhB,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,uDAAuD,CAAC,CAAC;QAC3E,CAAC;QAED,wBAAwB;QACxB,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,YAAY,EACZ,SAAS,EACT,gBAAoC,CACrC,CAAC;QAEF,mBAAmB;QACnB,MAAM,KAAK,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,CAAC;QAC/C,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAE/C,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CAAC,4CAA4C,UAAU,EAAE,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACM,OAAO;QACd,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CACX,+BAA+B,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,cAAc,CAC1E,CAAC;QACJ,CAAC;QAED,wBAAwB;QACxB,KAAK,MAAM,CAAC,UAAU,EAAE,EAAE,MAAM,EAAE,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACzE,IAAI,CAAC;gBACH,MAAM,CAAC,KAAK,EAAE,CAAC;YACjB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;oBAC3B,OAAO,CAAC,KAAK,CACX,8CAA8C,UAAU,GAAG,EAC3D,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;CACF"}