[Summary view] [Print] [Text view]

   1  <?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  
  12  namespace Twig\Node;
  13  
  14  use Twig\Compiler;
  15  use Twig\Node\Expression\ConstantExpression;
  16  
  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);
  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)
  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  }
 106  
 107  class_alias('Twig\Node\SetNode', 'Twig_Node_Set');