From 17134cdb291403df44b49620498b5f42a56ab1e1 Mon Sep 17 00:00:00 2001
From: Jan De Dobbeleer <jan.de.dobbeleer@gmail.com>
Date: Mon, 15 Jul 2024 21:58:40 +0200
Subject: [PATCH] fix(bash): support multiline prompt

---
 src/prompt/rprompt.go | 31 ++++++++++++++++++++-----------
 1 file changed, 20 insertions(+), 11 deletions(-)

diff --git a/src/prompt/rprompt.go b/src/prompt/rprompt.go
index 808872d4..a48692fa 100644
--- a/src/prompt/rprompt.go
+++ b/src/prompt/rprompt.go
@@ -11,17 +11,20 @@ import (
 )
 
 func (e *Engine) RPrompt() string {
-	filterRPromptBlock := func(blocks []*config.Block) *config.Block {
-		for _, block := range blocks {
-			if block.Type == config.RPrompt {
-				return block
-			}
+	var rprompt *config.Block
+	var lineCount int
+
+	for _, block := range e.Config.Blocks {
+		if block.Type == config.RPrompt {
+			rprompt = block
+		}
+
+		if block.Newline || block.Type == config.LineBreak {
+			lineCount++
 		}
-		return nil
 	}
 
-	block := filterRPromptBlock(e.Config.Blocks)
-	if block == nil {
+	if rprompt == nil {
 		return ""
 	}
 
@@ -29,13 +32,13 @@ func (e *Engine) RPrompt() string {
 		terminal.Init(shell.GENERIC)
 	}
 
-	block.Init(e.Env)
+	rprompt.Init(e.Env)
 
-	if !block.Enabled() {
+	if !rprompt.Enabled() {
 		return ""
 	}
 
-	text, length := e.renderBlockSegments(block)
+	text, length := e.renderBlockSegments(rprompt)
 	e.rpromptLength = length
 
 	if e.Env.Shell() != shell.BASH {
@@ -55,5 +58,11 @@ func (e *Engine) RPrompt() string {
 
 	text = fmt.Sprintf("%s%s\r", strings.Repeat(" ", padding), text)
 
+	// bash prints this on the same line as the prompt so we need to move the cursor down
+	// in case the prompt spans multiple lines
+	if lineCount > 0 {
+		return terminal.SaveCursorPosition() + strings.Repeat("\n", lineCount) + text + terminal.RestoreCursorPosition()
+	}
+
 	return text
 }