@google/gemini-cli-core 0.1.0-development.1

package/dist/src/tools/shell.test.js

@@ -0,0 +1,321 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { expect, describe, it } from 'vitest';
+import { ShellTool } from './shell.js';
+describe('ShellTool', () => {
+    it('should allow a command if no restrictions are provided', async () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => undefined,
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ls -l');
+        expect(result.allowed).toBe(true);
+    });
+    it('should allow a command if it is in the allowed list', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool(ls -l)'],
+            getExcludeTools: () => undefined,
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ls -l');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block a command if it is not in the allowed list', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool(ls -l)'],
+            getExcludeTools: () => undefined,
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is not in the allowed commands list");
+    });
+    it('should block a command if it is in the blocked list', async () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => ['ShellTool(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should allow a command if it is not in the blocked list', async () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => ['ShellTool(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ls -l');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block a command if it is in both the allowed and blocked lists', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool(rm -rf /)'],
+            getExcludeTools: () => ['ShellTool(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should allow any command when ShellTool is in coreTools without specific commands', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block any command when ShellTool is in excludeTools without specific commands', async () => {
+        const config = {
+            getCoreTools: () => [],
+            getExcludeTools: () => ['ShellTool'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe('Shell tool is globally disabled in configuration');
+    });
+    it('should allow a command if it is in the allowed list using the public-facing name', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(ls -l)'],
+            getExcludeTools: () => undefined,
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ls -l');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block a command if it is in the blocked list using the public-facing name', async () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => ['run_shell_command(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should block any command when ShellTool is in excludeTools using the public-facing name', async () => {
+        const config = {
+            getCoreTools: () => [],
+            getExcludeTools: () => ['run_shell_command'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe('Shell tool is globally disabled in configuration');
+    });
+    it('should block any command if coreTools contains an empty ShellTool command list using the public-facing name', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command()'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'any command' is not in the allowed commands list");
+    });
+    it('should block any command if coreTools contains an empty ShellTool command list', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool()'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'any command' is not in the allowed commands list");
+    });
+    it('should block a command with extra whitespace if it is in the blocked list', async () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => ['ShellTool(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed(' rm  -rf  / ');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should allow any command when ShellTool is present with specific commands', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool', 'ShellTool(ls)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('any command');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block a command on the blocklist even with a wildcard allow', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool'],
+            getExcludeTools: () => ['ShellTool(rm -rf /)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should allow a command that starts with an allowed command prefix', async () => {
+        const config = {
+            getCoreTools: () => ['ShellTool(gh issue edit)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue edit 1 --add-label "kind/feature"');
+        expect(result.allowed).toBe(true);
+    });
+    it('should allow a command that starts with an allowed command prefix using the public-facing name', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue edit)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue edit 1 --add-label "kind/feature"');
+        expect(result.allowed).toBe(true);
+    });
+    it('should not allow a command that starts with an allowed command prefix but is chained with another command', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue edit)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue edit&&rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is not in the allowed commands list");
+    });
+    it('should not allow a command that is a prefix of an allowed command', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue edit)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'gh issue' is not in the allowed commands list");
+    });
+    it('should not allow a command that is a prefix of a blocked command', async () => {
+        const config = {
+            getCoreTools: () => [],
+            getExcludeTools: () => ['run_shell_command(gh issue edit)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue');
+        expect(result.allowed).toBe(true);
+    });
+    it('should not allow a command that is chained with a pipe', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue list)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue list | rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is not in the allowed commands list");
+    });
+    it('should not allow a command that is chained with a semicolon', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue list)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue list; rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is not in the allowed commands list");
+    });
+    it('should block a chained command if any part is blocked', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(echo "hello")'],
+            getExcludeTools: () => ['run_shell_command(rm)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('echo "hello" && rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should block a command if its prefix is on the blocklist, even if the command itself is on the allowlist', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(git push)'],
+            getExcludeTools: () => ['run_shell_command(git)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('git push');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'git push' is blocked by configuration");
+    });
+    it('should be case-sensitive in its matching', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(echo)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ECHO "hello"');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe('Command \'ECHO "hello"\' is not in the allowed commands list');
+    });
+    it('should correctly handle commands with extra whitespace around chaining operators', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(ls -l)'],
+            getExcludeTools: () => ['run_shell_command(rm)'],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('ls -l  ;  rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is blocked by configuration");
+    });
+    it('should allow a chained command if all parts are allowed', async () => {
+        const config = {
+            getCoreTools: () => [
+                'run_shell_command(echo)',
+                'run_shell_command(ls -l)',
+            ],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('echo "hello" && ls -l');
+        expect(result.allowed).toBe(true);
+    });
+    it('should allow a command with command substitution using backticks', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(echo)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('echo `rm -rf /`');
+        expect(result.allowed).toBe(true);
+    });
+    it('should block a command with command substitution using $()', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(echo)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('echo $(rm -rf /)');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe('Command substitution using $() is not allowed for security reasons');
+    });
+    it('should allow a command with I/O redirection', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(echo)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('echo "hello" > file.txt');
+        expect(result.allowed).toBe(true);
+    });
+    it('should not allow a command that is chained with a double pipe', async () => {
+        const config = {
+            getCoreTools: () => ['run_shell_command(gh issue list)'],
+            getExcludeTools: () => [],
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.isCommandAllowed('gh issue list || rm -rf /');
+        expect(result.allowed).toBe(false);
+        expect(result.reason).toBe("Command 'rm -rf /' is not in the allowed commands list");
+    });
+});
+//# sourceMappingURL=shell.test.js.map

package/dist/src/tools/shell.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shell.test.js","sourceRoot":"","sources":["../../../src/tools/shell.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAGvC,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;SACvB,CAAC;QACZ,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kBAAkB,CAAC;YACxC,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;SACZ,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kBAAkB,CAAC;YACxC,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;SACZ,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;SAC1B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;SAC1B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;YAC3C,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;SAC1B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mFAAmF,EAAE,KAAK,IAAI,EAAE;QACjG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW,CAAC;YACjC,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sFAAsF,EAAE,KAAK,IAAI,EAAE;QACpG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,EAAE;YACtB,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW,CAAC;SAChB,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,kDAAkD,CACnD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,0BAA0B,CAAC;YAChD,eAAe,EAAE,GAAG,EAAE,CAAC,SAAS;SACZ,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,6BAA6B,CAAC;SAClC,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,EAAE;YACtB,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,mBAAmB,CAAC;SACxB,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,kDAAkD,CACnD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6GAA6G,EAAE,KAAK,IAAI,EAAE;QAC3H,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;YAC3C,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,2DAA2D,CAC5D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,aAAa,CAAC;YACnC,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,2DAA2D,CAC5D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;YAC7B,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;SAC1B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW,EAAE,eAAe,CAAC;YAClD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;QAClF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,WAAW,CAAC;YACjC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC;SAC1B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,0BAA0B,CAAC;YAChD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CACvC,4CAA4C,CAC7C,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gGAAgG,EAAE,KAAK,IAAI,EAAE;QAC9G,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CACvC,4CAA4C,CAC7C,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2GAA2G,EAAE,KAAK,IAAI,EAAE;QACzH,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,EAAE;YACtB,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;SACvC,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,0BAA0B,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;QAC3E,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,iCAAiC,CAAC;YACvD,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,uBAAuB,CAAC;SAC5B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,0BAA0B,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0GAA0G,EAAE,KAAK,IAAI,EAAE;QACxH,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,6BAA6B,CAAC;YACnD,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,wBAAwB,CAAC;SAC7B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,yBAAyB,CAAC;YAC/C,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,cAAc,CAAC,CAAC;QAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,8DAA8D,CAC/D,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,0BAA0B,CAAC;YAChD,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,uBAAuB,CAAC;SAC5B,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,oBAAoB,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,gDAAgD,CACjD,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC;gBAClB,yBAAyB;gBACzB,0BAA0B;aAC3B;YACD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,uBAAuB,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,yBAAyB,CAAC;YAC/C,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,yBAAyB,CAAC;YAC/C,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,kBAAkB,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,oEAAoE,CACrE,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,yBAAyB,CAAC;YAC/C,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,yBAAyB,CAAC,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,MAAM,GAAG;YACb,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,kCAAkC,CAAC;YACxD,eAAe,EAAE,GAAG,EAAE,CAAC,EAAE;SACL,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC,MAAM,CAAC,CAAC;QACxC,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,CAAC,2BAA2B,CAAC,CAAC;QACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACxB,wDAAwD,CACzD,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/tool-registry.d.ts

@@ -0,0 +1,67 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { FunctionDeclaration, Schema } from '@google/genai';
+import { Tool, ToolResult, BaseTool } from './tools.js';
+import { Config } from '../config/config.js';
+type ToolParams = Record<string, unknown>;
+export declare class DiscoveredTool extends BaseTool<ToolParams, ToolResult> {
+    private readonly config;
+    readonly name: string;
+    readonly description: string;
+    readonly parameterSchema: Record<string, unknown>;
+    constructor(config: Config, name: string, description: string, parameterSchema: Record<string, unknown>);
+    execute(params: ToolParams): Promise<ToolResult>;
+}
+export declare class ToolRegistry {
+    private tools;
+    private config;
+    constructor(config: Config);
+    /**
+     * Registers a tool definition.
+     * @param tool - The tool object containing schema and execution logic.
+     */
+    registerTool(tool: Tool): void;
+    /**
+     * Discovers tools from project (if available and configured).
+     * Can be called multiple times to update discovered tools.
+     */
+    discoverTools(): Promise<void>;
+    private discoverAndRegisterToolsFromCommand;
+    /**
+     * Retrieves the list of tool schemas (FunctionDeclaration array).
+     * Extracts the declarations from the ToolListUnion structure.
+     * Includes discovered (vs registered) tools if configured.
+     * @returns An array of FunctionDeclarations.
+     */
+    getFunctionDeclarations(): FunctionDeclaration[];
+    /**
+     * Returns an array of all registered and discovered tool instances.
+     */
+    getAllTools(): Tool[];
+    /**
+     * Returns an array of tools registered from a specific MCP server.
+     */
+    getToolsByServer(serverName: string): Tool[];
+    /**
+     * Get the definition of a specific tool.
+     */
+    getTool(name: string): Tool | undefined;
+}
+/**
+ * Sanitizes a schema object in-place to ensure compatibility with the Gemini API.
+ *
+ * NOTE: This function mutates the passed schema object.
+ *
+ * It performs the following actions:
+ * - Removes the `default` property when `anyOf` is present.
+ * - Removes unsupported `format` values from string properties, keeping only 'enum' and 'date-time'.
+ * - Recursively sanitizes nested schemas within `anyOf`, `items`, and `properties`.
+ * - Handles circular references within the schema to prevent infinite loops.
+ *
+ * @param schema The schema object to sanitize. It will be modified directly.
+ */
+export declare function sanitizeParameters(schema?: Schema): void;
+export {};