1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
package indent
import (
"bytes"
"io"
"strings"
"github.com/muesli/reflow/ansi"
)
type IndentFunc func(w io.Writer)
type Writer struct {
Indent uint
IndentFunc IndentFunc
ansiWriter *ansi.Writer
buf bytes.Buffer
skipIndent bool
ansi bool
}
func NewWriter(indent uint, indentFunc IndentFunc) *Writer {
w := &Writer{
Indent: indent,
IndentFunc: indentFunc,
}
w.ansiWriter = &ansi.Writer{
Forward: &w.buf,
}
return w
}
func NewWriterPipe(forward io.Writer, indent uint, indentFunc IndentFunc) *Writer {
return &Writer{
Indent: indent,
IndentFunc: indentFunc,
ansiWriter: &ansi.Writer{
Forward: forward,
},
}
}
// Bytes is shorthand for declaring a new default indent-writer instance,
// used to immediately indent a byte slice.
func Bytes(b []byte, indent uint) []byte {
f := NewWriter(indent, nil)
_, _ = f.Write(b)
return f.Bytes()
}
// String is shorthand for declaring a new default indent-writer instance,
// used to immediately indent a string.
func String(s string, indent uint) string {
return string(Bytes([]byte(s), indent))
}
// Write is used to write content to the indent buffer.
func (w *Writer) Write(b []byte) (int, error) {
for _, c := range string(b) {
if c == '\x1B' {
// ANSI escape sequence
w.ansi = true
} else if w.ansi {
if (c >= 0x41 && c <= 0x5a) || (c >= 0x61 && c <= 0x7a) {
// ANSI sequence terminated
w.ansi = false
}
} else {
if !w.skipIndent {
w.ansiWriter.ResetAnsi()
if w.IndentFunc != nil {
for i := 0; i < int(w.Indent); i++ {
w.IndentFunc(w.ansiWriter)
}
} else {
_, err := w.ansiWriter.Write([]byte(strings.Repeat(" ", int(w.Indent))))
if err != nil {
return 0, err
}
}
w.skipIndent = true
w.ansiWriter.RestoreAnsi()
}
if c == '\n' {
// end of current line
w.skipIndent = false
}
}
_, err := w.ansiWriter.Write([]byte(string(c)))
if err != nil {
return 0, err
}
}
return len(b), nil
}
// Bytes returns the indented result as a byte slice.
func (w *Writer) Bytes() []byte {
return w.buf.Bytes()
}
// String returns the indented result as a string.
func (w *Writer) String() string {
return w.buf.String()
}
|
