goparsify/html/html.go
2017-08-13 13:00:22 +10:00

60 lines
1.3 KiB
Go

package html
import (
. "github.com/vektah/goparsify"
)
func parse(input string) (result interface{}, err error) {
return Run(tag, input)
}
type htmlTag struct {
Name string
Attributes map[string]string
Body []interface{}
}
var (
tag Parser
identifier = Regex("[a-zA-Z][a-zA-Z0-9]*")
text = NotChars("<>").Map(func(n Result) Result {
return Result{Result: n.Token}
})
element = Any(text, &tag)
elements = Some(element).Map(func(n Result) Result {
ret := []interface{}{}
for _, child := range n.Child {
ret = append(ret, child.Result)
}
return Result{Result: ret}
})
attr = Seq(identifier, "=", StringLit(`"'`))
attrs = Some(attr).Map(func(node Result) Result {
attr := map[string]string{}
for _, attrNode := range node.Child {
attr[attrNode.Child[0].Token] = attrNode.Child[2].Result.(string)
}
return Result{Result: attr}
})
tstart = Seq("<", identifier, Cut(), attrs, ">")
tend = Seq("</", Cut(), identifier, ">")
)
func init() {
tag = Seq(tstart, Cut(), elements, tend).Map(func(node Result) Result {
openTag := node.Child[0]
return Result{Result: htmlTag{
Name: openTag.Child[1].Token,
Attributes: openTag.Child[3].Result.(map[string]string),
Body: node.Child[2].Result.([]interface{}),
}}
})
}