liquid/engine.go

package liquid

import (
	"io"

	"github.com/osteele/liquid/filters"
	"github.com/osteele/liquid/render"
	"github.com/osteele/liquid/tags"
)

type engine struct{ settings render.Settings }

// NewEngine returns a new template engine.
func NewEngine() Engine {
	e := engine{render.NewSettings()}
	filters.AddStandardFilters(e.settings.ExpressionSettings)
	tags.AddStandardTags(e.settings)
	return e
}

// RegisterBlock is in the Engine interface.
func (e engine) RegisterBlock(name string, td Renderer) {
	e.settings.AddBlock(name).Renderer(func(w io.Writer, ctx render.RenderContext) error {
		s, err := td(ctx)
		if err != nil {
			return err
		}
		_, err = w.Write([]byte(s))
		return err
	})
}

// RegisterFilter is in the Engine interface.
func (e engine) RegisterFilter(name string, fn interface{}) {
	e.settings.AddFilter(name, fn)
}

// RegisterTag is in the Engine interface.
func (e engine) RegisterTag(name string, td Renderer) {
	// For simplicity, don't expose the two stage parsing/rendering process to clients.
	// Client tags do everything at runtime.
	e.settings.AddTag(name, func(_ string) (func(io.Writer, render.RenderContext) error, error) {
		return func(w io.Writer, ctx render.RenderContext) error {
			s, err := td(ctx)
			if err != nil {
				return err
			}
			_, err = w.Write([]byte(s))
			return err
		}, nil
	})
}

// ParseTemplate is in the Engine interface.
func (e engine) ParseTemplate(text []byte) (Template, error) {
	ast, err := e.settings.Parse(string(text))
	if err != nil {
		return nil, err
	}
	return &template{ast, e.settings}, nil
}

// ParseAndRender is in the Engine interface.
func (e engine) ParseAndRender(text []byte, b Bindings) ([]byte, error) {
	t, err := e.ParseTemplate(text)
	if err != nil {
		return nil, err
	}
	return t.Render(b)
}

// ParseAndRenderString is in the Engine interface.
func (e engine) ParseAndRenderString(text string, b Bindings) (string, error) {
	bs, err := e.ParseAndRender([]byte(text), b)
	if err != nil {
		return "", err
	}
	return string(bs), nil
}