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

  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.     protected function doEnterNode(Node $node, Environment $env): Node
  41.     {
  42.         if ($node instanceof BlockNode || $node instanceof ModuleNode) {
  43.             $this->scope = $this->scope->enter();
  44.         }
  46.         if ($node instanceof TransDefaultDomainNode) {
  47.             if ($node->getNode('expr') instanceof ConstantExpression) {
  48.                 $this->scope->set('domain', $node->getNode('expr'));
  50.                 return $node;
  51.             } else {
  52.                 $var = $this->getVarName();
  53.                 $name = new AssignNameExpression($var, $node->getTemplateLine());
  54.                 $this->scope->set('domain', new NameExpression($var, $node->getTemplateLine()));
  56.                 return new SetNode(false, new Node([$name]), new Node([$node->getNode('expr')]), $node->getTemplateLine());
  57.             }
  58.         }
  60.         if (!$this->scope->has('domain')) {
  61.             return $node;
  62.         }
  64.         if ($node instanceof FilterExpression && 'trans' === $node->getNode('filter')->getAttribute('value')) {
  65.             $arguments = $node->getNode('arguments');
  66.             if ($this->isNamedArguments($arguments)) {
  67.                 if (!$arguments->hasNode('domain') && !$arguments->hasNode(1)) {
  68.                     $arguments->setNode('domain', $this->scope->get('domain'));
  69.                 }
  70.             } elseif (!$arguments->hasNode(1)) {
  71.                 if (!$arguments->hasNode(0)) {
  72.                     $arguments->setNode(0, new ArrayExpression([], $node->getTemplateLine()));
  73.                 }
  75.                 $arguments->setNode(1, $this->scope->get('domain'));
  76.             }
  77.         } elseif ($node instanceof TransNode) {
  78.             if (!$node->hasNode('domain')) {
  79.                 $node->setNode('domain', $this->scope->get('domain'));
  80.             }
  81.         }
  83.         return $node;
  84.     }
  86.     protected function doLeaveNode(Node $node, Environment $env): ?Node
  87.     {
  88.         if ($node instanceof TransDefaultDomainNode) {
  89.             return null;
  90.         }
  92.         if ($node instanceof BlockNode || $node instanceof ModuleNode) {
  93.             $this->scope = $this->scope->leave();
  94.         }
  96.         return $node;
  97.     }
  99.     public function getPriority(): int
  100.     {
  101.         return -10;
  102.     }
  104.     private function isNamedArguments(Node $arguments): bool
  105.     {
  106.         foreach ($arguments as $name => $node) {
  107.             if (!\is_int($name)) {
  108.                 return true;
  109.             }
  110.         }
  112.         return false;
  113.     }
  115.     private function getVarName(): string
  116.     {
  117.         return sprintf('__internal_%s', hash('sha256', uniqid(mt_rand(), true), false));
  118.     }
  119. }