MediaMetz/gemini-cli
1
0
Fork 0
mirror of https://github.com/google-gemini/gemini-cli.git synced 2026-04-12 22:31:12 -07:00
Code Issues Packages Projects Releases Wiki Activity

feat(mcp/extensions): Allow users to selectively enable/disable MCP servers included in an extension( Issue #11057 & #17402) (#17434)

Browse Source
This commit is contained in:
Jasmeet Bhatia
2026-01-26 17:02:14 -08:00
committed by GitHub
parent 3909ad67db
commit 57b57cc997
6 changed files with 145 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
packages/cli/src/ui/commands/extensionsCommand.ts
View File

@@ -29,6 +29,7 @@ import {
inferInstallMetadata,
} from '../../config/extension-manager.js';
import { SettingScope } from '../../config/settings.js';
import { McpServerEnablementManager } from '../../config/mcp/mcpServerEnablement.js';
import { theme } from '../semantic-colors.js';
import { stat } from 'node:fs/promises';
@@ -381,6 +382,38 @@ async function enableAction(context: CommandContext, args: string) {
type: MessageType.INFO,
text: `Extension "${name}" enabled for the scope "${scope}"`,
});
// Auto-enable any disabled MCP servers for this extension
const extension = extensionManager
.getExtensions()
.find((e) => e.name === name);
if (extension?.mcpServers) {
const mcpEnablementManager = McpServerEnablementManager.getInstance();
const mcpClientManager = context.services.config?.getMcpClientManager();
const enabledServers = await mcpEnablementManager.autoEnableServers(
Object.keys(extension.mcpServers ?? {}),
);
if (mcpClientManager && enabledServers.length > 0) {
const restartPromises = enabledServers.map((serverName) =>
mcpClientManager.restartServer(serverName).catch((error) => {
context.ui.addItem({
type: MessageType.WARNING,
text: `Failed to restart MCP server '${serverName}': ${getErrorMessage(error)}`,
});
}),
);
await Promise.all(restartPromises);
}
if (enabledServers.length > 0) {
context.ui.addItem({
type: MessageType.INFO,
text: `Re-enabled MCP servers: ${enabledServers.join(', ')}`,
});
}
}
}
}