liquid/chunks/parser.go

package chunks

import (
	"fmt"

	"github.com/osteele/liquid/expressions"
)

// Parse creates an AST from a sequence of Chunks.
func Parse(chunks []Chunk) (ASTNode, error) {
	type frame struct {
		cd *controlTagDefinition
		cn *ASTControlTag
		ap *[]ASTNode
	}
	var (
		root      = &ASTSeq{}
		ap        = &root.Children // pointer to current node accumulation slice
		ccd       *controlTagDefinition
		ccn       *ASTControlTag
		stack     []frame // stack of control structures
		rawTag    *ASTRaw
		inComment = false
		inRaw     = false
	)
	for _, c := range chunks {
		switch {
		case inComment:
			if c.Type == TagChunkType && c.Tag == "endcomment" {
				inComment = false
			}
		case inRaw:
			if c.Type == TagChunkType && c.Tag == "endraw" {
				inComment = false
			} else {
				rawTag.slices = append(rawTag.slices, c.Source)
			}
		case c.Type == ObjChunkType:
			expr, err := expressions.Parse(c.Args)
			if err != nil {
				return nil, err
			}
			*ap = append(*ap, &ASTObject{c, expr})
		case c.Type == TextChunkType:
			*ap = append(*ap, &ASTText{Chunk: c})
		case c.Type == TagChunkType:
			if cd, ok := findControlTagDefinition(c.Tag); ok {
				switch {
				case c.Tag == "comment":
					inComment = true
				case c.Tag == "raw":
					inRaw = true
					rawTag = &ASTRaw{}
					*ap = append(*ap, rawTag)
				case cd.requiresParent() && !cd.compatibleParent(ccd):
					suffix := ""
					if ccd != nil {
						suffix = "; immediate parent is " + ccd.name
					}
					return nil, fmt.Errorf("%s not inside %s%s", cd.name, cd.parent.name, suffix)
				case cd.isStartTag():
					stack = append(stack, frame{cd: ccd, cn: ccn, ap: ap})
					ccd, ccn = cd, &ASTControlTag{Chunk: c, cd: cd}
					*ap = append(*ap, ccn)
					ap = &ccn.Body
				case cd.isBranchTag:
					n := &ASTControlTag{Chunk: c, cd: cd}
					ccn.Branches = append(ccn.Branches, n)
					ap = &n.Body
				case cd.isEndTag:
					// if ccn != nil && cd.parser != nil {
					// 	renderer, err := cd.parser(*ccn)
					// 	if err != nil {
					// 		return nil, err
					// 	}
					// 	ccn.renderer = renderer
					// }
					f := stack[len(stack)-1]
					stack = stack[:len(stack)-1]
					ccd, ccn, ap = f.cd, f.cn, f.ap
				}
			} else if td, ok := FindTagDefinition(c.Tag); ok {
				f, err := td(c.Args)
				if err != nil {
					return nil, err
				}
				*ap = append(*ap, &ASTFunctional{c, f})
			} else {
				return nil, fmt.Errorf("unknown tag: %s", c.Tag)
			}
			// } else if len(*ap) > 0 {
			// 	switch n := ((*ap)[len(*ap)-1]).(type) {
			// 	case *ASTChunks:
			// 		n.chunks = append(n.chunks, c)
			// 	default:
			// 		*ap = append(*ap, &ASTChunks{chunks: []Chunk{c}})
			// 	}
			// } else {
			// 	*ap = append(*ap, &ASTChunks{chunks: []Chunk{c}})
			// }
		}
	}
	if ccd != nil {
		return nil, fmt.Errorf("unterminated %s tag", ccd.name)
	}
	if err := evaluateBuilders(root); err != nil {
		return nil, err
	}
	if len(root.Children) == 1 {
		return root.Children[0], nil
	}
	return root, nil
}

func evaluateBuilders(n ASTNode) error {
	switch n := n.(type) {
	case *ASTControlTag:
		for _, child := range n.Body {
			if err := evaluateBuilders(child); err != nil {
				return err
			}
		}
		for _, branch := range n.Branches {
			if err := evaluateBuilders(branch); err != nil {
				return err
			}
		}
		cd, ok := findControlTagDefinition(n.Tag)
		if ok && cd.parser != nil {
			renderer, err := cd.parser(*n)
			if err != nil {
				return err
			}
			n.renderer = renderer
		}
	case *ASTSeq:
		for _, child := range n.Children {
			if error := evaluateBuilders(child); error != nil {
				return error
			}
		}
	}
	return nil
}