Adding the Sequence function to the template context

internal/data/context_test.go

@@ -0,0 +1,40 @@
+package data
+
+import (
+	"github.com/AdikaStyle/go-report-builder/internal/models"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestTemplateContext_Sequence(t *testing.T) {
+	template := `{{- range $i, $item := .Sequence 3 }}
+a
+{{- end }}`
+
+	te := NewGolangTemplateEngine()
+	rendered, err := te.Render([]byte(template), &models.TemplateContext{})
+
+	assert.Nil(t, err)
+	assert.NotNil(t, rendered)
+	assert.EqualValues(t, "\na\na\na", string(rendered))
+}
+
+func TestTemplateContext_Sequence2(t *testing.T) {
+	template := `{{- range $i, $item := .Sequence 10 }}
+{{- if lt $i (len $.Values.array) }}
+{{ index $.Values.array $i }}
+{{- else }}
+0
+{{- end }}
+
+{{- end }}`
+
+	te := NewGolangTemplateEngine()
+	rendered, err := te.Render([]byte(template), &models.TemplateContext{Values: map[string]interface{}{
+		"array": []int{4, 5, 6},
+	}})
+
+	assert.Nil(t, err)
+	assert.NotNil(t, rendered)
+	assert.EqualValues(t, "\n4\n5\n6\n0\n0\n0\n0\n0\n0\n0", string(rendered))
+}

internal/models/context.go

@@ -3,3 +3,11 @@ package models
 type TemplateContext struct {
 	Values interface{}
 }
+
+func (ctx *TemplateContext) Sequence(size int) []int {
+	var out []int
+	for i := 0; i < size; i++ {
+		out = append(out, i+1)
+	}
+	return out
+}

main.go

@@ -20,7 +20,7 @@ func main() {
 	module := cmd.NewModule(config)
 	logrus.Info("module built successfully")
 
-	logrus.Info("server started at port: %d...", config.ServerPort)
+	logrus.Infof("server started at port: %d...", config.ServerPort)
 	panicOnError(module.Server.Start())
 }