File: tags_ifchanged.go

package info (click to toggle)
golang-github-flosch-pongo2.v4 4.0.2-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bookworm-backports, trixie
  • size: 860 kB
  • sloc: makefile: 3
file content (116 lines) | stat: -rw-r--r-- 2,600 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
package pongo2

import (
	"bytes"
)

type tagIfchangedNode struct {
	watchedExpr []IEvaluator
	lastValues  []*Value
	lastContent []byte
	thenWrapper *NodeWrapper
	elseWrapper *NodeWrapper
}

func (node *tagIfchangedNode) Execute(ctx *ExecutionContext, writer TemplateWriter) *Error {
	if len(node.watchedExpr) == 0 {
		// Check against own rendered body

		buf := bytes.NewBuffer(make([]byte, 0, 1024)) // 1 KiB
		err := node.thenWrapper.Execute(ctx, buf)
		if err != nil {
			return err
		}

		bufBytes := buf.Bytes()
		if !bytes.Equal(node.lastContent, bufBytes) {
			// Rendered content changed, output it
			writer.Write(bufBytes)
			node.lastContent = bufBytes
		}
	} else {
		nowValues := make([]*Value, 0, len(node.watchedExpr))
		for _, expr := range node.watchedExpr {
			val, err := expr.Evaluate(ctx)
			if err != nil {
				return err
			}
			nowValues = append(nowValues, val)
		}

		// Compare old to new values now
		changed := len(node.lastValues) == 0

		for idx, oldVal := range node.lastValues {
			if !oldVal.EqualValueTo(nowValues[idx]) {
				changed = true
				break // we can stop here because ONE value changed
			}
		}

		node.lastValues = nowValues

		if changed {
			// Render thenWrapper
			err := node.thenWrapper.Execute(ctx, writer)
			if err != nil {
				return err
			}
		} else {
			// Render elseWrapper
			err := node.elseWrapper.Execute(ctx, writer)
			if err != nil {
				return err
			}
		}
	}

	return nil
}

func tagIfchangedParser(doc *Parser, start *Token, arguments *Parser) (INodeTag, *Error) {
	ifchangedNode := &tagIfchangedNode{}

	for arguments.Remaining() > 0 {
		// Parse condition
		expr, err := arguments.ParseExpression()
		if err != nil {
			return nil, err
		}
		ifchangedNode.watchedExpr = append(ifchangedNode.watchedExpr, expr)
	}

	if arguments.Remaining() > 0 {
		return nil, arguments.Error("Ifchanged-arguments are malformed.", nil)
	}

	// Wrap then/else-blocks
	wrapper, endargs, err := doc.WrapUntilTag("else", "endifchanged")
	if err != nil {
		return nil, err
	}
	ifchangedNode.thenWrapper = wrapper

	if endargs.Count() > 0 {
		return nil, endargs.Error("Arguments not allowed here.", nil)
	}

	if wrapper.Endtag == "else" {
		// if there's an else in the if-statement, we need the else-Block as well
		wrapper, endargs, err = doc.WrapUntilTag("endifchanged")
		if err != nil {
			return nil, err
		}
		ifchangedNode.elseWrapper = wrapper

		if endargs.Count() > 0 {
			return nil, endargs.Error("Arguments not allowed here.", nil)
		}
	}

	return ifchangedNode, nil
}

func init() {
	RegisterTag("ifchanged", tagIfchangedParser)
}