Allow selectors to work with unknown tags

See https://godoc.org/golang.org/x/net/html#Node for more information
ericchiang · Feb 23, 2019 · f811b2e · f811b2e
1 parent 8f884ca
commit f811b2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/display.go b/display.go
@@ -272,7 +272,7 @@ func jsonify(node *html.Node) map[string]interface{} {
 			}
 		}
 	}
-	vals["tag"] = node.DataAtom.String()
+	vals["tag"] = node.Data
 	children := []interface{}{}
 	for child := node.FirstChild; child != nil; child = child.NextSibling {
 		switch child.Type {

diff --git a/selector.go b/selector.go
@@ -86,7 +86,7 @@ func (s CSSSelector) Match(node *html.Node) bool {
 		return false
 	}
 	if s.Tag != "" {
-		if s.Tag != node.DataAtom.String() {
+		if s.Tag != node.Data {
 			return false
 		}
 	}