Support raw tags in lambda.

cbroglie · Jul 12, 2022 · 5229524 · 5229524
1 parent c1e7de1
commit 5229524
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/mustache.go b/mustache.go
@@ -648,7 +648,11 @@ func getElementText(element interface{}, buf io.Writer) error {
 	case *textElement:
 		fmt.Fprintf(buf, "%s", elem.text)
 	case *varElement:
-		fmt.Fprintf(buf, "{{%s}}", elem.name)
+		if elem.raw {
+			fmt.Fprintf(buf, "{{{%s}}}", elem.name)
+		} else {
+			fmt.Fprintf(buf, "{{%s}}", elem.name)
+		}
 	case *sectionElement:
 		if elem.inverted {
 			fmt.Fprintf(buf, "{{^%s}}", elem.name)

diff --git a/mustache_test.go b/mustache_test.go
@@ -396,6 +396,28 @@ func TestLambdaStruct(t *testing.T) {
 	}
 }
 
+func TestLambdaRawTag(t *testing.T) {
+	tmpl := `{{#Lambda}}Hello {{{Name}}}.{{/Lambda}}`
+	data := struct {
+		Name   string
+		Lambda LambdaFunc
+	}{
+		Name: "<br>",
+		Lambda: func(text string, render RenderFunc) (string, error) {
+			return render(text)
+		},
+	}
+
+	output, err := Render(tmpl, data)
+	if err != nil {
+		t.Fatal(err)
+	}
+	expect := "Hello <br>."
+	if output != expect {
+		t.Fatalf("TestLambdaStruct expected %q got %q", expect, output)
+	}
+}
+
 func TestLambdaError(t *testing.T) {
 	tmpl := `{{#lambda}}{{/lambda}}`
 	data := map[string]interface{}{