69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package responses
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"net/http"
|
|
|
|
"git.kirsle.net/apps/gophertype/pkg/constants"
|
|
"git.kirsle.net/apps/gophertype/pkg/session"
|
|
)
|
|
|
|
// ExtraFuncs lets the core app inject extra template functions for all templates.
|
|
// Use cases include inserting comment or blog partials on other pages.
|
|
var ExtraFuncs template.FuncMap
|
|
|
|
// TemplateFuncs available to all templates.
|
|
func TemplateFuncs(r *http.Request) template.FuncMap {
|
|
funcs := template.FuncMap{
|
|
"CSRF": CSRF(r),
|
|
"CSRFToken": CSRFToken(r),
|
|
"FormValue": FormValue(r),
|
|
"TestFunction": TestFunction(r),
|
|
}
|
|
for k, v := range ExtraFuncs {
|
|
funcs[k] = v
|
|
}
|
|
return funcs
|
|
}
|
|
|
|
// CSRF returns the current CSRF token as an HTML hidden form field.
|
|
func CSRF(r *http.Request) func() template.HTML {
|
|
return func() template.HTML {
|
|
ctx := r.Context()
|
|
if token, ok := ctx.Value(session.CSRFKey).(string); ok {
|
|
return template.HTML(fmt.Sprintf(
|
|
`<input type="hidden" name="%s" value="%s">`,
|
|
constants.CSRFFormName,
|
|
token,
|
|
))
|
|
}
|
|
return template.HTML("[error: csrf token not found in request context]")
|
|
}
|
|
}
|
|
|
|
// CSRFToken returns the current CSRF token as a string.
|
|
func CSRFToken(r *http.Request) func() template.HTML {
|
|
return func() template.HTML {
|
|
ctx := r.Context()
|
|
if token, ok := ctx.Value(session.CSRFKey).(string); ok {
|
|
return template.HTML(token)
|
|
}
|
|
return template.HTML("[error: csrf token not found in request context]")
|
|
}
|
|
}
|
|
|
|
// FormValue returns a form value (1st item only).
|
|
func FormValue(r *http.Request) func(string) string {
|
|
return func(key string) string {
|
|
return r.FormValue(key)
|
|
}
|
|
}
|
|
|
|
// TestFunction is a "hello world" template function.
|
|
func TestFunction(r *http.Request) func() template.HTML {
|
|
return func() template.HTML {
|
|
return template.HTML("TestFunction() called")
|
|
}
|
|
}
|