package render

import (
	"fmt"
)

// A CompilationError is a parse error during template compilation.
type CompilationError string

func (e CompilationError) Error() string { return string(e) }

func compilationErrorf(format string, a ...interface{}) CompilationError {
	return CompilationError(fmt.Sprintf(format, a...))
}

// Compile parses a source template. It returns an AST root, that can be evaluated.
func (c Config) Compile(source string) (ASTNode, error) {
	root, err := c.Parse(source)
	if err != nil {
		return nil, err
	}
	return c.compileNode(root)
}

// nolint: gocyclo
func (c Config) compileNode(n ASTNode) (Node, error) {
	switch n := n.(type) {
	case *ASTBlock:
		body, err := c.compileNodes(n.Body)
		if err != nil {
			return nil, err
		}
		branches, err := c.compileBlocks(n.Branches)
		if err != nil {
			return nil, err
		}

		cd, ok := c.findBlockDef(n.Name)
		if !ok {
			return nil, compilationErrorf("undefined tag %q", n.Name)
		}
		node := BlockNode{
			Chunk:    n.Chunk,
			Body:     body,
			Branches: branches,
		}
		if cd.parser != nil {
			r, err := cd.parser(node)
			if err != nil {
				return nil, err
			}
			node.renderer = r
		}
		return &node, nil
	case *ASTRaw:
		return &RawNode{n.slices}, nil
	case *ASTSeq:
		children, err := c.compileNodes(n.Children)
		if err != nil {
			return nil, err
		}
		return &SeqNode{children}, nil
	case *ASTTag:
		if td, ok := c.FindTagDefinition(n.Name); ok {
			f, err := td(n.Args)
			if err != nil {
				return nil, err
			}
			return &TagNode{n.Chunk, f}, nil
		}
		return nil, compilationErrorf("unknown tag: %s", n.Name)
	case *ASTText:
		return &TextNode{n.Chunk}, nil
	case *ASTObject:
		return &ObjectNode{n.Chunk, n.expr}, nil
	default:
		panic(fmt.Errorf("un-compilable node type %T", n))
	}
}

func (c Config) compileBlocks(blocks []*ASTBlock) ([]*BlockNode, error) {
	out := make([]*BlockNode, 0, len(blocks))
	for _, child := range blocks {
		compiled, err := c.compileNode(child)
		if err != nil {
			return nil, err
		}
		out = append(out, compiled.(*BlockNode))
	}
	return out, nil
}

func (c Config) compileNodes(nodes []ASTNode) ([]Node, error) {
	out := make([]Node, 0, len(nodes))
	for _, child := range nodes {
		compiled, err := c.compileNode(child)
		if err != nil {
			return nil, err
		}
		out = append(out, compiled)
	}
	return out, nil
}