aboutsummaryrefslogtreecommitdiff
path: root/template.go
blob: 503af64338f8d2e30b3e37390cf9e54a642b2e44 (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
package main

import (
	html "html/template"
	"io"
	"os"
	text "text/template"
)

type Templater interface {
	Execute(wr io.Writer, data any) error
}

var (
	templateFunc = map[string]any{
		"DerefI": func(i *int) int { return *i },
		"DerefS": func(i *string) string { return *i },
	}
)

func GetTemplate(templateType, filePath string) (Templater, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return nil, err
	}

	tmpl, err := io.ReadAll(file)
	if err != nil {
		return nil, err
	}

	switch templateType {
	case "text":
		return text.New("text").
			Funcs(templateFunc).
			Parse(string(tmpl))
	case "html":
		return html.New("html").
			Funcs(templateFunc).
			Parse(string(tmpl))
	default:
		panic("Invalid template-type")
	}
}