123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142 |
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Bridge\Twig\NodeVisitor;
- use Symfony\Bridge\Twig\Node\TransNode;
- use Symfony\Bridge\Twig\Node\TransDefaultDomainNode;
- use Twig\Environment;
- use Twig\Node\BlockNode;
- use Twig\Node\Expression\ArrayExpression;
- use Twig\Node\Expression\AssignNameExpression;
- use Twig\Node\Expression\ConstantExpression;
- use Twig\Node\Expression\FilterExpression;
- use Twig\Node\Expression\NameExpression;
- use Twig\Node\ModuleNode;
- use Twig\Node\Node;
- use Twig\Node\SetNode;
- use Twig\NodeVisitor\AbstractNodeVisitor;
- /**
- * TranslationDefaultDomainNodeVisitor.
- *
- * @author Fabien Potencier <fabien@symfony.com>
- */
- class TranslationDefaultDomainNodeVisitor extends AbstractNodeVisitor
- {
- /**
- * @var Scope
- */
- private $scope;
- /**
- * Constructor.
- */
- public function __construct()
- {
- $this->scope = new Scope();
- }
- /**
- * {@inheritdoc}
- */
- protected function doEnterNode(Node $node, Environment $env)
- {
- if ($node instanceof BlockNode || $node instanceof ModuleNode) {
- $this->scope = $this->scope->enter();
- }
- if ($node instanceof TransDefaultDomainNode) {
- if ($node->getNode('expr') instanceof ConstantExpression) {
- $this->scope->set('domain', $node->getNode('expr'));
- return $node;
- } else {
- $var = $this->getVarName();
- $name = new AssignNameExpression($var, $node->getTemplateLine());
- $this->scope->set('domain', new NameExpression($var, $node->getTemplateLine()));
- return new SetNode(false, new Node(array($name)), new Node(array($node->getNode('expr'))), $node->getTemplateLine());
- }
- }
- if (!$this->scope->has('domain')) {
- return $node;
- }
- if ($node instanceof FilterExpression && in_array($node->getNode('filter')->getAttribute('value'), array('trans', 'transchoice'))) {
- $arguments = $node->getNode('arguments');
- $ind = 'trans' === $node->getNode('filter')->getAttribute('value') ? 1 : 2;
- if ($this->isNamedArguments($arguments)) {
- if (!$arguments->hasNode('domain') && !$arguments->hasNode($ind)) {
- $arguments->setNode('domain', $this->scope->get('domain'));
- }
- } else {
- if (!$arguments->hasNode($ind)) {
- if (!$arguments->hasNode($ind - 1)) {
- $arguments->setNode($ind - 1, new ArrayExpression(array(), $node->getTemplateLine()));
- }
- $arguments->setNode($ind, $this->scope->get('domain'));
- }
- }
- } elseif ($node instanceof TransNode) {
- if (!$node->hasNode('domain')) {
- $node->setNode('domain', $this->scope->get('domain'));
- }
- }
- return $node;
- }
- /**
- * {@inheritdoc}
- */
- protected function doLeaveNode(Node $node, Environment $env)
- {
- if ($node instanceof TransDefaultDomainNode) {
- return false;
- }
- if ($node instanceof BlockNode || $node instanceof ModuleNode) {
- $this->scope = $this->scope->leave();
- }
- return $node;
- }
- /**
- * {@inheritdoc}
- */
- public function getPriority()
- {
- return -10;
- }
- /**
- * @return bool
- */
- private function isNamedArguments($arguments)
- {
- foreach ($arguments as $name => $node) {
- if (!is_int($name)) {
- return true;
- }
- }
- return false;
- }
- private function getVarName()
- {
- return sprintf('__internal_%s', hash('sha256', uniqid(mt_rand(), true), false));
- }
- }
|