[WIP] Deep Research

Author: cte

package.json

@@ -75,15 +75,20 @@
 				"icon": "$(add)"
 			},
 			{
-				"command": "roo-cline.mcpButtonClicked",
-				"title": "MCP Servers",
-				"icon": "$(server)"
+				"command": "roo-cline.researchButtonClicked",
+				"title": "Research",
+				"icon": "$(telescope)"
 			},
 			{
 				"command": "roo-cline.promptsButtonClicked",
 				"title": "Prompts",
 				"icon": "$(notebook)"
 			},
+			{
+				"command": "roo-cline.mcpButtonClicked",
+				"title": "MCP Servers",
+				"icon": "$(server)"
+			},
 			{
 				"command": "roo-cline.historyButtonClicked",
 				"title": "History",
@@ -202,29 +207,34 @@
 					"when": "view == roo-cline.SidebarProvider"
 				},
 				{
-					"command": "roo-cline.promptsButtonClicked",
+					"command": "roo-cline.researchButtonClicked",
 					"group": "navigation@2",
 					"when": "view == roo-cline.SidebarProvider"
 				},
 				{
-					"command": "roo-cline.mcpButtonClicked",
+					"command": "roo-cline.promptsButtonClicked",
 					"group": "navigation@3",
 					"when": "view == roo-cline.SidebarProvider"
 				},
 				{
-					"command": "roo-cline.historyButtonClicked",
+					"command": "roo-cline.mcpButtonClicked",
 					"group": "navigation@4",
 					"when": "view == roo-cline.SidebarProvider"
 				},
 				{
-					"command": "roo-cline.popoutButtonClicked",
+					"command": "roo-cline.historyButtonClicked",
 					"group": "navigation@5",
 					"when": "view == roo-cline.SidebarProvider"
 				},
 				{
-					"command": "roo-cline.settingsButtonClicked",
+					"command": "roo-cline.popoutButtonClicked",
 					"group": "navigation@6",
 					"when": "view == roo-cline.SidebarProvider"
+				},
+				{
+					"command": "roo-cline.settingsButtonClicked",
+					"group": "navigation@7",
+					"when": "view == roo-cline.SidebarProvider"
 				}
 			]
 		},

src/activate/registerCommands.ts

@@ -10,36 +10,37 @@ export type RegisterCommandOptions = {
 }
 
 export const registerCommands = (options: RegisterCommandOptions) => {
-	const { context, outputChannel } = options
+	const { context } = options
 
 	for (const [command, callback] of Object.entries(getCommandsMap(options))) {
 		context.subscriptions.push(vscode.commands.registerCommand(command, callback))
 	}
 }
 
-const getCommandsMap = ({ context, outputChannel, provider }: RegisterCommandOptions) => {
-	return {
-		"roo-cline.plusButtonClicked": async () => {
-			await provider.clearTask()
-			await provider.postStateToWebview()
-			await provider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })
-		},
-		"roo-cline.mcpButtonClicked": () => {
-			provider.postMessageToWebview({ type: "action", action: "mcpButtonClicked" })
-		},
-		"roo-cline.promptsButtonClicked": () => {
-			provider.postMessageToWebview({ type: "action", action: "promptsButtonClicked" })
-		},
-		"roo-cline.popoutButtonClicked": () => openClineInNewTab({ context, outputChannel }),
-		"roo-cline.openInNewTab": () => openClineInNewTab({ context, outputChannel }),
-		"roo-cline.settingsButtonClicked": () => {
-			provider.postMessageToWebview({ type: "action", action: "settingsButtonClicked" })
-		},
-		"roo-cline.historyButtonClicked": () => {
-			provider.postMessageToWebview({ type: "action", action: "historyButtonClicked" })
-		},
-	}
-}
+const getCommandsMap = ({ context, outputChannel, provider }: RegisterCommandOptions) => ({
+	"roo-cline.plusButtonClicked": async () => {
+		await provider.clearTask()
+		await provider.postStateToWebview()
+		await provider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })
+	},
+	"roo-cline.researchButtonClicked": () => {
+		provider.postMessageToWebview({ type: "action", action: "researchButtonClicked" })
+	},
+	"roo-cline.promptsButtonClicked": () => {
+		provider.postMessageToWebview({ type: "action", action: "promptsButtonClicked" })
+	},
+	"roo-cline.mcpButtonClicked": () => {
+		provider.postMessageToWebview({ type: "action", action: "mcpButtonClicked" })
+	},
+	"roo-cline.historyButtonClicked": () => {
+		provider.postMessageToWebview({ type: "action", action: "historyButtonClicked" })
+	},
+	"roo-cline.popoutButtonClicked": () => openClineInNewTab({ context, outputChannel }),
+	"roo-cline.settingsButtonClicked": () => {
+		provider.postMessageToWebview({ type: "action", action: "settingsButtonClicked" })
+	},
+	"roo-cline.openInNewTab": () => openClineInNewTab({ context, outputChannel }),
+})
 
 const openClineInNewTab = async ({ context, outputChannel }: Omit<RegisterCommandOptions, "provider">) => {
 	outputChannel.appendLine("Opening Roo Code in new tab")

src/core/webview/ClineProvider.ts

@@ -20,11 +20,17 @@ import { ApiConfiguration, ApiProvider, ModelInfo } from "../../shared/api"
 import { findLast } from "../../shared/array"
 import { ApiConfigMeta, ExtensionMessage } from "../../shared/ExtensionMessage"
 import { HistoryItem } from "../../shared/HistoryItem"
-import { checkoutDiffPayloadSchema, checkoutRestorePayloadSchema, WebviewMessage } from "../../shared/WebviewMessage"
+import {
+	checkoutDiffPayloadSchema,
+	checkoutRestorePayloadSchema,
+	researchAppendPayloadSchema,
+	WebviewMessage,
+} from "../../shared/WebviewMessage"
 import { Mode, CustomModePrompts, PromptComponent, defaultModeSlug } from "../../shared/modes"
 import { SYSTEM_PROMPT } from "../prompts/system"
 import { fileExistsAtPath } from "../../utils/fs"
 import { Cline } from "../Cline"
+import { DeepResearchService } from "../../services/deep-research/DeepResearchService"
 import { openMention } from "../mentions"
 import { getNonce } from "./getNonce"
 import { getUri } from "./getUri"
@@ -40,11 +46,10 @@ import { CustomSupportPrompts, supportPrompt } from "../../shared/support-prompt
 import { ACTION_NAMES } from "../CodeActionProvider"
 import { McpServerManager } from "../../services/mcp/McpServerManager"
 
-/*
-https://github.com/microsoft/vscode-webview-ui-toolkit-samples/blob/main/default/weather-webview/src/providers/WeatherViewProvider.ts
-
-https://github.com/KumarVariable/vscode-extension-sidebar-html/blob/master/src/customSidebarViewProvider.ts
-*/
+/**
+ * https://github.com/microsoft/vscode-webview-ui-toolkit-samples/blob/main/default/weather-webview/src/providers/WeatherViewProvider.ts
+ * https://github.com/KumarVariable/vscode-extension-sidebar-html/blob/master/src/customSidebarViewProvider.ts
+ */
 
 type SecretKey =
 	| "apiKey"
@@ -60,6 +65,7 @@ type SecretKey =
 	| "mistralApiKey"
 	| "unboundApiKey"
 	| "requestyApiKey"
+
 type GlobalStateKey =
 	| "apiProvider"
 	| "apiModelId"
@@ -146,6 +152,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 	private view?: vscode.WebviewView | vscode.WebviewPanel
 	private isViewLaunched = false
 	private cline?: Cline
+	private deepResearchService?: DeepResearchService
 	private workspaceTracker?: WorkspaceTracker
 	protected mcpHub?: McpHub // Change from private to protected
 	private latestAnnouncementId = "jan-21-2025-custom-modes" // update to some unique identifier when we add a new announcement
@@ -158,6 +165,7 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 	) {
 		this.outputChannel.appendLine("ClineProvider instantiated")
 		ClineProvider.activeInstances.add(this)
+
 		this.workspaceTracker = new WorkspaceTracker(this)
 		this.configManager = new ConfigManager(this.context)
 		this.customModesManager = new CustomModesManager(this.context, async () => {
@@ -1515,10 +1523,43 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 							}
 
 							await this.customModesManager.deleteCustomMode(message.slug)
-							// Switch back to default mode after deletion
+							// Switch back to default mode after deletion.
 							await this.updateGlobalState("mode", defaultModeSlug)
 							await this.postStateToWebview()
 						}
+						break
+
+					case "research.start": {
+						const result = researchAppendPayloadSchema.safeParse(message.payload)
+						console.log("[ClineProvider] research.start", result)
+
+						if (result.success) {
+							if (this.deepResearchService) {
+								await this.deepResearchService.abort()
+							}
+
+							this.deepResearchService = new DeepResearchService(this, "o3-mini", 4, 2, 2)
+							this.deepResearchService.start(result.data.message.content)
+						}
+
+						break
+					}
+					case "research.append": {
+						const result = researchAppendPayloadSchema.safeParse(message.payload)
+						console.log("[ClineProvider] research.append", result)
+
+						if (result.success && this.deepResearchService) {
+							this.deepResearchService.append(result.data.message.content)
+						}
+
+						break
+					}
+					case "research.reload":
+						console.log("[ClineProvider] research.reload", message)
+						break
+					case "research.stop":
+						console.log("[ClineProvider] research.stop", message)
+						break
 				}
 			},
 			null,