checkpoint polish 💅🏻 (microsoft#258711)

some checkpoint cleanup

Author: justschen

src/vs/workbench/contrib/chat/browser/actions/chatClearActions.ts

@@ -211,7 +211,17 @@ export function registerNewChatActions() {
 			while (editingSession.canRedo.get()) {
 				await editingSession.redoInteraction();
 			}
-			widget.lastFocusedWidget?.viewModel?.model.setCheckpoint(undefined);
+
+			const currentWidget = widget.lastFocusedWidget;
+			const requestText = currentWidget?.viewModel?.model.checkpoint?.message.text;
+
+			// if the input has the same text that we just restored, clear it.
+			if (currentWidget?.inputEditor.getValue() === requestText) {
+				currentWidget?.input.setValue('', false);
+			}
+
+			currentWidget?.viewModel?.model.setCheckpoint(undefined);
+			currentWidget?.focusInput();
 		}
 	});
 }

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

@@ -429,7 +429,7 @@ registerAction2(class RestoreCheckpointAction extends Action2 {
 	constructor() {
 		super({
 			id: 'workbench.action.chat.restoreCheckpoint',
-			title: localize2('chat.restoreCheckpoint.label', "Restore checkpoint"),
+			title: localize2('chat.restoreCheckpoint.label', "Restore Checkpoint"),
 			tooltip: localize2('chat.restoreCheckpoint.tooltip', "Restores workspace and chat to this point"),
 			f1: false,
 			category: CHAT_CATEGORY,

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

@@ -430,7 +430,7 @@ export class ChatListItemRenderer extends Disposable implements ITreeRenderer<Ch
 		const codiconRestoreContainer = dom.append(checkpointRestoreContainer, $('.codicon-container'));
 		dom.append(codiconRestoreContainer, $('span.codicon.codicon-bookmark'));
 		const label = dom.append(checkpointRestoreContainer, $('span.checkpoint-label-text'));
-		label.textContent = localize('checkpointRestore', 'Checkpoint restored');
+		label.textContent = localize('checkpointRestore', 'Checkpoint Restored');
 		const checkpointRestoreToolbar = templateDisposables.add(scopedInstantiationService.createInstance(MenuWorkbenchToolBar, checkpointRestoreContainer, MenuId.ChatMessageRestoreCheckpoint, {
 			actionViewItemProvider: (action, options) => {
 				if (action instanceof MenuItemAction) {

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

@@ -2210,8 +2210,8 @@ display: none;
 			width: fit-content;
 			user-select: none;
 			position: absolute;
-			top: 5px;
-			left: 25px;
+			top: 4px;
+			margin-left: 25px;
 			background: var(--vscode-sideBar-background);
 		}
 
@@ -2228,14 +2228,14 @@ display: none;
 		.checkpoint-label-text {
 			font-size: 12px;
 			color: var(--vscode-descriptionForeground);
-			padding-right: 4px;
+			background-color: var(--vscode-sideBar-background);
+			padding: 4px;
 			display: flex;
 			align-items: center;
 			gap: 4px;
-		}
-
-		.monaco-toolbar {
-			left: 145px;
+			position: absolute;
+			margin-left: 71px;
+			margin-top: 2px;
 		}
 	}