feat: add programmatic MCP server refresh capability

packages/types/src/api.ts

@@ -1,144 +1,35 @@
-import type { EventEmitter } from "events"
-import type { Socket } from "net"
+/**
+ * API for programmatic MCP server operations
+ */
 
-import type { RooCodeEvents } from "./events.js"
-import type { RooCodeSettings } from "./global-settings.js"
-import type { ProviderSettingsEntry, ProviderSettings } from "./provider-settings.js"
-import type { IpcMessage, IpcServerEvents } from "./ipc.js"
-
-export type RooCodeAPIEvents = RooCodeEvents
-
-export interface RooCodeAPI extends EventEmitter<RooCodeAPIEvents> {
-	/**
-	 * Starts a new task with an optional initial message and images.
-	 * @param task Optional initial task message.
-	 * @param images Optional array of image data URIs (e.g., "data:image/webp;base64,...").
-	 * @returns The ID of the new task.
-	 */
-	startNewTask({
-		configuration,
-		text,
-		images,
-		newTab,
-	}: {
-		configuration?: RooCodeSettings
-		text?: string
-		images?: string[]
-		newTab?: boolean
-	}): Promise<string>
-	/**
-	 * Resumes a task with the given ID.
-	 * @param taskId The ID of the task to resume.
-	 * @throws Error if the task is not found in the task history.
-	 */
-	resumeTask(taskId: string): Promise<void>
-	/**
-	 * Checks if a task with the given ID is in the task history.
-	 * @param taskId The ID of the task to check.
-	 * @returns True if the task is in the task history, false otherwise.
-	 */
-	isTaskInHistory(taskId: string): Promise<boolean>
-	/**
-	 * Returns the current task stack.
-	 * @returns An array of task IDs.
-	 */
-	getCurrentTaskStack(): string[]
-	/**
-	 * Clears the current task.
-	 */
-	clearCurrentTask(lastMessage?: string): Promise<void>
-	/**
-	 * Cancels the current task.
-	 */
-	cancelCurrentTask(): Promise<void>
-	/**
-	 * Sends a message to the current task.
-	 * @param message Optional message to send.
-	 * @param images Optional array of image data URIs (e.g., "data:image/webp;base64,...").
-	 */
-	sendMessage(message?: string, images?: string[]): Promise<void>
-	/**
-	 * Simulates pressing the primary button in the chat interface.
-	 */
-	pressPrimaryButton(): Promise<void>
-	/**
-	 * Simulates pressing the secondary button in the chat interface.
-	 */
-	pressSecondaryButton(): Promise<void>
-	/**
-	 * Returns true if the API is ready to use.
-	 */
-	isReady(): boolean
+/**
+ * Interface for MCP operations that can be accessed programmatically
+ */
+export interface McpApi {
 	/**
-	 * Returns the current configuration.
-	 * @returns The current configuration.
+	 * Refresh all MCP server connections
+	 * @returns Promise that resolves when refresh is complete
 	 */
-	getConfiguration(): RooCodeSettings
-	/**
-	 * Sets the configuration for the current task.
-	 * @param values An object containing key-value pairs to set.
-	 */
-	setConfiguration(values: RooCodeSettings): Promise<void>
-	/**
-	 * Returns a list of all configured profile names
-	 * @returns Array of profile names
-	 */
-	getProfiles(): string[]
-	/**
-	 * Returns the profile entry for a given name
-	 * @param name The name of the profile
-	 * @returns The profile entry, or undefined if the profile does not exist
-	 */
-	getProfileEntry(name: string): ProviderSettingsEntry | undefined
-	/**
-	 * Creates a new API configuration profile
-	 * @param name The name of the profile
-	 * @param profile The profile to create; defaults to an empty object
-	 * @param activate Whether to activate the profile after creation; defaults to true
-	 * @returns The ID of the created profile
-	 * @throws Error if the profile already exists
-	 */
-	createProfile(name: string, profile?: ProviderSettings, activate?: boolean): Promise<string>
-	/**
-	 * Updates an existing API configuration profile
-	 * @param name The name of the profile
-	 * @param profile The profile to update
-	 * @param activate Whether to activate the profile after update; defaults to true
-	 * @returns The ID of the updated profile
-	 * @throws Error if the profile does not exist
-	 */
-	updateProfile(name: string, profile: ProviderSettings, activate?: boolean): Promise<string | undefined>
-	/**
-	 * Creates a new API configuration profile or updates an existing one
-	 * @param name The name of the profile
-	 * @param profile The profile to create or update; defaults to an empty object
-	 * @param activate Whether to activate the profile after upsert; defaults to true
-	 * @returns The ID of the upserted profile
-	 */
-	upsertProfile(name: string, profile: ProviderSettings, activate?: boolean): Promise<string | undefined>
-	/**
-	 * Deletes a profile by name
-	 * @param name The name of the profile to delete
-	 * @throws Error if the profile does not exist
-	 */
-	deleteProfile(name: string): Promise<void>
-	/**
-	 * Returns the name of the currently active profile
-	 * @returns The profile name, or undefined if no profile is active
-	 */
-	getActiveProfile(): string | undefined
-	/**
-	 * Changes the active API configuration profile
-	 * @param name The name of the profile to activate
-	 * @throws Error if the profile does not exist
-	 */
-	setActiveProfile(name: string): Promise<string | undefined>
+	refreshMcpServers(): Promise<void>
+}
+
+/**
+ * Global MCP API instance that will be set by the extension
+ */
+export let mcpApi: McpApi | undefined
+
+/**
+ * Set the global MCP API instance
+ * @param api The MCP API implementation
+ */
+export function setMcpApi(api: McpApi): void {
+	mcpApi = api
 }
 
-export interface RooCodeIpcServer extends EventEmitter<IpcServerEvents> {
-	listen(): void
-	broadcast(message: IpcMessage): void
-	send(client: string | Socket, message: IpcMessage): void
-	get socketPath(): string
-	get isListening(): boolean
+/**
+ * Get the global MCP API instance
+ * @returns The MCP API instance or undefined if not set
+ */
+export function getMcpApi(): McpApi | undefined {
+	return mcpApi
 }

packages/types/src/vscode.ts

@@ -54,6 +54,7 @@ export const commandIds = [
 	"acceptInput",
 	"focusPanel",
 	"toggleAutoApprove",
+	"refreshMcpServers",
 ] as const
 
 export type CommandId = (typeof commandIds)[number]

src/activate/__tests__/registerCommands.test.ts

@@ -0,0 +1,133 @@
+import { describe, it, expect, vi, beforeEach } from "vitest"
+import * as vscode from "vscode"
+import { registerCommands } from "../registerCommands"
+import { ClineProvider } from "../../core/webview/ClineProvider"
+
+// Mock vscode module
+vi.mock("vscode", () => ({
+	window: {
+		showInformationMessage: vi.fn(),
+		showWarningMessage: vi.fn(),
+	},
+	commands: {
+		registerCommand: vi.fn(),
+	},
+}))
+
+describe("registerCommands", () => {
+	let mockContext: vscode.ExtensionContext
+	let mockOutputChannel: vscode.OutputChannel
+	let mockProvider: ClineProvider
+
+	beforeEach(() => {
+		// Reset all mocks
+		vi.clearAllMocks()
+
+		// Create mock objects
+		mockContext = {
+			subscriptions: {
+				push: vi.fn(),
+			},
+		} as any
+
+		mockOutputChannel = {
+			appendLine: vi.fn(),
+		} as any
+
+		mockProvider = {
+			getMcpHub: vi.fn(),
+		} as any
+	})
+
+	describe("refreshMcpServers command", () => {
+		it("should refresh MCP servers when hub is available", async () => {
+			// Arrange
+			const mockMcpHub = {
+				refreshAllConnections: vi.fn().mockResolvedValue(undefined),
+			}
+			mockProvider.getMcpHub = vi.fn().mockReturnValue(mockMcpHub)
+
+			// Mock getVisibleProviderOrLog to return our mock provider
+			const getVisibleProviderOrLog = vi.fn().mockReturnValue(mockProvider)
+
+			// Register commands
+			const commands = registerCommands({
+				context: mockContext,
+				outputChannel: mockOutputChannel,
+				provider: mockProvider,
+			})
+
+			// Find and execute the refreshMcpServers command
+			const registerCommandCalls = vi.mocked(vscode.commands.registerCommand).mock.calls
+			const refreshCommand = registerCommandCalls.find(([cmd]) => cmd === "roo-cline.refreshMcpServers")
+
+			expect(refreshCommand).toBeDefined()
+
+			// Execute the command callback
+			const commandCallback = refreshCommand![1] as () => Promise<void>
+			await commandCallback()
+
+			// Assert
+			expect(mockProvider.getMcpHub).toHaveBeenCalled()
+			expect(mockMcpHub.refreshAllConnections).toHaveBeenCalled()
+			expect(vscode.window.showInformationMessage).toHaveBeenCalledWith("MCP servers refreshed successfully")
+		})
+
+		it("should show warning when MCP hub is not available", async () => {
+			// Arrange
+			mockProvider.getMcpHub = vi.fn().mockReturnValue(undefined)
+
+			// Mock getVisibleProviderOrLog to return our mock provider
+			const getVisibleProviderOrLog = vi.fn().mockReturnValue(mockProvider)
+
+			// Register commands
+			const commands = registerCommands({
+				context: mockContext,
+				outputChannel: mockOutputChannel,
+				provider: mockProvider,
+			})
+
+			// Find and execute the refreshMcpServers command
+			const registerCommandCalls = vi.mocked(vscode.commands.registerCommand).mock.calls
+			const refreshCommand = registerCommandCalls.find(([cmd]) => cmd === "roo-cline.refreshMcpServers")
+
+			expect(refreshCommand).toBeDefined()
+
+			// Execute the command callback
+			const commandCallback = refreshCommand![1] as () => Promise<void>
+			await commandCallback()
+
+			// Assert
+			expect(mockProvider.getMcpHub).toHaveBeenCalled()
+			expect(vscode.window.showWarningMessage).toHaveBeenCalledWith("MCP hub is not available")
+		})
+
+		it("should not execute when no visible provider is found", async () => {
+			// Arrange
+			// Mock getVisibleProviderOrLog to return undefined
+			const getVisibleProviderOrLog = vi.fn().mockReturnValue(undefined)
+
+			// Register commands
+			const commands = registerCommands({
+				context: mockContext,
+				outputChannel: mockOutputChannel,
+				provider: mockProvider,
+			})
+
+			// Find and execute the refreshMcpServers command
+			const registerCommandCalls = vi.mocked(vscode.commands.registerCommand).mock.calls
+			const refreshCommand = registerCommandCalls.find(([cmd]) => cmd === "roo-cline.refreshMcpServers")
+
+			expect(refreshCommand).toBeDefined()
+
+			// Execute the command callback
+			const commandCallback = refreshCommand![1] as () => Promise<void>
+			await commandCallback()
+
+			// Assert - should return early without calling getMcpHub
+			expect(mockProvider.getMcpHub).not.toHaveBeenCalled()
+			expect(vscode.window.showInformationMessage).not.toHaveBeenCalled()
+			expect(vscode.window.showWarningMessage).not.toHaveBeenCalled()
+		})
+	})
+})

src/activate/registerCommands.ts

@@ -233,6 +233,21 @@ const getCommandsMap = ({ context, outputChannel, provider }: RegisterCommandOpt
 			action: "toggleAutoApprove",
 		})
 	},
+	refreshMcpServers: async () => {
+		const visibleProvider = getVisibleProviderOrLog(outputChannel)
+
+		if (!visibleProvider) {
+			return
+		}
+
+		const mcpHub = visibleProvider.getMcpHub()
+		if (mcpHub) {
+			await mcpHub.refreshAllConnections()
+			vscode.window.showInformationMessage("MCP servers refreshed successfully")
+		} else {
+			vscode.window.showWarningMessage("MCP hub is not available")
+		}
+	},
 })
 
 export const openClineInNewTab = async ({ context, outputChannel }: Omit<RegisterCommandOptions, "provider">) => {