1 files changed, 12 insertions, 1 deletions
diff --git a/pkg/prompt/misc.go b/pkg/prompt/misc.go
index eaa1848..124015a 100644
--- a/pkg/prompt/misc.go
+++ b/pkg/prompt/misc.go
@@ -18,7 +18,10 @@
 
 package prompt
 
-import "os"
+import (
+	"os"
+	"strconv"
+)
 
 func getTerminalField() string {
 	termName := os.Getenv("TERM")
@@ -30,3 +33,11 @@ func getTerminalField() string {
 	}
 	return withType("term", termName)
 }
+
+func getExitCodeField(arg string) string {
+	exitCode, err := strconv.Atoi(arg)
+	if err == nil && exitCode > 0 {
+		return withColor("1;31", "exit:"+arg)
+	}
+	return ""
+}