summaryrefslogtreecommitdiff
path: root/internal/template.go
blob: 118f7a2b466e79545475cbcf0db8fa0af86519a6 (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
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
}