diff options
| author | Stefan Majewsky <majewsky@gmx.net> | 2017-07-11 15:55:34 +0200 |
|---|---|---|
| committer | Stefan Majewsky <majewsky@gmx.net> | 2017-07-11 15:55:34 +0200 |
| commit | e8b31652a0a0fa1767d5e5d86fa9f7016057262c (patch) | |
| tree | 3553f4e45cb9a479513b94177bb873cc671453a9 /pkg/prompt/login.go | |
| parent | 46a314c40c6cd4220c2fc24268008dd11a9110e4 (diff) | |
| download | gofu-e8b31652a0a0fa1767d5e5d86fa9f7016057262c.tar.gz | |
yagni
Diffstat (limited to 'pkg/prompt/login.go')
| -rw-r--r-- | pkg/prompt/login.go | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/pkg/prompt/login.go b/pkg/prompt/login.go index b5fcebd..6e7d8f7 100644 --- a/pkg/prompt/login.go +++ b/pkg/prompt/login.go @@ -24,7 +24,9 @@ import "os" //#include <pwd.h> import "C" -func addLogin(buf *LineBuffer) { +func getLoginField() string { + var result string + //show user name userName := getUserName() commonUser := getenvOrDefault("PRETTYPROMPT_COMMONUSER", "stefan") @@ -33,8 +35,7 @@ func addLogin(buf *LineBuffer) { if userName == "root" { color = "1;41" } - buf.WriteWithColor(userName, color) - buf.Write([]byte("@")) + result = withColor(color, userName) + "@" } //show hostname @@ -43,11 +44,10 @@ func addLogin(buf *LineBuffer) { handleError(err) hostname = "<unknown>" } - buf.WriteWithColor( + return result + withColor( getenvOrDefault("PRETTYPROMPT_HOSTCOLOR", "0;33"), hostname, - ) - buf.Write([]byte(" ")) + ) + " " } func getUserName() string { |