registerCommands.ts

import * as vscode from "vscode"
import delay from "delay"

import { ClineProvider } from "../core/webview/ClineProvider"

export type RegisterCommandOptions = {
	context: vscode.ExtensionContext
	outputChannel: vscode.OutputChannel
	provider: ClineProvider
}

export const registerCommands = (options: RegisterCommandOptions) => {
	const { context, outputChannel } = 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" })
		},
		"roo-cline.helpButtonClicked": () => {
			vscode.env.openExternal(vscode.Uri.parse("https://docs.roocode.com"))
		},
		"roo-cline.createInAgent": async (args: any) => {
			const sidebarProvider = ClineProvider.getSidebarInstance()
			if (sidebarProvider) {
				// Start a new chat in the sidebar
				vscode.commands.executeCommand("pearai-roo-cline.SidebarProvider.focus")
				await sidebarProvider.clearTask()
				await sidebarProvider.handleModeSwitch("code", args.creatorMode)
				await sidebarProvider.postStateToWebview()
				await sidebarProvider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })

				await sidebarProvider.initClineWithTask(args.text, undefined, args.creatorMode)
			}
		},
		"roo-cline.executeCreatorPlan": async (args: any) => {
			const sidebarProvider = ClineProvider.getSidebarInstance()
			if (sidebarProvider) {
				// Start a new chat in the sidebar
				vscode.commands.executeCommand("pearai-roo-cline.SidebarProvider.focus")
				await sidebarProvider.clearTask()
				await sidebarProvider.handleModeSwitch("code")
				await sidebarProvider.postStateToWebview()
				await sidebarProvider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })

				// Create the template message using the args
				// Todo: add structure to it i.e. You should have these sections: Architecutre, Features, etc.
				let executePlanTemplate = `This file contains detailed plan to my task. please read it and Execute the plan accordingly.
					File: ${args.filePath || "No file specified"}`

				if (args.code) {
					executePlanTemplate += `Code: \`\`\`
						${args.code}
						\`\`\`
					`
				}

				if (args.context) {
					executePlanTemplate += `Additional context: ${args.context}`
				}

				args.text = executePlanTemplate

				await sidebarProvider.initClineWithTask(args.text, undefined, true)
			}
		},
	}
}

const openClineInNewTab = async ({ context, outputChannel }: Omit<RegisterCommandOptions, "provider">) => {
	outputChannel.appendLine("Opening Roo Code in new tab")

	// (This example uses webviewProvider activation event which is necessary to
	// deserialize cached webview, but since we use retainContextWhenHidden, we
	// don't need to use that event).
	// https://github.com/microsoft/vscode-extension-samples/blob/main/webview-sample/src/extension.ts
	const tabProvider = new ClineProvider(context, outputChannel, true)
	// const column = vscode.window.activeTextEditor ? vscode.window.activeTextEditor.viewColumn : undefined
	// const lastCol = Math.max(...vscode.window.visibleTextEditors.map((editor) => editor.viewColumn || 0))

	// Check if there are any visible text editors, otherwise open a new group
	// to the right.
	// const hasVisibleEditors = vscode.window.visibleTextEditors.length > 0

	// if (!hasVisibleEditors) {
	// 	await vscode.commands.executeCommand("workbench.action.newGroupRight")
	// }

	// const targetCol = hasVisibleEditors ? Math.max(lastCol + 1, 1) : vscode.ViewColumn.Two

	const panel = vscode.window.createWebviewPanel(ClineProvider.tabPanelId, "Roo Code", vscode.ViewColumn.One, {
		enableScripts: true,
		retainContextWhenHidden: true,
		localResourceRoots: [context.extensionUri],
	})

	// TODO: use better svg icon with light and dark variants (see
	// https://stackoverflow.com/questions/58365687/vscode-extension-iconpath).
	panel.iconPath = {
		light: vscode.Uri.joinPath(context.extensionUri, "assets", "icons", "rocket.png"),
		dark: vscode.Uri.joinPath(context.extensionUri, "assets", "icons", "rocket.png"),
	}

	await tabProvider.resolveWebviewView(panel)

	// Lock the editor group so clicking on files doesn't open them over the panel
	await delay(100)
	// await vscode.commands.executeCommand("workbench.action.lockEditorGroup")
}