Merge branch 'feat/issue-5290-add-and-run-commands' of https://github.com/RooCodeInc/Roo-Code into feat/issue-5290-add-and-run-commands

Author: hannesrudolph

src/core/webview/__tests__/webviewMessageHandler.spec.ts

@@ -783,6 +783,36 @@ export const webviewMessageHandler = async (
 				.getConfiguration(Package.name)
 				.update("allowedCommands", validCommands, vscode.ConfigurationTarget.Global)
 
+			// Post state update to webview to reflect changes in UI
+			await provider.postStateToWebview()
+
+			break
+		}
+		case "addToWhitelist": {
+			// Handle adding a command pattern to the whitelist and then running it
+			if (message.pattern) {
+				// Get current allowed commands
+				const currentAllowedCommands = getGlobalState("allowedCommands") || []
+
+				// Add the new pattern if it's not already in the list
+				if (!currentAllowedCommands.includes(message.pattern)) {
+					const updatedCommands = [...currentAllowedCommands, message.pattern]
+
+					// Update global state
+					await updateGlobalState("allowedCommands", updatedCommands)
+
+					// Also update workspace settings
+					await vscode.workspace
+						.getConfiguration(Package.name)
+						.update("allowedCommands", updatedCommands, vscode.ConfigurationTarget.Global)
+
+					// Post state update to webview
+					await provider.postStateToWebview()
+				}
+
+				// After whitelisting, approve the pending command execution
+				provider.getCurrentCline()?.handleWebviewAskResponse("yesButtonClicked")
+			}
 			break
 		}
 		case "openCustomModesSettings": {

src/core/webview/webviewMessageHandler.ts

@@ -118,7 +118,13 @@ export interface ExtensionMessage {
 		| "didBecomeVisible"
 		| "focusInput"
 		| "switchTab"
-	invoke?: "newChat" | "sendMessage" | "primaryButtonClick" | "secondaryButtonClick" | "setChatBoxMessage"
+	invoke?:
+		| "newChat"
+		| "sendMessage"
+		| "primaryButtonClick"
+		| "secondaryButtonClick"
+		| "tertiaryButtonClick"
+		| "setChatBoxMessage"
 	state?: ExtensionState
 	images?: string[]
 	filePaths?: string[]

src/shared/ExtensionMessage.ts

@@ -12,7 +12,12 @@ import { marketplaceItemSchema } from "@roo-code/types"
 
 import { Mode } from "./modes"
 
-export type ClineAskResponse = "yesButtonClicked" | "noButtonClicked" | "messageResponse" | "objectResponse"
+export type ClineAskResponse =
+	| "yesButtonClicked"
+	| "noButtonClicked"
+	| "addAndRunButtonClicked"
+	| "messageResponse"
+	| "objectResponse"
 
 export type PromptMode = Mode | "enhance"
 
@@ -193,6 +198,7 @@ export interface WebviewMessage {
 		| "checkRulesDirectoryResult"
 		| "saveCodeIndexSettingsAtomic"
 		| "requestCodeIndexSecretStatus"
+		| "addToWhitelist"
 	text?: string
 	editedMessageContent?: string
 	tab?: "settings" | "history" | "mcp" | "modes" | "chat" | "marketplace" | "account"
@@ -234,6 +240,8 @@ export interface WebviewMessage {
 	visibility?: ShareVisibility // For share visibility
 	hasContent?: boolean // For checkRulesDirectoryResult
 	checkOnly?: boolean // For deleteCustomMode check
+	commandPattern?: string // For "Add & Run" button - the extracted command pattern to whitelist
+	pattern?: string // For "addToWhitelist" message - the command pattern to add to whitelist
 	codeIndexSettings?: {
 		// Global state settings
 		codebaseIndexEnabled: boolean