arno/pup
1
0
Fork 0
mirror of https://github.com/ericchiang/pup synced 2025-01-15 10:11:16 +00:00
Code Issues Releases Wiki Activity

selector is now an interface

Browse Source
This commit is contained in:
ericchiang 2014-09-14 20:34:21 -04:00
parent abd4c986bf
commit 66251a1061
2 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@ -11,7 +11,7 @@ import (
"strings" "strings"
) )
const VERSION string = "0.1.1" const VERSION string = "0.1.2a"
var ( var (
// Flags // Flags
@ -125,7 +125,7 @@ func main() {
PrintNode(root, 0) PrintNode(root, 0)
os.Exit(0) os.Exit(0)
} }
selectors := make([]*selector.Selector, len(cmds)) selectors := make([]selector.Selector, len(cmds))
for i, cmd := range cmds { for i, cmd := range cmds {
if i+1 == len(cmds) { if i+1 == len(cmds) {
d, err := funcs.NewDisplayFunc(cmd) d, err := funcs.NewDisplayFunc(cmd)

23
selector/selector.go
View File

@ -8,11 +8,20 @@ import (
) )
// A CSS Selector // A CSS Selector
type Selector struct { type BasicSelector struct {
Name *regexp.Regexp Name *regexp.Regexp
Attrs map[string]*regexp.Regexp Attrs map[string]*regexp.Regexp
} }
type Selector interface {
// Does this selector match a given node?
Match(node *html.Node) bool
// Find all nodes which match a selector. May return itself.
FindAll(node *html.Node) []*html.Node
// Find all child nodes which match a selector.
FindAllChildren(node *html.Node) []*html.Node
}
type selectorField int type selectorField int
const ( const (
@ -63,7 +72,7 @@ func parseAttrField(command string) (attrKey string, matcher *regexp.Regexp,
} }
// Set a field of this selector. // Set a field of this selector.
func (s *Selector) setFieldValue(f selectorField, v string) error { func (s *BasicSelector) setFieldValue(f selectorField, v string) error {
if v == "" { if v == "" {
return nil return nil
} }
@ -98,9 +107,9 @@ func (s *Selector) setFieldValue(f selectorField, v string) error {
} }
// Convert a string to a selector. // Convert a string to a selector.
func NewSelector(s string) (*Selector, error) { func NewSelector(s string) (Selector, error) {
attrs := map[string]*regexp.Regexp{} attrs := map[string]*regexp.Regexp{}
selector := &Selector{nil, attrs} selector := BasicSelector{nil, attrs}
nextField := NameField nextField := NameField
start := 0 start := 0
// Parse the selector character by character // Parse the selector character by character
@ -153,7 +162,7 @@ func NewSelector(s string) (*Selector, error) {
} }
// Find all nodes which match a selector. // Find all nodes which match a selector.
func (sel *Selector) FindAllChildren(node *html.Node) []*html.Node { func (sel BasicSelector) FindAllChildren(node *html.Node) []*html.Node {
selected := []*html.Node{} selected := []*html.Node{}
child := node.FirstChild child := node.FirstChild
for child != nil { for child != nil {
@ -165,7 +174,7 @@ func (sel *Selector) FindAllChildren(node *html.Node) []*html.Node {
} }
// Find all nodes which match a selector. May return itself. // Find all nodes which match a selector. May return itself.
func (sel *Selector) FindAll(node *html.Node) []*html.Node { func (sel BasicSelector) FindAll(node *html.Node) []*html.Node {
selected := []*html.Node{} selected := []*html.Node{}
if sel.Match(node) { if sel.Match(node) {
return []*html.Node{node} return []*html.Node{node}
@ -180,7 +189,7 @@ func (sel *Selector) FindAll(node *html.Node) []*html.Node {
} }
// Does this selector match a given node? // Does this selector match a given node?
func (sel *Selector) Match(node *html.Node) bool { func (sel BasicSelector) Match(node *html.Node) bool {
if node.Type != html.ElementNode { if node.Type != html.ElementNode {
return false return false
} }