undo button to undo requests, adds request in input (microsoft#251306)

change to undo, setting for exp

Author: justschen

src/vs/workbench/contrib/chat/browser/chat.contribution.ts

@@ -239,6 +239,12 @@ configurationRegistry.registerConfiguration({
 			type: 'boolean',
 			tags: ['experimental']
 		},
+		'chat.undoRequests.restoreInput': {
+			default: true,
+			markdownDescription: nls.localize('chat.undoRequests.restoreInput', "Controls whether the input of the chat should be restored when an undo request is made. The input will be filled with the text of the request that was restored."),
+			type: 'boolean',
+			tags: ['experimental']
+		},
 		[mcpEnabledSection]: {
 			type: 'boolean',
 			description: nls.localize('chat.mcp.enabled', "Enables integration with Model Context Protocol servers to provide additional tools and functionality."),

src/vs/workbench/contrib/chat/browser/chatEditing/chatEditingActions.ts

@@ -312,7 +312,7 @@ registerAction2(class RemoveAction extends Action2 {
 			title: localize2('chat.undoEdits.label', "Undo Requests"),
 			f1: false,
 			category: CHAT_CATEGORY,
-			icon: Codicon.x,
+			icon: Codicon.discard,
 			keybinding: {
 				primary: KeyCode.Delete,
 				mac: {
@@ -334,9 +334,9 @@ registerAction2(class RemoveAction extends Action2 {
 
 	async run(accessor: ServicesAccessor, ...args: any[]) {
 		let item: ChatTreeItem | undefined = args[0];
+		const chatWidgetService = accessor.get(IChatWidgetService);
+		const widget = chatWidgetService.lastFocusedWidget;
 		if (!isResponseVM(item) && !isRequestVM(item)) {
-			const chatWidgetService = accessor.get(IChatWidgetService);
-			const widget = chatWidgetService.lastFocusedWidget;
 			item = widget?.getFocus();
 		}
 
@@ -409,6 +409,11 @@ registerAction2(class RemoveAction extends Action2 {
 			const snapshotRequestId = chatRequests[itemIndex].id;
 			await session.restoreSnapshot(snapshotRequestId, undefined);
 		}
+
+		if (isRequestVM(item) && configurationService.getValue('chat.undoRequests.restoreInput')) {
+			widget?.focusInput();
+			widget?.input.setValue(item.messageText, false);
+		}
 	}
 });
 

src/vs/workbench/contrib/chat/browser/media/chat.css

@@ -1928,7 +1928,7 @@ have to be updated for changes to the rules above, or to support more deeply nes
 		height: 22px;
 	}
 
-	.request-hover .actions-container .action-label.codicon-x {
+	.request-hover .actions-container .action-label.codicon-discard {
 		margin-top: 4px;
 		padding: 3px 3px;
 	}