Report continuation prompt in bash

Author: Tyriar

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

@@ -112,8 +112,6 @@ export class PromptInputModel extends Disposable implements IPromptInputModel {
 			return;
 		}
 
-		// TODO: Support multi-line prompts
-
 		this._value = commandLine.substring(this._commandStartX);
 		this._cursorIndex = Math.max(buffer.cursorX - this._commandStartX, 0);
 

src/vs/platform/terminal/common/xterm/shellIntegrationAddon.ts

@@ -382,7 +382,12 @@ export class ShellIntegrationAddon extends Disposable implements IShellIntegrati
 				}
 				switch (key) {
 					case 'ContinuationPrompt': {
-						this._updateContinuationPrompt(value);
+						// Exclude escape sequences and values between \[ and \]
+						const sanitizedValue = (value
+							.replace(/\x1b\[[0-9;]*m/g, '')
+							.replace(/\\\[.*?\\\]/g, '')
+						);
+						this._updateContinuationPrompt(sanitizedValue);
 						return true;
 					}
 					case 'Cwd': {
@@ -411,8 +416,10 @@ export class ShellIntegrationAddon extends Disposable implements IShellIntegrati
 	}
 
 	private _updateContinuationPrompt(value: string) {
-		const commandDetection = this.capabilities.get(TerminalCapability.CommandDetection);
-		commandDetection?.setContinuationPrompt(value);
+		if (!this._terminal) {
+			return;
+		}
+		this._createOrGetCommandDetection(this._terminal).setContinuationPrompt(value);
 	}
 
 	private _updateCwd(value: string) {

src/vs/workbench/contrib/terminal/browser/media/shellIntegration-bash.sh

@@ -162,6 +162,9 @@ __vsc_current_command=""
 __vsc_nonce="$VSCODE_NONCE"
 unset VSCODE_NONCE
 
+# Report continuation prompt
+builtin printf "\e]633;P;ContinuationPrompt=$(echo "$PS2" | sed 's/\x1b/\\\\x1b/g')\a"
+
 __vsc_prompt_start() {
 	builtin printf '\e]633;A\a'
 }