Template.stage.funcs.go

package docxplate

import (
	"bytes"
	"encoding/xml"
	"strings"
	"sync"
)

// Collect and trigger placeholders with trigger but unset in `t.params`
// Placeholders with trigger `:empty` must be triggered
// otherwise they are left
func (t *Template) triggerMissingParams(xnode *xmlNode) {
	if t.params == nil {
		return
	}

	var triggerParams ParamList

	xnode.Walk(func(n *xmlNode) {
		if !n.isRowElement() || !n.HaveParams() {
			return
		}
		p := NewParamFromRaw(n.AllContents())
		if p != nil && p.Trigger != nil {
			triggerParams = append(triggerParams, p)
		}
	})

	if triggerParams == nil {
		return
	}

	// make sure not to "tint" original t.params
	_params := t.params
	t.params = triggerParams

	// do stuff only with filtered params
	t.replaceSingleParams(xnode, true)

	// back to original
	t.params = _params
}

// Expand complex placeholders
func (t *Template) expandPlaceholders(xnode *xmlNode) {
	if t.params == nil {
		return
	}
	xnode.WalkWithEnd(func(nrow *xmlNode) bool {
		var once sync.Once
		if nrow.isNew {
			return false
		}
		if !nrow.isRowElement() {
			return false
		}
		var max int
		contents := nrow.AllContents()
		rowParams := rowParams(contents)
		rowPlaceholders := make(map[string]*placeholder)
		for _, rowParam := range rowParams {
			placeholderType := rowPlaceholder
			if len(rowParam.Separator) > 0 {
				placeholderType = inlinePlaceholder
			}

			var formatter, trigger, params string
			if rowParam.Formatter != nil {
				formatter = rowParam.Formatter.String()
			}
			if rowParam.Trigger != nil {
				trigger = rowParam.Trigger.String()
			}
			if rowParam.Formatter != nil || rowParam.Trigger != nil {
				params = " " + formatter + trigger
			}

			paramData := t.params.FindAllByKey(rowParam.AbsoluteKey)
			if len(paramData) == 0 {
				continue
			}
			placeholders := make([]string, paramData[len(paramData)-1].Index)

			for _, param := range paramData {
				placeholders[param.Index-1] = "{{" + param.AbsoluteKey + params + "}}"
			}
			rowPlaceholders[rowParam.RowPlaceholder] = &placeholder{
				Type:         placeholderType,
				Placeholders: placeholders,
				Separator:    strings.TrimLeft(rowParam.Separator, " "),
			}
			if max < len(placeholders) {
				max = len(placeholders)
			}
		}
		nnews := make([]*xmlNode, max)
		for oldPlaceholder, newPlaceholder := range rowPlaceholders {
			switch newPlaceholder.Type {
			case inlinePlaceholder:
				nrow.Walk(func(n *xmlNode) {
					if !inSlice(n.XMLName.Local, []string{"w-t"}) || len(n.Content) == 0 {
						return
					}
					n.Content = bytes.ReplaceAll(n.Content, []byte(oldPlaceholder), []byte(strings.Join(newPlaceholder.Placeholders, newPlaceholder.Separator)))
				})
			case rowPlaceholder:
				defer once.Do(func() {
					nrow.delete()
				})
				for i := max - 1; i >= 0; i-- {
					if nnews[i] == nil {
						nnews[i] = nrow.cloneAndAppend()
					}

					nnews[i].Walk(func(n *xmlNode) {
						if !inSlice(n.XMLName.Local, []string{"w-t"}) || len(n.Content) == 0 {
							return
						}

						replaceData := oldPlaceholder
						if i < len(newPlaceholder.Placeholders) && newPlaceholder.Placeholders[i] != "" {
							replaceData = newPlaceholder.Placeholders[i]
						}
						n.Content = bytes.ReplaceAll(n.Content, []byte(oldPlaceholder), []byte(replaceData))
					})
				}
			}
		}
		return true
	})
}

// Replace single params by type
func (t *Template) replaceSingleParams(xnode *xmlNode, triggerParamOnly bool) {
	paramAbsoluteKeyMap := map[string]*Param{}

	t.params.Walk(func(p *Param) {
		if p.Type != StringParam && p.Type != ImageParam {
			return
		}
		// log.Printf("[%s][%s]", p.AbsoluteKey, p.Value)
		paramAbsoluteKeyMap[p.AbsoluteKey] = p
	})

	xnode.Walk(func(n *xmlNode) {

		for i := range n.Attrs {
			for _, key := range t.GetAttrParam(n.Attrs[i].Value) {
				p, ok := paramAbsoluteKeyMap[key]
				if !ok {
					continue
				}
				n.Attrs[i].Value = string(p.replaceIn([]byte(n.Attrs[i].Value)))
			}
		}

		for _, key := range n.GetContentPrefixList() {
			p, ok := paramAbsoluteKeyMap[key]
			if !ok {
				continue
			}
			t.replaceAndRunTrigger(p, n, triggerParamOnly)
		}

		// for _, p := range t.params {
		// 	p2, ok := paramAbsoluteKeyMap[p.Key]
		// 	if !ok {
		// 		continue
		// 	}
		// 	t.replaceAndRunTrigger(p2, n, triggerParamOnly)
		// }

	})
}

func (t *Template) replaceAndRunTrigger(p *Param, n *xmlNode, triggerParamOnly bool) {
	// log.Printf("replaceAndRunTrigger: %v", p.AbsoluteKey)

	// Trigger: does placeholder have trigger
	if p.Trigger = p.extractTriggerFrom(n.Content); p.Trigger != nil {
		defer func() {
			p.RunTrigger(n)
		}()
	}

	if triggerParamOnly {
		return
	}

	// Repalce by type
	switch p.Type {
	case StringParam:
		// log.Printf("-- StringParam: %v", p.AbsoluteKey)
		if p.Formatter = p.extractFormatter(n.Content); p.Formatter != nil {
			result := p.Formatter.ApplyFormat(p.Formatter.Format, []byte(p.Value))
			p.Value = string(result)
		}
		t.replaceTextParam(n, p)
	case ImageParam:
		t.replaceImageParams(n, p)
	}
}

// Enhance some markup (removed when building XML in the end)
// so easier to find some element
func (t *Template) enhanceMarkup(xnode *xmlNode) {

	// List items - add list item node `w-p` attributes
	// so it's recognized as listitem
	xnode.Walk(func(n *xmlNode) {
		if n.Tag() != "w-p" {
			return
		}

		isListItem, listID := n.IsListItem()
		if !isListItem {
			return
		}
		// n.XMLName.Local = "w-item"
		n.Attrs = append(n.Attrs, xml.Attr{
			Name:  xml.Name{Local: "list-id"},
			Value: listID,
		})

	})
}

// new line variable for reuse
var nl = []byte("\n")

// Enhance content
func (t *Template) enhanceContent(xnode *xmlNode) {

	// New lines from text as docx new lines
	xnode.Walk(func(n *xmlNode) {
		if !n.isSingle() {
			return
		}

		if !bytes.Contains(n.Content, nl) {
			return
		}

		nrow := n.closestUp([]string{"w-p"})
		// log.Printf("NEW LINE: %s..%s [%q] %d new lines", aurora.Cyan(nrow.Tag()), aurora.Blue(n.Tag()), aurora.Yellow(n.Content), bytes.Count(n.Content, nl))

		parts := bytes.Split(n.Content, nl)
		for i, buf := range parts {
			// clone the original node to preserve styles and append the cloned node
			nlast := nrow.cloneAndAppend()

			// first and last node can hold other text node prefixes, skip
			if i >= 1 && i <= len(parts) {
				nlast.Walk(func(n2 *xmlNode) {
					if n2.isSingle() && len(n2.Content) > 0 && !bytes.Contains(n.Content, n2.Content) {
						// delete all other text nodes because we need the same text node
						n2.delete()
					}
				})
			}
			nlast.ReplaceInContents(n.Content, buf)
			// nlast.printTree("NROW")
		}

		// delete the original node after cloning and adjusting (otherwise it shows at the end)
		nrow.delete()

	})
}

// This func is fixing broken placeholders by merging "w-t" nodes.
// "w-p" (Record) can hold multiple "w-r". And "w-r" holts "w-t" node
// -
// If these nodes not fixed than params replace can not be done as
// replacer process nodes one by one
func (t *Template) fixBrokenPlaceholders(xnode *xmlNode) {

	xnode.Walk(func(nrow *xmlNode) {
		if !nrow.isRowElement() {
			return
		}

		var brokenNode *xmlNode
		nrow.Walk(func(n *xmlNode) {
			// broken node state? merge next nodes

			if !n.isSingle() && len(n.AllContents()) > 0 {
				// fmt.Printf("\t RESET -- %s->%s [%s]\n", n.parent.Tag(), aurora.Blue(n.Tag()), aurora.Red(n.AllContents()))
				brokenNode = nil
				return
			}

			if brokenNode != nil {
				// fmt.Printf("OK [%s] + [%s]\n", aurora.Green(brokenNode.AllContents()), aurora.Green(n.AllContents()))
				brokenNode.Content = append(brokenNode.Content, n.AllContents()...)
				// aurora.Magenta("[%s] %v -- %v -- %v -- %v", brokenNode.Content, brokenNode.Tag(), brokenNode.parent.Tag(), brokenNode.parent.parent.Tag(), brokenNode.parent.parent.parent.Tag())
				n.delete()
				return
			}

			if t.matchBrokenLeftPlaceholder(string(n.Content)) {
				// nrow.printTree("BROKEN")
				brokenNode = n
				return
			}

			brokenNode = nil

		})
	})
}