use templ.Component in all utils

AccentDesign · Jan 29, 2025 · 36037f6 · 36037f6
1 parent 3cc256a
commit 36037f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/element_test.go b/element_test.go
@@ -3,11 +3,12 @@ package gtml
 import (
 	"bytes"
 	"context"
+	"github.com/a-h/templ"
 	"testing"
 )
 
 // Helper function to simplify rendering a Element and capturing the output.
-func renderElement(t *testing.T, node *Element) string {
+func renderElement(t *testing.T, node templ.Component) string {
 	var buf bytes.Buffer
 	err := node.Render(context.Background(), &buf)
 	if err != nil {

diff --git a/utils.go b/utils.go
@@ -5,15 +5,15 @@ import (
 )
 
 // If returns the element if the condition is true, otherwise returns an empty element.
-func If(condition bool, element *Element) *Element {
+func If(condition bool, element templ.Component) templ.Component {
 	if condition {
 		return element
 	}
 	return Empty()
 }
 
 // IfElse returns the truthy element if the condition is true, otherwise returns the falsy element.
-func IfElse(condition bool, truthyElement, falsyElement *Element) *Element {
+func IfElse(condition bool, truthyElement, falsyElement templ.Component) templ.Component {
 	if condition {
 		return truthyElement
 	}