udemy-go-web-2/cmd/web/render.go
2024-08-04 18:35:34 +02:00

101 lines
2.1 KiB
Go

package main
import (
"embed"
"fmt"
"html/template"
"net/http"
"strings"
)
type templateData struct {
StringMap map[string]string
IntMap map[string]int
FloatMap map[string]int
Data map[string]interface{}
CSRFToken string
Flash string
Warning string
Error string
API string
CSSVersion string
IsAuthenticated int
}
var functions = template.FuncMap{}
//go:embed templates
var templateFS embed.FS
func (app *application) addDefaultData(td *templateData, r *http.Request) *templateData {
td.API = app.config.api
return td
}
func (app *application) renderTemplate(
w http.ResponseWriter,
r *http.Request,
page string,
td *templateData,
partials ...string,
) error {
var t *template.Template
var err error
templateToRender := fmt.Sprintf("templates/%s.page.gohtml", page)
_, templateInMap := app.templateCache[templateToRender]
if app.config.env == "production" && templateInMap {
t = app.templateCache[templateToRender]
} else {
t, err = app.parseTemplate(partials, page, templateToRender)
if err != nil {
app.errorLog.Println(err)
return err
}
}
if td == nil {
td = &templateData{}
}
td = app.addDefaultData(td, r)
err = t.Execute(w, td)
if err != nil {
app.errorLog.Println(err)
return err
}
return nil
}
func (app *application) parseTemplate(
partials []string,
page, templateToRender string,
) (*template.Template, error) {
var t *template.Template
var err error
// build partials
if len(partials) > 0 {
for i, x := range partials {
partials[i] = fmt.Sprintf("tempaltes/%s.partial.gohtml", x)
}
t, err = template.New(fmt.Sprintf("%s.page.gohtml", page)).
Funcs(functions).
ParseFS(templateFS, "templates/base.layout.gohtml", strings.Join(partials, ","), templateToRender)
} else {
t, err = template.New(fmt.Sprintf("%s.page.gohtml", page)).
Funcs(functions).
ParseFS(templateFS, "templates/base.layout.gohtml", templateToRender)
}
if err != nil {
app.errorLog.Println(err)
return nil, err
}
app.templateCache[templateToRender] = t
return t, nil
}