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

  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): void
  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. }