From 340f345efe366764d057542d2194f7b44e7efe11 Mon Sep 17 00:00:00 2001 From: Christian Muehlhaeuser Date: Mon, 2 Dec 2019 15:39:12 +0100 Subject: [PATCH] Fixed rune handling in IndentWriter --- indent.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/indent.go b/indent.go index 24a8803..91a0924 100644 --- a/indent.go +++ b/indent.go @@ -14,6 +14,10 @@ type IndentWriter struct { // Write is used to write content to the indent buffer. func (w *IndentWriter) Write(b []byte) (int, error) { + if w.Indent == 0 { + return w.Forward.Write(b) + } + for _, c := range string(b) { if !w.skipIndent { _, err := w.Forward.Write([]byte(strings.Repeat(" ", int(w.Indent)))) @@ -23,7 +27,7 @@ func (w *IndentWriter) Write(b []byte) (int, error) { w.skipIndent = true } - _, err := w.Forward.Write([]byte{byte(c)}) + _, err := w.Forward.Write([]byte(string(c))) if err != nil { return 0, err }