From e8b31652a0a0fa1767d5e5d86fa9f7016057262c Mon Sep 17 00:00:00 2001
From: Stefan Majewsky <majewsky@gmx.net>
Date: Tue, 11 Jul 2017 15:55:34 +0200
Subject: yagni

---
 pkg/prompt/login.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

(limited to 'pkg/prompt/login.go')

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 {
-- 
cgit v1.3.1