Add IdentWriter

This commit is contained in:
Christian Muehlhaeuser
2019-11-30 06:45:33 +01:00
parent d37d023d5d
commit 72e53c38d2
+39
View File
@@ -0,0 +1,39 @@
package gold
import (
"io"
"strings"
"sync"
)
type IndentWriter struct {
Indent uint
Forward io.Writer
initialWrite sync.Once
}
// Write is used to write more content to the reflow buffer.
func (w *IndentWriter) Write(b []byte) (int, error) {
w.initialWrite.Do(func() {
w.Forward.Write([]byte(strings.Repeat(" ", int(w.Indent))))
})
for _, c := range string(b) {
w.Forward.Write([]byte{byte(c)})
if c == '\n' {
// end of current line
w.Forward.Write([]byte(strings.Repeat(" ", int(w.Indent))))
} else {
// any other character
}
}
return len(b), nil
}
// Close will finish the reflow operation. Always call it before trying to
// retrieve the final result.
func (w *IndentWriter) Close() error {
return nil
}