vendor/twig/twig/src/Node/SetNode.php line 24

Open in your IDE?
  1. <?php
  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. */
  12. namespace Twig\Node;
  14. use Twig\Compiler;
  15. use Twig\Node\Expression\ConstantExpression;
  17. /**
  18. * Represents a set node.
  19. *
  20. * @author Fabien Potencier <fabien@symfony.com>
  21. */
  22. class 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);
  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);
  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. }
  44. public function compile(Compiler $compiler)
  45. {
  46. $compiler->addDebugInfo($this);
  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. }
  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. }
  70. $compiler->subcompile($this->getNode('names'), false);
  72. if ($this->getAttribute('capture')) {
  73. $compiler->raw(" = ('' === \$tmp = ob_get_clean()) ? '' : new Markup(\$tmp, \$this->env->getCharset())");
  74. }
  75. }
  77. if (!$this->getAttribute('capture')) {
  78. $compiler->raw(' = ');
  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. }
  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. }
  103. $compiler->raw(";\n");
  104. }
  105. }
  107. class_alias('Twig\Node\SetNode', 'Twig_Node_Set');