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
|
package gohtml
import "bytes"
// A tagElement represents a tag element of an HTML document.
type tagElement struct {
tagName string
startTagRaw string
endTagRaw string
children []element
}
// Condense any tag with no child tags (only text or nothing) onto a single line
var Condense bool
// write writes a tag to the buffer.
func (e *tagElement) write(bf *bytes.Buffer, indent int) {
if Condense {
l := len(e.children)
if l == 0 {
writeLine(bf, indent, e.startTagRaw, e.endTagRaw)
return
} else if l == 1 && e.endTagRaw != "" {
if c, ok := e.children[0].(*textElement); ok {
writeLine(bf, indent, e.startTagRaw, c.text, e.endTagRaw)
return
}
}
}
writeLine(bf, indent, e.startTagRaw)
for _, c := range e.children {
var childIndent int
if e.endTagRaw != "" {
childIndent = indent + 1
} else {
childIndent = indent
}
c.write(bf, childIndent)
}
if e.endTagRaw != "" {
writeLine(bf, indent, e.endTagRaw)
}
}
// appendChild append an element to the element's children.
func (e *tagElement) appendChild(child element) {
e.children = append(e.children, child)
}
|
