prettyprompt: print second line with shell identification and prompt char

1 files changed, 11 insertions, 0 deletions

diff --git a/pkg/prompt/main.go b/pkg/prompt/main.go
index f46a7e4..e5096e8 100644
--- a/pkg/prompt/main.go
+++ b/pkg/prompt/main.go
@@ -61,6 +61,17 @@ func Exec() int {
 	}
 
 	os.Stdout.Write([]byte(line + "\n"))
+
+	//print second line: a letter identifying the shell, and the final "$ ")
+	shellIdent := ""
+	switch os.Getenv("PRETTYPROMPT_SHELL") {
+	case "zsh":
+		shellIdent = "Z"
+	case "bash":
+		shellIdent = "B"
+	}
+	os.Stdout.Write([]byte(shellIdent + "$ "))
+
 	return 0
 }