From 237bc19f687d6561f0c1405a9045a0b49f235f8a Mon Sep 17 00:00:00 2001
From: TorchedSammy <38820196+TorchedSammy@users.noreply.github.com>
Date: Fri, 4 Mar 2022 21:16:45 -0400
Subject: [PATCH] fix: prompt handling

---
 rl.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/rl.go b/rl.go
index a68189a..866d482 100644
--- a/rl.go
+++ b/rl.go
@@ -217,13 +217,12 @@ func (lr *lineReader) Read() (string, error) {
 func (lr *lineReader) SetPrompt(prompt string) {
 	halfPrompt := strings.Split(prompt, "\n")
 	if len(halfPrompt) > 1 {
+		lr.rl.Multiline = true
 		lr.rl.SetPrompt(strings.Join(halfPrompt[:len(halfPrompt) - 1], "\n"))
 		lr.rl.MultilinePrompt = halfPrompt[len(halfPrompt) - 1:][0]
 	} else {
-		// print cursor up ansi code
-		//fmt.Printf("\033[1A")
-		lr.rl.SetPrompt("")
-		lr.rl.MultilinePrompt = halfPrompt[len(halfPrompt) - 1:][0]
+		lr.rl.Multiline = false
+		lr.rl.SetPrompt(prompt)
 	}
 	if !running {
 		lr.rl.RefreshPromptInPlace("")