vendor/symfony/twig-bridge/NodeVisitor/TranslationDefaultDomainNodeVisitor.php line 37

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of the Symfony package.
  5.  *
  6.  * (c) Fabien Potencier <fabien@symfony.com>
  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 Symfony\Bridge\Twig\NodeVisitor;
  14. use Symfony\Bridge\Twig\Node\TransDefaultDomainNode;
  15. use Symfony\Bridge\Twig\Node\TransNode;
  16. use Twig\Environment;
  17. use Twig\Node\BlockNode;
  18. use Twig\Node\Expression\ArrayExpression;
  19. use Twig\Node\Expression\AssignNameExpression;
  20. use Twig\Node\Expression\ConstantExpression;
  21. use Twig\Node\Expression\FilterExpression;
  22. use Twig\Node\Expression\NameExpression;
  23. use Twig\Node\ModuleNode;
  24. use Twig\Node\Node;
  25. use Twig\Node\SetNode;
  26. use Twig\NodeVisitor\AbstractNodeVisitor;
  28. /**
  29.  * @author Fabien Potencier <fabien@symfony.com>
  30.  */
  31. final class TranslationDefaultDomainNodeVisitor extends AbstractNodeVisitor
  32. {
  33.     private Scope $scope;
  35.     public function __construct()
  36.     {
  37.         $this->scope = new Scope();
  38.     }
  40.     /**
  41.      * {@inheritdoc}
  42.      */
  43.     protected function doEnterNode(Node $node, Environment $env): Node
  44.     {
  45.         if ($node instanceof BlockNode || $node instanceof ModuleNode) {
  46.             $this->scope = $this->scope->enter();
  47.         }
  49.         if ($node instanceof TransDefaultDomainNode) {
  50.             if ($node->getNode('expr') instanceof ConstantExpression) {
  51.                 $this->scope->set('domain', $node->getNode('expr'));
  53.                 return $node;
  54.             } else {
  55.                 $var = $this->getVarName();
  56.                 $name = new AssignNameExpression($var, $node->getTemplateLine());
  57.                 $this->scope->set('domain', new NameExpression($var, $node->getTemplateLine()));
  59.                 return new SetNode(false, new Node([$name]), new Node([$node->getNode('expr')]), $node->getTemplateLine());
  60.             }
  61.         }
  63.         if (!$this->scope->has('domain')) {
  64.             return $node;
  65.         }
  67.         if ($node instanceof FilterExpression && 'trans' === $node->getNode('filter')->getAttribute('value')) {
  68.             $arguments = $node->getNode('arguments');
  69.             if ($this->isNamedArguments($arguments)) {
  70.                 if (!$arguments->hasNode('domain') && !$arguments->hasNode(1)) {
  71.                     $arguments->setNode('domain', $this->scope->get('domain'));
  72.                 }
  73.             } elseif (!$arguments->hasNode(1)) {
  74.                 if (!$arguments->hasNode(0)) {
  75.                     $arguments->setNode(0, new ArrayExpression([], $node->getTemplateLine()));
  76.                 }
  78.                 $arguments->setNode(1, $this->scope->get('domain'));
  79.             }
  80.         } elseif ($node instanceof TransNode) {
  81.             if (!$node->hasNode('domain')) {
  82.                 $node->setNode('domain', $this->scope->get('domain'));
  83.             }
  84.         }
  86.         return $node;
  87.     }
  89.     /**
  90.      * {@inheritdoc}
  91.      */
  92.     protected function doLeaveNode(Node $node, Environment $env): ?Node
  93.     {
  94.         if ($node instanceof TransDefaultDomainNode) {
  95.             return null;
  96.         }
  98.         if ($node instanceof BlockNode || $node instanceof ModuleNode) {
  99.             $this->scope = $this->scope->leave();
  100.         }
  102.         return $node;
  103.     }
  105.     /**
  106.      * {@inheritdoc}
  107.      */
  108.     public function getPriority(): int
  109.     {
  110.         return -10;
  111.     }
  113.     private function isNamedArguments(Node $arguments): bool
  114.     {
  115.         foreach ($arguments as $name => $node) {
  116.             if (!\is_int($name)) {
  117.                 return true;
  118.             }
  119.         }
  121.         return false;
  122.     }
  124.     private function getVarName(): string
  125.     {
  126.         return sprintf('__internal_%s', hash('sha256', uniqid(mt_rand(), true), false));
  127.     }
  128. }