prettyprompt: implement basic cwd display

1 files changed, 11 insertions, 1 deletions

diff --git a/pkg/prompt/main.go b/pkg/prompt/main.go
index 8ff56c9..dac8495 100644
--- a/pkg/prompt/main.go
+++ b/pkg/prompt/main.go
@@ -34,6 +34,9 @@ func Exec() int {
 	fields := []string{
 		getLoginField(),
 	}
+	cwd := CurrentDirectory()
+	fields = appendUnlessEmpty(fields, getDirectoryField(cwd))
+
 	line := strings.Join(fields, " ")
 	lineWidth := getPrintableLength(line)
 
@@ -51,7 +54,7 @@ func Exec() int {
 		for idx := range dashes {
 			dashes[idx] = '-'
 		}
-		line += string(dashes)
+		line += withColor("1", string(dashes))
 	}
 
 	os.Stdout.Write([]byte(line + "\n"))
@@ -66,6 +69,13 @@ func getenvOrDefault(key, defaultValue string) (value string) {
 	return
 }
 
+func appendUnlessEmpty(list []string, val string) []string {
+	if val == "" {
+		return list
+	}
+	return append(list, val)
+}
+
 func handleError(err error) {
 	if err != nil {
 		os.Stderr.Write([]byte("\x1B[1;31mPrompt error: " + err.Error() + "\x1B[0m\n"))