udemy-go-web-2/cmd/web/render.go

120 lines
2.5 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
UserID int
StripeSecretKey string
StripePubKey string
}
var functions = template.FuncMap{
"formatCurrency": formatCurrency,
}
func formatCurrency(n int) string {
f := float32(n) / float32(100)
return fmt.Sprintf("€%.2f", f)
}
//go:embed templates
var templateFS embed.FS
func (app *application) addDefaultData(td *templateData, r *http.Request) *templateData {
td.API = app.config.api
td.StripePubKey = app.config.stripe.key
td.StripeSecretKey = app.config.stripe.secret
if app.Session.Exists(r.Context(), "userID") {
td.IsAuthenticated = 1
td.UserID = app.Session.GetInt(r.Context(), "userID")
} else {
td.IsAuthenticated = 0
}
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("templates/%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
}