44 lines
889 B
Go
44 lines
889 B
Go
package utils
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
"text/template"
|
|
|
|
sprig "github.com/go-task/slim-sprig/v3"
|
|
)
|
|
|
|
func NewTemplateInject(r *http.Request, def map[string]any) map[string]any {
|
|
if def == nil {
|
|
def = make(map[string]any)
|
|
}
|
|
headers := make(map[string]string)
|
|
for k, vs := range r.Header.Clone() {
|
|
headers[k] = strings.Join(vs, ",")
|
|
}
|
|
def["Request"] = map[string]any{
|
|
"Host": r.Host,
|
|
"Path": r.URL.Path,
|
|
"Params": r.URL.Query(),
|
|
"Method": r.Method,
|
|
"RemoteIP": GetRemoteIP(r),
|
|
}
|
|
return def
|
|
}
|
|
|
|
func MustTemplate(data string) *template.Template {
|
|
newTemplate, err := NewTemplate(data)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return newTemplate
|
|
}
|
|
|
|
func NewTemplate(data string) (*template.Template, error) {
|
|
funcMap := sprig.FuncMap()
|
|
delete(funcMap, "env")
|
|
delete(funcMap, "expandenv")
|
|
t := template.New("tmpl").Funcs(funcMap)
|
|
return t.Parse(data)
|
|
}
|