Completed
Pull Request — master (#8)
by Alessandro
06:32
created

MongoDbBundleExtension::defineLoggers()   A

Complexity

Conditions 1
Paths 1

Size

Total Lines 8
Code Lines 5

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 6
CRAP Score 1

Importance

Changes 1
Bugs 0 Features 1
Metric Value
c 1
b 0
f 1
dl 0
loc 8
ccs 6
cts 6
cp 1
rs 9.4285
cc 1
eloc 5
nc 1
nop 0
crap 1
1
<?php
2
3
namespace Facile\MongoDbBundle\DependencyInjection;
4
5
use Facile\MongoDbBundle\Services\ClientRegistry;
6
use Facile\MongoDbBundle\Services\Loggers\DataCollectorLoggerInterface;
7
use MongoDB\Database;
8
use Symfony\Bundle\WebProfilerBundle\WebProfilerBundle;
9
use Symfony\Component\Config\FileLocator;
10
use Symfony\Component\DependencyInjection\ContainerBuilder;
11
use Symfony\Component\DependencyInjection\Definition;
12
use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
13
use Symfony\Component\DependencyInjection\Reference;
14
use Symfony\Component\HttpKernel\DependencyInjection\Extension;
15
16
/**
17
 * Class MongoDbBundleExtension.
18
 */
19
class MongoDbBundleExtension extends Extension
20
{
21
    /** @var ContainerBuilder */
22
    private $containerBuilder;
23
24
    private $env;
25
26
    /**
27
     * {@inheritdoc}
28
     */
29 2
    public function load(array $configs, ContainerBuilder $container)
30
    {
31 2
        $this->containerBuilder = $container;
32 2
        $configuration = new Configuration();
33 2
        $config = $this->processConfiguration($configuration, $configs);
34 2
        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
35 2
        $loader->load('factory.xml');
36
37 2
        $this->env = $container->getParameter("kernel.environment");
38 2
        if ($this->env === 'dev' && class_exists(WebProfilerBundle::class)) {
39 2
            $loader->load('web_profiler.xml');
40
        }
41
42 2
        $this->defineLoggers();
43 2
        $this->defineClientRegistry($config['clients']);
44 2
        $this->defineConnections($config['connections']);
45
46 2
        return $config;
47
    }
48
49
    /**
50
     * @param array $clientsConfig
51
     *
52
     * @return Definition
53
     */
54 2
    private function defineClientRegistry(array $clientsConfig)
55
    {
56 2
        $loggerReference = $this->env === 'dev' ?
57 2
            new Reference('facile_mongo_db.logger') :
58 2
            new Reference('facile_mongo_db.null_logger');
59
60 2
        $clientRegistryDefinition = new Definition(
61 2
            ClientRegistry::class,
62
            [
63 2
                $loggerReference
64
            ]
65
        );
66 2
        foreach ($clientsConfig as $name => $conf) {
67
            $clientRegistryDefinition
68 2
                ->addMethodCall(
69 2
                    'addClientConfiguration',
70
                    [
71 2
                        $name,
72 2
                        $conf,
73
                    ]
74
                );
75
        }
76
77 2
        $this->containerBuilder->setDefinition('mongo.client_registry', $clientRegistryDefinition);
78 2
    }
79
80
    /**
81
     * @param array $connections
82
     */
83 2
    private function defineConnections(array $connections)
84
    {
85
86 2
        foreach ($connections as $name => $conf) {
87 2
            $connectionDefinition = new Definition(
88 2
                Database::class,
89
                [
90 2
                    $conf['client_name'],
91 2
                    $conf['database_name'],
92
                ]
93
            );
94 2
            $connectionDefinition->setFactory([new Reference('mongo.connection_factory'), 'createConnection']);
95 2
            $this->containerBuilder->setDefinition('mongo.connection.'.$name, $connectionDefinition);
96
        }
97 2
        $this->containerBuilder->setAlias('mongo.connection', 'mongo.connection.'.array_keys($connections)[0]);
98 2
    }
99
100 2
    private function defineLoggers()
101
    {
102 2
        $loggerDefinition = new Definition(
103 2
            DataCollectorLoggerInterface::class
104
        );
105 2
        $loggerDefinition->setFactory([new Reference('mongo.logger_factory'), 'createLogger']);
106 2
        $this->containerBuilder->setDefinition('facile_mongo_db.logger', $loggerDefinition);
107 2
    }
108
}
109