Add support for JMSTranslatorBundle

Resources/config/core.xml

@@ -47,6 +47,12 @@
         <service id="sonata.admin.label.strategy.underscore" class="Sonata\AdminBundle\Translator\UnderscoreLabelTranslatorStrategy" />
         <service id="sonata.admin.label.strategy.form_component" class="Sonata\AdminBundle\Translator\FormLabelTranslatorStrategy" />
 
+        <!-- Translation extractor -->
+        <service id="sonata.admin.translator.extractor.jms_translator_bundle" class="Sonata\AdminBundle\Translator\Extractor\JMSTranslatorBundle\AdminExtractor" >
+            <tag name="jms_translation.extractor" alias="sonata_admin"/>
+            <argument type="service" id="logger" />
+            <argument type="service" id="sonata.admin.pool" />
+        </service>
     </services>
 </container>
 

Translator/Extractor/JMSTranslatorBundle/AdminExtractor.php

@@ -0,0 +1,197 @@
+<?php
+
+namespace Sonata\AdminBundle\Translator\Extractor\JMSTranslatorBundle;
+
+use Symfony\Component\Translation\TranslatorInterface;
+use Symfony\Component\HttpKernel\Log\LoggerInterface;
+
+use Sonata\AdminBundle\Admin\Pool;
+use Sonata\AdminBundle\Security\Handler\SecurityHandlerInterface;
+use Sonata\AdminBundle\Admin\AdminInterface;
+use Sonata\AdminBundle\Translator\LabelTranslatorStrategyInterface;
+
+use JMS\TranslationBundle\Translation\ExtractorInterface;
+use JMS\TranslationBundle\Model\MessageCatalogue;
+use JMS\TranslationBundle\Model\Message;
+use JMS\TranslationBundle\Model\FileSource;
+use JMS\TranslationBundle\Logger\LoggerAwareExtractorInterface;
+
+class AdminExtractor implements ExtractorInterface, TranslatorInterface, SecurityHandlerInterface, LabelTranslatorStrategyInterface
+{
+    private $logger;
+    private $adminPool;
+    private $catalogue;
+    private $translator;
+    private $labelStrategy;
+    private $domain;
+
+    /**
+     * @param \Symfony\Component\HttpKernel\Log\LoggerInterface $logger
+     * @param \Sonata\AdminBundle\Admin\Pool $adminPool
+     */
+    public function __construct(LoggerInterface $logger, Pool $adminPool)
+    {
+        $this->logger     = $logger;
+        $this->adminPool  = $adminPool;
+
+        // state variable
+        $this->catalogue  = false;
+        $this->translator = false;
+        $this->labelStrategy = false;
+        $this->domain      = false;
+    }
+
+    /**
+     * @param \Symfony\Component\HttpKernel\Log\LoggerInterface $logger
+     */
+    public function setLogger(LoggerInterface $logger)
+    {
+        $this->logger = $logger;
+    }
+
+    /**
+     * @return bool
+     * @throws \Exception|\RuntimeException
+     */
+    public function extract()
+    {
+        if ($this->catalogue) {
+            throw new \RuntimeException('Invalid state');
+        }
+
+        $this->catalogue = new MessageCatalogue;
+
+        foreach ($this->adminPool->getAdminServiceIds() as $id) {
+            $admin = $this->getAdmin($id);
+
+            $this->translator    = $admin->getTranslator();
+            $this->labelStrategy = $admin->getLabelTranslatorStrategy();
+            $this->domain        = $admin->getTranslationDomain();
+
+            $admin->setTranslator($this);
+            $admin->setSecurityHandler($this);
+            $admin->setLabelTranslatorStrategy($this);
+
+
+//            foreach ($admin->getChildren() as $child) {
+//                $child->setTranslator($this);
+//            }
+
+            // call the different public method
+            $methods = array(
+                'getShow'         => array(array()),
+                'getDatagrid'     => array(array()),
+                'getList'         => array(array()),
+                'getForm'         => array(array()),
+                'getBreadcrumbs'  => array(
+                    array('list'),
+                    array('edit'),
+                    array('create'),
+                    array('update'),
+                    array('batch'),
+                    array('delete'),
+                )
+            );
+
+            $this->logger->info(sprintf('Retrieving message from admin:%s - class: %s', $admin->getCode(), get_class($admin)));
+
+            foreach ($methods as $method => $calls) {
+                foreach ($calls as $args) {
+                    try {
+                        call_user_func_array(array($admin, $method), $args);
+                    } catch (\Exception $e) {
+                        $this->logger->err(sprintf('ERROR : admin:%s - Raise an exception : %s', $admin->getCode(), $e->getMessage()));
+
+                        throw $e;
+                    }
+                }
+            }
+        }
+
+        $catalogue = $this->catalogue;
+        $this->catalogue = false;
+
+        return $catalogue;
+    }
+
+    /**
+     * @param string $id
+     * @return \Sonata\AdminBundle\Admin\AdminInterface
+     */
+    private function getAdmin($id)
+    {
+        return $this->adminPool->getContainer()->get($id);
+    }
+
+    /**
+     * @param $id
+     * @param $domain
+     */
+    private function addMessage($id, $domain)
+    {
+        $message = new Message($id, $domain);
+
+//        $this->logger->debug(sprintf('extract: %s - domain:%s', $id, $domain));
+
+        $trace = debug_backtrace(false);
+        if (isset($trace[1]['file'])) {
+            $message->addSource(new FileSource($trace[1]['file']));
+        }
+
+        $this->catalogue->add($message);
+    }
+
+    public function trans($id, array $parameters = array(), $domain = null, $locale = null)
+    {
+        $this->addMessage($id, $domain);
+    }
+
+    public function transChoice($id, $number, array $parameters = array(), $domain = null, $locale = null)
+    {
+        $this->addMessage($id, $domain);
+    }
+
+    public function setLocale($locale)
+    {
+        $this->translator->setLocale($locale);
+    }
+
+    public function getLocale()
+    {
+        return $this->translator->getLocale();
+    }
+
+    /**
+     * @param string|array $attributes
+     * @param null $object
+     * @return boolean
+     */
+    public function isGranted(AdminInterface $admin, $attributes, $object = null)
+    {
+        return true;
+    }
+
+    /**
+     * @param \Sonata\AdminBundle\Admin\AdminInterface $admin
+     * @return void
+     */
+    public function buildSecurityInformation(AdminInterface $admin)
+    {
+
+    }
+
+    /**
+     * @param $label
+     * @param $context
+     * @param $type
+     * @return string
+     */
+    public function getLabel($label, $context = '', $type = '')
+    {
+        $label = $this->labelStrategy->getLabel($label, $context, $type);
+
+        $this->addMessage($label, $this->domain);
+
+        return $label;
+    }
+}