-
-
Notifications
You must be signed in to change notification settings - Fork 954
/
load_template.go
190 lines (163 loc) · 4.39 KB
/
load_template.go
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package template
import (
"bytes"
"context"
"embed"
htemplate "html/template"
"io"
"io/fs"
"path/filepath"
"text/template"
"github.com/ory/kratos/x"
"github.com/ory/x/fetcher"
"github.com/Masterminds/sprig/v3"
lru "github.com/hashicorp/golang-lru/v2"
"github.com/pkg/errors"
)
//go:embed courier/builtin/templates/*
var templates embed.FS
var Cache, _ = lru.New[string, Template](16)
type Template interface {
Execute(wr io.Writer, data interface{}) error
}
type templateDependencies interface {
x.HTTPClientProvider
}
func loadBuiltInTemplate(filesystem fs.FS, name string, html bool) (Template, error) {
if t, found := Cache.Get(name); found {
return t, nil
}
file, err := filesystem.Open(name)
if err != nil {
// try to fallback to bundled templates
var fallbackErr error
file, fallbackErr = templates.Open(filepath.Join("courier/builtin/templates", name))
if fallbackErr != nil {
// return original error from os.DirFS
return nil, errors.WithStack(err)
}
}
defer file.Close()
var b bytes.Buffer
if _, err := io.Copy(&b, file); err != nil {
return nil, errors.WithStack(err)
}
var tpl Template
if html {
t, err := htemplate.New(name).Funcs(sprig.HtmlFuncMap()).Parse(b.String())
if err != nil {
return nil, errors.WithStack(err)
}
tpl = t
} else {
t, err := template.New(name).Funcs(sprig.TxtFuncMap()).Parse(b.String())
if err != nil {
return nil, errors.WithStack(err)
}
tpl = t
}
_ = Cache.Add(name, tpl)
return tpl, nil
}
func loadRemoteTemplate(ctx context.Context, d templateDependencies, url string, html bool) (t Template, err error) {
if t, found := Cache.Get(url); found {
return t, nil
}
f := fetcher.NewFetcher(fetcher.WithClient(d.HTTPClient(ctx)))
bb, err := f.FetchContext(ctx, url)
if err != nil {
return nil, errors.WithStack(err)
}
b := bb.Bytes()
if html {
t, err = htemplate.New(url).Funcs(sprig.HermeticHtmlFuncMap()).Parse(string(b))
if err != nil {
return nil, errors.WithStack(err)
}
} else {
t, err = template.New(url).Funcs(sprig.HermeticTxtFuncMap()).Parse(string(b))
if err != nil {
return nil, errors.WithStack(err)
}
}
Cache.Add(url, t)
return t, nil
}
func loadTemplate(filesystem fs.FS, name, pattern string, html bool) (Template, error) {
if t, found := Cache.Get(name); found {
return t, nil
}
matches, _ := fs.Glob(filesystem, name)
// make sure the file exists in the fs, otherwise fallback to built in templates
if matches == nil {
return loadBuiltInTemplate(filesystem, name, html)
}
glob := name
if pattern != "" {
// pattern matching is used when we have more than one gotmpl for different use cases, such as i18n support
// e.g. some_template/template_name* will match some_template/template_name.body.en_US.gotmpl
matches, _ = fs.Glob(filesystem, pattern)
// set the glob string to match patterns
if matches != nil {
glob = pattern
}
}
var tpl Template
if html {
t, err := htemplate.New(filepath.Base(name)).Funcs(sprig.HermeticHtmlFuncMap()).ParseFS(filesystem, glob)
if err != nil {
return nil, errors.WithStack(err)
}
tpl = t
} else {
t, err := template.New(filepath.Base(name)).Funcs(sprig.HermeticTxtFuncMap()).ParseFS(filesystem, glob)
if err != nil {
return nil, errors.WithStack(err)
}
tpl = t
}
_ = Cache.Add(name, tpl)
return tpl, nil
}
func LoadText(ctx context.Context, d templateDependencies, filesystem fs.FS, name, pattern string, model interface{}, remoteURL string) (string, error) {
var t Template
var err error
if remoteURL != "" {
t, err = loadRemoteTemplate(ctx, d, remoteURL, false)
if err != nil {
return "", err
}
} else {
t, err = loadTemplate(filesystem, name, pattern, false)
if err != nil {
return "", err
}
}
var b bytes.Buffer
if err := t.Execute(&b, model); err != nil {
return "", err
}
return b.String(), nil
}
func LoadHTML(ctx context.Context, d templateDependencies, filesystem fs.FS, name, pattern string, model interface{}, remoteURL string) (string, error) {
var t Template
var err error
if remoteURL != "" {
t, err = loadRemoteTemplate(ctx, d, remoteURL, true)
if err != nil {
return "", err
}
} else {
t, err = loadTemplate(filesystem, name, pattern, true)
if err != nil {
return "", err
}
}
var b bytes.Buffer
if err := t.Execute(&b, model); err != nil {
return "", err
}
return b.String(), nil
}