blob: 96b6bd8bf58b05183bbd98e7f5e03c3e2954dd6d [file] [log] [blame]
Matthias Andreas Benkard12a57352021-12-28 18:02:04 +01001<?php
2
3/*
4 * This file is part of Twig.
5 *
6 * (c) Fabien Potencier
7 *
8 * For the full copyright and license information, please view the LICENSE
9 * file that was distributed with this source code.
10 */
11
12namespace Twig\Node;
13
14use Twig\Compiler;
15use Twig\Node\Expression\ConstantExpression;
16
17/**
18 * Represents a set node.
19 *
20 * @author Fabien Potencier <fabien@symfony.com>
21 */
22class SetNode extends Node implements NodeCaptureInterface
23{
24 public function __construct(bool $capture, Node $names, Node $values, int $lineno, string $tag = null)
25 {
26 parent::__construct(['names' => $names, 'values' => $values], ['capture' => $capture, 'safe' => false], $lineno, $tag);
27
28 /*
29 * Optimizes the node when capture is used for a large block of text.
30 *
31 * {% set foo %}foo{% endset %} is compiled to $context['foo'] = new Twig\Markup("foo");
32 */
33 if ($this->getAttribute('capture')) {
34 $this->setAttribute('safe', true);
35
36 $values = $this->getNode('values');
37 if ($values instanceof TextNode) {
38 $this->setNode('values', new ConstantExpression($values->getAttribute('data'), $values->getTemplateLine()));
39 $this->setAttribute('capture', false);
40 }
41 }
42 }
43
44 public function compile(Compiler $compiler): void
45 {
46 $compiler->addDebugInfo($this);
47
48 if (\count($this->getNode('names')) > 1) {
49 $compiler->write('list(');
50 foreach ($this->getNode('names') as $idx => $node) {
51 if ($idx) {
52 $compiler->raw(', ');
53 }
54
55 $compiler->subcompile($node);
56 }
57 $compiler->raw(')');
58 } else {
59 if ($this->getAttribute('capture')) {
60 if ($compiler->getEnvironment()->isDebug()) {
61 $compiler->write("ob_start();\n");
62 } else {
63 $compiler->write("ob_start(function () { return ''; });\n");
64 }
65 $compiler
66 ->subcompile($this->getNode('values'))
67 ;
68 }
69
70 $compiler->subcompile($this->getNode('names'), false);
71
72 if ($this->getAttribute('capture')) {
73 $compiler->raw(" = ('' === \$tmp = ob_get_clean()) ? '' : new Markup(\$tmp, \$this->env->getCharset())");
74 }
75 }
76
77 if (!$this->getAttribute('capture')) {
78 $compiler->raw(' = ');
79
80 if (\count($this->getNode('names')) > 1) {
81 $compiler->write('[');
82 foreach ($this->getNode('values') as $idx => $value) {
83 if ($idx) {
84 $compiler->raw(', ');
85 }
86
87 $compiler->subcompile($value);
88 }
89 $compiler->raw(']');
90 } else {
91 if ($this->getAttribute('safe')) {
92 $compiler
93 ->raw("('' === \$tmp = ")
94 ->subcompile($this->getNode('values'))
95 ->raw(") ? '' : new Markup(\$tmp, \$this->env->getCharset())")
96 ;
97 } else {
98 $compiler->subcompile($this->getNode('values'));
99 }
100 }
101 }
102
103 $compiler->raw(";\n");
104 }
105}