Print tail before resetting ANSI when truncating

muesli · May 2, 2021 · c80126e · c80126e
1 parent db37f10
commit c80126e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/truncate/truncate.go b/truncate/truncate.go
@@ -93,10 +93,11 @@ func (w *Writer) Write(b []byte) (int, error) {
 		}
 
 		if curWidth > w.width {
+			n, err := w.buf.WriteString(w.tail)
 			if w.ansiWriter.LastSequence() != "" {
 				w.ansiWriter.ResetAnsi()
 			}
-			return w.buf.WriteString(w.tail)
+			return n, err
 		}
 
 		_, err := w.ansiWriter.Write([]byte(string(c)))

diff --git a/truncate/truncate_test.go b/truncate/truncate_test.go
@@ -94,6 +94,13 @@ func TestTruncate(t *testing.T) {
 			"\x1B[7m--",
 			"\x1B[7m--",
 		},
+		// Tail is printed before reset sequence:
+		{
+			3,
+			"…",
+			"\x1B[38;5;219mHiya!",
+			"\x1B[38;5;219mHi…\x1B[0m",
+		},
 	}
 
 	for i, tc := range tt {