Implement color prompt #5 (#8)

sheepla · Mar 27, 2022 · 24f347a · 24f347a
1 parent 5e155ea
commit 24f347a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -191,9 +191,11 @@ REPL:
 			if result.Stderr != "" {
 				fmt.Fprintln(stderr, color.HiRedString(result.Stderr))
 			}
+			printPrompt(result)
 			line.AppendHistory(code)
 			continue
 		}
+
 	}
 
 	// Write history into file
@@ -215,3 +217,11 @@ func run(code string) (*client.Result, error) {
 func ping() (*client.PingResult, error) {
 	return client.Ping()
 }
+
+func printPrompt(result *client.Result) {
+	fmt.Fprintf(stdout,
+		"%s %s\n",
+		color.New(color.FgBlue, color.Bold).Sprint("websh"),
+		color.New(color.FgCyan, color.Bold).Sprint(result.ElapsedTime),
+	)
+}