@office-ai/aioncli-core 0.26.0 → 0.30.1

package/dist/src/code_assist/admin/admin_controls.test.js

@@ -0,0 +1,589 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { isDeepStrictEqual } from 'node:util';
+import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
+import { fetchAdminControls, fetchAdminControlsOnce, sanitizeAdminSettings, stopAdminControlsPolling, getAdminErrorMessage, getAdminBlockedMcpServersMessage, } from './admin_controls.js';
+import { getCodeAssistServer } from '../codeAssist.js';
+vi.mock('../codeAssist.js', () => ({
+    getCodeAssistServer: vi.fn(),
+}));
+describe('Admin Controls', () => {
+    let mockServer;
+    let mockOnSettingsChanged;
+    beforeEach(() => {
+        vi.resetAllMocks();
+        vi.useFakeTimers();
+        mockServer = {
+            projectId: 'test-project',
+            fetchAdminControls: vi.fn(),
+        };
+        mockOnSettingsChanged = vi.fn();
+    });
+    afterEach(() => {
+        stopAdminControlsPolling();
+        vi.useRealTimers();
+    });
+    describe('sanitizeAdminSettings', () => {
+        it('should strip unknown fields and pass through mcpConfigJson when valid', () => {
+            const mcpConfig = {
+                mcpServers: {
+                    'server-1': {
+                        url: 'http://example.com',
+                        type: 'sse',
+                        trust: true,
+                        includeTools: ['tool1'],
+                    },
+                },
+            };
+            const input = {
+                strictModeDisabled: false,
+                extraField: 'should be removed',
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfigJson: JSON.stringify(mcpConfig),
+                    unknownMcpField: 'remove me',
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result).toEqual({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfig,
+                },
+            });
+        });
+        it('should ignore mcpConfigJson if it is invalid JSON', () => {
+            const input = {
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfigJson: '{ invalid json }',
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.mcpSetting).toEqual({
+                mcpEnabled: true,
+                mcpConfig: {},
+            });
+        });
+        it('should ignore mcpConfigJson if it does not match schema', () => {
+            const invalidConfig = {
+                mcpServers: {
+                    'server-1': {
+                        url: 123, // should be string
+                        type: 'invalid-type', // should be sse or http
+                    },
+                },
+            };
+            const input = {
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfigJson: JSON.stringify(invalidConfig),
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.mcpSetting).toEqual({
+                mcpEnabled: true,
+                mcpConfig: {},
+            });
+        });
+        it('should apply default values when fields are missing', () => {
+            const input = {};
+            const result = sanitizeAdminSettings(input);
+            expect(result).toEqual({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+        });
+        it('should default mcpEnabled to false if mcpSetting is present but mcpEnabled is undefined', () => {
+            const input = { mcpSetting: {} };
+            const result = sanitizeAdminSettings(input);
+            expect(result.mcpSetting?.mcpEnabled).toBe(false);
+            expect(result.mcpSetting?.mcpConfig).toEqual({});
+        });
+        it('should default extensionsEnabled to false if extensionsSetting is present but extensionsEnabled is undefined', () => {
+            const input = {
+                cliFeatureSetting: {
+                    extensionsSetting: {},
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.cliFeatureSetting?.extensionsSetting?.extensionsEnabled).toBe(false);
+        });
+        it('should default unmanagedCapabilitiesEnabled to false if cliFeatureSetting is present but unmanagedCapabilitiesEnabled is undefined', () => {
+            const input = {
+                cliFeatureSetting: {},
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.cliFeatureSetting?.unmanagedCapabilitiesEnabled).toBe(false);
+        });
+        it('should reflect explicit values', () => {
+            const input = {
+                strictModeDisabled: true,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: true },
+                    unmanagedCapabilitiesEnabled: true,
+                },
+                mcpSetting: {
+                    mcpEnabled: true,
+                },
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result).toEqual({
+                strictModeDisabled: true,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: true },
+                    unmanagedCapabilitiesEnabled: true,
+                },
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfig: {},
+                },
+            });
+        });
+        it('should prioritize strictModeDisabled over secureModeEnabled', () => {
+            const input = {
+                strictModeDisabled: true,
+                secureModeEnabled: true, // Should be ignored because strictModeDisabled takes precedence for backwards compatibility if both exist (though usually they shouldn't)
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.strictModeDisabled).toBe(true);
+        });
+        it('should use secureModeEnabled if strictModeDisabled is undefined', () => {
+            const input = {
+                secureModeEnabled: false,
+            };
+            const result = sanitizeAdminSettings(input);
+            expect(result.strictModeDisabled).toBe(true);
+        });
+    });
+    describe('isDeepStrictEqual verification', () => {
+        it('should consider AdminControlsSettings with different key orders as equal', () => {
+            const settings1 = {
+                strictModeDisabled: false,
+                mcpSetting: { mcpEnabled: true },
+                cliFeatureSetting: { unmanagedCapabilitiesEnabled: true },
+            };
+            const settings2 = {
+                cliFeatureSetting: { unmanagedCapabilitiesEnabled: true },
+                mcpSetting: { mcpEnabled: true },
+                strictModeDisabled: false,
+            };
+            expect(isDeepStrictEqual(settings1, settings2)).toBe(true);
+        });
+        it('should consider nested settings objects with different key orders as equal', () => {
+            const settings1 = {
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfig: {
+                        mcpServers: {
+                            server1: { url: 'url', type: 'sse' },
+                        },
+                    },
+                },
+            };
+            // Order swapped in mcpConfig and mcpServers items
+            const settings2 = {
+                mcpSetting: {
+                    mcpConfig: {
+                        mcpServers: {
+                            server1: { type: 'sse', url: 'url' },
+                        },
+                    },
+                    mcpEnabled: true,
+                },
+            };
+            expect(isDeepStrictEqual(settings1, settings2)).toBe(true);
+        });
+        it('should consider arrays in options as order-independent and equal if shuffled after sanitization', () => {
+            const mcpConfig1 = {
+                mcpServers: {
+                    server1: { includeTools: ['a', 'b'] },
+                },
+            };
+            const mcpConfig2 = {
+                mcpServers: {
+                    server1: { includeTools: ['b', 'a'] },
+                },
+            };
+            const settings1 = sanitizeAdminSettings({
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfigJson: JSON.stringify(mcpConfig1),
+                },
+            });
+            const settings2 = sanitizeAdminSettings({
+                mcpSetting: {
+                    mcpEnabled: true,
+                    mcpConfigJson: JSON.stringify(mcpConfig2),
+                },
+            });
+            expect(isDeepStrictEqual(settings1, settings2)).toBe(true);
+        });
+    });
+    describe('fetchAdminControls', () => {
+        it('should return empty object and not poll if server is missing', async () => {
+            const result = await fetchAdminControls(undefined, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should return empty object if project ID is missing', async () => {
+            mockServer = {
+                fetchAdminControls: vi.fn(),
+            };
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should use cachedSettings and start polling if provided', async () => {
+            const cachedSettings = {
+                strictModeDisabled: false,
+                mcpSetting: { mcpEnabled: false, mcpConfig: {} },
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+            };
+            const result = await fetchAdminControls(mockServer, cachedSettings, true, mockOnSettingsChanged);
+            expect(result).toEqual(cachedSettings);
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+            // Should still start polling
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: true,
+                adminControlsApplicable: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should return empty object if admin controls are disabled', async () => {
+            const result = await fetchAdminControls(mockServer, undefined, false, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should fetch from server if no cachedSettings provided', async () => {
+            const serverResponse = {
+                strictModeDisabled: false,
+                adminControlsApplicable: true,
+            };
+            mockServer.fetchAdminControls.mockResolvedValue(serverResponse);
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should throw error on fetch error and NOT start polling', async () => {
+            const error = new Error('Network error');
+            mockServer.fetchAdminControls.mockRejectedValue(error);
+            await expect(fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged)).rejects.toThrow(error);
+            // Polling should NOT have been started
+            // Advance timers just to be absolutely sure
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1); // Only initial fetch
+        });
+        it('should return empty object on adminControlsApplicable false and STOP polling', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: false,
+            });
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({});
+            // Advance time - should NOT poll because of adminControlsApplicable: false
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1); // Only the initial call
+        });
+        it('should sanitize server response', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: false,
+                unknownField: 'bad',
+                adminControlsApplicable: true,
+            });
+            const result = await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(result).toEqual({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+            expect(result['unknownField']).toBeUndefined();
+        });
+        it('should reset polling interval if called again', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: true,
+            });
+            // First call
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Advance time, but not enough to trigger the poll
+            await vi.advanceTimersByTimeAsync(2 * 60 * 1000);
+            // Second call, should reset the timer
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+            // Advance time by 3 mins. If timer wasn't reset, it would have fired (2+3=5)
+            await vi.advanceTimersByTimeAsync(3 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2); // No new poll
+            // Advance time by another 2 mins. Now it should fire.
+            await vi.advanceTimersByTimeAsync(2 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(3); // Poll fires
+        });
+    });
+    describe('fetchAdminControlsOnce', () => {
+        it('should return empty object if server is missing', async () => {
+            const result = await fetchAdminControlsOnce(undefined, true);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should return empty object if project ID is missing', async () => {
+            mockServer = {
+                fetchAdminControls: vi.fn(),
+            };
+            const result = await fetchAdminControlsOnce(mockServer, true);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should return empty object if admin controls are disabled', async () => {
+            const result = await fetchAdminControlsOnce(mockServer, false);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).not.toHaveBeenCalled();
+        });
+        it('should fetch from server and sanitize the response', async () => {
+            const serverResponse = {
+                strictModeDisabled: true,
+                unknownField: 'should be removed',
+                adminControlsApplicable: true,
+            };
+            mockServer.fetchAdminControls.mockResolvedValue(serverResponse);
+            const result = await fetchAdminControlsOnce(mockServer, true);
+            expect(result).toEqual({
+                strictModeDisabled: true,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should return empty object on adminControlsApplicable false', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: false,
+            });
+            const result = await fetchAdminControlsOnce(mockServer, true);
+            expect(result).toEqual({});
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should throw error on any other fetch error', async () => {
+            const error = new Error('Network error');
+            mockServer.fetchAdminControls.mockRejectedValue(error);
+            await expect(fetchAdminControlsOnce(mockServer, true)).rejects.toThrow(error);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+        it('should not start or stop any polling timers', async () => {
+            const setIntervalSpy = vi.spyOn(global, 'setInterval');
+            const clearIntervalSpy = vi.spyOn(global, 'clearInterval');
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: true,
+            });
+            await fetchAdminControlsOnce(mockServer, true);
+            expect(setIntervalSpy).not.toHaveBeenCalled();
+            expect(clearIntervalSpy).not.toHaveBeenCalled();
+        });
+    });
+    describe('polling', () => {
+        it('should poll and emit changes', async () => {
+            // Initial fetch
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: true,
+                adminControlsApplicable: true,
+            });
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            // Update for next poll
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: false,
+                adminControlsApplicable: true,
+            });
+            // Fast forward
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockOnSettingsChanged).toHaveBeenCalledWith({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+        });
+        it('should NOT emit if settings are deeply equal but not the same instance', async () => {
+            const settings = {
+                strictModeDisabled: false,
+                adminControlsApplicable: true,
+            };
+            mockServer.fetchAdminControls.mockResolvedValue(settings);
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            mockOnSettingsChanged.mockClear();
+            // Next poll returns a different object with the same values
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: false,
+                adminControlsApplicable: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockOnSettingsChanged).not.toHaveBeenCalled();
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+        });
+        it('should continue polling after a fetch error', async () => {
+            // Initial fetch is successful
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: true,
+                adminControlsApplicable: true,
+            });
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Next poll fails
+            mockServer.fetchAdminControls.mockRejectedValue(new Error('Poll failed'));
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+            expect(mockOnSettingsChanged).not.toHaveBeenCalled(); // No changes on error
+            // Subsequent poll succeeds with new data
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: false,
+                adminControlsApplicable: true,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(3);
+            expect(mockOnSettingsChanged).toHaveBeenCalledWith({
+                strictModeDisabled: false,
+                cliFeatureSetting: {
+                    extensionsSetting: { extensionsEnabled: false },
+                    unmanagedCapabilitiesEnabled: false,
+                },
+                mcpSetting: {
+                    mcpEnabled: false,
+                    mcpConfig: {},
+                },
+            });
+        });
+        it('should STOP polling if server returns adminControlsApplicable false', async () => {
+            // Initial fetch is successful
+            mockServer.fetchAdminControls.mockResolvedValue({
+                strictModeDisabled: true,
+                adminControlsApplicable: true,
+            });
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Next poll returns adminControlsApplicable: false
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: false,
+            });
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+            // Advance time again - should NOT poll again
+            await vi.advanceTimersByTimeAsync(5 * 60 * 1000);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe('stopAdminControlsPolling', () => {
+        it('should stop polling after it has started', async () => {
+            mockServer.fetchAdminControls.mockResolvedValue({
+                adminControlsApplicable: true,
+            });
+            // Start polling
+            await fetchAdminControls(mockServer, undefined, true, mockOnSettingsChanged);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            // Stop polling
+            stopAdminControlsPolling();
+            // Advance timer well beyond the polling interval
+            await vi.advanceTimersByTimeAsync(10 * 60 * 1000);
+            // The poll should not have fired again
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+            expect(mockServer.fetchAdminControls).toHaveBeenCalledTimes(1);
+        });
+    });
+    describe('getAdminErrorMessage', () => {
+        let mockConfig;
+        beforeEach(() => {
+            mockConfig = {};
+        });
+        it('should include feature name and project ID when present', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue({
+                projectId: 'test-project-123',
+            });
+            const message = getAdminErrorMessage('Code Completion', mockConfig);
+            expect(message).toBe('Code Completion is disabled by your administrator. To enable it, please request an update to the settings at: https://goo.gle/manage-gemini-cli?project=test-project-123');
+        });
+        it('should include feature name but OMIT project ID when missing', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue({
+                projectId: undefined,
+            });
+            const message = getAdminErrorMessage('Chat', mockConfig);
+            expect(message).toBe('Chat is disabled by your administrator. To enable it, please request an update to the settings at: https://goo.gle/manage-gemini-cli');
+        });
+        it('should include feature name but OMIT project ID when server is undefined', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue(undefined);
+            const message = getAdminErrorMessage('Chat', mockConfig);
+            expect(message).toBe('Chat is disabled by your administrator. To enable it, please request an update to the settings at: https://goo.gle/manage-gemini-cli');
+        });
+        it('should include feature name but OMIT project ID when config is undefined', () => {
+            const message = getAdminErrorMessage('Chat', undefined);
+            expect(message).toBe('Chat is disabled by your administrator. To enable it, please request an update to the settings at: https://goo.gle/manage-gemini-cli');
+        });
+    });
+    describe('getAdminBlockedMcpServersMessage', () => {
+        let mockConfig;
+        beforeEach(() => {
+            mockConfig = {};
+        });
+        it('should show count for a single blocked server', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue({
+                projectId: 'test-project-123',
+            });
+            const message = getAdminBlockedMcpServersMessage(['server-1'], mockConfig);
+            expect(message).toBe('1 MCP server is not allowlisted by your administrator. To enable it, please request an update to the settings at: https://goo.gle/manage-gemini-cli?project=test-project-123');
+        });
+        it('should show count for multiple blocked servers', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue({
+                projectId: 'test-project-123',
+            });
+            const message = getAdminBlockedMcpServersMessage(['server-1', 'server-2', 'server-3'], mockConfig);
+            expect(message).toBe('3 MCP servers are not allowlisted by your administrator. To enable them, please request an update to the settings at: https://goo.gle/manage-gemini-cli?project=test-project-123');
+        });
+        it('should format message correctly with no project ID', () => {
+            vi.mocked(getCodeAssistServer).mockReturnValue(undefined);
+            const message = getAdminBlockedMcpServersMessage(['server-1', 'server-2'], mockConfig);
+            expect(message).toBe('2 MCP servers are not allowlisted by your administrator. To enable them, please request an update to the settings at: https://goo.gle/manage-gemini-cli');
+        });
+    });
+});
+//# sourceMappingURL=admin_controls.test.js.map

package/dist/src/code_assist/admin/admin_controls.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"admin_controls.test.js","sourceRoot":"","sources":["../../../../src/code_assist/admin/admin_controls.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AAC9C,OAAO,EACL,QAAQ,EACR,EAAE,EACF,MAAM,EACN,EAAE,EACF,UAAU,EACV,SAAS,GAEV,MAAM,QAAQ,CAAC;AAChB,OAAO,EACL,kBAAkB,EAClB,sBAAsB,EACtB,qBAAqB,EACrB,wBAAwB,EACxB,oBAAoB,EACpB,gCAAgC,GACjC,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAMvD,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC7B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,UAA4B,CAAC;IACjC,IAAI,qBAA2B,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;QACnB,EAAE,CAAC,aAAa,EAAE,CAAC;QAEnB,UAAU,GAAG;YACX,SAAS,EAAE,cAAc;YACzB,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;SACG,CAAC;QAEjC,qBAAqB,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,wBAAwB,EAAE,CAAC;QAC3B,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,SAAS,GAAG;gBAChB,UAAU,EAAE;oBACV,UAAU,EAAE;wBACV,GAAG,EAAE,oBAAoB;wBACzB,IAAI,EAAE,KAAc;wBACpB,KAAK,EAAE,IAAI;wBACX,YAAY,EAAE,CAAC,OAAO,CAAC;qBACxB;iBACF;aACF,CAAC;YAEF,MAAM,KAAK,GAAG;gBACZ,kBAAkB,EAAE,KAAK;gBACzB,UAAU,EAAE,mBAAmB;gBAC/B,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;oBACxC,eAAe,EAAE,WAAW;iBAC7B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAClC,KAA8C,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,SAAS;iBACV;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,KAAK,GAA+B;gBACxC,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,aAAa,EAAE,kBAAkB;iBAClC;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC,UAAU,EAAE,IAAI;gBAChB,SAAS,EAAE,EAAE;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,aAAa,GAAG;gBACpB,UAAU,EAAE;oBACV,UAAU,EAAE;wBACV,GAAG,EAAE,GAAG,EAAE,mBAAmB;wBAC7B,IAAI,EAAE,cAAc,EAAE,wBAAwB;qBAC/C;iBACF;aACF,CAAC;YACF,MAAM,KAAK,GAA+B;gBACxC,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;iBAC7C;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;gBAChC,UAAU,EAAE,IAAI;gBAChB,SAAS,EAAE,EAAE;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,KAAK,GAAG,EAAE,CAAC;YACjB,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAmC,CAAC,CAAC;YAE1E,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yFAAyF,EAAE,GAAG,EAAE;YACjG,MAAM,KAAK,GAAG,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAmC,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8GAA8G,EAAE,GAAG,EAAE;YACtH,MAAM,KAAK,GAAG;gBACZ,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE;iBACtB;aACF,CAAC;YACF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAmC,CAAC,CAAC;YAC1E,MAAM,CACJ,MAAM,CAAC,iBAAiB,EAAE,iBAAiB,EAAE,iBAAiB,CAC/D,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oIAAoI,EAAE,GAAG,EAAE;YAC5I,MAAM,KAAK,GAAG;gBACZ,iBAAiB,EAAE,EAAE;aACtB,CAAC;YACF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAmC,CAAC,CAAC;YAC1E,MAAM,CAAC,MAAM,CAAC,iBAAiB,EAAE,4BAA4B,CAAC,CAAC,IAAI,CACjE,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,KAAK,GAA+B;gBACxC,kBAAkB,EAAE,IAAI;gBACxB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE;oBAC9C,4BAA4B,EAAE,IAAI;iBACnC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;iBACjB;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,IAAI;gBACxB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE;oBAC9C,4BAA4B,EAAE,IAAI;iBACnC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,KAAK,GAA+B;gBACxC,kBAAkB,EAAE,IAAI;gBACxB,iBAAiB,EAAE,IAAI,EAAE,0IAA0I;aACpK,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,KAAK,GAA+B;gBACxC,iBAAiB,EAAE,KAAK;aACzB,CAAC;YAEF,MAAM,MAAM,GAAG,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gCAAgC,EAAE,GAAG,EAAE;QAC9C,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,SAAS,GAA0B;gBACvC,kBAAkB,EAAE,KAAK;gBACzB,UAAU,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE;gBAChC,iBAAiB,EAAE,EAAE,4BAA4B,EAAE,IAAI,EAAE;aAC1D,CAAC;YACF,MAAM,SAAS,GAA0B;gBACvC,iBAAiB,EAAE,EAAE,4BAA4B,EAAE,IAAI,EAAE;gBACzD,UAAU,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE;gBAChC,kBAAkB,EAAE,KAAK;aAC1B,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,SAAS,GAA0B;gBACvC,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,SAAS,EAAE;wBACT,UAAU,EAAE;4BACV,OAAO,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE;yBACrC;qBACF;iBACF;aACF,CAAC;YAEF,kDAAkD;YAClD,MAAM,SAAS,GAA0B;gBACvC,UAAU,EAAE;oBACV,SAAS,EAAE;wBACT,UAAU,EAAE;4BACV,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE;yBACrC;qBACF;oBACD,UAAU,EAAE,IAAI;iBACjB;aACF,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iGAAiG,EAAE,GAAG,EAAE;YACzG,MAAM,UAAU,GAAG;gBACjB,UAAU,EAAE;oBACV,OAAO,EAAE,EAAE,YAAY,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;iBACtC;aACF,CAAC;YACF,MAAM,UAAU,GAAG;gBACjB,UAAU,EAAE;oBACV,OAAO,EAAE,EAAE,YAAY,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;iBACtC;aACF,CAAC;YAEF,MAAM,SAAS,GAAG,qBAAqB,CAAC;gBACtC,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;iBAC1C;aACF,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,qBAAqB,CAAC;gBACtC,UAAU,EAAE;oBACV,UAAU,EAAE,IAAI;oBAChB,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;iBAC1C;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,SAAS,EACT,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,UAAU,GAAG;gBACX,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;aACG,CAAC;YAEjC,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,cAAc,GAAG;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,UAAU,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,EAAE,EAAE;gBAChD,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;aACF,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,cAAc,EACd,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAE7D,6BAA6B;YAC5B,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,IAAI;gBACxB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,KAAK,EACL,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,cAAc,GAAG;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,uBAAuB,EAAE,IAAI;aAC9B,CAAC;YACD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACxC,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAEjE,MAAM,MAAM,CACV,kBAAkB,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,EAAE,qBAAqB,CAAC,CACvE,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAEzB,uCAAuC;YACvC,4CAA4C;YAC5C,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,qBAAqB;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC3F,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,KAAK;aAC/B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAE3B,2EAA2E;YAC3E,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,wBAAwB;QAC1F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC9C,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,YAAY,EAAE,KAAK;gBACnB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACrC,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YACH,MAAM,CACH,MAAkC,CAAC,cAAc,CAAC,CACpD,CAAC,aAAa,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC5D,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YAEH,aAAa;YACb,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,mDAAmD;YACnD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,sCAAsC;YACtC,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,6EAA6E;YAC7E,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;YAE9E,sDAAsD;YACtD,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,UAAU,GAAG;gBACX,kBAAkB,EAAE,EAAE,CAAC,EAAE,EAAE;aACG,CAAC;YACjC,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,cAAc,GAAG;gBACrB,kBAAkB,EAAE,IAAI;gBACxB,YAAY,EAAE,mBAAmB;gBACjC,uBAAuB,EAAE,IAAI;aAC9B,CAAC;YACD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;YAE1E,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,kBAAkB,EAAE,IAAI;gBACxB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YACH,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC1E,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,KAAK;aAC/B,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC3B,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;YACxC,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACjE,MAAM,MAAM,CAAC,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACpE,KAAK,CACN,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,cAAc,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YACvD,MAAM,gBAAgB,GAAG,EAAE,CAAC,KAAK,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;YAE1D,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,sBAAsB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;YAE/C,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC9C,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,gBAAgB;YACf,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,IAAI;gBACxB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YAEF,uBAAuB;YACtB,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YAEH,eAAe;YACf,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;gBACjD,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;YACtF,MAAM,QAAQ,GAAG;gBACf,kBAAkB,EAAE,KAAK;gBACzB,uBAAuB,EAAE,IAAI;aAC9B,CAAC;YACD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;YAEpE,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,qBAAqB,CAAC,SAAS,EAAE,CAAC;YAElC,4DAA4D;YAC3D,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAEjD,MAAM,CAAC,qBAAqB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACrD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,8BAA8B;YAC7B,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,IAAI;gBACxB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,kBAAkB;YACjB,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CACvD,IAAI,KAAK,CAAC,aAAa,CAAC,CACzB,CAAC;YACF,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC,sBAAsB;YAE5E,yCAAyC;YACxC,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,KAAK;gBACzB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC;gBACjD,kBAAkB,EAAE,KAAK;gBACzB,iBAAiB,EAAE;oBACjB,iBAAiB,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE;oBAC/C,4BAA4B,EAAE,KAAK;iBACpC;gBACD,UAAU,EAAE;oBACV,UAAU,EAAE,KAAK;oBACjB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,8BAA8B;YAC7B,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,kBAAkB,EAAE,IAAI;gBACxB,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YACH,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,mDAAmD;YAClD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,KAAK;aAC/B,CAAC,CAAC;YAEH,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,6CAA6C;YAC7C,MAAM,EAAE,CAAC,wBAAwB,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YACjD,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACvD,UAAU,CAAC,kBAA2B,CAAC,iBAAiB,CAAC;gBACxD,uBAAuB,EAAE,IAAI;aAC9B,CAAC,CAAC;YAEH,gBAAgB;YAChB,MAAM,kBAAkB,CACtB,UAAU,EACV,SAAS,EACT,IAAI,EACJ,qBAAqB,CACtB,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAE/D,eAAe;YACf,wBAAwB,EAAE,CAAC;YAE3B,iDAAiD;YACjD,MAAM,EAAE,CAAC,wBAAwB,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC;YAElD,uCAAuC;YACvC,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YAC/D,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,IAAI,UAAkB,CAAC;QAEvB,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG,EAAY,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,kBAAkB;aACV,CAAC,CAAC;YAEvB,MAAM,OAAO,GAAG,oBAAoB,CAAC,iBAAiB,EAAE,UAAU,CAAC,CAAC;YAEpE,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,0KAA0K,CAC3K,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,SAAS;aACD,CAAC,CAAC;YAEvB,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,sIAAsI,CACvI,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAEzD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,sIAAsI,CACvI,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YAExD,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,sIAAsI,CACvI,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAChD,IAAI,UAAkB,CAAC;QAEvB,UAAU,CAAC,GAAG,EAAE;YACd,UAAU,GAAG,EAAY,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,kBAAkB;aACV,CAAC,CAAC;YAEvB,MAAM,OAAO,GAAG,gCAAgC,CAC9C,CAAC,UAAU,CAAC,EACZ,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,8KAA8K,CAC/K,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,kBAAkB;aACV,CAAC,CAAC;YAEvB,MAAM,OAAO,GAAG,gCAAgC,CAC9C,CAAC,UAAU,EAAE,UAAU,EAAE,UAAU,CAAC,EACpC,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,kLAAkL,CACnL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;YAE1D,MAAM,OAAO,GAAG,gCAAgC,CAC9C,CAAC,UAAU,EAAE,UAAU,CAAC,EACxB,UAAU,CACX,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAClB,yJAAyJ,CAC1J,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/code_assist/admin/mcpUtils.d.ts

@@ -0,0 +1,22 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { MCPServerConfig } from '../../config/config.js';
+/**
+ * Applies the admin allowlist to the local MCP servers.
+ *
+ * If an admin allowlist is provided and not empty, this function filters the
+ * local servers to only those present in the allowlist. It also overrides
+ * connection details (url, type, trust) with the admin configuration and
+ * removes local execution details (command, args, env, cwd).
+ *
+ * @param localMcpServers The locally configured MCP servers.
+ * @param adminAllowlist The admin allowlist configuration.
+ * @returns The filtered and merged MCP servers.
+ */
+export declare function applyAdminAllowlist(localMcpServers: Record<string, MCPServerConfig>, adminAllowlist: Record<string, MCPServerConfig> | undefined): {
+    mcpServers: Record<string, MCPServerConfig>;
+    blockedServerNames: string[];
+};