completed classList

Author: udan-jayanith

classList.go

@@ -0,0 +1,91 @@
+package GoHtml
+
+import (
+	"strings"
+	"sync"
+)
+
+type ClassList struct {
+	classes map[string]struct{}
+	rwMutex *sync.Mutex
+}
+
+// NewClassList returns a new empty ClassList.
+func NewClassList() ClassList {
+	classList := ClassList{
+		classes: make(map[string]struct{}),
+		rwMutex: &sync.Mutex{},
+	}
+
+	return classList
+}
+
+// AppendClass append className to classList. className that contains multiple classes is also a valid className.
+func (classList ClassList) AppendClass(className string) {
+	classList.rwMutex.Lock()
+	defer classList.rwMutex.Unlock()
+
+	classes := strings.SplitSeq(className, " ")
+	for v := range classes {
+		classList.classes[strings.TrimSpace(v)] = struct{}{}
+	}
+}
+
+// SetClass append classes in the node to classList.
+func (classList ClassList) SetClass(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 {
+	classList.rwMutex.Lock()
+	defer classList.rwMutex.Unlock()
+
+	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) {
+	classList.rwMutex.Lock()
+	defer classList.rwMutex.Unlock()
+
+	classes := strings.SplitSeq(className, " ")
+	for v := range classes {
+		delete(classList.classes, strings.TrimSpace(v))
+	}
+}
+
+// Encode returns the full className.
+func (classList ClassList) Encode() string {
+	classList.rwMutex.Lock()
+	defer classList.rwMutex.Unlock()
+
+	classes := ""
+	for v := range classList.classes {
+		if classes != ""{
+			classes+=" "
+		}
+		classes+=v
+	}
+	return classes
+}
+
+// EncodeTo encode className for the node.
+func (classList ClassList) EncodeTo(node *Node){
+	if node == nil {
+		return
+	}
+	node.SetAttribute("class", classList.Encode())
+}

classList_test.go

@@ -0,0 +1,37 @@
+package GoHtml_test
+
+import(
+	"testing"
+	"github.com/udan-jayanith/GoHTML"
+)
+
+func TestClasses(t *testing.T){
+	node := GoHtml.CreateNode("div")
+	node.SetAttribute("class", "div-container main")
+
+	classList := GoHtml.NewClassList()
+	classList.SetClass(node)
+	if !classList.Contains("main"){
+		t.Fatal("")
+		return
+	}
+	classList.DeleteClass("main")
+	if classList.Contains("main"){
+		t.Fatal("")
+		return
+	}
+
+	classList.AppendClass("main-div")
+	if !classList.Contains("main-div"){
+		t.Fatal("")
+		return
+	}
+
+	classList.EncodeTo(node)
+	
+	className, _ := node.GetAttribute("class")
+	if className != classList.Encode(){
+		t.Fatal("")
+		return
+	}
+}

main.go

@@ -37,6 +37,10 @@ func DeepCloneNode(node *Node) *Node{
 	attributes := node.attributes
 	node.rwMutex.Unlock()
 
+	if node == nil {
+		return node
+	}
+
 	newNode := Node{
 		childNode: node.GetChildNode(),
 		tagName: node.GetTagName(),

node-tree_test.go

@@ -109,4 +109,7 @@ func TestRemoveNode(t *testing.T){
 	}else if p.GetParent() != article {
 		t.Fatal("Unexpected parent.")
 	}
+
+	//p.RemoveNode()
+	//t.Log(GoHtml.NodeTreeToHTML(article))
 }