@google/gemini-cli 0.16.0-preview.4 → 0.17.0-nightly.20251116.e650a4ee5

package/dist/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@google/gemini-cli",
-    "version": "0.16.0-preview.4",
+    "version": "0.17.0-nightly.20251116.e650a4ee5",
     "description": "Gemini CLI",
     "repository": {
         "type": "git",
@@ -25,7 +25,7 @@
         "dist"
     ],
     "config": {
-        "sandboxImageUri": "us-docker.pkg.dev/gemini-code-dev/gemini-cli/sandbox:0.16.0-preview.4"
+        "sandboxImageUri": "us-docker.pkg.dev/gemini-code-dev/gemini-cli/sandbox:0.17.0-nightly.20251116.e650a4ee5"
     },
     "dependencies": {
         "@google/gemini-cli-core": "file:../core",

package/dist/src/commands/extensions/disable.test.d.ts

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

package/dist/src/commands/extensions/disable.test.js

@@ -0,0 +1,160 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+import {} from 'yargs';
+import { handleDisable, disableCommand } from './disable.js';
+import { ExtensionManager } from '../../config/extension-manager.js';
+import { loadSettings, SettingScope, } from '../../config/settings.js';
+import { getErrorMessage } from '../../utils/errors.js';
+// Mock dependencies
+vi.mock('../../config/extension-manager.js');
+vi.mock('../../config/settings.js');
+vi.mock('../../utils/errors.js');
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        debugLogger: {
+            log: vi.fn(),
+            error: vi.fn(),
+        },
+    };
+});
+vi.mock('../../config/extensions/consent.js', () => ({
+    requestConsentNonInteractive: vi.fn(),
+}));
+vi.mock('../../config/extensions/extensionSettings.js', () => ({
+    promptForSetting: vi.fn(),
+}));
+describe('extensions disable command', () => {
+    const mockLoadSettings = vi.mocked(loadSettings);
+    const mockGetErrorMessage = vi.mocked(getErrorMessage);
+    const mockExtensionManager = vi.mocked(ExtensionManager);
+    let mockDebugLogger;
+    beforeEach(async () => {
+        vi.clearAllMocks();
+        // We need to re-import the mocked module to get the fresh mock
+        mockDebugLogger = (await import('@google/gemini-cli-core'))
+            .debugLogger;
+        mockLoadSettings.mockReturnValue({
+            merged: {},
+        });
+        mockExtensionManager.prototype.loadExtensions = vi
+            .fn()
+            .mockResolvedValue(undefined);
+        mockExtensionManager.prototype.disableExtension = vi
+            .fn()
+            .mockResolvedValue(undefined);
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('handleDisable', () => {
+        it.each([
+            {
+                name: 'my-extension',
+                scope: undefined,
+                expectedScope: SettingScope.User,
+                expectedLog: 'Extension "my-extension" successfully disabled for scope "undefined".',
+            },
+            {
+                name: 'my-extension',
+                scope: 'user',
+                expectedScope: SettingScope.User,
+                expectedLog: 'Extension "my-extension" successfully disabled for scope "user".',
+            },
+            {
+                name: 'my-extension',
+                scope: 'workspace',
+                expectedScope: SettingScope.Workspace,
+                expectedLog: 'Extension "my-extension" successfully disabled for scope "workspace".',
+            },
+        ])('should disable an extension in the $expectedScope scope when scope is $scope', async ({ name, scope, expectedScope, expectedLog }) => {
+            const mockCwd = vi.spyOn(process, 'cwd').mockReturnValue('/test/dir');
+            await handleDisable({ name, scope });
+            expect(mockExtensionManager).toHaveBeenCalledWith(expect.objectContaining({
+                workspaceDir: '/test/dir',
+            }));
+            expect(mockExtensionManager.prototype.loadExtensions).toHaveBeenCalled();
+            expect(mockExtensionManager.prototype.disableExtension).toHaveBeenCalledWith(name, expectedScope);
+            expect(mockDebugLogger.log).toHaveBeenCalledWith(expectedLog);
+            mockCwd.mockRestore();
+        });
+        it('should log an error message and exit with code 1 when extension disabling fails', async () => {
+            const mockProcessExit = vi
+                .spyOn(process, 'exit')
+                .mockImplementation((() => { }));
+            const error = new Error('Disable failed');
+            mockExtensionManager.prototype.disableExtension.mockRejectedValue(error);
+            mockGetErrorMessage.mockReturnValue('Disable failed message');
+            await handleDisable({ name: 'my-extension' });
+            expect(mockDebugLogger.error).toHaveBeenCalledWith('Disable failed message');
+            expect(mockProcessExit).toHaveBeenCalledWith(1);
+            mockProcessExit.mockRestore();
+        });
+    });
+    describe('disableCommand', () => {
+        const command = disableCommand;
+        it('should have correct command and describe', () => {
+            expect(command.command).toBe('disable [--scope] <name>');
+            expect(command.describe).toBe('Disables an extension.');
+        });
+        describe('builder', () => {
+            let yargsMock;
+            beforeEach(() => {
+                yargsMock = {
+                    positional: vi.fn().mockReturnThis(),
+                    option: vi.fn().mockReturnThis(),
+                    check: vi.fn().mockReturnThis(),
+                };
+            });
+            it('should configure positional and option arguments', () => {
+                command.builder(yargsMock);
+                expect(yargsMock.positional).toHaveBeenCalledWith('name', {
+                    describe: 'The name of the extension to disable.',
+                    type: 'string',
+                });
+                expect(yargsMock.option).toHaveBeenCalledWith('scope', {
+                    describe: 'The scope to disable the extension in.',
+                    type: 'string',
+                    default: SettingScope.User,
+                });
+                expect(yargsMock.check).toHaveBeenCalled();
+            });
+            it('check function should throw for invalid scope', () => {
+                command.builder(yargsMock);
+                const checkCallback = yargsMock.check.mock.calls[0][0];
+                const expectedError = `Invalid scope: invalid. Please use one of ${Object.values(SettingScope)
+                    .map((s) => s.toLowerCase())
+                    .join(', ')}.`;
+                expect(() => checkCallback({ scope: 'invalid' })).toThrow(expectedError);
+            });
+            it.each(['user', 'workspace', 'USER', 'WorkSpace'])('check function should return true for valid scope "%s"', (scope) => {
+                command.builder(yargsMock);
+                const checkCallback = yargsMock.check.mock.calls[0][0];
+                expect(checkCallback({ scope })).toBe(true);
+            });
+        });
+        it('handler should trigger extension disabling', async () => {
+            const mockCwd = vi.spyOn(process, 'cwd').mockReturnValue('/test/dir');
+            const argv = {
+                name: 'test-ext',
+                scope: 'workspace',
+                _: [],
+                $0: '',
+            };
+            await command.handler(argv);
+            expect(mockExtensionManager).toHaveBeenCalledWith(expect.objectContaining({
+                workspaceDir: '/test/dir',
+            }));
+            expect(mockExtensionManager.prototype.loadExtensions).toHaveBeenCalled();
+            expect(mockExtensionManager.prototype.disableExtension).toHaveBeenCalledWith('test-ext', SettingScope.Workspace);
+            expect(mockDebugLogger.log).toHaveBeenCalledWith('Extension "test-ext" successfully disabled for scope "workspace".');
+            mockCwd.mockRestore();
+        });
+    });
+});
+//# sourceMappingURL=disable.test.js.map

package/dist/src/commands/extensions/disable.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"disable.test.js","sourceRoot":"","sources":["../../../../src/commands/extensions/disable.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,EAAE,EACF,QAAQ,EACR,EAAE,EACF,MAAM,EACN,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAEhB,OAAO,EAAiC,MAAM,OAAO,CAAC;AAEtD,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AAErE,OAAO,EACL,YAAY,EACZ,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,oBAAoB;AAEpB,EAAE,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;AAE7C,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AAEpC,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;AAEjC,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,MAAM,GACV,MAAM,cAAc,EAA4C,CAAC;IAEnE,OAAO;QACL,GAAG,MAAM;QAET,WAAW,EAAE;YACX,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YAEZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AAEH,EAAE,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACnD,4BAA4B,EAAE,EAAE,CAAC,EAAE,EAAE;CACtC,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,8CAA8C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC7D,gBAAgB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;IAC1C,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IAEjD,MAAM,mBAAmB,GAAG,EAAE,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;IAEvD,MAAM,oBAAoB,GAAG,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAMzD,IAAI,eAAgC,CAAC;IAErC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,+DAA+D;QAE/D,eAAe,GAAG,CAAC,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;aACxD,WAAyC,CAAC;QAE7C,gBAAgB,CAAC,eAAe,CAAC;YAC/B,MAAM,EAAE,EAAE;SACkB,CAAC,CAAC;QAEhC,oBAAoB,CAAC,SAAS,CAAC,cAAc,GAAG,EAAE;aAE/C,EAAE,EAAE;aAEJ,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAEhC,oBAAoB,CAAC,SAAS,CAAC,gBAAgB,GAAG,EAAE;aAEjD,EAAE,EAAE;aAEJ,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,IAAI,CAAC;YACN;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,SAAS;gBAChB,aAAa,EAAE,YAAY,CAAC,IAAI;gBAChC,WAAW,EACT,uEAAuE;aAC1E;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,MAAM;gBACb,aAAa,EAAE,YAAY,CAAC,IAAI;gBAChC,WAAW,EACT,kEAAkE;aACrE;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,WAAW;gBAClB,aAAa,EAAE,YAAY,CAAC,SAAS;gBACrC,WAAW,EACT,uEAAuE;aAC1E;SACF,CAAC,CACA,8EAA8E,EAC9E,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE;YACpD,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAEtE,MAAM,aAAa,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAErC,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,YAAY,EAAE,WAAW;aAC1B,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,cAAc,CAC9C,CAAC,gBAAgB,EAAE,CAAC;YAErB,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,gBAAgB,CAChD,CAAC,oBAAoB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAE5C,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAE9D,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CACF,CAAC;QAEF,EAAE,CAAC,iFAAiF,EAAE,KAAK,IAAI,EAAE;YAC/F,MAAM,eAAe,GAAG,EAAE;iBAEvB,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC;iBAEtB,kBAAkB,CAAC,CAAC,GAAG,EAAE,GAAE,CAAC,CAEnB,CAAC,CAAC;YAEd,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;YAGxC,oBAAoB,CAAC,SAAS,CAAC,gBAChC,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAE3B,mBAAmB,CAAC,eAAe,CAAC,wBAAwB,CAAC,CAAC;YAE9D,MAAM,aAAa,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;YAE9C,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,oBAAoB,CAChD,wBAAwB,CACzB,CAAC;YAEF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;YAEhD,eAAe,CAAC,WAAW,EAAE,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,MAAM,OAAO,GAAG,cAA+B,CAAC;QAEhD,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAEzD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;YASvB,IAAI,SAAoB,CAAC;YAEzB,UAAU,CAAC,GAAG,EAAE;gBACd,SAAS,GAAG;oBACV,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;oBAEpC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;oBAEhC,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;iBAChC,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBACzD,OAAO,CAAC,OAAiC,CACxC,SAA4B,CAC7B,CAAC;gBAEF,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE;oBACxD,QAAQ,EAAE,uCAAuC;oBAEjD,IAAI,EAAE,QAAQ;iBACf,CAAC,CAAC;gBAEH,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE;oBACrD,QAAQ,EAAE,wCAAwC;oBAElD,IAAI,EAAE,QAAQ;oBAEd,OAAO,EAAE,YAAY,CAAC,IAAI;iBAC3B,CAAC,CAAC;gBAEH,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC7C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;gBACtD,OAAO,CAAC,OAAiC,CACxC,SAA4B,CAC7B,CAAC;gBAEF,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEvD,MAAM,aAAa,GAAG,6CAA6C,MAAM,CAAC,MAAM,CAC9E,YAAY,CACb;qBAEE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;qBAE3B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;gBAEjB,MAAM,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,OAAO,CACvD,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CACjD,wDAAwD,EACxD,CAAC,KAAK,EAAE,EAAE;gBACP,OAAO,CAAC,OAAiC,CACxC,SAA4B,CAC7B,CAAC;gBAEF,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAEvD,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9C,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAOtE,MAAM,IAAI,GAAa;gBACrB,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,WAAW;gBAClB,CAAC,EAAE,EAAE;gBACL,EAAE,EAAE,EAAE;aACP,CAAC;YAEF,MAAO,OAAO,CAAC,OAA+C,CAAC,IAAI,CAAC,CAAC;YACrE,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,YAAY,EAAE,WAAW;aAC1B,CAAC,CACH,CAAC;YAEF,MAAM,CAAC,oBAAoB,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAEzE,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,gBAAgB,CAChD,CAAC,oBAAoB,CAAC,UAAU,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC;YAE3D,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAC9C,mEAAmE,CACpE,CAAC;YAEF,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/commands/extensions/enable.test.d.ts

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

package/dist/src/commands/extensions/enable.test.js

@@ -0,0 +1,140 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+import {} from 'yargs';
+import { handleEnable, enableCommand } from './enable.js';
+import { ExtensionManager } from '../../config/extension-manager.js';
+import { loadSettings, SettingScope, } from '../../config/settings.js';
+import { FatalConfigError } from '@google/gemini-cli-core';
+// Mock dependencies
+vi.mock('../../config/extension-manager.js');
+vi.mock('../../config/settings.js');
+vi.mock('@google/gemini-cli-core', async (importOriginal) => {
+    const actual = await importOriginal();
+    return {
+        ...actual,
+        debugLogger: {
+            log: vi.fn(),
+            error: vi.fn(),
+        },
+        getErrorMessage: vi.fn((error) => error.message),
+        FatalConfigError: class extends Error {
+            constructor(message) {
+                super(message);
+                this.name = 'FatalConfigError';
+            }
+        },
+    };
+});
+vi.mock('../../config/extensions/consent.js');
+vi.mock('../../config/extensions/extensionSettings.js');
+describe('extensions enable command', () => {
+    const mockLoadSettings = vi.mocked(loadSettings);
+    const mockExtensionManager = vi.mocked(ExtensionManager);
+    let mockDebugLogger;
+    beforeEach(async () => {
+        vi.clearAllMocks();
+        mockDebugLogger = (await import('@google/gemini-cli-core'))
+            .debugLogger;
+        mockLoadSettings.mockReturnValue({
+            merged: {},
+        });
+        mockExtensionManager.prototype.loadExtensions = vi
+            .fn()
+            .mockResolvedValue(undefined);
+        mockExtensionManager.prototype.enableExtension = vi.fn();
+    });
+    afterEach(() => {
+        vi.restoreAllMocks();
+    });
+    describe('handleEnable', () => {
+        it.each([
+            {
+                name: 'my-extension',
+                scope: undefined,
+                expectedScope: SettingScope.User,
+                expectedLog: 'Extension "my-extension" successfully enabled in all scopes.',
+            },
+            {
+                name: 'my-extension',
+                scope: 'workspace',
+                expectedScope: SettingScope.Workspace,
+                expectedLog: 'Extension "my-extension" successfully enabled for scope "workspace".',
+            },
+        ])('should enable an extension in the $expectedScope scope when scope is $scope', async ({ name, scope, expectedScope, expectedLog }) => {
+            const mockCwd = vi.spyOn(process, 'cwd').mockReturnValue('/test/dir');
+            await handleEnable({ name, scope });
+            expect(mockExtensionManager).toHaveBeenCalledWith(expect.objectContaining({
+                workspaceDir: '/test/dir',
+            }));
+            expect(mockExtensionManager.prototype.loadExtensions).toHaveBeenCalled();
+            expect(mockExtensionManager.prototype.enableExtension).toHaveBeenCalledWith(name, expectedScope);
+            expect(mockDebugLogger.log).toHaveBeenCalledWith(expectedLog);
+            mockCwd.mockRestore();
+        });
+        it('should throw FatalConfigError when extension enabling fails', async () => {
+            const mockCwd = vi.spyOn(process, 'cwd').mockReturnValue('/test/dir');
+            const error = new Error('Enable failed');
+            mockExtensionManager.prototype.enableExtension.mockImplementation(() => {
+                throw error;
+            });
+            const promise = handleEnable({ name: 'my-extension' });
+            await expect(promise).rejects.toThrow(FatalConfigError);
+            await expect(promise).rejects.toThrow('Enable failed');
+            mockCwd.mockRestore();
+        });
+    });
+    describe('enableCommand', () => {
+        const command = enableCommand;
+        it('should have correct command and describe', () => {
+            expect(command.command).toBe('enable [--scope] <name>');
+            expect(command.describe).toBe('Enables an extension.');
+        });
+        describe('builder', () => {
+            let yargsMock;
+            beforeEach(() => {
+                yargsMock = {
+                    positional: vi.fn().mockReturnThis(),
+                    option: vi.fn().mockReturnThis(),
+                    check: vi.fn().mockReturnThis(),
+                };
+            });
+            it('should configure positional and option arguments', () => {
+                command.builder(yargsMock);
+                expect(yargsMock.positional).toHaveBeenCalledWith('name', {
+                    describe: 'The name of the extension to enable.',
+                    type: 'string',
+                });
+                expect(yargsMock.option).toHaveBeenCalledWith('scope', {
+                    describe: 'The scope to enable the extension in. If not set, will be enabled in all scopes.',
+                    type: 'string',
+                });
+                expect(yargsMock.check).toHaveBeenCalled();
+            });
+            it('check function should throw for invalid scope', () => {
+                command.builder(yargsMock);
+                const checkCallback = yargsMock.check.mock.calls[0][0];
+                const expectedError = `Invalid scope: invalid. Please use one of ${Object.values(SettingScope)
+                    .map((s) => s.toLowerCase())
+                    .join(', ')}.`;
+                expect(() => checkCallback({ scope: 'invalid' })).toThrow(expectedError);
+            });
+        });
+        it('handler should call handleEnable', async () => {
+            const mockCwd = vi.spyOn(process, 'cwd').mockReturnValue('/test/dir');
+            const argv = {
+                name: 'test-ext',
+                scope: 'workspace',
+                _: [],
+                $0: '',
+            };
+            await command.handler(argv);
+            expect(mockExtensionManager.prototype.enableExtension).toHaveBeenCalledWith('test-ext', SettingScope.Workspace);
+            mockCwd.mockRestore();
+        });
+    });
+});
+//# sourceMappingURL=enable.test.js.map

package/dist/src/commands/extensions/enable.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enable.test.js","sourceRoot":"","sources":["../../../../src/commands/extensions/enable.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,EAAiC,MAAM,OAAO,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EACL,YAAY,EACZ,YAAY,GAEb,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAE3D,oBAAoB;AACpB,EAAE,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;AAC7C,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AACpC,EAAE,CAAC,IAAI,CAAC,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE,EAAE;IAC1D,MAAM,MAAM,GACV,MAAM,cAAc,EAA4C,CAAC;IACnE,OAAO;QACL,GAAG,MAAM;QACT,WAAW,EAAE;YACX,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;YACZ,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE;SACf;QACD,eAAe,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAA0B,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;QACrE,gBAAgB,EAAE,KAAM,SAAQ,KAAK;YACnC,YAAY,OAAe;gBACzB,KAAK,CAAC,OAAO,CAAC,CAAC;gBACf,IAAI,CAAC,IAAI,GAAG,kBAAkB,CAAC;YACjC,CAAC;SACF;KACF,CAAC;AACJ,CAAC,CAAC,CAAC;AACH,EAAE,CAAC,IAAI,CAAC,oCAAoC,CAAC,CAAC;AAC9C,EAAE,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC;AAExD,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,MAAM,gBAAgB,GAAG,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;IACjD,MAAM,oBAAoB,GAAG,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAKzD,IAAI,eAAgC,CAAC;IAErC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,eAAe,GAAG,CAAC,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;aACxD,WAAyC,CAAC;QAC7C,gBAAgB,CAAC,eAAe,CAAC;YAC/B,MAAM,EAAE,EAAE;SACkB,CAAC,CAAC;QAChC,oBAAoB,CAAC,SAAS,CAAC,cAAc,GAAG,EAAE;aAC/C,EAAE,EAAE;aACJ,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAChC,oBAAoB,CAAC,SAAS,CAAC,eAAe,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,IAAI,CAAC;YACN;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,SAAS;gBAChB,aAAa,EAAE,YAAY,CAAC,IAAI;gBAChC,WAAW,EACT,8DAA8D;aACjE;YACD;gBACE,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,WAAW;gBAClB,aAAa,EAAE,YAAY,CAAC,SAAS;gBACrC,WAAW,EACT,sEAAsE;aACzE;SACF,CAAC,CACA,6EAA6E,EAC7E,KAAK,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE;YACpD,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACtE,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAEpC,MAAM,CAAC,oBAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,gBAAgB,CAAC;gBACtB,YAAY,EAAE,WAAW;aAC1B,CAAC,CACH,CAAC;YACF,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,cAAc,CAC9C,CAAC,gBAAgB,EAAE,CAAC;YACrB,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,eAAe,CAC/C,CAAC,oBAAoB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAC5C,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAC9D,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CACF,CAAC;QAEF,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YACtE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YAEvC,oBAAoB,CAAC,SAAS,CAAC,eAChC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACxB,MAAM,KAAK,CAAC;YACd,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;YACvD,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;YAEvD,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,MAAM,OAAO,GAAG,aAA8B,CAAC;QAE/C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;YACxD,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;YAOvB,IAAI,SAAoB,CAAC;YACzB,UAAU,CAAC,GAAG,EAAE;gBACd,SAAS,GAAG;oBACV,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;oBACpC,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;oBAChC,KAAK,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,cAAc,EAAE;iBAChC,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBACzD,OAAO,CAAC,OAAiC,CACxC,SAA4B,CAC7B,CAAC;gBACF,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE;oBACxD,QAAQ,EAAE,sCAAsC;oBAChD,IAAI,EAAE,QAAQ;iBACf,CAAC,CAAC;gBACH,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,OAAO,EAAE;oBACrD,QAAQ,EACN,kFAAkF;oBACpF,IAAI,EAAE,QAAQ;iBACf,CAAC,CAAC;gBACH,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC7C,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;gBACtD,OAAO,CAAC,OAAiC,CACxC,SAA4B,CAC7B,CAAC;gBACF,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvD,MAAM,aAAa,GAAG,6CAA6C,MAAM,CAAC,MAAM,CAC9E,YAAY,CACb;qBACE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;qBAC3B,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;gBACjB,MAAM,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,OAAO,CACvD,aAAa,CACd,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,OAAO,GAAG,EAAE,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAMtE,MAAM,IAAI,GAAa;gBACrB,IAAI,EAAE,UAAU;gBAChB,KAAK,EAAE,WAAW;gBAClB,CAAC,EAAE,EAAE;gBACL,EAAE,EAAE,EAAE;aACP,CAAC;YACF,MAAO,OAAO,CAAC,OAA+C,CAAC,IAAI,CAAC,CAAC;YAErE,MAAM,CACJ,oBAAoB,CAAC,SAAS,CAAC,eAAe,CAC/C,CAAC,oBAAoB,CAAC,UAAU,EAAE,YAAY,CAAC,SAAS,CAAC,CAAC;YAC3D,OAAO,CAAC,WAAW,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/commands/extensions/examples/mcp-server/example.test.d.ts

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

package/dist/src/commands/extensions/examples/mcp-server/example.test.js

@@ -0,0 +1,111 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+// Mock the MCP server and transport
+const mockRegisterTool = vi.fn();
+const mockRegisterPrompt = vi.fn();
+const mockConnect = vi.fn();
+vi.mock('@modelcontextprotocol/sdk/server/mcp.js', () => ({
+    McpServer: vi.fn().mockImplementation(() => ({
+        registerTool: mockRegisterTool,
+        registerPrompt: mockRegisterPrompt,
+        connect: mockConnect,
+    })),
+}));
+vi.mock('@modelcontextprotocol/sdk/server/stdio.js', () => ({
+    StdioServerTransport: vi.fn(),
+}));
+describe('MCP Server Example', () => {
+    beforeEach(async () => {
+        // Dynamically import the server setup after mocks are in place
+        await import('./example.js');
+    });
+    afterEach(() => {
+        vi.clearAllMocks();
+        vi.resetModules();
+    });
+    it('should create an McpServer with the correct name and version', () => {
+        expect(McpServer).toHaveBeenCalledWith({
+            name: 'prompt-server',
+            version: '1.0.0',
+        });
+    });
+    it('should register the "fetch_posts" tool', () => {
+        expect(mockRegisterTool).toHaveBeenCalledWith('fetch_posts', {
+            description: 'Fetches a list of posts from a public API.',
+            inputSchema: z.object({}).shape,
+        }, expect.any(Function));
+    });
+    it('should register the "poem-writer" prompt', () => {
+        expect(mockRegisterPrompt).toHaveBeenCalledWith('poem-writer', {
+            title: 'Poem Writer',
+            description: 'Write a nice haiku',
+            argsSchema: expect.any(Object),
+        }, expect.any(Function));
+    });
+    it('should connect the server to an StdioServerTransport', () => {
+        expect(StdioServerTransport).toHaveBeenCalled();
+        expect(mockConnect).toHaveBeenCalledWith(expect.any(StdioServerTransport));
+    });
+    describe('fetch_posts tool implementation', () => {
+        it('should fetch posts and return a formatted response', async () => {
+            const mockPosts = [
+                { id: 1, title: 'Post 1' },
+                { id: 2, title: 'Post 2' },
+            ];
+            global.fetch = vi.fn().mockResolvedValue({
+                json: vi.fn().mockResolvedValue(mockPosts),
+            });
+            const toolFn = mockRegisterTool.mock.calls[0][2];
+            const result = await toolFn();
+            expect(global.fetch).toHaveBeenCalledWith('https://jsonplaceholder.typicode.com/posts');
+            expect(result).toEqual({
+                content: [
+                    {
+                        type: 'text',
+                        text: JSON.stringify({ posts: mockPosts }),
+                    },
+                ],
+            });
+        });
+    });
+    describe('poem-writer prompt implementation', () => {
+        it('should generate a prompt with a title', () => {
+            const promptFn = mockRegisterPrompt.mock.calls[0][2];
+            const result = promptFn({ title: 'My Poem' });
+            expect(result).toEqual({
+                messages: [
+                    {
+                        role: 'user',
+                        content: {
+                            type: 'text',
+                            text: 'Write a haiku called My Poem. Note that a haiku is 5 syllables followed by 7 syllables followed by 5 syllables ',
+                        },
+                    },
+                ],
+            });
+        });
+        it('should generate a prompt with a title and mood', () => {
+            const promptFn = mockRegisterPrompt.mock.calls[0][2];
+            const result = promptFn({ title: 'My Poem', mood: 'sad' });
+            expect(result).toEqual({
+                messages: [
+                    {
+                        role: 'user',
+                        content: {
+                            type: 'text',
+                            text: 'Write a haiku with the mood sad called My Poem. Note that a haiku is 5 syllables followed by 7 syllables followed by 5 syllables ',
+                        },
+                    },
+                ],
+            });
+        });
+    });
+});
+//# sourceMappingURL=example.test.js.map

package/dist/src/commands/extensions/examples/mcp-server/example.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"example.test.js","sourceRoot":"","sources":["../../../../../../src/commands/extensions/examples/mcp-server/example.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AACpE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,oCAAoC;AACpC,MAAM,gBAAgB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACjC,MAAM,kBAAkB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AACnC,MAAM,WAAW,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;AAE5B,EAAE,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,SAAS,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC3C,YAAY,EAAE,gBAAgB;QAC9B,cAAc,EAAE,kBAAkB;QAClC,OAAO,EAAE,WAAW;KACrB,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,EAAE,CAAC,IAAI,CAAC,2CAA2C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,oBAAoB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC9B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,+DAA+D;QAC/D,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,YAAY,EAAE,CAAC;IACpB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC;YACrC,IAAI,EAAE,eAAe;YACrB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,gBAAgB,CAAC,CAAC,oBAAoB,CAC3C,aAAa,EACb;YACE,WAAW,EAAE,4CAA4C;YACzD,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,KAAK;SAChC,EACD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,kBAAkB,CAAC,CAAC,oBAAoB,CAC7C,aAAa,EACb;YACE,KAAK,EAAE,aAAa;YACpB,WAAW,EAAE,oBAAoB;YACjC,UAAU,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC;SAC/B,EACD,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CACrB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,oBAAoB,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAChD,MAAM,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,SAAS,GAAG;gBAChB,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE;gBAC1B,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE;aAC3B,CAAC;YACF,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBACvC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC;aAC3C,CAAC,CAAC;YAEH,MAAM,MAAM,GAAI,gBAAyB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,MAAM,MAAM,GAAG,MAAM,MAAM,EAAE,CAAC;YAE9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,oBAAoB,CACvC,4CAA4C,CAC7C,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;qBAC3C;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,QAAQ,GAAI,kBAA2B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,iHAAiH;yBACxH;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,QAAQ,GAAI,kBAA2B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,QAAQ,EAAE;oBACR;wBACE,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE;4BACP,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,mIAAmI;yBAC1I;qBACF;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/commands/extensions/examples/mcp-server/example.test.ts

@@ -0,0 +1,143 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import {
+  describe,
+  it,
+  expect,
+  vi,
+  beforeEach,
+  afterEach,
+  type Mock,
+} from 'vitest';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { z } from 'zod';
+
+// Mock the MCP server and transport
+const mockRegisterTool = vi.fn();
+const mockRegisterPrompt = vi.fn();
+const mockConnect = vi.fn();
+
+vi.mock('@modelcontextprotocol/sdk/server/mcp.js', () => ({
+  McpServer: vi.fn().mockImplementation(() => ({
+    registerTool: mockRegisterTool,
+    registerPrompt: mockRegisterPrompt,
+    connect: mockConnect,
+  })),
+}));
+
+vi.mock('@modelcontextprotocol/sdk/server/stdio.js', () => ({
+  StdioServerTransport: vi.fn(),
+}));
+
+describe('MCP Server Example', () => {
+  beforeEach(async () => {
+    // Dynamically import the server setup after mocks are in place
+    await import('./example.js');
+  });
+
+  afterEach(() => {
+    vi.clearAllMocks();
+    vi.resetModules();
+  });
+
+  it('should create an McpServer with the correct name and version', () => {
+    expect(McpServer).toHaveBeenCalledWith({
+      name: 'prompt-server',
+      version: '1.0.0',
+    });
+  });
+
+  it('should register the "fetch_posts" tool', () => {
+    expect(mockRegisterTool).toHaveBeenCalledWith(
+      'fetch_posts',
+      {
+        description: 'Fetches a list of posts from a public API.',
+        inputSchema: z.object({}).shape,
+      },
+      expect.any(Function),
+    );
+  });
+
+  it('should register the "poem-writer" prompt', () => {
+    expect(mockRegisterPrompt).toHaveBeenCalledWith(
+      'poem-writer',
+      {
+        title: 'Poem Writer',
+        description: 'Write a nice haiku',
+        argsSchema: expect.any(Object),
+      },
+      expect.any(Function),
+    );
+  });
+
+  it('should connect the server to an StdioServerTransport', () => {
+    expect(StdioServerTransport).toHaveBeenCalled();
+    expect(mockConnect).toHaveBeenCalledWith(expect.any(StdioServerTransport));
+  });
+
+  describe('fetch_posts tool implementation', () => {
+    it('should fetch posts and return a formatted response', async () => {
+      const mockPosts = [
+        { id: 1, title: 'Post 1' },
+        { id: 2, title: 'Post 2' },
+      ];
+      global.fetch = vi.fn().mockResolvedValue({
+        json: vi.fn().mockResolvedValue(mockPosts),
+      });
+
+      const toolFn = (mockRegisterTool as Mock).mock.calls[0][2];
+      const result = await toolFn();
+
+      expect(global.fetch).toHaveBeenCalledWith(
+        'https://jsonplaceholder.typicode.com/posts',
+      );
+      expect(result).toEqual({
+        content: [
+          {
+            type: 'text',
+            text: JSON.stringify({ posts: mockPosts }),
+          },
+        ],
+      });
+    });
+  });
+
+  describe('poem-writer prompt implementation', () => {
+    it('should generate a prompt with a title', () => {
+      const promptFn = (mockRegisterPrompt as Mock).mock.calls[0][2];
+      const result = promptFn({ title: 'My Poem' });
+      expect(result).toEqual({
+        messages: [
+          {
+            role: 'user',
+            content: {
+              type: 'text',
+              text: 'Write a haiku called My Poem. Note that a haiku is 5 syllables followed by 7 syllables followed by 5 syllables ',
+            },
+          },
+        ],
+      });
+    });
+
+    it('should generate a prompt with a title and mood', () => {
+      const promptFn = (mockRegisterPrompt as Mock).mock.calls[0][2];
+      const result = promptFn({ title: 'My Poem', mood: 'sad' });
+      expect(result).toEqual({
+        messages: [
+          {
+            role: 'user',
+            content: {
+              type: 'text',
+              text: 'Write a haiku with the mood sad called My Poem. Note that a haiku is 5 syllables followed by 7 syllables followed by 5 syllables ',
+            },
+          },
+        ],
+      });
+    });
+  });
+});

package/dist/src/commands/extensions/link.test.d.ts

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