statusline fanciness

1 files changed, 15 insertions, 6 deletions

diff --git a/go/src/lib/statusline/statuslinue.go b/go/src/lib/statusline/statuslinue.go
index 90ec1af..323c178 100644
--- a/go/src/lib/statusline/statuslinue.go
+++ b/go/src/lib/statusline/statuslinue.go
@@ -5,20 +5,29 @@ import (
 	"io"
 )
 
-type StatusLine struct {
+type StatusLine interface {
+	Put(line string)
+	End(keep bool)
+}
+
+type statusLine struct {
 	out io.Writer
 	prevLen int
 }
 
-func NewStatusLine(out io.Writer) *StatusLine {
-	return &StatusLine{out: out}
+func New(out io.Writer) StatusLine {
+	return &statusLine{out: out}
 }
 
-func (sl *StatusLine) Put(line string) {
+func (sl *statusLine) Put(line string) {
 	fmt.Fprintf(sl.out, "\r%-[1]*[2]s", sl.prevLen, line)
 	sl.prevLen = len(line)
 }
 
-func (sl *StatusLine) End() {
-	fmt.Fprintf(sl.out, "\r%-[1]*[2]s\r", sl.prevLen, "")
+func (sl *statusLine) End(keep bool) {
+	if (keep) {
+		io.WriteString(sl.out, "\n")
+	} else {
+		fmt.Fprintf(sl.out, "\r%-[1]*[2]s\r", sl.prevLen, "")
+	}
 }