diff options
| -rw-r--r-- | pkg/prompt/main.go | 11 |
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 } |