mcp: auto-start server on install

Author: connor4312

src/vs/workbench/contrib/mcp/browser/mcpServerActions.ts

@@ -26,6 +26,7 @@ import { errorIcon, infoIcon, manageExtensionIcon, trustIcon, warningIcon } from
 import { McpCommandIds } from '../common/mcpCommandIds.js';
 import { IMcpRegistry } from '../common/mcpRegistryTypes.js';
 import { IMcpSamplingService, IMcpServer, IMcpServerContainer, IMcpService, IMcpWorkbenchService, IWorkbenchMcpServer, McpCapability, McpConnectionState, McpServerEditorTab, McpServerInstallState } from '../common/mcpTypes.js';
+import { startServerByFilter } from '../common/mcpTypesUtils.js';
 
 export abstract class McpServerAction extends Action implements IMcpServerContainer {
 
@@ -108,6 +109,7 @@ export class InstallAction extends McpServerAction {
 		private readonly editor: boolean,
 		@IMcpWorkbenchService private readonly mcpWorkbenchService: IMcpWorkbenchService,
 		@ITelemetryService private readonly telemetryService: ITelemetryService,
+		@IMcpService private readonly mcpService: IMcpService,
 	) {
 		super('extensions.install', localize('install', "Install"), InstallAction.CLASS, false);
 		this.update();
@@ -149,7 +151,11 @@ export class InstallAction extends McpServerAction {
 		};
 		this.telemetryService.publicLog2<McpServerInstall, McpServerInstallClassification>('mcp:action:install', { name: this.mcpServer.gallery?.name });
 
-		await this.mcpWorkbenchService.install(this.mcpServer);
+		const installed = await this.mcpWorkbenchService.install(this.mcpServer);
+
+		await startServerByFilter(this.mcpService, s => {
+			return s.definition.label === installed.name;
+		});
 	}
 }
 

src/vs/workbench/contrib/mcp/common/mcpTypesUtils.ts

@@ -3,10 +3,42 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
+import { disposableTimeout } from '../../../../base/common/async.js';
 import { CancellationToken } from '../../../../base/common/cancellation.js';
+import { CancellationError } from '../../../../base/common/errors.js';
 import { DisposableStore } from '../../../../base/common/lifecycle.js';
 import { autorun } from '../../../../base/common/observable.js';
-import { IMcpServer, IMcpServerStartOpts, McpConnectionState, McpServerCacheState } from './mcpTypes.js';
+import { IMcpServer, IMcpServerStartOpts, IMcpService, McpConnectionState, McpServerCacheState } from './mcpTypes.js';
+
+/**
+ * Waits up to `timeout` for a server passing the filter to be discovered,
+ * and then starts it.
+ */
+export function startServerByFilter(mcpService: IMcpService, filter: (s: IMcpServer) => boolean, timeout = 5000) {
+	return new Promise<void>((resolve, reject) => {
+		const store = new DisposableStore();
+		store.add(autorun(reader => {
+			const servers = mcpService.servers.read(reader);
+			const server = servers.find(filter);
+
+			if (server) {
+				server.start({ promptType: 'all-untrusted' }).then(state => {
+					if (state.state === McpConnectionState.Kind.Error) {
+						server.showOutput();
+					}
+				});
+
+				resolve();
+				store.dispose();
+			}
+		}));
+
+		store.add(disposableTimeout(() => {
+			store.dispose();
+			reject(new CancellationError());
+		}, timeout));
+	});
+}
 
 /**
  * Starts a server (if needed) and waits for its tools to be live. Returns