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.Wrapperf("load template file failed: %v", path)

	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
}