@office-ai/aioncli-core 0.26.0 → 0.30.0

package/dist/src/scheduler/policy.js

@@ -0,0 +1,119 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { ToolErrorType } from '../tools/tool-error.js';
+import { ApprovalMode, PolicyDecision, } from '../policy/types.js';
+import { MessageBusType, } from '../confirmation-bus/types.js';
+import { ToolConfirmationOutcome, } from '../tools/tools.js';
+import { DiscoveredMCPTool } from '../tools/mcp-tool.js';
+import { EDIT_TOOL_NAMES } from '../tools/tool-names.js';
+/**
+ * Helper to format the policy denial error.
+ */
+export function getPolicyDenialError(config, rule) {
+    const denyMessage = rule?.denyMessage ? ` ${rule.denyMessage}` : '';
+    return {
+        errorMessage: `Tool execution denied by policy.${denyMessage}`,
+        errorType: ToolErrorType.POLICY_VIOLATION,
+    };
+}
+/**
+ * Queries the system PolicyEngine to determine tool allowance.
+ * @returns The PolicyDecision.
+ * @throws Error if policy requires ASK_USER but the CLI is non-interactive.
+ */
+export async function checkPolicy(toolCall, config) {
+    const serverName = toolCall.tool instanceof DiscoveredMCPTool
+        ? toolCall.tool.serverName
+        : undefined;
+    const result = await config
+        .getPolicyEngine()
+        .check({ name: toolCall.request.name, args: toolCall.request.args }, serverName);
+    const { decision } = result;
+    /*
+     * Return the full check result including the rule that matched.
+     * This is necessary to access metadata like custom deny messages.
+     */
+    if (decision === PolicyDecision.ASK_USER) {
+        if (!config.isInteractive()) {
+            throw new Error(`Tool execution for "${toolCall.tool.displayName || toolCall.tool.name}" requires user confirmation, which is not supported in non-interactive mode.`);
+        }
+    }
+    return { decision, rule: result.rule };
+}
+/**
+ * Evaluates the outcome of a user confirmation and dispatches
+ * policy config updates.
+ */
+export async function updatePolicy(tool, outcome, confirmationDetails, deps) {
+    // Mode Transitions (AUTO_EDIT)
+    if (isAutoEditTransition(tool, outcome)) {
+        deps.config.setApprovalMode(ApprovalMode.AUTO_EDIT);
+        return;
+    }
+    // Specialized Tools (MCP)
+    if (confirmationDetails?.type === 'mcp') {
+        await handleMcpPolicyUpdate(tool, outcome, confirmationDetails, deps.messageBus);
+        return;
+    }
+    // Generic Fallback (Shell, Info, etc.)
+    await handleStandardPolicyUpdate(tool, outcome, confirmationDetails, deps.messageBus);
+}
+/**
+ * Returns true if the user's 'Always Allow' selection for a specific tool
+ * should trigger a session-wide transition to AUTO_EDIT mode.
+ */
+function isAutoEditTransition(tool, outcome) {
+    // TODO: This is a temporary fix to enable AUTO_EDIT mode for specific
+    // tools. We should refactor this so that callbacks can be removed from
+    // tools.
+    return (outcome === ToolConfirmationOutcome.ProceedAlways &&
+        EDIT_TOOL_NAMES.has(tool.name));
+}
+/**
+ * Handles policy updates for standard tools (Shell, Info, etc.), including
+ * session-level and persistent approvals.
+ */
+async function handleStandardPolicyUpdate(tool, outcome, confirmationDetails, messageBus) {
+    if (outcome === ToolConfirmationOutcome.ProceedAlways ||
+        outcome === ToolConfirmationOutcome.ProceedAlwaysAndSave) {
+        const options = {};
+        if (confirmationDetails?.type === 'exec') {
+            options.commandPrefix = confirmationDetails.rootCommands;
+        }
+        await messageBus.publish({
+            type: MessageBusType.UPDATE_POLICY,
+            toolName: tool.name,
+            persist: outcome === ToolConfirmationOutcome.ProceedAlwaysAndSave,
+            ...options,
+        });
+    }
+}
+/**
+ * Handles policy updates specifically for MCP tools, including session-level
+ * and persistent approvals.
+ */
+async function handleMcpPolicyUpdate(tool, outcome, confirmationDetails, messageBus) {
+    const isMcpAlways = outcome === ToolConfirmationOutcome.ProceedAlways ||
+        outcome === ToolConfirmationOutcome.ProceedAlwaysTool ||
+        outcome === ToolConfirmationOutcome.ProceedAlwaysServer ||
+        outcome === ToolConfirmationOutcome.ProceedAlwaysAndSave;
+    if (!isMcpAlways) {
+        return;
+    }
+    let toolName = tool.name;
+    const persist = outcome === ToolConfirmationOutcome.ProceedAlwaysAndSave;
+    // If "Always allow all tools from this server", use the wildcard pattern
+    if (outcome === ToolConfirmationOutcome.ProceedAlwaysServer) {
+        toolName = `${confirmationDetails.serverName}__*`;
+    }
+    await messageBus.publish({
+        type: MessageBusType.UPDATE_POLICY,
+        toolName,
+        mcpName: confirmationDetails.serverName,
+        persist,
+    });
+}
+//# sourceMappingURL=policy.js.map

package/dist/src/scheduler/policy.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy.js","sourceRoot":"","sources":["../../../src/scheduler/policy.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EACL,YAAY,EACZ,cAAc,GAGf,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EACL,cAAc,GAEf,MAAM,8BAA8B,CAAC;AACtC,OAAO,EACL,uBAAuB,GAGxB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAGzD;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,MAAc,EACd,IAAiB;IAEjB,MAAM,WAAW,GAAG,IAAI,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACpE,OAAO;QACL,YAAY,EAAE,mCAAmC,WAAW,EAAE;QAC9D,SAAS,EAAE,aAAa,CAAC,gBAAgB;KAC1C,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAC/B,QAA4B,EAC5B,MAAc;IAEd,MAAM,UAAU,GACd,QAAQ,CAAC,IAAI,YAAY,iBAAiB;QACxC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU;QAC1B,CAAC,CAAC,SAAS,CAAC;IAEhB,MAAM,MAAM,GAAG,MAAM,MAAM;SACxB,eAAe,EAAE;SACjB,KAAK,CACJ,EAAE,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,EAC5D,UAAU,CACX,CAAC;IAEJ,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;IAE5B;;;OAGG;IACH,IAAI,QAAQ,KAAK,cAAc,CAAC,QAAQ,EAAE,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CACb,uBACE,QAAQ,CAAC,IAAI,CAAC,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,IAC7C,+EAA+E,CAChF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC;AACzC,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,IAAwB,EACxB,OAAgC,EAChC,mBAAgE,EAChE,IAAgD;IAEhD,+BAA+B;IAC/B,IAAI,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACpD,OAAO;IACT,CAAC;IAED,0BAA0B;IAC1B,IAAI,mBAAmB,EAAE,IAAI,KAAK,KAAK,EAAE,CAAC;QACxC,MAAM,qBAAqB,CACzB,IAAI,EACJ,OAAO,EACP,mBAAmB,EACnB,IAAI,CAAC,UAAU,CAChB,CAAC;QACF,OAAO;IACT,CAAC;IAED,uCAAuC;IACvC,MAAM,0BAA0B,CAC9B,IAAI,EACJ,OAAO,EACP,mBAAmB,EACnB,IAAI,CAAC,UAAU,CAChB,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,SAAS,oBAAoB,CAC3B,IAAwB,EACxB,OAAgC;IAEhC,sEAAsE;IACtE,uEAAuE;IACvE,SAAS;IACT,OAAO,CACL,OAAO,KAAK,uBAAuB,CAAC,aAAa;QACjD,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAC/B,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,0BAA0B,CACvC,IAAwB,EACxB,OAAgC,EAChC,mBAAgE,EAChE,UAAsB;IAEtB,IACE,OAAO,KAAK,uBAAuB,CAAC,aAAa;QACjD,OAAO,KAAK,uBAAuB,CAAC,oBAAoB,EACxD,CAAC;QACD,MAAM,OAAO,GAAwB,EAAE,CAAC;QAExC,IAAI,mBAAmB,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACzC,OAAO,CAAC,aAAa,GAAG,mBAAmB,CAAC,YAAY,CAAC;QAC3D,CAAC;QAED,MAAM,UAAU,CAAC,OAAO,CAAC;YACvB,IAAI,EAAE,cAAc,CAAC,aAAa;YAClC,QAAQ,EAAE,IAAI,CAAC,IAAI;YACnB,OAAO,EAAE,OAAO,KAAK,uBAAuB,CAAC,oBAAoB;YACjE,GAAG,OAAO;SACX,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,qBAAqB,CAClC,IAAwB,EACxB,OAAgC,EAChC,mBAGC,EACD,UAAsB;IAEtB,MAAM,WAAW,GACf,OAAO,KAAK,uBAAuB,CAAC,aAAa;QACjD,OAAO,KAAK,uBAAuB,CAAC,iBAAiB;QACrD,OAAO,KAAK,uBAAuB,CAAC,mBAAmB;QACvD,OAAO,KAAK,uBAAuB,CAAC,oBAAoB,CAAC;IAE3D,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO;IACT,CAAC;IAED,IAAI,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IACzB,MAAM,OAAO,GAAG,OAAO,KAAK,uBAAuB,CAAC,oBAAoB,CAAC;IAEzE,yEAAyE;IACzE,IAAI,OAAO,KAAK,uBAAuB,CAAC,mBAAmB,EAAE,CAAC;QAC5D,QAAQ,GAAG,GAAG,mBAAmB,CAAC,UAAU,KAAK,CAAC;IACpD,CAAC;IAED,MAAM,UAAU,CAAC,OAAO,CAAC;QACvB,IAAI,EAAE,cAAc,CAAC,aAAa;QAClC,QAAQ;QACR,OAAO,EAAE,mBAAmB,CAAC,UAAU;QACvC,OAAO;KACR,CAAC,CAAC;AACL,CAAC"}

package/dist/src/scheduler/policy.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/scheduler/policy.test.js

@@ -0,0 +1,419 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { checkPolicy, updatePolicy, getPolicyDenialError } from './policy.js';
+import { MessageBusType } from '../confirmation-bus/types.js';
+import { ApprovalMode, PolicyDecision } from '../policy/types.js';
+import { ToolConfirmationOutcome, } from '../tools/tools.js';
+import { DiscoveredMCPTool } from '../tools/mcp-tool.js';
+import { CoreToolScheduler } from '../core/coreToolScheduler.js';
+import { Scheduler } from './scheduler.js';
+import { ROOT_SCHEDULER_ID } from './types.js';
+import { ToolErrorType } from '../tools/tool-error.js';
+describe('policy.ts', () => {
+    describe('checkPolicy', () => {
+        it('should return the decision from the policy engine', async () => {
+            const mockPolicyEngine = {
+                check: vi.fn().mockResolvedValue({ decision: PolicyDecision.ALLOW }),
+            };
+            const mockConfig = {
+                getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+            };
+            const toolCall = {
+                request: { name: 'test-tool', args: {} },
+                tool: { name: 'test-tool' },
+            };
+            const result = await checkPolicy(toolCall, mockConfig);
+            expect(result.decision).toBe(PolicyDecision.ALLOW);
+            expect(mockPolicyEngine.check).toHaveBeenCalledWith({ name: 'test-tool', args: {} }, undefined);
+        });
+        it('should pass serverName for MCP tools', async () => {
+            const mockPolicyEngine = {
+                check: vi.fn().mockResolvedValue({ decision: PolicyDecision.ALLOW }),
+            };
+            const mockConfig = {
+                getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+            };
+            const mcpTool = Object.create(DiscoveredMCPTool.prototype);
+            mcpTool.serverName = 'my-server';
+            const toolCall = {
+                request: { name: 'mcp-tool', args: {} },
+                tool: mcpTool,
+            };
+            await checkPolicy(toolCall, mockConfig);
+            expect(mockPolicyEngine.check).toHaveBeenCalledWith({ name: 'mcp-tool', args: {} }, 'my-server');
+        });
+        it('should throw if ASK_USER is returned in non-interactive mode', async () => {
+            const mockPolicyEngine = {
+                check: vi.fn().mockResolvedValue({ decision: PolicyDecision.ASK_USER }),
+            };
+            const mockConfig = {
+                getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+                isInteractive: vi.fn().mockReturnValue(false),
+            };
+            const toolCall = {
+                request: { name: 'test-tool', args: {} },
+                tool: { name: 'test-tool' },
+            };
+            await expect(checkPolicy(toolCall, mockConfig)).rejects.toThrow(/not supported in non-interactive mode/);
+        });
+        it('should return DENY without throwing', async () => {
+            const mockPolicyEngine = {
+                check: vi.fn().mockResolvedValue({ decision: PolicyDecision.DENY }),
+            };
+            const mockConfig = {
+                getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+            };
+            const toolCall = {
+                request: { name: 'test-tool', args: {} },
+                tool: { name: 'test-tool' },
+            };
+            const result = await checkPolicy(toolCall, mockConfig);
+            expect(result.decision).toBe(PolicyDecision.DENY);
+        });
+        it('should return ASK_USER without throwing in interactive mode', async () => {
+            const mockPolicyEngine = {
+                check: vi.fn().mockResolvedValue({ decision: PolicyDecision.ASK_USER }),
+            };
+            const mockConfig = {
+                getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+                isInteractive: vi.fn().mockReturnValue(true),
+            };
+            const toolCall = {
+                request: { name: 'test-tool', args: {} },
+                tool: { name: 'test-tool' },
+            };
+            const result = await checkPolicy(toolCall, mockConfig);
+            expect(result.decision).toBe(PolicyDecision.ASK_USER);
+        });
+    });
+    describe('updatePolicy', () => {
+        it('should set AUTO_EDIT mode for auto-edit transition tools', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'replace' }; // 'replace' is in EDIT_TOOL_NAMES
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlways, undefined, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockConfig.setApprovalMode).toHaveBeenCalledWith(ApprovalMode.AUTO_EDIT);
+            expect(mockMessageBus.publish).not.toHaveBeenCalled();
+        });
+        it('should handle standard policy updates (persist=false)', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'test-tool' };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlways, undefined, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'test-tool',
+                persist: false,
+            }));
+        });
+        it('should handle standard policy updates with persistence', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'test-tool' };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlwaysAndSave, undefined, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'test-tool',
+                persist: true,
+            }));
+        });
+        it('should handle shell command prefixes', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'run_shell_command' };
+            const details = {
+                type: 'exec',
+                command: 'ls -la',
+                rootCommand: 'ls',
+                rootCommands: ['ls'],
+                title: 'Shell',
+                onConfirm: vi.fn(),
+            };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlways, details, {
+                config: mockConfig,
+                messageBus: mockMessageBus,
+            });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'run_shell_command',
+                commandPrefix: ['ls'],
+            }));
+        });
+        it('should handle MCP policy updates (server scope)', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'mcp-tool' };
+            const details = {
+                type: 'mcp',
+                serverName: 'my-server',
+                toolName: 'mcp-tool',
+                toolDisplayName: 'My Tool',
+                title: 'MCP',
+                onConfirm: vi.fn(),
+            };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlwaysServer, details, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'my-server__*',
+                mcpName: 'my-server',
+                persist: false,
+            }));
+        });
+        it('should NOT publish update for ProceedOnce', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'test-tool' };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedOnce, undefined, {
+                config: mockConfig,
+                messageBus: mockMessageBus,
+            });
+            expect(mockMessageBus.publish).not.toHaveBeenCalled();
+            expect(mockConfig.setApprovalMode).not.toHaveBeenCalled();
+        });
+        it('should NOT publish update for Cancel', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'test-tool' };
+            await updatePolicy(tool, ToolConfirmationOutcome.Cancel, undefined, {
+                config: mockConfig,
+                messageBus: mockMessageBus,
+            });
+            expect(mockMessageBus.publish).not.toHaveBeenCalled();
+        });
+        it('should NOT publish update for ModifyWithEditor', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'test-tool' };
+            await updatePolicy(tool, ToolConfirmationOutcome.ModifyWithEditor, undefined, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).not.toHaveBeenCalled();
+        });
+        it('should handle MCP ProceedAlwaysTool (specific tool name)', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'mcp-tool' };
+            const details = {
+                type: 'mcp',
+                serverName: 'my-server',
+                toolName: 'mcp-tool',
+                toolDisplayName: 'My Tool',
+                title: 'MCP',
+                onConfirm: vi.fn(),
+            };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlwaysTool, details, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'mcp-tool', // Specific name, not wildcard
+                mcpName: 'my-server',
+                persist: false,
+            }));
+        });
+        it('should handle MCP ProceedAlways (persist: false)', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'mcp-tool' };
+            const details = {
+                type: 'mcp',
+                serverName: 'my-server',
+                toolName: 'mcp-tool',
+                toolDisplayName: 'My Tool',
+                title: 'MCP',
+                onConfirm: vi.fn(),
+            };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlways, details, {
+                config: mockConfig,
+                messageBus: mockMessageBus,
+            });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'mcp-tool',
+                mcpName: 'my-server',
+                persist: false,
+            }));
+        });
+        it('should handle MCP ProceedAlwaysAndSave (persist: true)', async () => {
+            const mockConfig = {
+                setApprovalMode: vi.fn(),
+            };
+            const mockMessageBus = {
+                publish: vi.fn(),
+            };
+            const tool = { name: 'mcp-tool' };
+            const details = {
+                type: 'mcp',
+                serverName: 'my-server',
+                toolName: 'mcp-tool',
+                toolDisplayName: 'My Tool',
+                title: 'MCP',
+                onConfirm: vi.fn(),
+            };
+            await updatePolicy(tool, ToolConfirmationOutcome.ProceedAlwaysAndSave, details, { config: mockConfig, messageBus: mockMessageBus });
+            expect(mockMessageBus.publish).toHaveBeenCalledWith(expect.objectContaining({
+                type: MessageBusType.UPDATE_POLICY,
+                toolName: 'mcp-tool',
+                mcpName: 'my-server',
+                persist: true,
+            }));
+        });
+    });
+    describe('getPolicyDenialError', () => {
+        it('should return default denial message when no rule provided', () => {
+            const mockConfig = {
+                getApprovalMode: vi.fn().mockReturnValue(ApprovalMode.DEFAULT),
+            };
+            const { errorMessage, errorType } = getPolicyDenialError(mockConfig);
+            expect(errorMessage).toBe('Tool execution denied by policy.');
+            expect(errorType).toBe(ToolErrorType.POLICY_VIOLATION);
+        });
+        it('should return custom deny message if provided', () => {
+            const mockConfig = {
+                getApprovalMode: vi.fn().mockReturnValue(ApprovalMode.DEFAULT),
+            };
+            const rule = {
+                decision: PolicyDecision.DENY,
+                denyMessage: 'Custom Deny',
+            };
+            const { errorMessage, errorType } = getPolicyDenialError(mockConfig, rule);
+            expect(errorMessage).toBe('Tool execution denied by policy. Custom Deny');
+            expect(errorType).toBe(ToolErrorType.POLICY_VIOLATION);
+        });
+    });
+});
+describe('Plan Mode Denial Consistency', () => {
+    let mockConfig;
+    let mockMessageBus;
+    let mockPolicyEngine;
+    let mockToolRegistry;
+    let mockTool;
+    let mockInvocation;
+    const req = {
+        callId: 'call-1',
+        name: 'test-tool',
+        args: { foo: 'bar' },
+        isClientInitiated: false,
+        prompt_id: 'prompt-1',
+        schedulerId: ROOT_SCHEDULER_ID,
+    };
+    beforeEach(() => {
+        mockTool = {
+            name: 'test-tool',
+            build: vi.fn(),
+        };
+        mockInvocation = {
+            shouldConfirmExecute: vi.fn(),
+        };
+        vi.mocked(mockTool.build).mockReturnValue(mockInvocation);
+        mockPolicyEngine = {
+            check: vi.fn().mockResolvedValue({ decision: PolicyDecision.DENY }), // Default to DENY for this test
+        };
+        mockToolRegistry = {
+            getTool: vi.fn().mockReturnValue(mockTool),
+            getAllToolNames: vi.fn().mockReturnValue(['test-tool']),
+        };
+        mockMessageBus = {
+            publish: vi.fn(),
+            subscribe: vi.fn(),
+        };
+        mockConfig = {
+            getPolicyEngine: vi.fn().mockReturnValue(mockPolicyEngine),
+            getToolRegistry: vi.fn().mockReturnValue(mockToolRegistry),
+            getMessageBus: vi.fn().mockReturnValue(mockMessageBus),
+            isInteractive: vi.fn().mockReturnValue(true),
+            getEnableHooks: vi.fn().mockReturnValue(false),
+            getApprovalMode: vi.fn().mockReturnValue(ApprovalMode.PLAN), // Key: Plan Mode
+            setApprovalMode: vi.fn(),
+            getUsageStatisticsEnabled: vi.fn().mockReturnValue(false),
+        };
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+    });
+    describe.each([
+        { enableEventDrivenScheduler: false, name: 'Legacy CoreToolScheduler' },
+        { enableEventDrivenScheduler: true, name: 'Event-Driven Scheduler' },
+    ])('$name', ({ enableEventDrivenScheduler }) => {
+        it('should return the correct Plan Mode denial message when policy denies execution', async () => {
+            let resultMessage;
+            let resultErrorType;
+            const signal = new AbortController().signal;
+            if (enableEventDrivenScheduler) {
+                const scheduler = new Scheduler({
+                    config: mockConfig,
+                    messageBus: mockMessageBus,
+                    getPreferredEditor: () => undefined,
+                    schedulerId: ROOT_SCHEDULER_ID,
+                });
+                const results = await scheduler.schedule(req, signal);
+                const result = results[0];
+                expect(result.status).toBe('error');
+                if (result.status === 'error') {
+                    resultMessage = result.response.error?.message;
+                    resultErrorType = result.response.errorType;
+                }
+            }
+            else {
+                let capturedCalls = [];
+                const scheduler = new CoreToolScheduler({
+                    config: mockConfig,
+                    getPreferredEditor: () => undefined,
+                    onAllToolCallsComplete: async (calls) => {
+                        capturedCalls = calls;
+                    },
+                });
+                await scheduler.schedule(req, signal);
+                expect(capturedCalls.length).toBeGreaterThan(0);
+                const call = capturedCalls[0];
+                if (call.status === 'error') {
+                    resultMessage = call.response.error?.message;
+                    resultErrorType = call.response.errorType;
+                }
+            }
+            expect(resultMessage).toBe('Tool execution denied by policy.');
+            expect(resultErrorType).toBe(ToolErrorType.POLICY_VIOLATION);
+        });
+    });
+});
+//# sourceMappingURL=policy.test.js.map

package/dist/src/scheduler/policy.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"policy.test.js","sourceRoot":"","sources":["../../../src/scheduler/policy.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EAEF,UAAU,EACV,SAAS,GACV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAG9E,OAAO,EAAE,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAClE,OAAO,EACL,uBAAuB,GAKxB,MAAM,mBAAmB,CAAC;AAO3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAGvD,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,KAAK,EAAE,CAAC;aAClC,CAAC;YAErC,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;aAC9B,CAAC;YAE/B,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACxC,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACN,CAAC;YAExB,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACnD,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACjD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE,EAC/B,SAAS,CACV,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,KAAK,EAAE,CAAC;aAClC,CAAC;YAErC,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;aAC9B,CAAC;YAE/B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAC3D,OAAO,CAAC,UAAU,GAAG,WAAW,CAAC;YAEjC,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE;gBACvC,IAAI,EAAE,OAAO;aACQ,CAAC;YAExB,MAAM,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACxC,MAAM,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACjD,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,EAAE,EAC9B,WAAW,CACZ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC;aACrC,CAAC;YAErC,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;gBAC1D,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;aACjB,CAAC;YAE/B,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACxC,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACN,CAAC;YAExB,MAAM,MAAM,CAAC,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC7D,uCAAuC,CACxC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC;aACjC,CAAC;YAErC,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;aAC9B,CAAC;YAE/B,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACxC,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACN,CAAC;YAExB,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,gBAAgB,GAAG;gBACvB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC;aACrC,CAAC;YAErC,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;gBAC1D,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;aAChB,CAAC;YAE/B,MAAM,QAAQ,GAAG;gBACf,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,EAAE;gBACxC,IAAI,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;aACN,CAAC;YAExB,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YAEnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,SAAS,EAAwB,CAAC,CAAC,kCAAkC;YAE1F,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,aAAa,EACrC,SAAS,EACT,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,oBAAoB,CACrD,YAAY,CAAC,SAAS,CACvB,CAAC;YACF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAwB,CAAC;YAEzD,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,aAAa,EACrC,SAAS,EACT,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,WAAW;gBACrB,OAAO,EAAE,KAAK;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAwB,CAAC;YAEzD,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,oBAAoB,EAC5C,SAAS,EACT,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,WAAW;gBACrB,OAAO,EAAE,IAAI;aACd,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,mBAAmB,EAAwB,CAAC;YACjE,MAAM,OAAO,GAAmC;gBAC9C,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,QAAQ;gBACjB,WAAW,EAAE,IAAI;gBACjB,YAAY,EAAE,CAAC,IAAI,CAAC;gBACpB,KAAK,EAAE,OAAO;gBACd,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;aACnB,CAAC;YAEF,MAAM,YAAY,CAAC,IAAI,EAAE,uBAAuB,CAAC,aAAa,EAAE,OAAO,EAAE;gBACvE,MAAM,EAAE,UAAU;gBAClB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,mBAAmB;gBAC7B,aAAa,EAAE,CAAC,IAAI,CAAC;aACtB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,UAAU,EAAwB,CAAC;YACxD,MAAM,OAAO,GAA+B;gBAC1C,IAAI,EAAE,KAAK;gBACX,UAAU,EAAE,WAAW;gBACvB,QAAQ,EAAE,UAAU;gBACpB,eAAe,EAAE,SAAS;gBAC1B,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;aACnB,CAAC;YAEF,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,mBAAmB,EAC3C,OAAO,EACP,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,cAAc;gBACxB,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,KAAK;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAwB,CAAC;YAEzD,MAAM,YAAY,CAAC,IAAI,EAAE,uBAAuB,CAAC,WAAW,EAAE,SAAS,EAAE;gBACvE,MAAM,EAAE,UAAU;gBAClB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAwB,CAAC;YAEzD,MAAM,YAAY,CAAC,IAAI,EAAE,uBAAuB,CAAC,MAAM,EAAE,SAAS,EAAE;gBAClE,MAAM,EAAE,UAAU;gBAClB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,WAAW,EAAwB,CAAC;YAEzD,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,gBAAgB,EACxC,SAAS,EACT,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,UAAU,EAAwB,CAAC;YACxD,MAAM,OAAO,GAA+B;gBAC1C,IAAI,EAAE,KAAK;gBACX,UAAU,EAAE,WAAW;gBACvB,QAAQ,EAAE,UAAU;gBACpB,eAAe,EAAE,SAAS;gBAC1B,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;aACnB,CAAC;YAEF,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,iBAAiB,EACzC,OAAO,EACP,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,UAAU,EAAE,8BAA8B;gBACpD,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,KAAK;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,UAAU,EAAwB,CAAC;YACxD,MAAM,OAAO,GAA+B;gBAC1C,IAAI,EAAE,KAAK;gBACX,UAAU,EAAE,WAAW;gBACvB,QAAQ,EAAE,UAAU;gBACpB,eAAe,EAAE,SAAS;gBAC1B,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;aACnB,CAAC;YAEF,MAAM,YAAY,CAAC,IAAI,EAAE,uBAAuB,CAAC,aAAa,EAAE,OAAO,EAAE;gBACvE,MAAM,EAAE,UAAU;gBAClB,UAAU,EAAE,cAAc;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,UAAU;gBACpB,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,KAAK;aACf,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;aACI,CAAC;YAC/B,MAAM,cAAc,GAAG;gBACrB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;aACgB,CAAC;YACnC,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,UAAU,EAAwB,CAAC;YACxD,MAAM,OAAO,GAA+B;gBAC1C,IAAI,EAAE,KAAK;gBACX,UAAU,EAAE,WAAW;gBACvB,QAAQ,EAAE,UAAU;gBACpB,eAAe,EAAE,SAAS;gBAC1B,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;aACnB,CAAC;YAEF,MAAM,YAAY,CAChB,IAAI,EACJ,uBAAuB,CAAC,oBAAoB,EAC5C,OAAO,EACP,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CACnD,CAAC;YAEF,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,oBAAoB,CACjD,MAAM,CAAC,gBAAgB,CAAC;gBACtB,IAAI,EAAE,cAAc,CAAC,aAAa;gBAClC,QAAQ,EAAE,UAAU;gBACpB,OAAO,EAAE,WAAW;gBACpB,OAAO,EAAE,IAAI;aACd,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC;aAC1C,CAAC;YAEvB,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAErE,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC9D,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,UAAU,GAAG;gBACjB,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,OAAO,CAAC;aAC1C,CAAC;YACvB,MAAM,IAAI,GAAG;gBACX,QAAQ,EAAE,cAAc,CAAC,IAAI;gBAC7B,WAAW,EAAE,aAAa;aAC3B,CAAC;YAEF,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,GAAG,oBAAoB,CACtD,UAAU,EACV,IAAI,CACL,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;YAC1E,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,8BAA8B,EAAE,GAAG,EAAE;IAC5C,IAAI,UAA0B,CAAC;IAC/B,IAAI,cAAkC,CAAC;IACvC,IAAI,gBAAsC,CAAC;IAC3C,IAAI,gBAAsC,CAAC;IAC3C,IAAI,QAA4B,CAAC;IACjC,IAAI,cAAiC,CAAC;IAEtC,MAAM,GAAG,GAAwB;QAC/B,MAAM,EAAE,QAAQ;QAChB,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;QACpB,iBAAiB,EAAE,KAAK;QACxB,SAAS,EAAE,UAAU;QACrB,WAAW,EAAE,iBAAiB;KAC/B,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACkB,CAAC;QAEnC,cAAc,GAAG;YACf,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;SACE,CAAC;QAClC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC;QAE1D,gBAAgB,GAAG;YACjB,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,QAAQ,EAAE,cAAc,CAAC,IAAI,EAAE,CAAC,EAAE,gCAAgC;SACnE,CAAC;QAErC,gBAAgB,GAAG;YACjB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;YAC1C,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,WAAW,CAAC,CAAC;SACrB,CAAC;QAErC,cAAc,GAAG;YACf,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;YAChB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;SACc,CAAC;QAEnC,UAAU,GAAG;YACX,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;YAC1D,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,gBAAgB,CAAC;YAC1D,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,cAAc,CAAC;YACtD,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;YAC9C,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,iBAAiB;YAC9E,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;YACxB,yBAAyB,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;SAC7B,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,IAAI,CAAC;QACZ,EAAE,0BAA0B,EAAE,KAAK,EAAE,IAAI,EAAE,0BAA0B,EAAE;QACvE,EAAE,0BAA0B,EAAE,IAAI,EAAE,IAAI,EAAE,wBAAwB,EAAE;KACrE,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,0BAA0B,EAAE,EAAE,EAAE;QAC7C,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,IAAI,aAAiC,CAAC;YACtC,IAAI,eAA0C,CAAC;YAE/C,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;YAE5C,IAAI,0BAA0B,EAAE,CAAC;gBAC/B,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;oBAC9B,MAAM,EAAE,UAAU;oBAClB,UAAU,EAAE,cAAc;oBAC1B,kBAAkB,EAAE,GAAG,EAAE,CAAC,SAAS;oBACnC,WAAW,EAAE,iBAAiB;iBAC/B,CAAC,CAAC;gBAEH,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;gBACtD,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;gBAE1B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpC,IAAI,MAAM,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAC9B,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;oBAC/C,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC9C,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,aAAa,GAAwB,EAAE,CAAC;gBAC5C,MAAM,SAAS,GAAG,IAAI,iBAAiB,CAAC;oBACtC,MAAM,EAAE,UAAU;oBAClB,kBAAkB,EAAE,GAAG,EAAE,CAAC,SAAS;oBACnC,sBAAsB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;wBACtC,aAAa,GAAG,KAAK,CAAC;oBACxB,CAAC;iBACF,CAAC,CAAC;gBAEH,MAAM,SAAS,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;gBAEtC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;gBAChD,MAAM,IAAI,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;gBAC9B,IAAI,IAAI,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;oBAC5B,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,CAAC;oBAC7C,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;YAC/D,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/scheduler/scheduler.d.ts

@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Config } from '../config/config.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+import { type ToolCallRequestInfo, type CompletedToolCall } from './types.js';
+import type { EditorType } from '../utils/editor.js';
+export interface SchedulerOptions {
+    config: Config;
+    messageBus: MessageBus;
+    getPreferredEditor: () => EditorType | undefined;
+    schedulerId: string;
+    parentCallId?: string;
+    onWaitingForConfirmation?: (waiting: boolean) => void;
+}
+/**
+ * Event-Driven Orchestrator for Tool Execution.
+ * Coordinates execution via state updates and event listening.
+ */
+export declare class Scheduler {
+    private static subscribedMessageBuses;
+    private readonly state;
+    private readonly executor;
+    private readonly modifier;
+    private readonly config;
+    private readonly messageBus;
+    private readonly getPreferredEditor;
+    private readonly schedulerId;
+    private readonly parentCallId?;
+    private readonly onWaitingForConfirmation?;
+    private isProcessing;
+    private isCancelling;
+    private readonly requestQueue;
+    constructor(options: SchedulerOptions);
+    private setupMessageBusListener;
+    /**
+     * Schedules a batch of tool calls.
+     * @returns A promise that resolves with the results of the completed batch.
+     */
+    schedule(request: ToolCallRequestInfo | ToolCallRequestInfo[], signal: AbortSignal): Promise<CompletedToolCall[]>;
+    private _enqueueRequest;
+    cancelAll(): void;
+    get completedCalls(): CompletedToolCall[];
+    private isTerminal;
+    private _startBatch;
+    private _createToolNotFoundErroredToolCall;
+    private _validateAndCreateToolCall;
+    private _processQueue;
+    /**
+     * Processes the next item in the queue.
+     * @returns true if the loop should continue, false if it should terminate.
+     */
+    private _processNextItem;
+    private _processValidatingCall;
+    private _processToolCall;
+    /**
+     * Executes the tool and records the result.
+     */
+    private _execute;
+    private _processNextInRequestQueue;
+}