/
xmlutil.go
61 lines (57 loc) · 1.19 KB
/
xmlutil.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
package main
import (
"encoding/xml"
"io"
)
type Tag struct {
Name xml.Name
Attr []xml.Attr
Children []interface{}
}
func (t *Tag) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
start.Name = t.Name
start.Attr = t.Attr
e.EncodeToken(start)
for _, v := range t.Children {
switch v.(type) {
case *Tag:
child := v.(*Tag)
if err := e.Encode(child); err != nil {
return err
}
case xml.CharData:
e.EncodeToken(v.(xml.CharData))
case xml.Comment:
e.EncodeToken(v.(xml.Comment))
}
}
e.EncodeToken(start.End())
return nil
}
func (t *Tag) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
t.Name = start.Name
t.Attr = start.Attr
for {
token, err := d.Token()
if err != nil {
if err == io.EOF {
return nil
}
return err
}
switch token.(type) {
case xml.StartElement:
tok := token.(xml.StartElement)
var data *Tag
if err := d.DecodeElement(&data, &tok); err != nil {
return err
}
t.Children = append(t.Children, data)
case xml.CharData:
t.Children = append(t.Children, token.(xml.CharData).Copy())
case xml.Comment:
t.Children = append(t.Children, token.(xml.Comment).Copy())
}
}
return nil
}