From abc1027743055c0ff9e423c7e173207f8afa219a Mon Sep 17 00:00:00 2001
From: early <me@earlybird.gay>
Date: Mon, 30 Dec 2024 22:57:31 -0700
Subject: [PATCH] add range template blocks

---
 web/htmlt/parse/parse.go      | 42 +++++++++++++++++++++++++++++++++++
 web/htmlt/parse/parse_test.go |  2 ++
 2 files changed, 44 insertions(+)

diff --git a/web/htmlt/parse/parse.go b/web/htmlt/parse/parse.go
index 6c1e630..cb0f682 100644
--- a/web/htmlt/parse/parse.go
+++ b/web/htmlt/parse/parse.go
@@ -150,6 +150,8 @@ readTemplate:
 					p.tokens.discardUntil(TEMPLATE_CLOSE)
 				}
 				return &TemplateElse{}, nil
+			case "range":
+				return p.templateRangeNode(callback)
 			case "with":
 				return p.templateWithNode(callback)
 			case "end":
@@ -294,6 +296,46 @@ func (p *parser) templateWithNode(callback contentFunc) (*TemplateWith, error) {
 	return node, nil
 }
 
+type TemplateRange struct {
+	Expression string
+	Content    []any
+}
+
+func (node *TemplateRange) String() string {
+	return fmt.Sprintf("{{ range %s }}%s{{ end }}", node.Expression, fmt.Sprint(node.Content...))
+}
+
+func (p *parser) templateRangeNode(callback contentFunc) (*TemplateRange, error) {
+	p.debug.Println("reading template range")
+	node := new(TemplateRange)
+	for tok := range p.tokens.until(TEMPLATE_CLOSE) {
+		switch tok.Type {
+		case TEXT, WHITESPACE:
+			node.Expression += tok.Literal
+		}
+	}
+	node.Expression = strings.TrimSpace(node.Expression)
+
+	acc := make([]any, 0)
+	for {
+		then, err := callback(TEMPLATE_OPEN)
+		if err != nil {
+			return nil, err
+		}
+		if len(then) == 0 {
+			break
+		}
+		acc = append(acc, then...)
+		last := acc[len(acc)-1]
+		if _, ok := last.(*TemplateEnd); ok {
+			node.Content = acc[:len(acc)-1]
+			break
+		}
+	}
+	p.debug.Printf("read template range %s", node)
+	return node, nil
+}
+
 var voidElems = []string{
 	"area", "base", "br", "col",
 	"embed", "hr", "img", "input",
diff --git a/web/htmlt/parse/parse_test.go b/web/htmlt/parse/parse_test.go
index 44c68fa..c359826 100644
--- a/web/htmlt/parse/parse_test.go
+++ b/web/htmlt/parse/parse_test.go
@@ -30,6 +30,8 @@ func TestParseBasic(t *testing.T) {
 		{name: "if-elif-else-nested", input: "{{ if .m1 }}{{ .m1 }}{{ else if m2 }}{{ .m2 }}{{ else }}{{ .m3 }}{{ end }}"},
 		{name: "with", input: "{{ with .message }}hello{{ end }}"},
 		{name: "with-nested", input: "{{ with .message }}{{ . }}{{ end }}"},
+		{name: "range", input: "{{ range .messages }}hellp{{ end }}"},
+		{name: "range-nested", input: "{{ range .messages }}{{ . }}{{ end }}"},
 	}
 	/*
 		testStrings := map[string]string{
-- 
2.39.5