feat: add command to switch mode

Author: aheizi

package.json

@@ -153,6 +153,11 @@
 				"command": "roo-cline.terminalExplainCommandInCurrentTask",
 				"title": "Roo Code: Explain This Command (Current Task)",
 				"category": "Terminal"
+			},
+			{
+				"command": "roo-cline.cycleModes",
+				"title": "Cycle Through Modes",
+				"category": "Roo Code"
 			}
 		],
 		"menus": {
@@ -271,7 +276,17 @@
 					"description": "Settings for VSCode Language Model API"
 				}
 			}
-		}
+		},
+		"keybindings": [
+			{
+				"command": "roo-cline.cycleModes",
+				"key": ".",
+				"win": "ctrl+.",
+				"linux": "ctrl+.",
+				"mac": "cmd+.",
+				"when": "true"
+			}
+		]
 	},
 	"scripts": {
 		"build": "npm run build:webview && npm run vsix",

src/commands/mode-switching.ts

@@ -0,0 +1,30 @@
+import * as vscode from "vscode"
+import { modes } from "../shared/modes"
+import { ClineProvider } from "../core/webview/ClineProvider"
+
+export function registerModeSwitchingCommands(context: vscode.ExtensionContext) {
+	let currentModeIndex = 0
+
+	context.subscriptions.push(
+		vscode.commands.registerCommand("roo-cline.cycleModes", async () => {
+			try {
+				// Get next mode index
+				currentModeIndex = (currentModeIndex + 1) % modes.length
+				const nextMode = modes[currentModeIndex]
+
+				// Get visible instance and send message
+				const provider = await ClineProvider.getInstance()
+				if (provider) {
+					await provider.postMessageToWebview({
+						type: "mode",
+						text: nextMode.slug,
+					})
+				}
+
+				vscode.window.showInformationMessage(`Switched to ${nextMode.name} mode`)
+			} catch (error) {
+				vscode.window.showErrorMessage(`Failed to switch mode: ${error}`)
+			}
+		}),
+	)
+}

src/extension.ts

@@ -7,6 +7,7 @@ import { CodeActionProvider } from "./core/CodeActionProvider"
 import { DIFF_VIEW_URI_SCHEME } from "./integrations/editor/DiffViewProvider"
 import { handleUri, registerCommands, registerCodeActions, registerTerminalActions } from "./activate"
 import { McpServerManager } from "./services/mcp/McpServerManager"
+import { registerModeSwitchingCommands } from "./commands/mode-switching"
 
 /**
  * Built using https://github.com/microsoft/vscode-webview-ui-toolkit
@@ -83,6 +84,9 @@ export function activate(context: vscode.ExtensionContext) {
 	registerCodeActions(context)
 	registerTerminalActions(context)
 
+	// Register mode switching commands
+	registerModeSwitchingCommands(context)
+
 	return createClineAPI(outputChannel, sidebarProvider)
 }
 

src/services/checkpoints/__tests__/ShadowCheckpointService.test.ts

@@ -13,6 +13,9 @@ jest.mock("globby", () => ({
 	globby: jest.fn().mockResolvedValue([]),
 }))
 
+// Set timeout for all tests
+jest.setTimeout(30000)
+
 describe("ShadowCheckpointService", () => {
 	const taskId = "test-task"
 

src/shared/ExtensionMessage.ts

@@ -20,6 +20,7 @@ export interface ExtensionMessage {
 	type:
 		| "action"
 		| "state"
+		| "mode"
 		| "selectedImages"
 		| "ollamaModels"
 		| "lmStudioModels"

webview-ui/src/App.tsx

@@ -27,6 +27,7 @@ const App = () => {
 	const [showAnnouncement, setShowAnnouncement] = useState(false)
 	const [tab, setTab] = useState<Tab>("chat")
 	const settingsRef = useRef<SettingsViewRef>(null)
+	const chatViewRef = useRef<any>(null)
 
 	const switchTab = useCallback((newTab: Tab) => {
 		if (settingsRef.current?.checkUnsaveChanges) {
@@ -47,8 +48,18 @@ const App = () => {
 					switchTab(newTab)
 				}
 			}
+
+			if (message.type === "mode" && message.text) {
+				console.log("Received mode switch message:", message.text)
+				if (tab !== "chat") {
+					switchTab("chat")
+				}
+				if (chatViewRef.current?.switchMode) {
+					chatViewRef.current.switchMode(message.text)
+				}
+			}
 		},
-		[switchTab],
+		[switchTab, tab],
 	)
 
 	useEvent("message", onMessage)
@@ -75,6 +86,7 @@ const App = () => {
 			{tab === "mcp" && <McpView onDone={() => switchTab("chat")} />}
 			{tab === "prompts" && <PromptsView onDone={() => switchTab("chat")} />}
 			<ChatView
+				ref={chatViewRef}
 				isHidden={tab !== "chat"}
 				showAnnouncement={showAnnouncement}
 				hideAnnouncement={() => setShowAnnouncement(false)}