<?php

/*

 * This file is part of the FOSUserBundle package.

 *

 * (c) FriendsOfSymfony <http://friendsofsymfony.github.com/>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

namespace FOS\UserBundle\DependencyInjection;

use Symfony\Component\Config\Definition\Processor;

use Symfony\Component\HttpKernel\DependencyInjection\Extension;

use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;

use Symfony\Component\DependencyInjection\ContainerBuilder;

use Symfony\Component\DependencyInjection\Alias;

use Symfony\Component\Config\FileLocator;

class FOSUserExtension extends Extension

{

    public function load(array $configs, ContainerBuilder $container)

    {

        $processor = new Processor();

        $configuration = new Configuration();

        $config = $processor->processConfiguration($configuration, $configs);

        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));

        if (!in_array(strtolower($config['db_driver']), array('orm', 'mongodb', 'couchdb'))) {

            throw new \InvalidArgumentException(sprintf('Invalid db driver "%s".', $config['db_driver']));

        }

        $loader->load(sprintf('%s.xml', $config['db_driver']));

        foreach (array('validator', 'security', 'util', 'mailer') as $basename) {

            $loader->load(sprintf('%s.xml', $basename));

        }

        $container->setAlias('fos_user.mailer', $config['service']['mailer']);

        $container->setAlias('fos_user.util.email_canonicalizer', $config['service']['email_canonicalizer']);

        $container->setAlias('fos_user.util.username_canonicalizer', $config['service']['username_canonicalizer']);

        $container->setAlias('fos_user.user_manager', $config['service']['user_manager']);

        if ($config['use_listener']) {

            switch ($config['db_driver']) {

                case 'orm':

                    $container->getDefinition('fos_user.user_listener')->addTag('doctrine.event_subscriber');

                    break;

                case 'mongodb':

                    $container->getDefinition('fos_user.user_listener')->addTag('doctrine.common.event_subscriber');

                    break;

                case 'couchdb':

                    $container->getDefinition('fos_user.user_listener')->addTag('doctrine_couchdb.event_subscriber');

                    break;

                default:

                    break;

            }

        }

        if ($config['use_username_form_type']) {

            $loader->load('username_form_type.xml');

        }

        $this->remapParametersNamespaces($config, $container, array(

            ''          => array(

                'firewall_name' => 'fos_user.firewall_name',

                'model_manager_name' => 'fos_user.model_manager_name',

                'user_class' => 'fos_user.model.user.class',

            ),

            'encoder'   => 'fos_user.encoder.%s',

            'template'  => 'fos_user.template.%s',

        ));

        $container->setParameter(

            'fos_user.registration.confirmation.from_email',

            array($config['from_email']['address'] => $config['from_email']['sender_name'])

        );

        $container->setParameter(

            'fos_user.resetting.email.from_email',

            array($config['from_email']['address'] => $config['from_email']['sender_name'])

        );

        // handle the MongoDB document manager name in a specific way as it does not have a registry to make it easy

        // TODO: change it if https://github.com/symfony/DoctrineMongoDBBundle/pull/31 is merged

        if ('mongodb' === $config['db_driver']) {

            if (null === $config['model_manager_name']) {

                $container->setAlias(new Alias('fos_user.document_manager', false), 'doctrine.odm.mongodb.document_manager');

            } else {

                $container->setAlias(new Alias('fos_user.document_manager', false), sprintf('doctrine.odm.%s_mongodb.document_manager', $config['model_manager_name']));

            }

        }

        if (!empty($config['profile'])) {

            $loader->load('profile.xml');

            $container->setAlias('fos_user.profile.form.handler', $config['profile']['form']['handler']);

            unset($config['profile']['form']['handler']);

            $this->remapParametersNamespaces($config['profile'], $container, array(

                'form' => 'fos_user.profile.form.%s',

            ));

        }

        if (!empty($config['registration'])) {

            $loader->load('registration.xml');

            $container->setAlias('fos_user.registration.form.handler', $config['registration']['form']['handler']);

            unset($config['registration']['form']['handler']);

            if (!empty($config['registration']['confirmation']['from_email'])) {

                $container->setParameter(

                    'fos_user.registration.confirmation.from_email',

                    array($config['registration']['confirmation']['from_email']['address'] => $config['registration']['confirmation']['from_email']['sender_name'])

                );

            }

            unset($config['registration']['confirmation']['from_email']);

            $this->remapParametersNamespaces($config['registration'], $container, array(

                'confirmation' => 'fos_user.registration.confirmation.%s',

                'form' => 'fos_user.registration.form.%s',

            ));

        }

        if (!empty($config['change_password'])) {

            $loader->load('change_password.xml');

            $container->setAlias('fos_user.change_password.form.handler', $config['change_password']['form']['handler']);

            unset($config['change_password']['form']['handler']);

            $this->remapParametersNamespaces($config['change_password'], $container, array(

                'form' => 'fos_user.change_password.form.%s',

            ));

        }

        if (!empty($config['resetting'])) {

            $loader->load('resetting.xml');

            $container->setAlias('fos_user.resetting.form.handler', $config['resetting']['form']['handler']);

            unset($config['resetting']['form']['handler']);

            if (!empty($config['resetting']['email']['from_email'])) {

                $container->setParameter(

                    'fos_user.resetting.email.from_email',

                    array($config['resetting']['email']['from_email']['address'] => $config['resetting']['email']['from_email']['sender_name'])

                );

            }

            unset($config['resetting']['email']['from_email']);

            $this->remapParametersNamespaces($config['resetting'], $container, array(

                '' => array (

                    'token_ttl' => 'fos_user.resetting.token_ttl',

                ),

                'email' => 'fos_user.resetting.email.%s',

                'form' => 'fos_user.resetting.form.%s',

            ));

        }

        if (!empty($config['group'])) {

            $loader->load('group.xml');

            $loader->load(sprintf('%s_group.xml', $config['db_driver']));

            $container->setAlias('fos_user.group_manager', $config['group']['group_manager']);

            $container->setAlias('fos_user.group.form.handler', $config['group']['form']['handler']);

            unset($config['group']['form']['handler']);

            $this->remapParametersNamespaces($config['group'], $container, array(

                '' => array(

                    'group_class' => 'fos_user.model.group.class',

                ),

                'form' => 'fos_user.group.form.%s',

            ));

        }

    }

    protected function remapParameters(array $config, ContainerBuilder $container, array $map)

    {

        foreach ($map as $name => $paramName) {

            if (array_key_exists($name, $config)) {

                $container->setParameter($paramName, $config[$name]);

            }

        }

    }

    protected function remapParametersNamespaces(array $config, ContainerBuilder $container, array $namespaces)

    {

        foreach ($namespaces as $ns => $map) {

            if ($ns) {

                if (!array_key_exists($ns, $config)) {

                    continue;

                }

                $namespaceConfig = $config[$ns];

            } else {

                $namespaceConfig = $config;

            }

            if (is_array($map)) {

                $this->remapParameters($namespaceConfig, $container, $map);

            } else {

                foreach ($namespaceConfig as $name => $value) {

                    $container->setParameter(sprintf($map, $name), $value);

                }

            }

        }

    }

}