Update on write parsed as well

Author: Tyriar

src/vs/platform/terminal/common/capabilities/commandDetection/promptInputModel.ts

@@ -3,11 +3,11 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-import { Emitter, type Event } from 'vs/base/common/event';
+import { Emitter, Event } from 'vs/base/common/event';
 import { Disposable } from 'vs/base/common/lifecycle';
 import { ILogService, LogLevel } from 'vs/platform/log/common/log';
 import type { ITerminalCommand } from 'vs/platform/terminal/common/capabilities/capabilities';
-import { debounce } from 'vs/base/common/decorators';
+import { throttle } from 'vs/base/common/decorators';
 
 // Importing types is safe in any layer
 // eslint-disable-next-line local/code-import-patterns
@@ -67,7 +67,11 @@ export class PromptInputModel extends Disposable implements IPromptInputModel {
 		super();
 
 		this._register(this._xterm.onData(e => this._handleInput(e)));
-		this._register(this._xterm.onCursorMove(() => this._sync()));
+		this._register(Event.any(
+			// TODO: Upstream me to headless
+			(this._xterm as any).onWriteParsed,
+			this._xterm.onCursorMove,
+		)(() => this._sync()));
 
 		this._register(onCommandStart(e => this._handleCommandStart(e as { marker: IMarker })));
 		this._register(onCommandExecuted(() => this._handleCommandExecuted()));
@@ -115,7 +119,7 @@ export class PromptInputModel extends Disposable implements IPromptInputModel {
 		this._sync();
 	}
 
-	@debounce(50)
+	@throttle(0)
 	private _sync() {
 		this._syncNow();
 	}

src/vs/platform/terminal/test/common/capabilities/commandDetection/promptInputModel.test.ts

@@ -12,6 +12,7 @@ import type { ITerminalCommand } from 'vs/platform/terminal/common/capabilities/
 // eslint-disable-next-line local/code-import-patterns, local/code-amd-node-module
 import { Terminal } from '@xterm/headless';
 import { strictEqual } from 'assert';
+import { timeout } from 'vs/base/common/async';
 
 class TestPromptInputModel extends PromptInputModel {
 	forceSync() {
@@ -285,6 +286,42 @@ suite('PromptInputModel', () => {
 				fireCommandStart();
 				assertPromptInput('|');
 			});
+
+			test.only('input, go to start (ctrl+home), delete word in front (ctrl+delete)', async () => {
+				await replayEvents([
+					'[?25l]0;C:\Program Files\WindowsApps\Microsoft.PowerShell_7.4.2.0_x64__8wekyb3d8bbwe\pwsh.exe[?25h',
+					'[?25l\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n\r\n[?25h',
+					']633;P;IsWindows=True',
+					']633;P;ContinuationPrompt=\x1b[38\x3b5\x3b8m∙\x1b[0m ',
+					']633;A]633;P;Cwd=C:\x5cGithub\x5cmicrosoft\x5cvscode]633;B',
+					'\r\n16:07:06  vscode   tyriar/210662  $!  via  v18.18.2 \r\n❯ ',
+				]);
+				fireCommandStart();
+				assertPromptInput('|');
+
+				await replayEvents([
+					'[?25lGit push[?25h',
+					'',
+					'[?25lGet-ChildItem -Path a[?25h',
+					'',
+					'[?25lGet-ChildItem -Path a[?25h',
+				]);
+				assertPromptInput('Get|[-ChildItem -Path a]');
+
+				await replayEvents([
+					'',
+					'[?25l[?25h',
+					'',
+				]);
+
+				// Don't force a sync, the prompt input model should update by itself
+				await timeout(0);
+				const actualValueWithCursor = promptInputModel.getCombinedString();
+				strictEqual(
+					actualValueWithCursor,
+					'|'.replaceAll('\n', '\u23CE')
+				);
+			});
 		});
 	});
 });