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\Translation\MessageCatalogue;

/**
 * Upload all translations for specified locales from given bundle to memcache
 *
 * @author Viktor Novikov <viktor.novikov95@gmail.com>
 */
class ImportCommand extends ContainerAwareCommand
{
    /**
     * {@inheritdoc}
     */
    protected function configure()
    {
        $this
            ->setName('uber:translations:import')
            ->setDefinition(array(
                new InputArgument('bundle', InputArgument::REQUIRED, 'Name of the bundle'),
            ))
            ->setDescription('Import translations into memcached')
            ->setHelp("
The <info>uber:translations:import</info> command imports translations of given bundle into memcache:

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

Command example:

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

            ");
    }

    /**
     * {@inheritdoc}
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $container = $this->getContainer();
        $locales = $container->getParameter('sleepness_uber_translation.supported_locales'); // prepare locales from parameters
        $bundle = $container->get('kernel')->getBundle($input->getArgument('bundle')); // get bundle props

        if (!is_dir($bundle->getPath() . '/Resources/translations')) {
            $output->writeln("\033[37;43m There is no folder with translations in " . $input->getArgument('bundle') . " \033[0m");
            return;
        }

        $loader = $container->get('translation.loader'); // get translator loader
        $uberMemcached = $container->get('uber.memcached'); // get uber memcached
        $catalogues = array(); // prepare array for catalogues

        foreach ($locales as $locale) { // run through locales
            if (!preg_match('/^[a-z]{2}_[a-zA-Z]{2}$|[a-z]{2}/', $locale)) {
                $output->writeln("\033[37;43m Make sure you define all locales properly \033[0m   \n");
                return;
            }
            $currentCatalogue = new MessageCatalogue($locale); // Load defined messages for given locale
            $loader->loadMessages($bundle->getPath() . '/Resources/translations', $currentCatalogue); // load messages from catalogue
            $catalogues[$locale] = $currentCatalogue; // pass MessageCatalogue instance into $catalogues array
        }

        foreach ($catalogues as $locale => $catalogue) { // run over all catalogues
            $catalogueMessages = $catalogue->all(); // get messages from current catalogue
            $memcacheMessages = $uberMemcached->getItem($locale); // get existing messages from the memcache by locale

            if (!$memcacheMessages) { // if for now no messages in memcache
                $uberMemcached->addItem($locale, $catalogueMessages); // just upload them
            } else {
                foreach ($memcacheMessages as $domain => $messagesArray) { // run over messages in cache
                    foreach ($catalogueMessages as $d => $catMess) { // run over all in catalogue
                        if ($domain == $d) { // if domains equals
                            foreach ($messagesArray as $messKey => $trans) { //run over messages
                                foreach ($catMess as $ymlKey => $transl) {
                                    if ($messKey == $ymlKey) { // if keys equals
                                        unset($catalogueMessages[$d][$messKey]); // unset the value because we don`t want to repeated values
                                    }
                                }
                            }
                        }
                    }
                }
                $mergedMessages = array_merge_recursive($memcacheMessages, $catalogueMessages); // merge recursive message arrays
                $uberMemcached->addItem($locale, $mergedMessages); // set merged messages to memcache
            }
        }

        $output->writeln("\033[37;42m Translations from " . $input->getArgument('bundle') . " imported successfully! \033[0m");
    }
}