fix wrong variable

models/output.go

@@ -0,0 +1,5 @@
+package models
+
+type Outputs []Output
+
+type Output []byte

templateBuilder.go

@@ -1,8 +1,11 @@
 package templatebuilder
 
 import (
+	"bytes"
 	"html/template"
 	"os"
+
+	"gitea.tecamino.com/paadi/template-builder/models"
 )
 
 type TemplateBuilder struct {
@@ -12,7 +15,7 @@ func NewTemplateBuilder() *TemplateBuilder {
 	return &TemplateBuilder{}
 }
 
-func (tb *TemplateBuilder) Generate(templatePath, outputFile string, data any) error {
+func (tb *TemplateBuilder) GenerateFile(templatePath, outputFile string, data any) error {
 	tmpl, err := template.ParseFiles(templatePath)
 	if err != nil {
 		return err
@@ -30,3 +33,35 @@ func (tb *TemplateBuilder) Generate(templatePath, outputFile string, data any) e
 	}
 	return nil
 }
+
+func (tb *TemplateBuilder) Generate(templatePath string, data any) (b []byte, err error) {
+	tmpl, err := template.ParseFiles(templatePath)
+	if err != nil {
+		return
+	}
+
+	var buf bytes.Buffer
+	err = tmpl.Execute(&buf, data)
+	if err != nil {
+		return
+	}
+	b = buf.Bytes()
+
+	return
+}
+
+func (tb *TemplateBuilder) GenerateMultiple(templatePath string, data ...any) (output models.Outputs, err error) {
+	tmpl, err := template.ParseFiles(templatePath)
+	if err != nil {
+		return
+	}
+	for _, d := range data {
+		var buf bytes.Buffer
+		err = tmpl.Execute(&buf, d)
+		if err != nil {
+			return
+		}
+		output = append(output, buf.Bytes())
+	}
+	return
+}

test/templateBuilder_test.go

@@ -11,13 +11,24 @@ type Inventory struct {
 	Count    uint
 }
 
-func TestTemplateBuilder(t *testing.T) {
+func TestTemplateBuilderFile(t *testing.T) {
-
 	sweaters := Inventory{"wool", 17}
 	t.Log("start test")
 	tmplBuilder := templatebuilder.NewTemplateBuilder()
-	err := tmplBuilder.Generate("test.tmpl", "test.txt", sweaters)
+	err := tmplBuilder.GenerateFile("test.tmpl", "test.txt", sweaters)
 	if err != nil {
 		t.Fatal(err)
 	}
 }
+
+func TestTemplateBuilderData(t *testing.T) {
+	sweaters := Inventory{"wool", 17}
+	t.Log("start test")
+	tmplBuilder := templatebuilder.NewTemplateBuilder()
+	data, err := tmplBuilder.Generate("test.tmpl", sweaters)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	t.Log(data)
+}