package selector
import (
"slices"
"strings"
"golang.org/x/net/html"
)
func GetInitCSSSelectors(node *html.Node, selectors []string) []string {
if node.Type == html.ElementNode {
selectors = append(selectors, node.Data)
for _, attr := range node.Attr {
if attr.Key == "class" {
classes := strings.Split(attr.Val, " ")
for _, class := range classes {
selectors = append(selectors, "."+class)
}
} else if attr.Key == "id" {
selectors = append(selectors, "#"+attr.Val)
} else {
selectors = append(selectors, "["+attr.Key+"=\""+attr.Val+"\"]")
}
}
}
return selectors
}
func SplitSelector(s string) []string {
var result []string
var current string
for _, char := range s {
switch char {
case '.', '#', '[', ']', ':':
if current != "" {
if string(char) == "]" {
current += string(char)
}
result = append(result, current)
}
current = ""
if string(char) != "]" {
current += string(char)
}
default:
current += string(char)
}
}
if current != "" && current != "]" {
result = append(result, current)
}
return result
}
func Contains(selector []string, node []string) bool {
has := true
for _, s := range selector {
if !slices.Contains(node, s) {
has = false
}
}
return has
}