mirror of https://github.com/mickael-menu/zk
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
package handlebars
|
|
|
|
import (
|
|
"html"
|
|
"path/filepath"
|
|
|
|
"github.com/aymerick/raymond"
|
|
"github.com/mickael-menu/zk/adapter/handlebars/helpers"
|
|
"github.com/mickael-menu/zk/util"
|
|
"github.com/mickael-menu/zk/util/date"
|
|
"github.com/mickael-menu/zk/util/errors"
|
|
)
|
|
|
|
func Init(lang string, logger util.Logger, date date.Provider) {
|
|
helpers.RegisterSlug(logger, lang)
|
|
helpers.RegisterDate(logger, date)
|
|
helpers.RegisterShell(logger)
|
|
}
|
|
|
|
// HandlebarsRenderer holds parsed handlebars template and renders them.
|
|
type HandlebarsRenderer struct {
|
|
templates map[string]*raymond.Template
|
|
}
|
|
|
|
// NewRenderer creates a new instance of HandlebarsRenderer.
|
|
func NewRenderer() *HandlebarsRenderer {
|
|
return &HandlebarsRenderer{
|
|
templates: make(map[string]*raymond.Template),
|
|
}
|
|
}
|
|
|
|
// Render renders a handlebars string template with the given context.
|
|
func (hr *HandlebarsRenderer) Render(template string, context interface{}) (string, error) {
|
|
templ, err := hr.loadTemplate(template)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return hr.render(templ, context)
|
|
}
|
|
|
|
// RenderFile renders a handlebars template file with the given context.
|
|
func (hr *HandlebarsRenderer) RenderFile(path string, context interface{}) (string, error) {
|
|
templ, err := hr.loadFileTemplate(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return hr.render(templ, context)
|
|
}
|
|
|
|
func (hr *HandlebarsRenderer) render(template *raymond.Template, context interface{}) (string, error) {
|
|
res, err := template.Exec(context)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "render template failed")
|
|
}
|
|
return html.UnescapeString(res), nil
|
|
}
|
|
|
|
// loadTemplate loads the template with the given content into the renderer if needed.
|
|
// Returns the parsed template.
|
|
func (hr *HandlebarsRenderer) loadTemplate(content string) (*raymond.Template, error) {
|
|
wrap := errors.Wrapperf("load template failed")
|
|
|
|
// Already loaded?
|
|
templ, ok := hr.templates[content]
|
|
if ok {
|
|
return templ, nil
|
|
}
|
|
|
|
// Load new template.
|
|
templ, err := raymond.Parse(content)
|
|
if err != nil {
|
|
return nil, wrap(err)
|
|
}
|
|
hr.templates[content] = templ
|
|
return templ, nil
|
|
}
|
|
|
|
// loadFileTemplate loads the template at given path into the renderer if needed.
|
|
// Returns the parsed template.
|
|
func (hr *HandlebarsRenderer) loadFileTemplate(path string) (*raymond.Template, error) {
|
|
wrap := errors.Wrapper("load template file failed")
|
|
|
|
path, err := filepath.Abs(path)
|
|
if err != nil {
|
|
return nil, wrap(err)
|
|
}
|
|
|
|
// Already loaded?
|
|
templ, ok := hr.templates[path]
|
|
if ok {
|
|
return templ, nil
|
|
}
|
|
|
|
// Load new template.
|
|
templ, err = raymond.ParseFile(path)
|
|
if err != nil {
|
|
return nil, wrap(err)
|
|
}
|
|
hr.templates[path] = templ
|
|
return templ, nil
|
|
}
|