Sleepness/UberTranslationBundle

<?php

namespace Sleepness\UberTranslationBundle\Command;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;
use Symfony\Component\Console\Input\InputArgument;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Yaml\Dumper;

/**
 * Export all translations from memcache into YAML files of given bundle
 *
 * @author Alexandr Zhulev <alexandrzhulev@gmail.com>
 * @author Viktor Novikov <viktor.novikov95@gmail.com>
 */
class ExportCommand extends ContainerAwareCommand
{
    /**
     * {@inheritdoc}
     */
    protected function configure()
    {
        $this
            ->setName('uber:translations:export')
            ->setDefinition(array(
                new InputArgument('bundle', InputArgument::REQUIRED, 'Name of the bundle'),
            ))
            ->setDescription('Export translations into YAML files')
            ->setHelp("
The <info>uber:translations:export</info> command exports translations from memcache into YAML files of given bundle:

  <info>./app/console uber:translations:export VendorNameYourBundle</info>

Command example:

  <info>./app/console uber:translations:export AcmeDemoBundle</info>

            ");
    }

    /**
     * {@inheritdoc}
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $container = $this->getContainer();
        $bundleName = $input->getArgument('bundle');
        $bundlePath = $container->get('kernel')->getBundle($bundleName)->getPath();
        $translationDirPath = $bundlePath . '/Resources/translations/';
        if (!file_exists($translationDirPath)) {
            mkdir($translationDirPath, 0777);
        }
        $uberMemcached = $container->get('uber.memcached');
        $locales = $uberMemcached->getAllKeys();
        $response = "\033[37;43m No translations in Memcache! \033[0m";
        $numberOfLocales = 0;
        foreach ($locales as $locale) {
            $numberOfLocales++;
            $memcacheMessages = $uberMemcached->getItem($locale);
            foreach ($memcacheMessages as $domain => $messagesArray) {
                $yamlArr = array();
                $indent = 0;
                foreach ($messagesArray as $key => $value) {
                    $delimArray = explode('.', $key);
                    $indent = count($delimArray);
                    array_push($delimArray, $value);
                    $expArray = $this->expand($delimArray);
                    foreach ($expArray as $expArrayKey => $expArrayVal) {
                        if (array_key_exists($expArrayKey, $yamlArr)) {
                            $yamlArr[$expArrayKey] = array_replace_recursive($yamlArr[$expArrayKey], $expArrayVal);
                        } else {
                            $yamlArr[$expArrayKey] = $expArrayVal;
                        }
                    }
                }
                $dumper = new Dumper();
                $yaml = $dumper->dump($yamlArr, $indent);
                file_put_contents($translationDirPath . '/' . $domain . '.' . $locale . '.yml', $yaml);
            }
        }
        if (!empty($numberOfLocales)) {
            $response = "\033[37;42m Translations exported successfully into " . $bundleName . "/Resources/translations/ ! \033[0m";
        }
        $output->writeln($response);
    }

    /**
     * Expand array to multidimensional array
     *
     * @param $array
     * @param int $level
     * @return array
     */
    private function expand($array, $level = 0)
    {
        $result = array();
        $next = $level + 1;
        $value = (count($array) == $level + 2) ? $array[$next] : $this->expand($array, $next);
        $result[$array[$level]] = $value;

        return $result;
    }
}