[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\NodeVisitor;
  13  
  14  use Twig\Environment;
  15  use Twig\Node\Expression\AssignNameExpression;
  16  use Twig\Node\Expression\ConstantExpression;
  17  use Twig\Node\Expression\GetAttrExpression;
  18  use Twig\Node\Expression\MethodCallExpression;
  19  use Twig\Node\Expression\NameExpression;
  20  use Twig\Node\ImportNode;
  21  use Twig\Node\ModuleNode;
  22  use Twig\Node\Node;
  23  
  24  /**
  25   * @author Fabien Potencier <fabien@symfony.com>
  26   */
  27  final class MacroAutoImportNodeVisitor implements NodeVisitorInterface
  28  {
  29      private $inAModule = false;
  30      private $hasMacroCalls = false;
  31  
  32      public function enterNode(Node $node, Environment $env)
  33      {
  34          if ($node instanceof ModuleNode) {
  35              $this->inAModule = true;
  36              $this->hasMacroCalls = false;
  37          }
  38  
  39          return $node;
  40      }
  41  
  42      public function leaveNode(Node $node, Environment $env)
  43      {
  44          if ($node instanceof ModuleNode) {
  45              $this->inAModule = false;
  46              if ($this->hasMacroCalls) {
  47                  $node->getNode('constructor_end')->setNode('_auto_macro_import', new ImportNode(new NameExpression('_self', 0), new AssignNameExpression('_self', 0), 0, 'import', true));
  48              }
  49          } elseif ($this->inAModule) {
  50              if (
  51                  $node instanceof GetAttrExpression &&
  52                  $node->getNode('node') instanceof NameExpression &&
  53                  '_self' === $node->getNode('node')->getAttribute('name') &&
  54                  $node->getNode('attribute') instanceof ConstantExpression
  55              ) {
  56                  $this->hasMacroCalls = true;
  57  
  58                  $name = $node->getNode('attribute')->getAttribute('value');
  59                  $node = new MethodCallExpression($node->getNode('node'), 'macro_'.$name, $node->getNode('arguments'), $node->getTemplateLine());
  60                  $node->setAttribute('safe', true);
  61              }
  62          }
  63  
  64          return $node;
  65      }
  66  
  67      public function getPriority()
  68      {
  69          // we must be ran before auto-escaping
  70          return -10;
  71      }
  72  }