css: fix class selector for elements with multiple classes

ericchiang · Jun 25, 2022 · 6e0f06c · 6e0f06c
1 parent afb36d4
commit 6e0f06c
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/css.go b/css.go
@@ -378,8 +378,12 @@ func (s *subclassSelectorMatcher) match(n *html.Node) bool {
 
 	if s.classSelector != "" {
 		for _, a := range n.Attr {
-			if a.Key == "class" && a.Val == s.classSelector {
-				return true
+			if a.Key == "class" {
+				for _, val := range strings.Fields(a.Val) {
+					if val == s.classSelector {
+						return true
+					}
+				}
 			}
 		}
 		return false

diff --git a/css_test.go b/css_test.go
@@ -139,6 +139,14 @@ func TestSelector(t *testing.T) {
 				`<div class="foo"></div>`,
 			},
 		},
+		{
+			".foo",
+			`<h1><h2 class="foo bar"></h2><div class="bar foo"></div><div id="foo"></div></h1>`,
+			[]string{
+				`<h2 class="foo bar"></h2>`,
+				`<div class="bar foo"></div>`,
+			},
+		},
 		{
 			"div.foo",
 			`<h1><h2 class="foo"></h2><div class="foo"></div><div id="foo"></div></h1>`,