@office-ai/aioncli-core 0.8.1 → 0.18.4

package/dist/src/policy/toml-loader.test.js

@@ -0,0 +1,376 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { ApprovalMode, PolicyDecision } from './types.js';
+import * as fs from 'node:fs/promises';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import { loadPoliciesFromToml } from './toml-loader.js';
+describe('policy-toml-loader', () => {
+    let tempDir;
+    beforeEach(async () => {
+        tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'policy-test-'));
+    });
+    afterEach(async () => {
+        if (tempDir) {
+            await fs.rm(tempDir, {
+                recursive: true,
+                force: true,
+                maxRetries: 3,
+                retryDelay: 10,
+            });
+        }
+    });
+    async function runLoadPoliciesFromToml(tomlContent, fileName = 'test.toml') {
+        await fs.writeFile(path.join(tempDir, fileName), tomlContent);
+        const getPolicyTier = (_dir) => 1;
+        return loadPoliciesFromToml(ApprovalMode.DEFAULT, [tempDir], getPolicyTier);
+    }
+    describe('loadPoliciesFromToml', () => {
+        it('should load and parse a simple policy file', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "glob"
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(1);
+            expect(result.rules[0]).toEqual({
+                toolName: 'glob',
+                decision: PolicyDecision.ALLOW,
+                priority: 1.1, // tier 1 + 100/1000
+            });
+            expect(result.checkers).toHaveLength(0);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should expand commandPrefix array to multiple rules', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandPrefix = ["git status", "git log"]
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(2);
+            expect(result.rules[0].toolName).toBe('run_shell_command');
+            expect(result.rules[1].toolName).toBe('run_shell_command');
+            expect(result.rules[0].argsPattern?.test('{"command":"git status"}')).toBe(true);
+            expect(result.rules[1].argsPattern?.test('{"command":"git log"}')).toBe(true);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should transform commandRegex to argsPattern', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandRegex = "git (status|log).*"
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(1);
+            expect(result.rules[0].argsPattern?.test('{"command":"git status"}')).toBe(true);
+            expect(result.rules[0].argsPattern?.test('{"command":"git log --all"}')).toBe(true);
+            expect(result.rules[0].argsPattern?.test('{"command":"git branch"}')).toBe(false);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should expand toolName array', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = ["glob", "grep", "read"]
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(3);
+            expect(result.rules.map((r) => r.toolName)).toEqual([
+                'glob',
+                'grep',
+                'read',
+            ]);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should transform mcpName to composite toolName', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+mcpName = "google-workspace"
+toolName = ["calendar.list", "calendar.get"]
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(2);
+            expect(result.rules[0].toolName).toBe('google-workspace__calendar.list');
+            expect(result.rules[1].toolName).toBe('google-workspace__calendar.get');
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should filter rules by mode', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "glob"
+decision = "allow"
+priority = 100
+modes = ["default", "yolo"]
+
+[[rule]]
+toolName = "grep"
+decision = "allow"
+priority = 100
+modes = ["yolo"]
+`);
+            // Only the first rule should be included (modes includes "default")
+            expect(result.rules).toHaveLength(1);
+            expect(result.rules[0].toolName).toBe('glob');
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should handle TOML parse errors', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]
+toolName = "glob"
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(0);
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].errorType).toBe('toml_parse');
+            expect(result.errors[0].fileName).toBe('test.toml');
+        });
+        it('should handle schema validation errors', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "glob"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(0);
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].errorType).toBe('schema_validation');
+            expect(result.errors[0].details).toContain('decision');
+        });
+        it('should reject commandPrefix without run_shell_command', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "glob"
+commandPrefix = "git status"
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].errorType).toBe('rule_validation');
+            expect(result.errors[0].details).toContain('run_shell_command');
+        });
+        it('should reject commandPrefix + argsPattern combination', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandPrefix = "git status"
+argsPattern = "test"
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].errorType).toBe('rule_validation');
+            expect(result.errors[0].details).toContain('mutually exclusive');
+        });
+        it('should handle invalid regex patterns', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandRegex = "git (status|branch"
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(0);
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].errorType).toBe('regex_compilation');
+            expect(result.errors[0].details).toContain('git (status|branch');
+        });
+        it('should escape regex special characters in commandPrefix', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandPrefix = "git log *.txt"
+decision = "allow"
+priority = 100
+`);
+            expect(result.rules).toHaveLength(1);
+            // The regex should have escaped the * and .
+            expect(result.rules[0].argsPattern?.test('{"command":"git log file.txt"}')).toBe(false);
+            expect(result.rules[0].argsPattern?.test('{"command":"git log *.txt"}')).toBe(true);
+            expect(result.errors).toHaveLength(0);
+        });
+        it('should handle a mix of valid and invalid policy files', async () => {
+            await fs.writeFile(path.join(tempDir, 'valid.toml'), `
+[[rule]]
+toolName = "glob"
+decision = "allow"
+priority = 100
+`);
+            await fs.writeFile(path.join(tempDir, 'invalid.toml'), `
+[[rule]]
+toolName = "grep"
+decision = "allow"
+priority = -1
+`);
+            const getPolicyTier = (_dir) => 1;
+            const result = await loadPoliciesFromToml(ApprovalMode.DEFAULT, [tempDir], getPolicyTier);
+            expect(result.rules).toHaveLength(1);
+            expect(result.rules[0].toolName).toBe('glob');
+            expect(result.errors).toHaveLength(1);
+            expect(result.errors[0].fileName).toBe('invalid.toml');
+            expect(result.errors[0].errorType).toBe('schema_validation');
+        });
+    });
+    describe('Negative Tests', () => {
+        it('should return a schema_validation error if priority is missing', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+        });
+        it('should return a schema_validation error if priority is a float', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+priority = 1.5
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+            expect(error.details).toContain('integer');
+        });
+        it('should return a schema_validation error if priority is negative', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+priority = -1
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+            expect(error.details).toContain('>= 0');
+        });
+        it('should return a schema_validation error if priority is much lower than 0', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+priority = -9999
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+            expect(error.details).toContain('>= 0');
+        });
+        it('should return a schema_validation error if priority is >= 1000', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+priority = 1000
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+            expect(error.details).toContain('<= 999');
+        });
+        it('should return a schema_validation error if priority is much higher than 1000', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "allow"
+priority = 9999
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('priority');
+            expect(error.details).toContain('<= 999');
+        });
+        it('should return a schema_validation error if decision is invalid', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+decision = "maybe"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('decision');
+        });
+        it('should return a schema_validation error if toolName is not a string or array', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = 123
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('schema_validation');
+            expect(error.details).toContain('toolName');
+        });
+        it('should return a rule_validation error if commandRegex is used with wrong toolName', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "not_shell"
+commandRegex = ".*"
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('rule_validation');
+            expect(error.details).toContain('run_shell_command');
+        });
+        it('should return a rule_validation error if commandPrefix and commandRegex are combined', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "run_shell_command"
+commandPrefix = "git"
+commandRegex = ".*"
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('rule_validation');
+            expect(error.details).toContain('mutually exclusive');
+        });
+        it('should return a regex_compilation error for invalid argsPattern', async () => {
+            const result = await runLoadPoliciesFromToml(`
+[[rule]]
+toolName = "test"
+argsPattern = "([a-z)"
+decision = "allow"
+priority = 100
+`);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('regex_compilation');
+            expect(error.message).toBe('Invalid regex pattern');
+        });
+        it('should return a file_read error if readdir fails', async () => {
+            // Create a file and pass it as a directory to trigger ENOTDIR
+            const filePath = path.join(tempDir, 'not-a-dir');
+            await fs.writeFile(filePath, 'content');
+            const getPolicyTier = (_dir) => 1;
+            const result = await loadPoliciesFromToml(ApprovalMode.DEFAULT, [filePath], getPolicyTier);
+            expect(result.errors).toHaveLength(1);
+            const error = result.errors[0];
+            expect(error.errorType).toBe('file_read');
+            expect(error.message).toContain('Failed to read policy directory');
+        });
+    });
+});
+//# sourceMappingURL=toml-loader.test.js.map

package/dist/src/policy/toml-loader.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toml-loader.test.js","sourceRoot":"","sources":["../../../src/policy/toml-loader.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC1D,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAGxD,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,IAAI,OAAe,CAAC;IAEpB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,OAAO,EAAE,CAAC;YACZ,MAAM,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE;gBACnB,SAAS,EAAE,IAAI;gBACf,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,CAAC;gBACb,UAAU,EAAE,EAAE;aACf,CAAC,CAAC;QACL,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,uBAAuB,CACpC,WAAmB,EACnB,QAAQ,GAAG,WAAW;QAEtB,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,EAAE,WAAW,CAAC,CAAC;QAC9D,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1C,OAAO,oBAAoB,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,CAAC;IAC9E,CAAC;IAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAC9B,QAAQ,EAAE,MAAM;gBAChB,QAAQ,EAAE,cAAc,CAAC,KAAK;gBAC9B,QAAQ,EAAE,GAAG,EAAE,oBAAoB;aACpC,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACxC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC3D,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAC9D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CACrE,IAAI,CACL,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAC9D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,6BAA6B,CAAC,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAC9D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACd,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC;gBAClD,MAAM;gBACN,MAAM;gBACN,MAAM;aACP,CAAC,CAAC;YACH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;YACzE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;;;;;;;CAYlD,CAAC,CAAC;YAEG,oEAAoE;YACpE,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;CAIlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;;CAOlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YAEG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,4CAA4C;YAC5C,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,gCAAgC,CAAC,CACpE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACd,MAAM,CACJ,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,CAAC,6BAA6B,CAAC,CACjE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACb,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,YAAY,CAAC,EAChC;;;;;CAKP,CACM,CAAC;YAEF,MAAM,EAAE,CAAC,SAAS,CAChB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,EAClC;;;;;CAKP,CACM,CAAC;YAEF,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,oBAAoB,CACvC,YAAY,CAAC,OAAO,EACpB,CAAC,OAAO,CAAC,EACT,aAAa,CACd,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;CAIlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;YACxF,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;CAKlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;YACjG,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;YACpG,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;;CAOlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAChD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC;;;;;;CAMlD,CAAC,CAAC;YACG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAClD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,8DAA8D;YAC9D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;YACjD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;YAExC,MAAM,aAAa,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1C,MAAM,MAAM,GAAG,MAAM,oBAAoB,CACvC,YAAY,CAAC,OAAO,EACpB,CAAC,QAAQ,CAAC,EACV,aAAa,CACd,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACtC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/policy/types.d.ts

@@ -3,11 +3,55 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import type { SafetyCheckInput } from '../safety/protocol.js';
 export declare enum PolicyDecision {
     ALLOW = "allow",
     DENY = "deny",
     ASK_USER = "ask_user"
 }
+export declare enum ApprovalMode {
+    DEFAULT = "default",
+    AUTO_EDIT = "autoEdit",
+    YOLO = "yolo"
+}
+/**
+ * Configuration for the built-in allowed-path checker.
+ */
+export interface AllowedPathConfig {
+    /**
+     * Explicitly include argument keys to be checked as paths.
+     */
+    included_args?: string[];
+    /**
+     * Explicitly exclude argument keys from being checked as paths.
+     */
+    excluded_args?: string[];
+}
+/**
+ * Base interface for external checkers.
+ */
+export interface ExternalCheckerConfig {
+    type: 'external';
+    name: string;
+    config?: unknown;
+    required_context?: Array<keyof SafetyCheckInput['context']>;
+}
+export declare enum InProcessCheckerType {
+    ALLOWED_PATH = "allowed-path"
+}
+/**
+ * Base interface for in-process checkers.
+ */
+export interface InProcessCheckerConfig {
+    type: 'in-process';
+    name: InProcessCheckerType;
+    config?: AllowedPathConfig;
+    required_context?: Array<keyof SafetyCheckInput['context']>;
+}
+/**
+ * A discriminated union for all safety checker configurations.
+ */
+export type SafetyCheckerConfig = ExternalCheckerConfig | InProcessCheckerConfig;
 export interface PolicyRule {
     /**
      * The name of the tool this rule applies to.
@@ -29,11 +73,37 @@ export interface PolicyRule {
      */
     priority?: number;
 }
+export interface SafetyCheckerRule {
+    /**
+     * The name of the tool this rule applies to.
+     * If undefined, the rule applies to all tools.
+     */
+    toolName?: string;
+    /**
+     * Pattern to match against tool arguments.
+     * Can be used for more fine-grained control.
+     */
+    argsPattern?: RegExp;
+    /**
+     * Priority of this checker. Higher numbers run first.
+     * Default is 0.
+     */
+    priority?: number;
+    /**
+     * Specifies an external or built-in safety checker to execute for
+     * additional validation of a tool call.
+     */
+    checker: SafetyCheckerConfig;
+}
 export interface PolicyEngineConfig {
     /**
      * List of policy rules to apply.
      */
     rules?: PolicyRule[];
+    /**
+     * List of safety checkers to apply.
+     */
+    checkers?: SafetyCheckerRule[];
     /**
      * Default decision when no rules match.
      * Defaults to ASK_USER.
@@ -45,3 +115,16 @@ export interface PolicyEngineConfig {
      */
     nonInteractive?: boolean;
 }
+export interface PolicySettings {
+    mcp?: {
+        excluded?: string[];
+        allowed?: string[];
+    };
+    tools?: {
+        exclude?: string[];
+        allowed?: string[];
+    };
+    mcpServers?: Record<string, {
+        trust?: boolean;
+    }>;
+}

package/dist/src/policy/types.js

@@ -9,4 +9,14 @@ export var PolicyDecision;
     PolicyDecision["DENY"] = "deny";
     PolicyDecision["ASK_USER"] = "ask_user";
 })(PolicyDecision || (PolicyDecision = {}));
+export var ApprovalMode;
+(function (ApprovalMode) {
+    ApprovalMode["DEFAULT"] = "default";
+    ApprovalMode["AUTO_EDIT"] = "autoEdit";
+    ApprovalMode["YOLO"] = "yolo";
+})(ApprovalMode || (ApprovalMode = {}));
+export var InProcessCheckerType;
+(function (InProcessCheckerType) {
+    InProcessCheckerType["ALLOWED_PATH"] = "allowed-path";
+})(InProcessCheckerType || (InProcessCheckerType = {}));
 //# sourceMappingURL=types.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/policy/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,MAAM,CAAN,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,iCAAe,CAAA;IACf,+BAAa,CAAA;IACb,uCAAqB,CAAA;AACvB,CAAC,EAJW,cAAc,KAAd,cAAc,QAIzB"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/policy/types.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,CAAN,IAAY,cAIX;AAJD,WAAY,cAAc;IACxB,iCAAe,CAAA;IACf,+BAAa,CAAA;IACb,uCAAqB,CAAA;AACvB,CAAC,EAJW,cAAc,KAAd,cAAc,QAIzB;AAED,MAAM,CAAN,IAAY,YAIX;AAJD,WAAY,YAAY;IACtB,mCAAmB,CAAA;IACnB,sCAAsB,CAAA;IACtB,6BAAa,CAAA;AACf,CAAC,EAJW,YAAY,KAAZ,YAAY,QAIvB;AA2BD,MAAM,CAAN,IAAY,oBAEX;AAFD,WAAY,oBAAoB;IAC9B,qDAA6B,CAAA;AAC/B,CAAC,EAFW,oBAAoB,KAApB,oBAAoB,QAE/B"}

package/dist/src/prompts/mcp-prompts.test.d.ts

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

package/dist/src/prompts/mcp-prompts.test.js

@@ -0,0 +1,39 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { getMCPServerPrompts } from './mcp-prompts.js';
+import { PromptRegistry } from './prompt-registry.js';
+describe('getMCPServerPrompts', () => {
+    it('should return prompts from the registry for a given server', () => {
+        const mockPrompts = [
+            {
+                name: 'prompt1',
+                serverName: 'server1',
+                invoke: async () => ({
+                    messages: [
+                        { role: 'assistant', content: { type: 'text', text: '' } },
+                    ],
+                }),
+            },
+        ];
+        const mockRegistry = new PromptRegistry();
+        vi.spyOn(mockRegistry, 'getPromptsByServer').mockReturnValue(mockPrompts);
+        const mockConfig = {
+            getPromptRegistry: () => mockRegistry,
+        };
+        const result = getMCPServerPrompts(mockConfig, 'server1');
+        expect(mockRegistry.getPromptsByServer).toHaveBeenCalledWith('server1');
+        expect(result).toEqual(mockPrompts);
+    });
+    it('should return an empty array if there is no prompt registry', () => {
+        const mockConfig = {
+            getPromptRegistry: () => undefined,
+        };
+        const result = getMCPServerPrompts(mockConfig, 'server1');
+        expect(result).toEqual([]);
+    });
+});
+//# sourceMappingURL=mcp-prompts.test.js.map

package/dist/src/prompts/mcp-prompts.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-prompts.test.js","sourceRoot":"","sources":["../../../src/prompts/mcp-prompts.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAEvD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,WAAW,GAA0B;YACzC;gBACE,IAAI,EAAE,SAAS;gBACf,UAAU,EAAE,SAAS;gBACrB,MAAM,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;oBACnB,QAAQ,EAAE;wBACR,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE;qBAC3D;iBACF,CAAC;aACH;SACF,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,cAAc,EAAE,CAAC;QAC1C,EAAE,CAAC,KAAK,CAAC,YAAY,EAAE,oBAAoB,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QAE1E,MAAM,UAAU,GAAG;YACjB,iBAAiB,EAAE,GAAG,EAAE,CAAC,YAAY;SACjB,CAAC;QAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAE1D,MAAM,CAAC,YAAY,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACxE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,UAAU,GAAG;YACjB,iBAAiB,EAAE,GAAG,EAAE,CAAC,SAAS;SACd,CAAC;QAEvB,MAAM,MAAM,GAAG,mBAAmB,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/prompts/prompt-registry.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { debugLogger } from '../utils/debugLogger.js';
 export class PromptRegistry {
     prompts = new Map();
     /**
@@ -12,7 +13,7 @@ export class PromptRegistry {
     registerPrompt(prompt) {
         if (this.prompts.has(prompt.name)) {
             const newName = `${prompt.serverName}_${prompt.name}`;
-            console.warn(`Prompt with name "${prompt.name}" is already registered. Renaming to "${newName}".`);
+            debugLogger.warn(`Prompt with name "${prompt.name}" is already registered. Renaming to "${newName}".`);
             this.prompts.set(newName, { ...prompt, name: newName });
         }
         else {

package/dist/src/prompts/prompt-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"prompt-registry.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,MAAM,OAAO,cAAc;IACjB,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IAE9D;;;OAGG;IACH,cAAc,CAAC,MAA2B;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACtD,OAAO,CAAC,IAAI,CACV,qBAAqB,MAAM,CAAC,IAAI,yCAAyC,OAAO,IAAI,CACrF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,UAAkB;QACnC,MAAM,aAAa,GAA0B,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,qBAAqB,CAAC,UAAkB;QACtC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}
+{"version":3,"file":"prompt-registry.js","sourceRoot":"","sources":["../../../src/prompts/prompt-registry.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAEtD,MAAM,OAAO,cAAc;IACjB,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IAE9D;;;OAGG;IACH,cAAc,CAAC,MAA2B;QACxC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,GAAG,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YACtD,WAAW,CAAC,IAAI,CACd,qBAAqB,MAAM,CAAC,IAAI,yCAAyC,OAAO,IAAI,CACrF,CAAC;YACF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACrD,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAC7B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,UAAkB;QACnC,MAAM,aAAa,GAA0B,EAAE,CAAC;QAChD,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC7B,CAAC;QACH,CAAC;QACD,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACpE,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,qBAAqB,CAAC,UAAkB;QACtC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACpD,IAAI,MAAM,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;gBACrC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;CACF"}

package/dist/src/prompts/prompt-registry.test.d.ts

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