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

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

@@ -0,0 +1,78 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Permission Controller
+ *
+ * Handles permission-related control requests:
+ * - can_use_tool: Check if tool usage is allowed
+ * - set_permission_mode: Change permission mode at runtime
+ *
+ * Abstracts all permission logic from the session manager to keep it clean.
+ */
+import type { ToolCallRequestInfo } from '@umsai/ums-code-core';
+import type { ControlRequestPayload, PermissionSuggestion } from '../../types.js';
+import { BaseController } from './baseController.js';
+export declare class PermissionController extends BaseController {
+    private pendingOutgoingRequests;
+    /**
+     * Handle permission control requests
+     */
+    protected handleRequestPayload(payload: ControlRequestPayload, _signal: AbortSignal): Promise<Record<string, unknown>>;
+    /**
+     * Handle can_use_tool request
+     *
+     * Comprehensive permission evaluation based on:
+     * - Permission mode (approval level)
+     * - Tool registry validation
+     * - Error handling with safe defaults
+     */
+    private handleCanUseTool;
+    /**
+     * Check permission mode for tool execution
+     */
+    private checkPermissionMode;
+    /**
+     * Check if tool exists in registry
+     */
+    private checkToolRegistry;
+    /**
+     * Handle set_permission_mode request
+     *
+     * Updates the permission mode in the context
+     */
+    private handleSetPermissionMode;
+    /**
+     * Build permission suggestions for tool confirmation UI
+     *
+     * This method creates UI suggestions based on tool confirmation details,
+     * helping the host application present appropriate permission options.
+     */
+    buildPermissionSuggestions(confirmationDetails: unknown): PermissionSuggestion[] | null;
+    /**
+     * Check if a tool should be executed based on current permission settings
+     *
+     * This is a convenience method for direct tool execution checks without
+     * going through the control request flow.
+     */
+    shouldAllowTool(toolRequest: ToolCallRequestInfo, confirmationDetails?: unknown): Promise<{
+        allowed: boolean;
+        message?: string;
+        updatedArgs?: Record<string, unknown>;
+    }>;
+    /**
+     * Get callback for monitoring tool calls and handling outgoing permission requests
+     * This is passed to executeToolCall to hook into CoreToolScheduler updates
+     */
+    getToolCallUpdateCallback(): (toolCalls: unknown[]) => void;
+    /**
+     * Handle outgoing permission request
+     *
+     * Behavior depends on input format:
+     * - stream-json mode: Send can_use_tool to SDK and await response
+     * - Other modes: Check local approval mode and decide immediately
+     */
+    private handleOutgoingPermissionRequest;
+}

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

@@ -0,0 +1,358 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { InputFormat, ToolConfirmationOutcome, } from '@umsai/ums-code-core';
+import { BaseController } from './baseController.js';
+export class PermissionController extends BaseController {
+    pendingOutgoingRequests = new Set();
+    /**
+     * Handle permission control requests
+     */
+    async handleRequestPayload(payload, _signal) {
+        switch (payload.subtype) {
+            case 'can_use_tool':
+                return this.handleCanUseTool(payload);
+            case 'set_permission_mode':
+                return this.handleSetPermissionMode(payload);
+            default:
+                throw new Error(`Unsupported request subtype in PermissionController`);
+        }
+    }
+    /**
+     * Handle can_use_tool request
+     *
+     * Comprehensive permission evaluation based on:
+     * - Permission mode (approval level)
+     * - Tool registry validation
+     * - Error handling with safe defaults
+     */
+    async handleCanUseTool(payload) {
+        const toolName = payload.tool_name;
+        if (!toolName ||
+            typeof toolName !== 'string' ||
+            toolName.trim().length === 0) {
+            return {
+                subtype: 'can_use_tool',
+                behavior: 'deny',
+                message: 'Missing or invalid tool_name in can_use_tool request',
+            };
+        }
+        let behavior = 'allow';
+        let message;
+        try {
+            // Check permission mode first
+            const permissionResult = this.checkPermissionMode();
+            if (!permissionResult.allowed) {
+                behavior = 'deny';
+                message = permissionResult.message;
+            }
+            // Check tool registry if permission mode allows
+            if (behavior === 'allow') {
+                const registryResult = this.checkToolRegistry(toolName);
+                if (!registryResult.allowed) {
+                    behavior = 'deny';
+                    message = registryResult.message;
+                }
+            }
+        }
+        catch (error) {
+            behavior = 'deny';
+            message =
+                error instanceof Error
+                    ? `Failed to evaluate tool permission: ${error.message}`
+                    : 'Failed to evaluate tool permission';
+        }
+        const response = {
+            subtype: 'can_use_tool',
+            behavior,
+        };
+        if (message) {
+            response['message'] = message;
+        }
+        return response;
+    }
+    /**
+     * Check permission mode for tool execution
+     */
+    checkPermissionMode() {
+        const mode = this.context.permissionMode;
+        // Map permission modes to approval logic (aligned with VALID_APPROVAL_MODE_VALUES)
+        switch (mode) {
+            case 'yolo': // Allow all tools
+            case 'auto-edit': // Auto-approve edit operations
+            case 'plan': // Auto-approve planning operations
+                return { allowed: true };
+            case 'default': // TODO: allow all tools for test
+            default:
+                return {
+                    allowed: false,
+                    message: 'Tool execution requires manual approval. Update permission mode or approve via host.',
+                };
+        }
+    }
+    /**
+     * Check if tool exists in registry
+     */
+    checkToolRegistry(toolName) {
+        try {
+            // Access tool registry through config
+            const config = this.context.config;
+            const registryProvider = config;
+            if (typeof registryProvider.getToolRegistry === 'function') {
+                const registry = registryProvider.getToolRegistry();
+                if (registry &&
+                    typeof registry.getTool === 'function' &&
+                    !registry.getTool(toolName)) {
+                    return {
+                        allowed: false,
+                        message: `Tool "${toolName}" is not registered.`,
+                    };
+                }
+            }
+            return { allowed: true };
+        }
+        catch (error) {
+            return {
+                allowed: false,
+                message: `Failed to check tool registry: ${error instanceof Error ? error.message : 'Unknown error'}`,
+            };
+        }
+    }
+    /**
+     * Handle set_permission_mode request
+     *
+     * Updates the permission mode in the context
+     */
+    async handleSetPermissionMode(payload) {
+        const mode = payload.mode;
+        const validModes = [
+            'default',
+            'plan',
+            'auto-edit',
+            'yolo',
+        ];
+        if (!validModes.includes(mode)) {
+            throw new Error(`Invalid permission mode: ${mode}. Valid values are: ${validModes.join(', ')}`);
+        }
+        this.context.permissionMode = mode;
+        if (this.context.debugMode) {
+            console.error(`[PermissionController] Permission mode updated to: ${mode}`);
+        }
+        return { status: 'updated', mode };
+    }
+    /**
+     * Build permission suggestions for tool confirmation UI
+     *
+     * This method creates UI suggestions based on tool confirmation details,
+     * helping the host application present appropriate permission options.
+     */
+    buildPermissionSuggestions(confirmationDetails) {
+        if (!confirmationDetails ||
+            typeof confirmationDetails !== 'object' ||
+            !('type' in confirmationDetails)) {
+            return null;
+        }
+        const details = confirmationDetails;
+        const type = String(details['type'] ?? '');
+        const title = typeof details['title'] === 'string' ? details['title'] : undefined;
+        // Ensure type matches ToolCallConfirmationDetails union
+        const confirmationType = type;
+        switch (confirmationType) {
+            case 'exec': // ToolExecuteConfirmationDetails
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Allow Command',
+                        description: `Execute: ${details['command']}`,
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Deny',
+                        description: 'Block this command execution',
+                    },
+                ];
+            case 'edit': // ToolEditConfirmationDetails
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Allow Edit',
+                        description: `Edit file: ${details['fileName']}`,
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Deny',
+                        description: 'Block this file edit',
+                    },
+                    {
+                        type: 'modify',
+                        label: 'Review Changes',
+                        description: 'Review the proposed changes before applying',
+                    },
+                ];
+            case 'plan': // ToolPlanConfirmationDetails
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Approve Plan',
+                        description: title || 'Execute the proposed plan',
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Reject Plan',
+                        description: 'Do not execute this plan',
+                    },
+                ];
+            case 'mcp': // ToolMcpConfirmationDetails
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Allow MCP Call',
+                        description: `${details['serverName']}: ${details['toolName']}`,
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Deny',
+                        description: 'Block this MCP server call',
+                    },
+                ];
+            case 'info': // ToolInfoConfirmationDetails
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Allow Info Request',
+                        description: title || 'Allow information request',
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Deny',
+                        description: 'Block this information request',
+                    },
+                ];
+            default:
+                // Fallback for unknown types
+                return [
+                    {
+                        type: 'allow',
+                        label: 'Allow',
+                        description: title || `Allow ${type} operation`,
+                    },
+                    {
+                        type: 'deny',
+                        label: 'Deny',
+                        description: `Block ${type} operation`,
+                    },
+                ];
+        }
+    }
+    /**
+     * Check if a tool should be executed based on current permission settings
+     *
+     * This is a convenience method for direct tool execution checks without
+     * going through the control request flow.
+     */
+    async shouldAllowTool(toolRequest, confirmationDetails) {
+        // Check permission mode
+        const modeResult = this.checkPermissionMode();
+        if (!modeResult.allowed) {
+            return {
+                allowed: false,
+                message: modeResult.message,
+            };
+        }
+        // Check tool registry
+        const registryResult = this.checkToolRegistry(toolRequest.name);
+        if (!registryResult.allowed) {
+            return {
+                allowed: false,
+                message: registryResult.message,
+            };
+        }
+        // If we have confirmation details, we could potentially modify args
+        // This is a hook for future enhancement
+        if (confirmationDetails) {
+            // Future: handle argument modifications based on confirmation details
+        }
+        return { allowed: true };
+    }
+    /**
+     * Get callback for monitoring tool calls and handling outgoing permission requests
+     * This is passed to executeToolCall to hook into CoreToolScheduler updates
+     */
+    getToolCallUpdateCallback() {
+        return (toolCalls) => {
+            for (const call of toolCalls) {
+                if (call &&
+                    typeof call === 'object' &&
+                    call.status === 'awaiting_approval') {
+                    const awaiting = call;
+                    if (typeof awaiting.confirmationDetails?.onConfirm === 'function' &&
+                        !this.pendingOutgoingRequests.has(awaiting.request.callId)) {
+                        this.pendingOutgoingRequests.add(awaiting.request.callId);
+                        void this.handleOutgoingPermissionRequest(awaiting);
+                    }
+                }
+            }
+        };
+    }
+    /**
+     * Handle outgoing permission request
+     *
+     * Behavior depends on input format:
+     * - stream-json mode: Send can_use_tool to SDK and await response
+     * - Other modes: Check local approval mode and decide immediately
+     */
+    async handleOutgoingPermissionRequest(toolCall) {
+        try {
+            const inputFormat = this.context.config.getInputFormat?.();
+            const isStreamJsonMode = inputFormat === InputFormat.STREAM_JSON;
+            if (!isStreamJsonMode) {
+                // No SDK available - use local permission check
+                const modeCheck = this.checkPermissionMode();
+                const outcome = modeCheck.allowed
+                    ? ToolConfirmationOutcome.ProceedOnce
+                    : ToolConfirmationOutcome.Cancel;
+                await toolCall.confirmationDetails.onConfirm(outcome);
+                return;
+            }
+            // Stream-json mode: ask SDK for permission
+            const permissionSuggestions = this.buildPermissionSuggestions(toolCall.confirmationDetails);
+            const response = await this.sendControlRequest({
+                subtype: 'can_use_tool',
+                tool_name: toolCall.request.name,
+                tool_use_id: toolCall.request.callId,
+                input: toolCall.request.args,
+                permission_suggestions: permissionSuggestions,
+                blocked_path: null,
+            }, 30000);
+            if (response.subtype !== 'success') {
+                await toolCall.confirmationDetails.onConfirm(ToolConfirmationOutcome.Cancel);
+                return;
+            }
+            const payload = (response.response || {});
+            const behavior = String(payload['behavior'] || '').toLowerCase();
+            if (behavior === 'allow') {
+                // Handle updated input if provided
+                const updatedInput = payload['updatedInput'];
+                if (updatedInput && typeof updatedInput === 'object') {
+                    toolCall.request.args = updatedInput;
+                }
+                await toolCall.confirmationDetails.onConfirm(ToolConfirmationOutcome.ProceedOnce);
+            }
+            else {
+                await toolCall.confirmationDetails.onConfirm(ToolConfirmationOutcome.Cancel);
+            }
+        }
+        catch (error) {
+            if (this.context.debugMode) {
+                console.error('[PermissionController] Outgoing permission failed:', error);
+            }
+            await toolCall.confirmationDetails.onConfirm(ToolConfirmationOutcome.Cancel);
+        }
+        finally {
+            this.pendingOutgoingRequests.delete(toolCall.request.callId);
+        }
+    }
+}
+//# sourceMappingURL=permissionController.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"permissionController.js","sourceRoot":"","sources":["../../../../../src/nonInteractive/control/controllers/permissionController.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgBH,OAAO,EACL,WAAW,EACX,uBAAuB,GACxB,MAAM,sBAAsB,CAAC;AAQ9B,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAKrD,MAAM,OAAO,oBAAqB,SAAQ,cAAc;IAC9C,uBAAuB,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpD;;OAEG;IACO,KAAK,CAAC,oBAAoB,CAClC,OAA8B,EAC9B,OAAoB;QAEpB,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,KAAK,cAAc;gBACjB,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAsC,CAAC,CAAC;YAEvE,KAAK,qBAAqB;gBACxB,OAAO,IAAI,CAAC,uBAAuB,CACjC,OAA6C,CAC9C,CAAC;YAEJ;gBACE,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QAC3E,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,gBAAgB,CAC5B,OAAoC;QAEpC,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IACE,CAAC,QAAQ;YACT,OAAO,QAAQ,KAAK,QAAQ;YAC5B,QAAQ,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAC5B,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,cAAc;gBACvB,QAAQ,EAAE,MAAM;gBAChB,OAAO,EAAE,sDAAsD;aAChE,CAAC;QACJ,CAAC;QAED,IAAI,QAAQ,GAAqB,OAAO,CAAC;QACzC,IAAI,OAA2B,CAAC;QAEhC,IAAI,CAAC;YACH,8BAA8B;YAC9B,MAAM,gBAAgB,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACpD,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;gBAC9B,QAAQ,GAAG,MAAM,CAAC;gBAClB,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC;YACrC,CAAC;YAED,gDAAgD;YAChD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;gBACzB,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBACxD,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;oBAC5B,QAAQ,GAAG,MAAM,CAAC;oBAClB,OAAO,GAAG,cAAc,CAAC,OAAO,CAAC;gBACnC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,QAAQ,GAAG,MAAM,CAAC;YAClB,OAAO;gBACL,KAAK,YAAY,KAAK;oBACpB,CAAC,CAAC,uCAAuC,KAAK,CAAC,OAAO,EAAE;oBACxD,CAAC,CAAC,oCAAoC,CAAC;QAC7C,CAAC;QAED,MAAM,QAAQ,GAA4B;YACxC,OAAO,EAAE,cAAc;YACvB,QAAQ;SACT,CAAC;QAEF,IAAI,OAAO,EAAE,CAAC;YACZ,QAAQ,CAAC,SAAS,CAAC,GAAG,OAAO,CAAC;QAChC,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,mBAAmB;QACzB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC;QAEzC,mFAAmF;QACnF,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,MAAM,CAAC,CAAC,kBAAkB;YAC/B,KAAK,WAAW,CAAC,CAAC,+BAA+B;YACjD,KAAK,MAAM,EAAE,mCAAmC;gBAC9C,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,SAAS,CAAC,CAAC,iCAAiC;YACjD;gBACE,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,OAAO,EACL,sFAAsF;iBACzF,CAAC;QACN,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,QAAgB;QAIxC,IAAI,CAAC;YACH,sCAAsC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;YACnC,MAAM,gBAAgB,GAAG,MAIxB,CAAC;YAEF,IAAI,OAAO,gBAAgB,CAAC,eAAe,KAAK,UAAU,EAAE,CAAC;gBAC3D,MAAM,QAAQ,GAAG,gBAAgB,CAAC,eAAe,EAAE,CAAC;gBACpD,IACE,QAAQ;oBACR,OAAO,QAAQ,CAAC,OAAO,KAAK,UAAU;oBACtC,CAAC,QAAQ,CAAC,OAAO,CAAC,QAAQ,CAAC,EAC3B,CAAC;oBACD,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,SAAS,QAAQ,sBAAsB;qBACjD,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE;aACtG,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,uBAAuB,CACnC,OAA2C;QAE3C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QAC1B,MAAM,UAAU,GAAqB;YACnC,SAAS;YACT,MAAM;YACN,WAAW;YACX,MAAM;SACP,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,4BAA4B,IAAI,uBAAuB,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAC/E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC;QAEnC,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CACX,sDAAsD,IAAI,EAAE,CAC7D,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IACrC,CAAC;IAED;;;;;OAKG;IACH,0BAA0B,CACxB,mBAA4B;QAE5B,IACE,CAAC,mBAAmB;YACpB,OAAO,mBAAmB,KAAK,QAAQ;YACvC,CAAC,CAAC,MAAM,IAAI,mBAAmB,CAAC,EAChC,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,mBAA8C,CAAC;QAC/D,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3C,MAAM,KAAK,GACT,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEtE,wDAAwD;QACxD,MAAM,gBAAgB,GAAG,IAA4B,CAAC;QAEtD,QAAQ,gBAAgB,EAAE,CAAC;YACzB,KAAK,MAAM,EAAE,iCAAiC;gBAC5C,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,eAAe;wBACtB,WAAW,EAAE,YAAY,OAAO,CAAC,SAAS,CAAC,EAAE;qBAC9C;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,MAAM;wBACb,WAAW,EAAE,8BAA8B;qBAC5C;iBACF,CAAC;YAEJ,KAAK,MAAM,EAAE,8BAA8B;gBACzC,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,YAAY;wBACnB,WAAW,EAAE,cAAc,OAAO,CAAC,UAAU,CAAC,EAAE;qBACjD;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,MAAM;wBACb,WAAW,EAAE,sBAAsB;qBACpC;oBACD;wBACE,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,gBAAgB;wBACvB,WAAW,EAAE,6CAA6C;qBAC3D;iBACF,CAAC;YAEJ,KAAK,MAAM,EAAE,8BAA8B;gBACzC,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,cAAc;wBACrB,WAAW,EAAE,KAAK,IAAI,2BAA2B;qBAClD;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,aAAa;wBACpB,WAAW,EAAE,0BAA0B;qBACxC;iBACF,CAAC;YAEJ,KAAK,KAAK,EAAE,6BAA6B;gBACvC,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,gBAAgB;wBACvB,WAAW,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,KAAK,OAAO,CAAC,UAAU,CAAC,EAAE;qBAChE;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,MAAM;wBACb,WAAW,EAAE,4BAA4B;qBAC1C;iBACF,CAAC;YAEJ,KAAK,MAAM,EAAE,8BAA8B;gBACzC,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,oBAAoB;wBAC3B,WAAW,EAAE,KAAK,IAAI,2BAA2B;qBAClD;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,MAAM;wBACb,WAAW,EAAE,gCAAgC;qBAC9C;iBACF,CAAC;YAEJ;gBACE,6BAA6B;gBAC7B,OAAO;oBACL;wBACE,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,OAAO;wBACd,WAAW,EAAE,KAAK,IAAI,SAAS,IAAI,YAAY;qBAChD;oBACD;wBACE,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,MAAM;wBACb,WAAW,EAAE,SAAS,IAAI,YAAY;qBACvC;iBACF,CAAC;QACN,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,eAAe,CACnB,WAAgC,EAChC,mBAA6B;QAM7B,wBAAwB;QACxB,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAC9C,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC;QACJ,CAAC;QAED,sBAAsB;QACtB,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAChE,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;YAC5B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,OAAO,EAAE,cAAc,CAAC,OAAO;aAChC,CAAC;QACJ,CAAC;QAED,oEAAoE;QACpE,wCAAwC;QACxC,IAAI,mBAAmB,EAAE,CAAC;YACxB,sEAAsE;QACxE,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED;;;OAGG;IACH,yBAAyB;QACvB,OAAO,CAAC,SAAoB,EAAE,EAAE;YAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,IACE,IAAI;oBACJ,OAAO,IAAI,KAAK,QAAQ;oBACvB,IAA4B,CAAC,MAAM,KAAK,mBAAmB,EAC5D,CAAC;oBACD,MAAM,QAAQ,GAAG,IAAuB,CAAC;oBACzC,IACE,OAAO,QAAQ,CAAC,mBAAmB,EAAE,SAAS,KAAK,UAAU;wBAC7D,CAAC,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,EAC1D,CAAC;wBACD,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;wBAC1D,KAAK,IAAI,CAAC,+BAA+B,CAAC,QAAQ,CAAC,CAAC;oBACtD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACK,KAAK,CAAC,+BAA+B,CAC3C,QAAyB;QAEzB,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3D,MAAM,gBAAgB,GAAG,WAAW,KAAK,WAAW,CAAC,WAAW,CAAC;YAEjE,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,gDAAgD;gBAChD,MAAM,SAAS,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC7C,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO;oBAC/B,CAAC,CAAC,uBAAuB,CAAC,WAAW;oBACrC,CAAC,CAAC,uBAAuB,CAAC,MAAM,CAAC;gBAEnC,MAAM,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;gBACtD,OAAO;YACT,CAAC;YAED,2CAA2C;YAC3C,MAAM,qBAAqB,GAAG,IAAI,CAAC,0BAA0B,CAC3D,QAAQ,CAAC,mBAAmB,CAC7B,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAC5C;gBACE,OAAO,EAAE,cAAc;gBACvB,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI;gBAChC,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;gBACpC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI;gBAC5B,sBAAsB,EAAE,qBAAqB;gBAC7C,YAAY,EAAE,IAAI;aACY,EAChC,KAAK,CACN,CAAC;YAEF,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBACnC,MAAM,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAC1C,uBAAuB,CAAC,MAAM,CAC/B,CAAC;gBACF,OAAO;YACT,CAAC;YAED,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAA4B,CAAC;YACrE,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YAEjE,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;gBACzB,mCAAmC;gBACnC,MAAM,YAAY,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;gBAC7C,IAAI,YAAY,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACrD,QAAQ,CAAC,OAAO,CAAC,IAAI,GAAG,YAAuC,CAAC;gBAClE,CAAC;gBACD,MAAM,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAC1C,uBAAuB,CAAC,WAAW,CACpC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,MAAM,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAC1C,uBAAuB,CAAC,MAAM,CAC/B,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CACX,oDAAoD,EACpD,KAAK,CACN,CAAC;YACJ,CAAC;YACD,MAAM,QAAQ,CAAC,mBAAmB,CAAC,SAAS,CAC1C,uBAAuB,CAAC,MAAM,CAC/B,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC;CACF"}

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

@@ -0,0 +1,56 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * System Controller
+ *
+ * Handles system-level control requests:
+ * - initialize: Setup session and return system info
+ * - interrupt: Cancel current operations
+ * - set_model: Switch model (placeholder)
+ */
+import { BaseController } from './baseController.js';
+import type { ControlRequestPayload } from '../../types.js';
+export declare class SystemController extends BaseController {
+    /**
+     * Handle system control requests
+     */
+    protected handleRequestPayload(payload: ControlRequestPayload, _signal: AbortSignal): Promise<Record<string, unknown>>;
+    /**
+     * Handle initialize request
+     *
+     * Registers SDK MCP servers and returns capabilities
+     */
+    private handleInitialize;
+    /**
+     * Build control capabilities for initialize control response
+     *
+     * This method constructs the control capabilities object that indicates
+     * what control features are available. It is used exclusively in the
+     * initialize control response.
+     */
+    buildControlCapabilities(): Record<string, unknown>;
+    /**
+     * Handle interrupt request
+     *
+     * Triggers the interrupt callback to cancel current operations
+     */
+    private handleInterrupt;
+    /**
+     * Handle set_model request
+     *
+     * Implements actual model switching with validation and error handling
+     */
+    private handleSetModel;
+    /**
+     * Handle supported_commands request
+     *
+     * Returns list of supported control commands
+     *
+     * Note: This list should match the ControlRequestType enum in
+     * packages/sdk/typescript/src/types/controlRequests.ts
+     */
+    private handleSupportedCommands;
+}

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

@@ -0,0 +1,166 @@
+/**
+ * @license
+ * Copyright 2025 Qwen Team
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * System Controller
+ *
+ * Handles system-level control requests:
+ * - initialize: Setup session and return system info
+ * - interrupt: Cancel current operations
+ * - set_model: Switch model (placeholder)
+ */
+import { BaseController } from './baseController.js';
+export class SystemController extends BaseController {
+    /**
+     * Handle system control requests
+     */
+    async handleRequestPayload(payload, _signal) {
+        switch (payload.subtype) {
+            case 'initialize':
+                return this.handleInitialize(payload);
+            case 'interrupt':
+                return this.handleInterrupt();
+            case 'set_model':
+                return this.handleSetModel(payload);
+            case 'supported_commands':
+                return this.handleSupportedCommands();
+            default:
+                throw new Error(`Unsupported request subtype in SystemController`);
+        }
+    }
+    /**
+     * Handle initialize request
+     *
+     * Registers SDK MCP servers and returns capabilities
+     */
+    async handleInitialize(payload) {
+        // Register SDK MCP servers if provided
+        if (payload.sdkMcpServers && Array.isArray(payload.sdkMcpServers)) {
+            for (const serverName of payload.sdkMcpServers) {
+                this.context.sdkMcpServers.add(serverName);
+            }
+        }
+        // Build capabilities for response
+        const capabilities = this.buildControlCapabilities();
+        if (this.context.debugMode) {
+            console.error(`[SystemController] Initialized with ${this.context.sdkMcpServers.size} SDK MCP servers`);
+        }
+        return {
+            subtype: 'initialize',
+            capabilities,
+        };
+    }
+    /**
+     * Build control capabilities for initialize control response
+     *
+     * This method constructs the control capabilities object that indicates
+     * what control features are available. It is used exclusively in the
+     * initialize control response.
+     */
+    buildControlCapabilities() {
+        const capabilities = {
+            can_handle_can_use_tool: true,
+            can_handle_hook_callback: true,
+            can_set_permission_mode: typeof this.context.config.setApprovalMode === 'function',
+            can_set_model: typeof this.context.config.setModel === 'function',
+        };
+        // Check if MCP message handling is available
+        try {
+            const mcpProvider = this.context.config;
+            if (typeof mcpProvider.getMcpServers === 'function') {
+                const servers = mcpProvider.getMcpServers();
+                capabilities['can_handle_mcp_message'] = Boolean(servers && Object.keys(servers).length > 0);
+            }
+            else {
+                capabilities['can_handle_mcp_message'] = false;
+            }
+        }
+        catch (error) {
+            if (this.context.debugMode) {
+                console.error('[SystemController] Failed to determine MCP capability:', error);
+            }
+            capabilities['can_handle_mcp_message'] = false;
+        }
+        return capabilities;
+    }
+    /**
+     * Handle interrupt request
+     *
+     * Triggers the interrupt callback to cancel current operations
+     */
+    async handleInterrupt() {
+        // Trigger interrupt callback if available
+        if (this.context.onInterrupt) {
+            this.context.onInterrupt();
+        }
+        // Abort the main signal to cancel ongoing operations
+        if (this.context.abortSignal && !this.context.abortSignal.aborted) {
+            // Note: We can't directly abort the signal, but the onInterrupt callback should handle this
+            if (this.context.debugMode) {
+                console.error('[SystemController] Interrupt signal triggered');
+            }
+        }
+        if (this.context.debugMode) {
+            console.error('[SystemController] Interrupt handled');
+        }
+        return { subtype: 'interrupt' };
+    }
+    /**
+     * Handle set_model request
+     *
+     * Implements actual model switching with validation and error handling
+     */
+    async handleSetModel(payload) {
+        const model = payload.model;
+        // Validate model parameter
+        if (typeof model !== 'string' || model.trim() === '') {
+            throw new Error('Invalid model specified for set_model request');
+        }
+        try {
+            // Attempt to set the model using config
+            await this.context.config.setModel(model);
+            if (this.context.debugMode) {
+                console.error(`[SystemController] Model switched to: ${model}`);
+            }
+            return {
+                subtype: 'set_model',
+                model,
+            };
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : 'Failed to set model';
+            if (this.context.debugMode) {
+                console.error(`[SystemController] Failed to set model ${model}:`, error);
+            }
+            throw new Error(errorMessage);
+        }
+    }
+    /**
+     * Handle supported_commands request
+     *
+     * Returns list of supported control commands
+     *
+     * Note: This list should match the ControlRequestType enum in
+     * packages/sdk/typescript/src/types/controlRequests.ts
+     */
+    async handleSupportedCommands() {
+        const commands = [
+            'initialize',
+            'interrupt',
+            'set_model',
+            'supported_commands',
+            'can_use_tool',
+            'set_permission_mode',
+            'mcp_message',
+            'mcp_server_status',
+            'hook_callback',
+        ];
+        return {
+            subtype: 'supported_commands',
+            commands,
+        };
+    }
+}
+//# sourceMappingURL=systemController.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"systemController.js","sourceRoot":"","sources":["../../../../../src/nonInteractive/control/controllers/systemController.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAOrD,MAAM,OAAO,gBAAiB,SAAQ,cAAc;IAClD;;OAEG;IACO,KAAK,CAAC,oBAAoB,CAClC,OAA8B,EAC9B,OAAoB;QAEpB,QAAQ,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,KAAK,YAAY;gBACf,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAsC,CAAC,CAAC;YAEvE,KAAK,WAAW;gBACd,OAAO,IAAI,CAAC,eAAe,EAAE,CAAC;YAEhC,KAAK,WAAW;gBACd,OAAO,IAAI,CAAC,cAAc,CAAC,OAAoC,CAAC,CAAC;YAEnE,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAExC;gBACE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,gBAAgB,CAC5B,OAAoC;QAEpC,uCAAuC;QACvC,IAAI,OAAO,CAAC,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YAClE,KAAK,MAAM,UAAU,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;gBAC/C,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,kCAAkC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAErD,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CACX,uCAAuC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,kBAAkB,CACzF,CAAC;QACJ,CAAC;QAED,OAAO;YACL,OAAO,EAAE,YAAY;YACrB,YAAY;SACb,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,wBAAwB;QACtB,MAAM,YAAY,GAA4B;YAC5C,uBAAuB,EAAE,IAAI;YAC7B,wBAAwB,EAAE,IAAI;YAC9B,uBAAuB,EACrB,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,KAAK,UAAU;YAC3D,aAAa,EAAE,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,KAAK,UAAU;SAClE,CAAC;QAEF,6CAA6C;QAC7C,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,MAEhC,CAAC;YACF,IAAI,OAAO,WAAW,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;gBACpD,MAAM,OAAO,GAAG,WAAW,CAAC,aAAa,EAAE,CAAC;gBAC5C,YAAY,CAAC,wBAAwB,CAAC,GAAG,OAAO,CAC9C,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAC3C,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,wBAAwB,CAAC,GAAG,KAAK,CAAC;YACjD,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CACX,wDAAwD,EACxD,KAAK,CACN,CAAC;YACJ,CAAC;YACD,YAAY,CAAC,wBAAwB,CAAC,GAAG,KAAK,CAAC;QACjD,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,eAAe;QAC3B,0CAA0C;QAC1C,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC7B,CAAC;QAED,qDAAqD;QACrD,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YAClE,4FAA4F;YAC5F,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CAAC,+CAA+C,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3B,OAAO,CAAC,KAAK,CAAC,sCAAsC,CAAC,CAAC;QACxD,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;IAClC,CAAC;IAED;;;;OAIG;IACK,KAAK,CAAC,cAAc,CAC1B,OAAkC;QAElC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAE5B,2BAA2B;QAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,CAAC;YACH,wCAAwC;YACxC,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAE1C,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CAAC,yCAAyC,KAAK,EAAE,CAAC,CAAC;YAClE,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,WAAW;gBACpB,KAAK;aACN,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,CAAC;YAEjE,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;gBAC3B,OAAO,CAAC,KAAK,CACX,0CAA0C,KAAK,GAAG,EAClD,KAAK,CACN,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;;;;;;OAOG;IACK,KAAK,CAAC,uBAAuB;QACnC,MAAM,QAAQ,GAAG;YACf,YAAY;YACZ,WAAW;YACX,WAAW;YACX,oBAAoB;YACpB,cAAc;YACd,qBAAqB;YACrB,aAAa;YACb,mBAAmB;YACnB,eAAe;SAChB,CAAC;QAEF,OAAO;YACL,OAAO,EAAE,oBAAoB;YAC7B,QAAQ;SACT,CAAC;IACJ,CAAC;CACF"}