html element setters to return the element

maxence-charriere · Oct 3, 2023 · 57ae52e · 57ae52e
1 parent c063b65
commit 57ae52e
Show file tree

Hide file tree

Showing 4 changed files with 1,685 additions and 13 deletions.
diff --git a/pkg/app/gen/html.go b/pkg/app/gen/html.go
@@ -2342,12 +2342,30 @@ func writeStruct(w io.Writer, t tag) {
 	fmt.Fprintln(w)
 
 	fmt.Fprintf(w, `
+	func (e *html%s) setDepth(v uint) UI {
+		e.depth = v
+		return e
+	}
+
+	func (e *html%s) setJSElement(v Value) HTML {
+		e.jsElement = v
+		return e
+	}
+
+	func (e *html%s) setAttrs(v attributes) HTML {
+		e.attributes = v
+		return e
+	}
+
 	func (e *html%s) setParent(v UI) UI {
 		e.parentElement = v
 		return e
 	}
 	`,
 		t.Name,
+		t.Name,
+		t.Name,
+		t.Name,
 	)
 }
 

diff --git a/pkg/app/html.go b/pkg/app/html.go
@@ -27,9 +27,10 @@ type HTML interface {
 	Depth() uint
 
 	attrs() attributes
+	setAttrs(attributes) HTML
 	events() eventHandlers
-	setDepth(uint)
-	setJSElement(Value)
+	setDepth(uint) UI
+	setJSElement(Value) HTML
 	parent() UI
 	body() []UI
 }
@@ -432,14 +433,6 @@ func (e *htmlElement) events() eventHandlers {
 	return e.eventHandlers
 }
 
-func (e *htmlElement) setDepth(v uint) {
-	e.depth = v
-}
-
-func (e *htmlElement) setJSElement(v Value) {
-	e.jsElement = v
-}
-
 func (e *htmlElement) parent() UI {
 	return e.parentElement
 }