46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package handler
|
|
|
|
import (
|
|
"html/template"
|
|
"io/fs"
|
|
"net/http"
|
|
"path/filepath"
|
|
|
|
"github.com/gorilla/csrf"
|
|
"github.com/zhang2092/go-url-shortener/pkg/logger"
|
|
)
|
|
|
|
// renderLayout 渲染方法 带框架
|
|
func renderLayout(w http.ResponseWriter, r *http.Request, templates fs.FS, data any, tmpl string) {
|
|
t := template.New(filepath.Base(tmpl))
|
|
t = t.Funcs(template.FuncMap{
|
|
"csrfField": func() template.HTML {
|
|
return csrf.TemplateField(r)
|
|
},
|
|
"currentUser": func() *Authorize {
|
|
return withUser(r.Context())
|
|
},
|
|
"genShortUrl": func(url string) string {
|
|
scheme := "http://"
|
|
if r.TLS != nil {
|
|
scheme = "https://"
|
|
}
|
|
return scheme + r.Host + "/" + url
|
|
},
|
|
})
|
|
|
|
tpl := template.Must(t.Clone())
|
|
tpl, err := tpl.ParseFS(templates, tmpl, "base/header.html.tmpl", "base/footer.html.tmpl")
|
|
if err != nil {
|
|
logger.Logger.Errorf("template parse: %s, %v", tmpl, err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
if err := tpl.Execute(w, data); err != nil {
|
|
logger.Logger.Errorf("template execute: %s, %v", tmpl, err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|