propel-bundle/DependencyInjection/PropelExtension.php

<?php

namespace Propel\PropelBundle\DependencyInjection;

use Symfony\Component\HttpKernel\DependencyInjection\Extension;
use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
use Symfony\Component\DependencyInjection\ContainerBuilder;
use Symfony\Component\DependencyInjection\Definition;
use Symfony\Component\DependencyInjection\Reference;
use Symfony\Component\Config\FileLocator;

class PropelExtension extends Extension
{
    /**
     * Loads the Propel configuration.
     *
     * @param array            $configs   An array of configuration settings
     * @param ContainerBuilder $container A ContainerBuilder instance
     */
    public function load(array $configs, ContainerBuilder $container)
    {
        $dbal = array();
        foreach ($configs as $config) {
            if (isset($config['dbal'])) {
                $dbal[] = $config['dbal'];
            }
        }

        $config = $configs[0];

        if (!$container->hasDefinition('propel')) {
            $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
            $loader->load('propel.xml');
        }

        if (!$container->hasParameter('propel.path')) {
            if (!isset($config['path'])) {
                throw new \InvalidArgumentException('The "path" parameter is mandatory.');
            }

            $container->setParameter('propel.path', $config['path']);
        }

        if (isset($config['path'])) {
            $container->setParameter('propel.path', $config['path']);
        }

        if (isset($config['phing_path'])) {
            $container->setParameter('propel.phing_path', $config['phing_path']);
        }

        if (isset($config['charset'])) {
            $charset = $config['charset'];
        } else {
            $charset = 'UTF8';
        }
        $container->setParameter('propel.charset', $charset);

        if (!empty($dbal)) {
            $this->dbalLoad($dbal, $container);
        }
    }

    /**
     * Loads the DBAL configuration.
     *
     * @param array            $configs   An array of configuration settings
     * @param ContainerBuilder $container A ContainerBuilder instance
     */
    protected function dbalLoad(array $configs, ContainerBuilder $container)
    {
        if (!$container->hasDefinition('propel')) {
            $loader = new XmlFileLoader($container, __DIR__.'/../Resources/config');
            $loader->load('propel.xml');
        }

        $mergedConfig = array(
            'default_connection'  => 'default',
        );

        if (!$container->hasParameter('propel.charset')) {
            $container->setParameter('propel.charset', 'utf8'); 
        }

        if ($container->hasParameter('kernel.debug')) {
            $className = $container->getParameter('kernel.debug') ? 'DebugPDO' : 'PropelPDO';
        } else {
            $className = 'PropelPDO';
        }

        $defaultConnection = array(
            'driver'              => 'mysql',
            'user'                => 'root',
            'password'            => '',
            'dsn'                 => '',
            'classname'           => $className,
            'options'             => array(),
            'attributes'          => array(),
            'settings'            => array('charset' => array('value' => $container->getParameter('propel.charset'))),
        );

        foreach ($configs as $config) {
            if (isset($config['default-connection'])) {
                $mergedConfig['default_connection'] = $config['default-connection'];
            } else if (isset($config['default_connection'])) {
                $mergedConfig['default_connection'] = $config['default_connection'];
            }
        }

        foreach ($configs as $config) {
            if (isset($config['connections'])) {
                $configConnections = $config['connections'];
                if (isset($config['connections']['connection']) && isset($config['connections']['connection'][0])) {
                    $configConnections = $config['connections']['connection'];
                }
            } else {
                $configConnections[$mergedConfig['default_connection']] = $config;
            }

            foreach ($configConnections as $name => $connection) {
                $connectionName = isset($connection['name']) ? $connection['name'] : $name;
                if (!isset($mergedConfig['connections'][$connectionName])) {
                    $mergedConfig['connections'][$connectionName] = $defaultConnection;
                }

                $mergedConfig['connections'][$connectionName]['name'] = $connectionName;

                foreach ($connection as $k => $v) {
                    if (isset($defaultConnection[$k])) {
                        $mergedConfig['connections'][$connectionName][$k] = null !== $v ? $v : '';
                    }
                }
            }
        }

        $config = $mergedConfig;
        $connectionName = $config['default_connection'];

        $container->setParameter('propel.dbal.default_connection', $connectionName);

        $c = array();

        foreach ($config['connections'] as $name => $conf) {
            $c['datasources'][$name]['adapter'] = $config['connections'][$name]['driver'];

            foreach (array('dsn', 'user', 'password', 'classname', 'options', 'attributes', 'settings') as $att) {
                if (isset($config['connections'][$name][$att])) {
                    $c['datasources'][$name]['connection'][$att] = $config['connections'][$name][$att];
                }
            }
        }

        $container->getDefinition('propel.configuration')->setArguments(array($c));
    }

    /**
     * Returns the base path for the XSD files.
     *
     * @return string The XSD base path
     */
    public function getXsdValidationBasePath()
    {
        return __DIR__.'/../Resources/config/schema';
    }

    /**
     * Returns the namespace to be used for this extension (XML namespace).
     *
     * @return string The XML namespace
     */
    public function getNamespace()
    {
        return 'http://www.symfony-project.org/schema/dic/propel';
    }

    /**
     * Returns the recommended alias to use in XML.
     *
     * This alias is also the mandatory prefix to use when using YAML.
     *
     * @return string The alias
     */
    public function getAlias()
    {
        return 'propel';
    }
}