mirror of
https://github.com/strongdm/comply
synced 2024-08-28 10:03:33 +00:00
52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package ace
|
|
|
|
import "bytes"
|
|
import "strings"
|
|
|
|
// File represents a file.
|
|
type outputFormatter interface {
|
|
OpeningElement(*bytes.Buffer, element) (int, error)
|
|
ClosingElement(*bytes.Buffer, element) (int, error)
|
|
WritingTextValue(*bytes.Buffer, element) (int, error)
|
|
}
|
|
|
|
type Formatter struct {
|
|
indent string
|
|
}
|
|
|
|
func newFormatter(indent string) outputFormatter {
|
|
f := &Formatter{
|
|
indent: indent,
|
|
}
|
|
return f
|
|
}
|
|
|
|
func (f *Formatter) OpeningElement(bf *bytes.Buffer, e element) (int, error) {
|
|
if e.IsControlElement() {
|
|
return 0, nil
|
|
}
|
|
|
|
base := e.Base()
|
|
if base.parent != nil && base.parent.IsBlockElement() {
|
|
return f.writeIndent(bf, base.ln.indent)
|
|
}
|
|
return 0, nil
|
|
}
|
|
func (f *Formatter) ClosingElement(bf *bytes.Buffer, e element) (int, error) {
|
|
if e.IsBlockElement() {
|
|
return f.writeIndent(bf, e.Base().ln.indent)
|
|
}
|
|
return 0, nil
|
|
}
|
|
func (f *Formatter) WritingTextValue(bf *bytes.Buffer, e element) (int, error) {
|
|
if e.IsBlockElement() {
|
|
return f.writeIndent(bf, e.Base().ln.indent+1)
|
|
}
|
|
return 0, nil
|
|
}
|
|
|
|
func (f *Formatter) writeIndent(bf *bytes.Buffer, depth int) (int, error) {
|
|
indent := "\n" + strings.Repeat(f.indent, depth)
|
|
return bf.WriteString(indent)
|
|
}
|