summaryrefslogtreecommitdiff
path: root/internal/template.go
blob: 3800aefb7325db3b68919b3133f06dfbf6aeb4c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package internal

import (
	"bytes"
	"fmt"
	"html/template"
	"io"
	"log"

	"github.com/labstack/echo/v4"
)

type TemplateEngine struct {
	tmpl *template.Template
}

func NewTemplateEngine() (*TemplateEngine, error) {
	tmpl := template.New("")

	if _, err := tmpl.ParseGlob("views/*.html"); err != nil {
		return nil, fmt.Errorf("parsing layout templates: %w", err)
	}

	if _, err := tmpl.ParseGlob("views/pages/*/*.html"); err != nil {
		return nil, fmt.Errorf("parsing page templates: %w", err)
	}

	log.Println("Templates loaded successfully")
	return &TemplateEngine{tmpl: tmpl}, nil
}

func (te *TemplateEngine) Render(
	w io.Writer,
	name string,
	data interface{},
	c echo.Context,
) error {
	return te.tmpl.ExecuteTemplate(w, name, data)
}

func (te *TemplateEngine) GetContent(
	page,
	section string,
) (template.HTML, error) {
	templateName := fmt.Sprintf("%s_%s", page, section)

	var buf bytes.Buffer
	if err := te.tmpl.ExecuteTemplate(&buf, templateName, nil); err != nil {
		return "", err
	}

	return template.HTML(buf.String()), nil
}