aw/goutil
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow chaining of node compare functions

Browse Source
This commit is contained in:
gutmet 2020-11-08 19:28:25 +01:00
parent ed273f9485
commit c19893df11
1 changed files with 34 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
html/html.go
View File

@ -7,6 +7,8 @@ import (
type HtmlNode html.Node type HtmlNode html.Node
type NodeComparer func(n *HtmlNode) bool
func (n *HtmlNode) GetAttribute(key string) ([]string, bool) { func (n *HtmlNode) GetAttribute(key string) ([]string, bool) {
if n != nil { if n != nil {
for _, attr := range n.Attr { for _, attr := range n.Attr {
@ -18,6 +20,12 @@ func (n *HtmlNode) GetAttribute(key string) ([]string, bool) {
return []string{}, false return []string{}, false
} }
func (n *HtmlNode) HasAttributeVal(attr string, val string) NodeComparer {
return func(n *HtmlNode) bool {
return n.CheckAttribute(attr, val)
}
}
func (n *HtmlNode) CheckAttribute(attr string, val string) bool { func (n *HtmlNode) CheckAttribute(attr string, val string) bool {
if n != nil && n.Type == html.ElementNode { if n != nil && n.Type == html.ElementNode {
attrvals, ok := n.GetAttribute(attr) attrvals, ok := n.GetAttribute(attr)
@ -32,7 +40,13 @@ func (n *HtmlNode) CheckAttribute(attr string, val string) bool {
return false return false
} }
func (n *HtmlNode) Find(f func(n *HtmlNode) bool) *HtmlNode { func (f NodeComparer) And(f2 NodeComparer) NodeComparer {
return func(n *HtmlNode) bool {
return f(n) && f2(n)
}
}
func (n *HtmlNode) Find(f NodeComparer) *HtmlNode {
if n != nil { if n != nil {
if f(n) { if f(n) {
return n return n
@ -47,7 +61,7 @@ func (n *HtmlNode) Find(f func(n *HtmlNode) bool) *HtmlNode {
return nil return nil
} }
func (n *HtmlNode) FindAll(f func(n *HtmlNode) bool) []*HtmlNode { func (n *HtmlNode) FindAll(f NodeComparer) []*HtmlNode {
all := []*HtmlNode{} all := []*HtmlNode{}
if n != nil { if n != nil {
if f(n) { if f(n) {
@ -63,9 +77,25 @@ func (n *HtmlNode) FindAll(f func(n *HtmlNode) bool) []*HtmlNode {
return all return all
} }
func IsText() NodeComparer {
return func(n *HtmlNode) bool { return n.Type == html.TextNode }
}
func IsTag(name string) NodeComparer {
return func(n *HtmlNode) bool { return n.Type == html.ElementNode && n.Data == name }
}
func HasClass(name string) NodeComparer {
return func(n *HtmlNode) bool { return n.CheckAttribute("class", name) }
}
func HasID(id string) NodeComparer {
return func(n *HtmlNode) bool { return n.CheckAttribute("id", id) }
}
func (n *HtmlNode) GetElementById(id string) *HtmlNode { func (n *HtmlNode) GetElementById(id string) *HtmlNode {
if n != nil { if n != nil {
return n.Find(func(n *HtmlNode) bool { return n.CheckAttribute("id", id) }) return n.Find(HasID(id))
} else { } else {
return nil return nil
} }
@ -73,7 +103,7 @@ func (n *HtmlNode) GetElementById(id string) *HtmlNode {
func (n *HtmlNode) GetElementsByClass(class string) []*HtmlNode { func (n *HtmlNode) GetElementsByClass(class string) []*HtmlNode {
if n != nil { if n != nil {
return n.FindAll(func(n *HtmlNode) bool { return n.CheckAttribute("class", class) }) return n.FindAll(HasClass(class))
} else { } else {
return nil return nil
} }