src/Twig/TransExtension.php line 107

Open in your IDE?
  1. <?php
  3. namespace App\Twig;
  5. use Symfony\Bridge\Twig\NodeVisitor\TranslationDefaultDomainNodeVisitor;
  6. use Symfony\Bridge\Twig\NodeVisitor\TranslationNodeVisitor;
  7. use Symfony\Bridge\Twig\TokenParser\TransDefaultDomainTokenParser;
  8. use Symfony\Bridge\Twig\TokenParser\TransTokenParser;
  9. use Symfony\Component\Security\Core\Security;
  10. use Symfony\Component\Translation\TranslatableMessage;
  11. use Symfony\Contracts\Translation\TranslatableInterface;
  12. use Symfony\Contracts\Translation\TranslatorInterface;
  13. use Symfony\Contracts\Translation\TranslatorTrait;
  14. use Twig\Environment;
  15. use Twig\Extension\AbstractExtension;
  16. use Twig\TwigFilter;
  17. use Twig\TwigFunction;
  19. class TransExtension extends AbstractExtension
  20. {
  21.     private $translator;
  22.     private $translationNodeVisitor;
  23.     private $security;
  25.     public function __construct(TranslatorInterface $translator = null, TranslationNodeVisitor $translationNodeVisitor = null, Security $security, Environment $twig)
  26.     {
  27.         $this->translator = $translator;
  28.         $this->translationNodeVisitor = $translationNodeVisitor;
  29.         $this->security = $security;
  30.         $this->twig = $twig;
  32.     }
  34.     /**
  35.      * {@inheritdoc}
  36.      */
  37.     public function getFunctions(): array
  38.     {
  39.         return [
  40.             new TwigFunction('t', [$this, 'createTranslatable']),
  41.         ];
  42.     }
  44.     /**
  45.      * {@inheritdoc}
  46.      */
  47.     public function getFilters(): array
  48.     {
  49.         return [
  50.             new TwigFilter('admintrans', [$this, 'trans']),
  51.         ];
  52.     }
  54.     /**
  55.      * {@inheritdoc}
  56.      */
  57.     public function getTokenParsers(): array
  58.     {
  59.         return [
  60.             // {% trans %}Symfony is great!{% endtrans %}
  61.             new TransTokenParser(),
  63.             // {% trans_default_domain "foobar" %}
  64.             new TransDefaultDomainTokenParser(),
  65.         ];
  66.     }
  68.     /**
  69.      * {@inheritdoc}
  70.      */
  71.     public function getNodeVisitors(): array
  72.     {
  73.         return [$this->getTranslationNodeVisitor(), new TranslationDefaultDomainNodeVisitor()];
  74.     }
  76.     public function getTranslationNodeVisitor(): TranslationNodeVisitor
  77.     {
  78.         return $this->translationNodeVisitor ?: $this->translationNodeVisitor = new TranslationNodeVisitor();
  79.     }
  81.     /**
  82.      * @param string|\Stringable|TranslatableInterface|null $message
  83.      * @param array|string $arguments Can be the locale as a string when $message is a TranslatableInterface
  84.      */
  85.     public function trans($message, $arguments = [], string $domain = null, string $locale = null, int $count = null): string
  86.     {
  87.         if ($message instanceof TranslatableInterface) {
  88.             if ([] !== $arguments && !\is_string($arguments)) {
  89.                 throw new \TypeError(sprintf('Argument 2 passed to "%s()" must be a locale passed as a string when the message is a "%s", "%s" given.', __METHOD__, TranslatableInterface::class, get_debug_type($arguments)));
  90.             }
  92.             return $message->trans($this->getTranslator(), $locale ?? (\is_string($arguments) ? $arguments : null));
  93.         }
  95.         if (!\is_array($arguments)) {
  96.             throw new \TypeError(sprintf('Unless the message is a "%s", argument 2 passed to "%s()" must be an array of parameters, "%s" given.', TranslatableInterface::class, __METHOD__, get_debug_type($arguments)));
  97.         }
  99.         if ('' === $message = (string)$message) {
  100.             return '';
  101.         }
  103.         if (null !== $count) {
  104.             $arguments['%count%'] = $count;
  105.         }
  107.         $user = $this->security->getUser();
  108.         $isAdmin = !is_null($user) && in_array('ROLE_ADMIN', $user->getRoles());
  109.         return $this->getTranslator()->trans($message, $arguments, $domain, $locale) . ($isAdmin ? $this->twig->render('components/text-edit-markup.html.twig', [
  110.                 'message' => $message
  111.             ]) : '');
  112.     }
  114.     public function getTranslator(): TranslatorInterface
  115.     {
  116.         if (null === $this->translator) {
  117.             if (!interface_exists(TranslatorInterface::class)) {
  118.                 throw new \LogicException(sprintf('You cannot use the "%s" if the Translation Contracts are not available. Try running "composer require symfony/translation".', __CLASS__));
  119.             }
  121.             $this->translator = new class() implements TranslatorInterface {
  122.                 use TranslatorTrait;
  123.             };
  124.         }
  126.         return $this->translator;
  127.     }
  129.     public function createTranslatable(string $message, array $parameters = [], string $domain = null): TranslatableMessage
  130.     {
  131.         if (!class_exists(TranslatableMessage::class)) {
  132.             throw new \LogicException(sprintf('You cannot use the "%s" as the Translation Component is not installed. Try running "composer require symfony/translation".', __CLASS__));
  133.         }
  135.         return new TranslatableMessage($message, $parameters, $domain);
  136.     }
  138. }