classList.go

package GoHtml

import (
	"strings"
)

type ClassList struct {
	classes map[string]struct{}
}

// NewClassList returns a new empty ClassList.
func NewClassList() ClassList {
	classList := ClassList{
		classes: make(map[string]struct{}),
	}

	return classList
}

// AppendClass append className to classList. className that contains multiple classes is also a valid className.
func (classList ClassList) AppendClass(className string) {
	classes := strings.SplitSeq(className, " ")
	for v := range classes {
		classList.classes[strings.TrimSpace(v)] = struct{}{}
	}
}

// DecodeFrom append classes in the node to classList.
// If node is nil SetClass does nothing.
func (classList ClassList) DecodeFrom(node *Node) {
	if node == nil {
		return 
	}
	classes, _ := node.GetAttribute("class")
	classList.AppendClass(classes)
}

// Contains returns whether the className exists or not.
func (classList ClassList) Contains(className string) bool {
	classes := strings.SplitSeq(className, " ")
	for v := range classes {
		_, ok := classList.classes[strings.TrimSpace(v)]
		if !ok {
			return false
		}
	}

	return true
}

// DeleteClass deletes the specified classes in className.
func (classList ClassList) DeleteClass(className string) {
	classes := strings.SplitSeq(className, " ")
	for v := range classes {
		delete(classList.classes, strings.TrimSpace(v))
	}
}

// Encode returns the full className.
func (classList ClassList) Encode() string {
	classes := ""
	for v := range classList.classes {
		if classes != ""{
			classes+=" "
		}
		classes+=v
	}
	return classes
}

// EncodeTo encodes classNames for the node.
// If node is nil EncodeTo does nothing.
func (classList ClassList) EncodeTo(node *Node){
	if node == nil {
		return
	}
	node.SetAttribute("class", classList.Encode())
}