69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package pkg
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"net/http"
|
|
)
|
|
|
|
const overrideMethodName = "_method"
|
|
|
|
func templateFile(name string) string {
|
|
return "web/template/" + name
|
|
}
|
|
|
|
func mustRenderTemplate(wr io.Writer, r *http.Request, layout string, filename string, data interface{}) {
|
|
locale := getLocale(r)
|
|
company := getCompany(r)
|
|
user := getUser(r)
|
|
t := template.New(filename)
|
|
t.Funcs(template.FuncMap{
|
|
"gettext": locale.Get,
|
|
"pgettext": locale.GetC,
|
|
"currentLocale": func() string {
|
|
return locale.Language.String()
|
|
},
|
|
"companyURI": func(uri string) string {
|
|
if company == nil {
|
|
return uri
|
|
}
|
|
return "/company/" + company.Slug + uri
|
|
},
|
|
"csrfToken": func() template.HTML {
|
|
return template.HTML(fmt.Sprintf(`<input type="hidden" name="%s" value="%s">`, csrfTokenField, user.CsrfToken))
|
|
},
|
|
"addInputAttr": func(attr string, field *InputField) *InputField {
|
|
field.Attributes = append(field.Attributes, template.HTMLAttr(attr))
|
|
return field
|
|
},
|
|
"addSelectAttr": func(attr string, field *SelectField) *SelectField {
|
|
field.Attributes = append(field.Attributes, template.HTMLAttr(attr))
|
|
return field
|
|
},
|
|
"deleteMethod": func() template.HTML {
|
|
return overrideMethodField(http.MethodDelete)
|
|
},
|
|
"putMethod": func() template.HTML {
|
|
return overrideMethodField(http.MethodPut)
|
|
},
|
|
})
|
|
if _, err := t.ParseFiles(templateFile(filename), templateFile(layout), templateFile("form.gohtml")); err != nil {
|
|
panic(err)
|
|
}
|
|
if err := t.ExecuteTemplate(wr, layout, data); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func overrideMethodField(method string) template.HTML {
|
|
return template.HTML(fmt.Sprintf(`<input type="hidden" name="%s" value="%s">`, overrideMethodName, method))
|
|
}
|
|
func mustRenderAppTemplate(w io.Writer, r *http.Request, filename string, data interface{}) {
|
|
mustRenderTemplate(w, r, "app.gohtml", filename, data)
|
|
}
|
|
|
|
func mustRenderWebTemplate(w io.Writer, r *http.Request, filename string, data interface{}) {
|
|
mustRenderTemplate(w, r, "web.gohtml", filename, data)
|
|
}
|