-
Notifications
You must be signed in to change notification settings - Fork 3
/
printer.go
67 lines (59 loc) · 1.24 KB
/
printer.go
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
// Copyright 2014 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package bind
import (
"bytes"
"fmt"
)
type Printer struct {
Buf *bytes.Buffer
IndentEach []byte
indentText []byte
needIndent bool
}
func (p *Printer) writeIndent() error {
if !p.needIndent {
return nil
}
p.needIndent = false
_, err := p.Buf.Write(p.indentText)
return err
}
func (p *Printer) Write(b []byte) (n int, err error) {
wrote := 0
for len(b) > 0 {
if err := p.writeIndent(); err != nil {
return wrote, err
}
i := bytes.IndexByte(b, '\n')
if i < 0 {
break
}
n, err = p.Buf.Write(b[0 : i+1])
wrote += n
if err != nil {
return wrote, err
}
b = b[i+1:]
p.needIndent = true
}
if len(b) > 0 {
n, err = p.Buf.Write(b)
wrote += n
}
return wrote, err
}
func (p *Printer) Printf(format string, args ...interface{}) {
if _, err := fmt.Fprintf(p, format, args...); err != nil {
panic(fmt.Sprintf("printer: %v", err))
}
}
func (p *Printer) Indent() {
p.indentText = append(p.indentText, p.IndentEach...)
}
func (p *Printer) Outdent() {
if len(p.indentText) > len(p.IndentEach)-1 {
p.indentText = p.indentText[len(p.IndentEach):]
}
}