@zgsm/costrict-cli 0.0.1

package/dist/src/commands/mcp/remove.test.js

@@ -0,0 +1,189 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+import yargs, {} from 'yargs';
+import { SettingScope } from '../../config/settings.js';
+import { removeCommand } from './remove.js';
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import { GEMINI_DIR, debugLogger } from '@google/gemini-cli-core';
+vi.mock('fs/promises', () => ({
+    readFile: vi.fn(),
+    writeFile: vi.fn(),
+}));
+vi.mock('../utils.js', () => ({
+    exitCli: vi.fn(),
+}));
+describe('mcp remove command', () => {
+    describe('unit tests with mocks', () => {
+        let parser;
+        let mockSetValue;
+        let mockSettings;
+        beforeEach(async () => {
+            vi.resetAllMocks();
+            mockSetValue = vi.fn();
+            mockSettings = {
+                mcpServers: {
+                    'test-server': {
+                        command: 'echo "hello"',
+                    },
+                },
+            };
+            vi.spyOn(await import('../../config/settings.js'), 'loadSettings').mockReturnValue({
+                forScope: () => ({ settings: mockSettings }),
+                setValue: mockSetValue,
+                workspace: { path: '/path/to/project' },
+                user: { path: '/home/user' },
+            });
+            const yargsInstance = yargs([]).command(removeCommand);
+            parser = yargsInstance;
+        });
+        it('should remove a server from project settings', async () => {
+            await parser.parseAsync('remove test-server');
+            expect(mockSetValue).toHaveBeenCalledWith(SettingScope.Workspace, 'mcpServers', {});
+        });
+        it('should show a message if server not found', async () => {
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await parser.parseAsync('remove non-existent-server');
+            expect(mockSetValue).not.toHaveBeenCalled();
+            expect(debugLogSpy).toHaveBeenCalledWith('Server "non-existent-server" not found in project settings.');
+            debugLogSpy.mockRestore();
+        });
+    });
+    describe('integration tests with real file I/O', () => {
+        let tempDir;
+        let settingsDir;
+        let settingsPath;
+        let parser;
+        let cwdSpy;
+        beforeEach(() => {
+            vi.resetAllMocks();
+            vi.restoreAllMocks();
+            tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'mcp-remove-test-'));
+            settingsDir = path.join(tempDir, GEMINI_DIR);
+            settingsPath = path.join(settingsDir, 'settings.json');
+            fs.mkdirSync(settingsDir, { recursive: true });
+            cwdSpy = vi.spyOn(process, 'cwd').mockReturnValue(tempDir);
+            parser = yargs([]).command(removeCommand);
+        });
+        afterEach(() => {
+            cwdSpy.mockRestore();
+            if (fs.existsSync(tempDir)) {
+                fs.rmSync(tempDir, { recursive: true, force: true });
+            }
+        });
+        it('should actually remove a server from the settings file', async () => {
+            const originalContent = `{
+        "mcpServers": {
+          "server-to-keep": {
+            "command": "node",
+            "args": ["keep.js"]
+          },
+          "server-to-remove": {
+            "command": "node",
+            "args": ["remove.js"]
+          }
+        }
+      }`;
+            fs.writeFileSync(settingsPath, originalContent, 'utf-8');
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await parser.parseAsync('remove server-to-remove');
+            const updatedContent = fs.readFileSync(settingsPath, 'utf-8');
+            expect(updatedContent).toContain('"server-to-keep"');
+            expect(updatedContent).not.toContain('"server-to-remove"');
+            expect(debugLogSpy).toHaveBeenCalledWith('Server "server-to-remove" removed from project settings.');
+            debugLogSpy.mockRestore();
+        });
+        it('should preserve comments when removing a server', async () => {
+            const originalContent = `{
+        "mcpServers": {
+          // Server to keep
+          "context7": {
+            "command": "node",
+            "args": ["server.js"]
+          },
+          // Server to remove
+          "oldServer": {
+            "command": "old",
+            "args": ["old.js"]
+          }
+        }
+      }`;
+            fs.writeFileSync(settingsPath, originalContent, 'utf-8');
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await parser.parseAsync('remove oldServer');
+            const updatedContent = fs.readFileSync(settingsPath, 'utf-8');
+            expect(updatedContent).toContain('// Server to keep');
+            expect(updatedContent).toContain('"context7"');
+            expect(updatedContent).not.toContain('"oldServer"');
+            expect(updatedContent).toContain('// Server to remove');
+            debugLogSpy.mockRestore();
+        });
+        it('should handle removing the only server', async () => {
+            const originalContent = `{
+        "mcpServers": {
+          "only-server": {
+            "command": "node",
+            "args": ["server.js"]
+          }
+        }
+      }`;
+            fs.writeFileSync(settingsPath, originalContent, 'utf-8');
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await parser.parseAsync('remove only-server');
+            const updatedContent = fs.readFileSync(settingsPath, 'utf-8');
+            expect(updatedContent).toContain('"mcpServers"');
+            expect(updatedContent).not.toContain('"only-server"');
+            expect(updatedContent).toMatch(/"mcpServers"\s*:\s*\{\s*\}/);
+            debugLogSpy.mockRestore();
+        });
+        it('should preserve other settings when removing a server', async () => {
+            // Create settings file with other settings
+            // Note: "model" will be migrated to "model": { "name": ... } format
+            const originalContent = `{
+        "model": {
+          "name": "gemini-2.5-pro"
+        },
+        "mcpServers": {
+          "server1": {
+            "command": "node",
+            "args": ["s1.js"]
+          },
+          "server2": {
+            "command": "node",
+            "args": ["s2.js"]
+          }
+        },
+        "ui": {
+          "theme": "dark"
+        }
+      }`;
+            fs.writeFileSync(settingsPath, originalContent, 'utf-8');
+            const debugLogSpy = vi
+                .spyOn(debugLogger, 'log')
+                .mockImplementation(() => { });
+            await parser.parseAsync('remove server1');
+            const updatedContent = fs.readFileSync(settingsPath, 'utf-8');
+            expect(updatedContent).toContain('"model"');
+            expect(updatedContent).toContain('"gemini-2.5-pro"');
+            expect(updatedContent).toContain('"server2"');
+            expect(updatedContent).toContain('"ui"');
+            expect(updatedContent).toContain('"theme": "dark"');
+            expect(updatedContent).not.toContain('"server1"');
+            debugLogSpy.mockRestore();
+        });
+    });
+});
+//# sourceMappingURL=remove.test.js.map

package/dist/src/commands/mcp/remove.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"remove.test.js","sourceRoot":"","sources":["../../../../src/commands/mcp/remove.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,EAAE,EACF,QAAQ,EACR,EAAE,EACF,MAAM,EACN,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,KAAK,EAAE,EAAa,MAAM,OAAO,CAAC;AACzC,OAAO,EAAE,YAAY,EAAuB,MAAM,0BAA0B,CAAC;AAC7E,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAElE,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,QAAQ,EAAE,EAAE,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE;CACnB,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC;IAC5B,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE;CACjB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,IAAI,MAAY,CAAC;QACjB,IAAI,YAAkB,CAAC;QACvB,IAAI,YAAqC,CAAC;QAE1C,UAAU,CAAC,KAAK,IAAI,EAAE;YACpB,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,YAAY,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;YACvB,YAAY,GAAG;gBACb,UAAU,EAAE;oBACV,aAAa,EAAE;wBACb,OAAO,EAAE,cAAc;qBACxB;iBACF;aACF,CAAC;YAEF,EAAE,CAAC,KAAK,CACN,MAAM,MAAM,CAAC,0BAA0B,CAAC,EACxC,cAAc,CACf,CAAC,eAAe,CAAC;gBAChB,QAAQ,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;gBAC5C,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE;gBACvC,IAAI,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE;aACA,CAAC,CAAC;YAEhC,MAAM,aAAa,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YACvD,MAAM,GAAG,aAAa,CAAC;QACzB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;YAE9C,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CACvC,YAAY,CAAC,SAAS,EACtB,YAAY,EACZ,EAAE,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,UAAU,CAAC,4BAA4B,CAAC,CAAC;YAEtD,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC5C,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,6DAA6D,CAC9D,CAAC;YACF,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;QACpD,IAAI,OAAe,CAAC;QACpB,IAAI,WAAmB,CAAC;QACxB,IAAI,YAAoB,CAAC;QACzB,IAAI,MAAY,CAAC;QACjB,IAAI,MAAmC,CAAC;QAExC,UAAU,CAAC,GAAG,EAAE;YACd,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,EAAE,CAAC,eAAe,EAAE,CAAC;YAErB,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,kBAAkB,CAAC,CAAC,CAAC;YACrE,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YAC7C,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,eAAe,CAAC,CAAC;YACvD,EAAE,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAE/C,MAAM,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YAE3D,MAAM,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,MAAM,CAAC,WAAW,EAAE,CAAC;YAErB,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,eAAe,GAAG;;;;;;;;;;;QAWtB,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC;YAEnD,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YACrD,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YAE3D,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CACtC,0DAA0D,CAC3D,CAAC;YAEF,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,eAAe,GAAG;;;;;;;;;;;;;QAatB,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAE5C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YACtD,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;YAC/C,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACpD,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAExD,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,eAAe,GAAG;;;;;;;QAOtB,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC;YAE9C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;YACtD,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,4BAA4B,CAAC,CAAC;YAE7D,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,2CAA2C;YAC3C,oEAAoE;YACpE,MAAM,eAAe,GAAG;;;;;;;;;;;;;;;;;QAiBtB,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,YAAY,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,WAAW,GAAG,EAAE;iBACnB,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC;iBACzB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;YAE1C,MAAM,cAAc,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAC9D,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;YACrD,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YAC9C,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,CAAC,cAAc,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YACpD,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;YAElD,WAAW,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/commands/mcp.d.ts

@@ -0,0 +1,7 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { CommandModule } from 'yargs';
+export declare const mcpCommand: CommandModule;

package/dist/src/commands/mcp.js

@@ -0,0 +1,25 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { addCommand } from './mcp/add.js';
+import { removeCommand } from './mcp/remove.js';
+import { listCommand } from './mcp/list.js';
+import { initializeOutputListenersAndFlush } from '../gemini.js';
+export const mcpCommand = {
+    command: 'mcp',
+    describe: 'Manage MCP servers',
+    builder: (yargs) => yargs
+        .middleware(() => initializeOutputListenersAndFlush())
+        .command(addCommand)
+        .command(removeCommand)
+        .command(listCommand)
+        .demandCommand(1, 'You need at least one command before continuing.')
+        .version(false),
+    handler: () => {
+        // yargs will automatically show help if no subcommand is provided
+        // thanks to demandCommand(1) in the builder.
+    },
+};
+//# sourceMappingURL=mcp.js.map

package/dist/src/commands/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../../../src/commands/mcp.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAIH,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,iCAAiC,EAAE,MAAM,cAAc,CAAC;AAEjE,MAAM,CAAC,MAAM,UAAU,GAAkB;IACvC,OAAO,EAAE,KAAK;IACd,QAAQ,EAAE,oBAAoB;IAC9B,OAAO,EAAE,CAAC,KAAW,EAAE,EAAE,CACvB,KAAK;SACF,UAAU,CAAC,GAAG,EAAE,CAAC,iCAAiC,EAAE,CAAC;SACrD,OAAO,CAAC,UAAU,CAAC;SACnB,OAAO,CAAC,aAAa,CAAC;SACtB,OAAO,CAAC,WAAW,CAAC;SACpB,aAAa,CAAC,CAAC,EAAE,kDAAkD,CAAC;SACpE,OAAO,CAAC,KAAK,CAAC;IACnB,OAAO,EAAE,GAAG,EAAE;QACZ,kEAAkE;QAClE,6CAA6C;IAC/C,CAAC;CACF,CAAC"}

package/dist/src/commands/mcp.test.d.ts

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

package/dist/src/commands/mcp.test.js

@@ -0,0 +1,63 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi } from 'vitest';
+import { mcpCommand } from './mcp.js';
+import {} from 'yargs';
+import yargs from 'yargs';
+describe('mcp command', () => {
+    it('should have correct command definition', () => {
+        expect(mcpCommand.command).toBe('mcp');
+        expect(mcpCommand.describe).toBe('Manage MCP servers');
+        expect(typeof mcpCommand.builder).toBe('function');
+        expect(typeof mcpCommand.handler).toBe('function');
+    });
+    it('should show help when no subcommand is provided', async () => {
+        const yargsInstance = yargs();
+        mcpCommand.builder(yargsInstance);
+        const parser = yargsInstance.command(mcpCommand).help();
+        // Mock console.log and console.error to catch help output
+        const consoleLogMock = vi
+            .spyOn(console, 'log')
+            .mockImplementation(() => { });
+        const consoleErrorMock = vi
+            .spyOn(console, 'error')
+            .mockImplementation(() => { });
+        try {
+            await parser.parse('mcp');
+        }
+        catch (_error) {
+            // yargs might throw an error when demandCommand is not met
+        }
+        // Check if help output is shown
+        const helpOutput = consoleLogMock.mock.calls.join('\n') +
+            consoleErrorMock.mock.calls.join('\n');
+        expect(helpOutput).toContain('Manage MCP servers');
+        expect(helpOutput).toContain('Commands:');
+        expect(helpOutput).toContain('add');
+        expect(helpOutput).toContain('remove');
+        expect(helpOutput).toContain('list');
+        consoleLogMock.mockRestore();
+        consoleErrorMock.mockRestore();
+    });
+    it('should register add, remove, and list subcommands', () => {
+        const mockYargs = {
+            command: vi.fn().mockReturnThis(),
+            demandCommand: vi.fn().mockReturnThis(),
+            version: vi.fn().mockReturnThis(),
+            middleware: vi.fn().mockReturnThis(),
+        };
+        mcpCommand.builder(mockYargs);
+        expect(mockYargs.command).toHaveBeenCalledTimes(3);
+        // Verify that the specific subcommands are registered
+        const commandCalls = mockYargs.command.mock.calls;
+        const commandNames = commandCalls.map((call) => call[0].command);
+        expect(commandNames).toContain('add <name> <commandOrUrl> [args...]');
+        expect(commandNames).toContain('remove <name>');
+        expect(commandNames).toContain('list');
+        expect(mockYargs.demandCommand).toHaveBeenCalledWith(1, 'You need at least one command before continuing.');
+    });
+});
+//# sourceMappingURL=mcp.test.js.map

package/dist/src/commands/mcp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.test.js","sourceRoot":"","sources":["../../../src/commands/mcp.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,UAAU,CAAC;AACtC,OAAO,EAAa,MAAM,OAAO,CAAC;AAClC,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACvC,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,MAAM,aAAa,GAAG,KAAK,EAAE,CAAC;QAC7B,UAAU,CAAC,OAA6B,CAAC,aAAa,CAAC,CAAC;QAEzD,MAAM,MAAM,GAAG,aAAa,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD,0DAA0D;QAC1D,MAAM,cAAc,GAAG,EAAE;aACtB,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC;aACrB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChC,MAAM,gBAAgB,GAAG,EAAE;aACxB,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC;aACvB,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEhC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YAChB,2DAA2D;QAC7D,CAAC;QAED,gCAAgC;QAChC,MAAM,UAAU,GACd,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;YACpC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QACnD,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACpC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAErC,cAAc,CAAC,WAAW,EAAE,CAAC;QAC7B,gBAAgB,CAAC,WAAW,EAAE,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,SAAS,GAAG;YAChB,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YACjC,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YACvC,OAAO,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;YACjC,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;SACrC,CAAC;QAED,UAAU,CAAC,OAA6B,CAAC,SAA4B,CAAC,CAAC;QAExE,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAEnD,sDAAsD;QACtD,MAAM,YAAY,GAAG,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC;QAClD,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAEjE,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;QAChD,MAAM,CAAC,YAAY,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAEvC,MAAM,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAClD,CAAC,EACD,kDAAkD,CACnD,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/commands/utils.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare function exitCli(exitCode?: number): Promise<void>;

package/dist/src/commands/utils.js

@@ -0,0 +1,11 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { runExitCleanup } from '../utils/cleanup.js';
+export async function exitCli(exitCode = 0) {
+    await runExitCleanup();
+    process.exit(exitCode);
+}
+//# sourceMappingURL=utils.js.map

package/dist/src/commands/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/commands/utils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,QAAQ,GAAG,CAAC;IACxC,MAAM,cAAc,EAAE,CAAC;IACvB,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACzB,CAAC"}

package/dist/src/commands/utils.test.d.ts

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

package/dist/src/commands/utils.test.js

@@ -0,0 +1,35 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { exitCli } from './utils.js';
+import { runExitCleanup } from '../utils/cleanup.js';
+vi.mock('../utils/cleanup.js', () => ({
+    runExitCleanup: vi.fn(),
+}));
+describe('utils', () => {
+    const originalProcessExit = process.exit;
+    beforeEach(() => {
+        // @ts-expect-error - Mocking process.exit
+        process.exit = vi.fn();
+    });
+    afterEach(() => {
+        process.exit = originalProcessExit;
+        vi.clearAllMocks();
+    });
+    describe('exitCli', () => {
+        it('should call runExitCleanup and process.exit with default exit code 0', async () => {
+            await exitCli();
+            expect(runExitCleanup).toHaveBeenCalled();
+            expect(process.exit).toHaveBeenCalledWith(0);
+        });
+        it('should call runExitCleanup and process.exit with specified exit code', async () => {
+            await exitCli(1);
+            expect(runExitCleanup).toHaveBeenCalled();
+            expect(process.exit).toHaveBeenCalledWith(1);
+        });
+    });
+});
+//# sourceMappingURL=utils.test.js.map

package/dist/src/commands/utils.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.test.js","sourceRoot":"","sources":["../../../src/commands/utils.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACrC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAErD,EAAE,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACpC,cAAc,EAAE,EAAE,CAAC,EAAE,EAAE;CACxB,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;IACrB,MAAM,mBAAmB,GAAG,OAAO,CAAC,IAAI,CAAC;IAEzC,UAAU,CAAC,GAAG,EAAE;QACd,0CAA0C;QAC1C,OAAO,CAAC,IAAI,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,IAAI,GAAG,mBAAmB,CAAC;QACnC,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,OAAO,EAAE,CAAC;YAChB,MAAM,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,OAAO,CAAC,CAAC,CAAC,CAAC;YACjB,MAAM,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/config/auth.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare function validateAuthMethod(authMethod: string): string | null;

package/dist/src/config/auth.js

@@ -0,0 +1,53 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '@google/gemini-cli-core';
+import { loadEnvironment, loadSettings } from './settings.js';
+export function validateAuthMethod(authMethod) {
+    loadEnvironment(loadSettings().merged);
+    if (authMethod === AuthType.LOGIN_WITH_GOOGLE ||
+        authMethod === AuthType.COMPUTE_ADC) {
+        return null;
+    }
+    if (authMethod === AuthType.USE_GEMINI) {
+        if (!process.env['GEMINI_API_KEY']) {
+            return ('When using Gemini API, you must specify the GEMINI_API_KEY environment variable.\n' +
+                'Update your environment and try again (no reload needed if using .env)!');
+        }
+        return null;
+    }
+    if (authMethod === AuthType.USE_VERTEX_AI) {
+        const hasVertexProjectLocationConfig = !!process.env['GOOGLE_CLOUD_PROJECT'] &&
+            !!process.env['GOOGLE_CLOUD_LOCATION'];
+        const hasGoogleApiKey = !!process.env['GOOGLE_API_KEY'];
+        if (!hasVertexProjectLocationConfig && !hasGoogleApiKey) {
+            return ('When using Vertex AI, you must specify either:\n' +
+                '• GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION environment variables.\n' +
+                '• GOOGLE_API_KEY environment variable (if using express mode).\n' +
+                'Update your environment and try again (no reload needed if using .env)!');
+        }
+        return null;
+    }
+    if (authMethod === AuthType.CUSTOM_LLM_API) {
+        const hasApiKey = !!process.env['CUSTOM_LLM_API_KEY'];
+        const hasEndpoint = !!process.env['CUSTOM_LLM_ENDPOINT'];
+        const hasModelName = !!process.env['CUSTOM_LLM_MODEL_NAME'];
+        if (!hasApiKey || !hasEndpoint || !hasModelName) {
+            const missing = [];
+            if (!hasApiKey)
+                missing.push('CUSTOM_LLM_API_KEY');
+            if (!hasEndpoint)
+                missing.push('CUSTOM_LLM_ENDPOINT');
+            if (!hasModelName)
+                missing.push('CUSTOM_LLM_MODEL_NAME');
+            return ('When using Custom LLM (OpenAI Protocol), you must specify:\n' +
+                `• ${missing.join('\n• ')} environment variable${missing.length > 1 ? 's' : ''}.\n` +
+                'Update your environment and try again (no reload needed if using .env)!');
+        }
+        return null;
+    }
+    return 'Invalid auth method selected.';
+}
+//# sourceMappingURL=auth.js.map

package/dist/src/config/auth.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.js","sourceRoot":"","sources":["../../../src/config/auth.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAE9D,MAAM,UAAU,kBAAkB,CAAC,UAAkB;IACnD,eAAe,CAAC,YAAY,EAAE,CAAC,MAAM,CAAC,CAAC;IACvC,IACE,UAAU,KAAK,QAAQ,CAAC,iBAAiB;QACzC,UAAU,KAAK,QAAQ,CAAC,WAAW,EACnC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,UAAU,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC;QACvC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACnC,OAAO,CACL,oFAAoF;gBACpF,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,UAAU,KAAK,QAAQ,CAAC,aAAa,EAAE,CAAC;QAC1C,MAAM,8BAA8B,GAClC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;YACrC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QACzC,MAAM,eAAe,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACxD,IAAI,CAAC,8BAA8B,IAAI,CAAC,eAAe,EAAE,CAAC;YACxD,OAAO,CACL,kDAAkD;gBAClD,2EAA2E;gBAC3E,kEAAkE;gBAClE,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,UAAU,KAAK,QAAQ,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QACtD,MAAM,WAAW,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;QAE5D,IAAI,CAAC,SAAS,IAAI,CAAC,WAAW,IAAI,CAAC,YAAY,EAAE,CAAC;YAChD,MAAM,OAAO,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,SAAS;gBAAE,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;YACnD,IAAI,CAAC,WAAW;gBAAE,OAAO,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACtD,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YAEzD,OAAO,CACL,8DAA8D;gBAC9D,KAAK,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,wBAAwB,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK;gBACnF,yEAAyE,CAC1E,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,OAAO,+BAA+B,CAAC;AACzC,CAAC"}

package/dist/src/config/auth.test.d.ts

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

package/dist/src/config/auth.test.js

@@ -0,0 +1,145 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { AuthType } from '@google/gemini-cli-core';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+import { validateAuthMethod } from './auth.js';
+vi.mock('./settings.js', () => ({
+    loadEnvironment: vi.fn(),
+    loadSettings: vi.fn().mockReturnValue({
+        merged: vi.fn().mockReturnValue({}),
+    }),
+}));
+describe('validateAuthMethod', () => {
+    beforeEach(() => {
+        vi.stubEnv('GEMINI_API_KEY', undefined);
+        vi.stubEnv('GOOGLE_CLOUD_PROJECT', undefined);
+        vi.stubEnv('GOOGLE_CLOUD_LOCATION', undefined);
+        vi.stubEnv('GOOGLE_API_KEY', undefined);
+        vi.stubEnv('CUSTOM_LLM_API_KEY', undefined);
+        vi.stubEnv('CUSTOM_LLM_ENDPOINT', undefined);
+        vi.stubEnv('CUSTOM_LLM_MODEL_NAME', undefined);
+    });
+    afterEach(() => {
+        vi.unstubAllEnvs();
+    });
+    it.each([
+        {
+            description: 'should return null for LOGIN_WITH_GOOGLE',
+            authType: AuthType.LOGIN_WITH_GOOGLE,
+            envs: {},
+            expected: null,
+        },
+        {
+            description: 'should return null for COMPUTE_ADC',
+            authType: AuthType.COMPUTE_ADC,
+            envs: {},
+            expected: null,
+        },
+        {
+            description: 'should return null for USE_GEMINI if GEMINI_API_KEY is set',
+            authType: AuthType.USE_GEMINI,
+            envs: { GEMINI_API_KEY: 'test-key' },
+            expected: null,
+        },
+        {
+            description: 'should return an error message for USE_GEMINI if GEMINI_API_KEY is not set',
+            authType: AuthType.USE_GEMINI,
+            envs: {},
+            expected: 'When using Gemini API, you must specify the GEMINI_API_KEY environment variable.\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return null for USE_VERTEX_AI if GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION are set',
+            authType: AuthType.USE_VERTEX_AI,
+            envs: {
+                GOOGLE_CLOUD_PROJECT: 'test-project',
+                GOOGLE_CLOUD_LOCATION: 'test-location',
+            },
+            expected: null,
+        },
+        {
+            description: 'should return null for USE_VERTEX_AI if GOOGLE_API_KEY is set',
+            authType: AuthType.USE_VERTEX_AI,
+            envs: { GOOGLE_API_KEY: 'test-api-key' },
+            expected: null,
+        },
+        {
+            description: 'should return an error message for USE_VERTEX_AI if no required environment variables are set',
+            authType: AuthType.USE_VERTEX_AI,
+            envs: {},
+            expected: 'When using Vertex AI, you must specify either:\n' +
+                '• GOOGLE_CLOUD_PROJECT and GOOGLE_CLOUD_LOCATION environment variables.\n' +
+                '• GOOGLE_API_KEY environment variable (if using express mode).\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return null for CUSTOM_LLM_API if all required environment variables are set',
+            authType: AuthType.CUSTOM_LLM_API,
+            envs: {
+                CUSTOM_LLM_API_KEY: 'test-api-key',
+                CUSTOM_LLM_ENDPOINT: 'https://api.example.com/v1',
+                CUSTOM_LLM_MODEL_NAME: 'gpt-4',
+            },
+            expected: null,
+        },
+        {
+            description: 'should return an error message for CUSTOM_LLM_API if API key is missing',
+            authType: AuthType.CUSTOM_LLM_API,
+            envs: {
+                CUSTOM_LLM_ENDPOINT: 'https://api.example.com/v1',
+                CUSTOM_LLM_MODEL_NAME: 'gpt-4',
+            },
+            expected: 'When using Custom LLM (OpenAI Protocol), you must specify:\n' +
+                '• CUSTOM_LLM_API_KEY environment variable.\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return an error message for CUSTOM_LLM_API if endpoint is missing',
+            authType: AuthType.CUSTOM_LLM_API,
+            envs: {
+                CUSTOM_LLM_API_KEY: 'test-api-key',
+                CUSTOM_LLM_MODEL_NAME: 'gpt-4',
+            },
+            expected: 'When using Custom LLM (OpenAI Protocol), you must specify:\n' +
+                '• CUSTOM_LLM_ENDPOINT environment variable.\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return an error message for CUSTOM_LLM_API if model name is missing',
+            authType: AuthType.CUSTOM_LLM_API,
+            envs: {
+                CUSTOM_LLM_API_KEY: 'test-api-key',
+                CUSTOM_LLM_ENDPOINT: 'https://api.example.com/v1',
+            },
+            expected: 'When using Custom LLM (OpenAI Protocol), you must specify:\n' +
+                '• CUSTOM_LLM_MODEL_NAME environment variable.\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return an error message for CUSTOM_LLM_API if all required environment variables are missing',
+            authType: AuthType.CUSTOM_LLM_API,
+            envs: {},
+            expected: 'When using Custom LLM (OpenAI Protocol), you must specify:\n' +
+                '• CUSTOM_LLM_API_KEY\n' +
+                '• CUSTOM_LLM_ENDPOINT\n' +
+                '• CUSTOM_LLM_MODEL_NAME environment variables.\n' +
+                'Update your environment and try again (no reload needed if using .env)!',
+        },
+        {
+            description: 'should return an error message for an invalid auth method',
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            authType: 'invalid-method',
+            envs: {},
+            expected: 'Invalid auth method selected.',
+        },
+    ])('$description', ({ authType, envs, expected }) => {
+        for (const [key, value] of Object.entries(envs)) {
+            vi.stubEnv(key, value);
+        }
+        expect(validateAuthMethod(authType)).toBe(expected);
+    });
+});
+//# sourceMappingURL=auth.test.js.map

package/dist/src/config/auth.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auth.test.js","sourceRoot":"","sources":["../../../src/config/auth.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,yBAAyB,CAAC;AACnD,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,WAAW,CAAC;AAE/C,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9B,eAAe,EAAE,EAAE,CAAC,EAAE,EAAE;IACxB,YAAY,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;QACpC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC;KACpC,CAAC;CACH,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACxC,EAAE,CAAC,OAAO,CAAC,sBAAsB,EAAE,SAAS,CAAC,CAAC;QAC9C,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,SAAS,CAAC,CAAC;QAC/C,EAAE,CAAC,OAAO,CAAC,gBAAgB,EAAE,SAAS,CAAC,CAAC;QACxC,EAAE,CAAC,OAAO,CAAC,oBAAoB,EAAE,SAAS,CAAC,CAAC;QAC5C,EAAE,CAAC,OAAO,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;QAC7C,EAAE,CAAC,OAAO,CAAC,uBAAuB,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,IAAI,CAAC;QACN;YACE,WAAW,EAAE,0CAA0C;YACvD,QAAQ,EAAE,QAAQ,CAAC,iBAAiB;YACpC,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EAAE,oCAAoC;YACjD,QAAQ,EAAE,QAAQ,CAAC,WAAW;YAC9B,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EAAE,4DAA4D;YACzE,QAAQ,EAAE,QAAQ,CAAC,UAAU;YAC7B,IAAI,EAAE,EAAE,cAAc,EAAE,UAAU,EAAE;YACpC,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EACT,4EAA4E;YAC9E,QAAQ,EAAE,QAAQ,CAAC,UAAU;YAC7B,IAAI,EAAE,EAAE;YACR,QAAQ,EACN,oFAAoF;gBACpF,yEAAyE;SAC5E;QACD;YACE,WAAW,EACT,gGAAgG;YAClG,QAAQ,EAAE,QAAQ,CAAC,aAAa;YAChC,IAAI,EAAE;gBACJ,oBAAoB,EAAE,cAAc;gBACpC,qBAAqB,EAAE,eAAe;aACvC;YACD,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EACT,+DAA+D;YACjE,QAAQ,EAAE,QAAQ,CAAC,aAAa;YAChC,IAAI,EAAE,EAAE,cAAc,EAAE,cAAc,EAAE;YACxC,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EACT,+FAA+F;YACjG,QAAQ,EAAE,QAAQ,CAAC,aAAa;YAChC,IAAI,EAAE,EAAE;YACR,QAAQ,EACN,kDAAkD;gBAClD,2EAA2E;gBAC3E,kEAAkE;gBAClE,yEAAyE;SAC5E;QACD;YACE,WAAW,EACT,qFAAqF;YACvF,QAAQ,EAAE,QAAQ,CAAC,cAAc;YACjC,IAAI,EAAE;gBACJ,kBAAkB,EAAE,cAAc;gBAClC,mBAAmB,EAAE,4BAA4B;gBACjD,qBAAqB,EAAE,OAAO;aAC/B;YACD,QAAQ,EAAE,IAAI;SACf;QACD;YACE,WAAW,EACT,yEAAyE;YAC3E,QAAQ,EAAE,QAAQ,CAAC,cAAc;YACjC,IAAI,EAAE;gBACJ,mBAAmB,EAAE,4BAA4B;gBACjD,qBAAqB,EAAE,OAAO;aAC/B;YACD,QAAQ,EACN,8DAA8D;gBAC9D,8CAA8C;gBAC9C,yEAAyE;SAC5E;QACD;YACE,WAAW,EACT,0EAA0E;YAC5E,QAAQ,EAAE,QAAQ,CAAC,cAAc;YACjC,IAAI,EAAE;gBACJ,kBAAkB,EAAE,cAAc;gBAClC,qBAAqB,EAAE,OAAO;aAC/B;YACD,QAAQ,EACN,8DAA8D;gBAC9D,+CAA+C;gBAC/C,yEAAyE;SAC5E;QACD;YACE,WAAW,EACT,4EAA4E;YAC9E,QAAQ,EAAE,QAAQ,CAAC,cAAc;YACjC,IAAI,EAAE;gBACJ,kBAAkB,EAAE,cAAc;gBAClC,mBAAmB,EAAE,4BAA4B;aAClD;YACD,QAAQ,EACN,8DAA8D;gBAC9D,iDAAiD;gBACjD,yEAAyE;SAC5E;QACD;YACE,WAAW,EACT,qGAAqG;YACvG,QAAQ,EAAE,QAAQ,CAAC,cAAc;YACjC,IAAI,EAAE,EAAE;YACR,QAAQ,EACN,8DAA8D;gBAC9D,wBAAwB;gBACxB,yBAAyB;gBACzB,kDAAkD;gBAClD,yEAAyE;SAC5E;QACD;YACE,WAAW,EAAE,2DAA2D;YACxE,8DAA8D;YAC9D,QAAQ,EAAE,gBAAuB;YACjC,IAAI,EAAE,EAAE;YACR,QAAQ,EAAE,+BAA+B;SAC1C;KACF,CAAC,CAAC,cAAc,EAAE,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;QAClD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,KAAe,CAAC,CAAC;QACnC,CAAC;QACD,MAAM,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}