Completed
Pull Request — master (#1020)
by Eugene
06:11
created

FOSElasticaExtension::loadSerializer()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 12
Code Lines 7

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 7
CRAP Score 2.0078

Importance

Changes 1
Bugs 1 Features 0
Metric Value
c 1
b 1
f 0
dl 0
loc 12
ccs 7
cts 8
cp 0.875
rs 9.4285
cc 2
eloc 7
nc 2
nop 2
crap 2.0078
1
<?php
2
3
namespace FOS\ElasticaBundle\DependencyInjection;
4
5
use Symfony\Component\HttpKernel\DependencyInjection\Extension;
6
use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
7
use Symfony\Component\DependencyInjection\ContainerBuilder;
8
use Symfony\Component\DependencyInjection\DefinitionDecorator;
9
use Symfony\Component\DependencyInjection\Reference;
10
use Symfony\Component\Config\FileLocator;
11
use InvalidArgumentException;
12
13
class FOSElasticaExtension extends Extension
14
{
15
    /**
16
     * Definition of elastica clients as configured by this extension.
17
     *
18
     * @var array
19
     */
20
    private $clients = array();
21
22
    /**
23
     * An array of indexes as configured by the extension.
24
     *
25
     * @var array
26
     */
27
    private $indexConfigs = array();
28
29
    /**
30
     * If we've encountered a type mapped to a specific persistence driver, it will be loaded
31
     * here.
32
     *
33
     * @var array
34
     */
35
    private $loadedDrivers = array();
36
37 13
    public function load(array $configs, ContainerBuilder $container)
38
    {
39 13
        $configuration = $this->getConfiguration($configs, $container);
40 13
        $config = $this->processConfiguration($configuration, $configs);
41
42 13
        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
43
44 13
        if (empty($config['clients']) || empty($config['indexes'])) {
45
            // No Clients or indexes are defined
46
            return;
47
        }
48
49 13
        foreach (array('config', 'index', 'persister', 'provider', 'source', 'transformer') as $basename) {
50 13
            $loader->load(sprintf('%s.xml', $basename));
51
        }
52
53 13
        if (empty($config['default_client'])) {
54 13
            $keys = array_keys($config['clients']);
55 13
            $config['default_client'] = reset($keys);
56
        }
57
58 13
        if (empty($config['default_index'])) {
59 13
            $keys = array_keys($config['indexes']);
60 13
            $config['default_index'] = reset($keys);
61
        }
62
63 13
        if (isset($config['serializer'])) {
64 2
            $loader->load('serializer.xml');
65
66 2
            $this->loadSerializer($config['serializer'], $container);
67
        }
68
69 13
        $this->loadClients($config['clients'], $container);
70 13
        $container->setAlias('fos_elastica.client', sprintf('fos_elastica.client.%s', $config['default_client']));
71
72 13
        $this->loadIndexes($config['indexes'], $container);
73 13
        $container->setAlias('fos_elastica.index', sprintf('fos_elastica.index.%s', $config['default_index']));
74
75 13
        $container->getDefinition('fos_elastica.config_source.container')->replaceArgument(0, $this->indexConfigs);
76
77 13
        $this->loadIndexManager($container);
78
79 13
        $this->createDefaultManagerAlias($config['default_manager'], $container);
80 13
    }
81
82
    /**
83
     * @param array            $config
84
     * @param ContainerBuilder $container
85
     *
86
     * @return Configuration
87
     */
88 13
    public function getConfiguration(array $config, ContainerBuilder $container)
89
    {
90 13
        return new Configuration($container->getParameter('kernel.debug'));
91
    }
92
93
    /**
94
     * Loads the configured clients.
95
     *
96
     * @param array            $clients   An array of clients configurations
97
     * @param ContainerBuilder $container A ContainerBuilder instance
98
     *
99
     * @return array
100
     */
101 13
    private function loadClients(array $clients, ContainerBuilder $container)
102
    {
103 13
        foreach ($clients as $name => $clientConfig) {
104 13
            $clientId = sprintf('fos_elastica.client.%s', $name);
105
106 13
            $clientDef = new DefinitionDecorator('fos_elastica.client_prototype');
107 13
            $clientDef->replaceArgument(0, $clientConfig);
108
109 13
            $logger = $clientConfig['connections'][0]['logger'];
110 13
            if (false !== $logger) {
111 13
                $clientDef->addMethodCall('setLogger', array(new Reference($logger)));
112
            }
113 13
            $clientDef->addTag('fos_elastica.client');
114
115 13
            $container->setDefinition($clientId, $clientDef);
116
117 13
            $this->clients[$name] = array(
118 13
                'id' => $clientId,
119 13
                'reference' => new Reference($clientId),
120
            );
121
        }
122 13
    }
123
124
    /**
125
     * Loads the configured indexes.
126
     *
127
     * @param array            $indexes   An array of indexes configurations
128
     * @param ContainerBuilder $container A ContainerBuilder instance
129
     *
130
     * @throws \InvalidArgumentException
131
     *
132
     * @return array
133
     */
134 13
    private function loadIndexes(array $indexes, ContainerBuilder $container)
135
    {
136 13
        $indexableCallbacks = array();
137
138 13
        foreach ($indexes as $name => $index) {
139 13
            $indexId = sprintf('fos_elastica.index.%s', $name);
140 13
            $indexName = isset($index['index_name']) ? $index['index_name'] : $name;
141
142 13
            $indexDef = new DefinitionDecorator('fos_elastica.index_prototype');
143 13
            $indexDef->replaceArgument(0, $indexName);
144 13
            $indexDef->addTag('fos_elastica.index', array(
145 13
                'name' => $name,
146
            ));
147
148 13
            if (method_exists($indexDef, 'setFactory')) {
149 13
                $indexDef->setFactory(array(new Reference('fos_elastica.client'), 'getIndex'));
150
            } else {
151
                // To be removed when dependency on Symfony DependencyInjection is bumped to 2.6
152
                $indexDef->setFactoryService('fos_elastica.client');
0 ignored issues
show
Bug introduced by
The method setFactoryService() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
153
                $indexDef->setFactoryMethod('getIndex');
0 ignored issues
show
Bug introduced by
The method setFactoryMethod() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
154
            }
155
156 13
            if (isset($index['client'])) {
157 2
                $client = $this->getClient($index['client']);
158
159 2
                if (method_exists($indexDef, 'setFactory')) {
160 2
                    $indexDef->setFactory(array($client, 'getIndex'));
161
                } else {
162
                    // To be removed when dependency on Symfony DependencyInjection is bumped to 2.6
163
                    $indexDef->setFactoryService($client);
0 ignored issues
show
Bug introduced by
The method setFactoryService() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
164
                    $indexDef->setFactoryMethod('getIndex');
0 ignored issues
show
Bug introduced by
The method setFactoryMethod() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
165
                }
166
            }
167
168 13
            $container->setDefinition($indexId, $indexDef);
169 13
            $reference = new Reference($indexId);
170
171 13
            $this->indexConfigs[$name] = array(
172 13
                'elasticsearch_name' => $indexName,
173 13
                'reference' => $reference,
174 13
                'name' => $name,
175 13
                'settings' => $index['settings'],
176 13
                'type_prototype' => isset($index['type_prototype']) ? $index['type_prototype'] : array(),
177 13
                'use_alias' => $index['use_alias'],
178
            );
179
180 13
            if ($index['finder']) {
181
                $this->loadIndexFinder($container, $name, $reference);
182
            }
183
184 13
            $this->loadTypes((array) $index['types'], $container, $this->indexConfigs[$name], $indexableCallbacks);
185
        }
186
187 13
        $indexable = $container->getDefinition('fos_elastica.indexable');
188 13
        $indexable->replaceArgument(0, $indexableCallbacks);
189 13
    }
190
191
    /**
192
     * Loads the configured index finders.
193
     *
194
     * @param \Symfony\Component\DependencyInjection\ContainerBuilder $container
195
     * @param string                                                  $name      The index name
196
     * @param Reference                                               $index     Reference to the related index
197
     *
198
     * @return string
199
     */
200
    private function loadIndexFinder(ContainerBuilder $container, $name, Reference $index)
201
    {
202
        /* Note: transformer services may conflict with "collection.index", if
203
         * an index and type names were "collection" and an index, respectively.
204
         */
205
        $transformerId = sprintf('fos_elastica.elastica_to_model_transformer.collection.%s', $name);
206
        $transformerDef = new DefinitionDecorator('fos_elastica.elastica_to_model_transformer.collection');
207
        $container->setDefinition($transformerId, $transformerDef);
208
209
        $finderId = sprintf('fos_elastica.finder.%s', $name);
210
        $finderDef = new DefinitionDecorator('fos_elastica.finder');
211
        $finderDef->replaceArgument(0, $index);
212
        $finderDef->replaceArgument(1, new Reference($transformerId));
213
214
        $container->setDefinition($finderId, $finderDef);
215
    }
216
217
    /**
218
     * Loads the configured types.
219
     *
220
     * @param array            $types
221
     * @param ContainerBuilder $container
222
     * @param array            $indexConfig
223
     * @param array            $indexableCallbacks
224
     */
225 13
    private function loadTypes(array $types, ContainerBuilder $container, array $indexConfig, array &$indexableCallbacks)
226
    {
227 13
        foreach ($types as $name => $type) {
228 13
            $indexName = $indexConfig['name'];
229
230 13
            $typeId = sprintf('%s.%s', $indexConfig['reference'], $name);
231 13
            $typeDef = new DefinitionDecorator('fos_elastica.type_prototype');
232 13
            $typeDef->replaceArgument(0, $name);
233
234 13
            if (method_exists($typeDef, 'setFactory')) {
235 13
                $typeDef->setFactory(array($indexConfig['reference'], 'getType'));
236
            } else {
237
                // To be removed when dependency on Symfony DependencyInjection is bumped to 2.6
238
                $typeDef->setFactoryService($indexConfig['reference']);
0 ignored issues
show
Bug introduced by
The method setFactoryService() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
239
                $typeDef->setFactoryMethod('getType');
0 ignored issues
show
Bug introduced by
The method setFactoryMethod() does not exist on Symfony\Component\Depend...ion\DefinitionDecorator. Did you maybe mean setFactory()?

This check marks calls to methods that do not seem to exist on an object.

This is most likely the result of a method being renamed without all references to it being renamed likewise.

Loading history...
240
            }
241
242 13
            $container->setDefinition($typeId, $typeDef);
243
244
            $typeConfig = array(
245 13
                'name' => $name,
246
                'mapping' => array(), // An array containing anything that gets sent directly to ElasticSearch
247
                'config' => array(),
248
            );
249
250
            foreach (array(
251 13
                'dynamic_templates',
252
                'properties',
253
                '_all',
254
                '_boost',
255
                '_id',
256
                '_parent',
257
                '_routing',
258
                '_source',
259
                '_timestamp',
260
                '_ttl',
261
            ) as $field) {
262 13
                if (isset($type[$field])) {
263 13
                    $typeConfig['mapping'][$field] = $type[$field];
264
                }
265
            }
266
267
            foreach (array(
268 13
                'persistence',
269
                'serializer',
270
                'index_analyzer',
271
                'search_analyzer',
272
                'dynamic',
273
                'date_detection',
274
                'dynamic_date_formats',
275
                'numeric_detection',
276
            ) as $field) {
277 13
                $typeConfig['config'][$field] = array_key_exists($field, $type) ?
278 13
                    $type[$field] :
279 13
                    null;
280
            }
281
282 13
            $this->indexConfigs[$indexName]['types'][$name] = $typeConfig;
283
284 13
            if (isset($type['persistence'])) {
285 9
                $this->loadTypePersistenceIntegration($type['persistence'], $container, new Reference($typeId), $indexName, $name);
286
287 9
                $typeConfig['persistence'] = $type['persistence'];
288
            }
289
290 13
            if (isset($type['indexable_callback'])) {
291 5
                $indexableCallbacks[sprintf('%s/%s', $indexName, $name)] = $type['indexable_callback'];
292
            }
293
294 13
            if ($container->hasDefinition('fos_elastica.serializer_callback_prototype')) {
295 2
                $typeSerializerId = sprintf('%s.serializer.callback', $typeId);
296 2
                $typeSerializerDef = new DefinitionDecorator('fos_elastica.serializer_callback_prototype');
297
298 2
                if (isset($type['serializer']['groups'])) {
299 2
                    $typeSerializerDef->addMethodCall('setGroups', array($type['serializer']['groups']));
300
                }
301 2
                if (isset($type['serializer']['version'])) {
302 2
                    $typeSerializerDef->addMethodCall('setVersion', array($type['serializer']['version']));
303
                }
304
305 2
                $typeDef->addMethodCall('setSerializer', array(array(new Reference($typeSerializerId), 'serialize')));
306 13
                $container->setDefinition($typeSerializerId, $typeSerializerDef);
307
            }
308
        }
309 13
    }
310
311
    /**
312
     * Loads the optional provider and finder for a type.
313
     *
314
     * @param array            $typeConfig
315
     * @param ContainerBuilder $container
316
     * @param Reference        $typeRef
317
     * @param string           $indexName
318
     * @param string           $typeName
319
     */
320 9
    private function loadTypePersistenceIntegration(array $typeConfig, ContainerBuilder $container, Reference $typeRef, $indexName, $typeName)
321
    {
322 9
        if (isset($typeConfig['driver'])) {
323 8
            $this->loadDriver($container, $typeConfig['driver']);
324
        }
325
326 9
        $elasticaToModelTransformerId = $this->loadElasticaToModelTransformer($typeConfig, $container, $indexName, $typeName);
327 9
        $modelToElasticaTransformerId = $this->loadModelToElasticaTransformer($typeConfig, $container, $indexName, $typeName);
328 9
        $objectPersisterId = $this->loadObjectPersister($typeConfig, $typeRef, $container, $indexName, $typeName, $modelToElasticaTransformerId);
329
330 9
        if (isset($typeConfig['provider'])) {
331 5
            $this->loadTypeProvider($typeConfig, $container, $objectPersisterId, $indexName, $typeName);
332
        }
333 9
        if (isset($typeConfig['finder'])) {
334 5
            $this->loadTypeFinder($typeConfig, $container, $elasticaToModelTransformerId, $typeRef, $indexName, $typeName);
335
        }
336 9
        if (isset($typeConfig['listener'])) {
337 5
            $this->loadTypeListener($typeConfig, $container, $objectPersisterId, $indexName, $typeName);
338
        }
339 9
    }
340
341
    /**
342
     * Creates and loads an ElasticaToModelTransformer.
343
     *
344
     * @param array            $typeConfig
345
     * @param ContainerBuilder $container
346
     * @param string           $indexName
347
     * @param string           $typeName
348
     *
349
     * @return string
350
     */
351 9
    private function loadElasticaToModelTransformer(array $typeConfig, ContainerBuilder $container, $indexName, $typeName)
352
    {
353 9
        if (isset($typeConfig['elastica_to_model_transformer']['service'])) {
354 1
            return $typeConfig['elastica_to_model_transformer']['service'];
355
        }
356
357
        /* Note: transformer services may conflict with "prototype.driver", if
358
         * the index and type names were "prototype" and a driver, respectively.
359
         */
360 8
        $abstractId = sprintf('fos_elastica.elastica_to_model_transformer.prototype.%s', $typeConfig['driver']);
361 8
        $serviceId = sprintf('fos_elastica.elastica_to_model_transformer.%s.%s', $indexName, $typeName);
362 8
        $serviceDef = new DefinitionDecorator($abstractId);
363 8
        $serviceDef->addTag('fos_elastica.elastica_to_model_transformer', array('type' => $typeName, 'index' => $indexName));
364
365
        // Doctrine has a mandatory service as first argument
366 8
        $argPos = ('propel' === $typeConfig['driver']) ? 0 : 1;
367
368 8
        $serviceDef->replaceArgument($argPos, $typeConfig['model']);
369 8
        $serviceDef->replaceArgument($argPos + 1, array_merge($typeConfig['elastica_to_model_transformer'], array(
370 8
            'identifier' => $typeConfig['identifier'],
371
        )));
372 8
        $container->setDefinition($serviceId, $serviceDef);
373
374 8
        return $serviceId;
375
    }
376
377
    /**
378
     * Creates and loads a ModelToElasticaTransformer for an index/type.
379
     *
380
     * @param array            $typeConfig
381
     * @param ContainerBuilder $container
382
     * @param string           $indexName
383
     * @param string           $typeName
384
     *
385
     * @return string
386
     */
387 9
    private function loadModelToElasticaTransformer(array $typeConfig, ContainerBuilder $container, $indexName, $typeName)
388
    {
389 9
        if (isset($typeConfig['model_to_elastica_transformer']['service'])) {
390
            return $typeConfig['model_to_elastica_transformer']['service'];
391
        }
392
393 9
        $abstractId = $container->hasDefinition('fos_elastica.serializer_callback_prototype') ?
394 2
            'fos_elastica.model_to_elastica_identifier_transformer' :
395 9
            'fos_elastica.model_to_elastica_transformer';
396
397 9
        $serviceId = sprintf('fos_elastica.model_to_elastica_transformer.%s.%s', $indexName, $typeName);
398 9
        $serviceDef = new DefinitionDecorator($abstractId);
399 9
        $serviceDef->replaceArgument(0, array(
400 9
            'identifier' => $typeConfig['identifier'],
401
        ));
402 9
        $container->setDefinition($serviceId, $serviceDef);
403
404 9
        return $serviceId;
405
    }
406
407
    /**
408
     * Creates and loads an object persister for a type.
409
     *
410
     * @param array            $typeConfig
411
     * @param Reference        $typeRef
412
     * @param ContainerBuilder $container
413
     * @param string           $indexName
414
     * @param string           $typeName
415
     * @param string           $transformerId
416
     *
417
     * @return string
418
     */
419 9
    private function loadObjectPersister(array $typeConfig, Reference $typeRef, ContainerBuilder $container, $indexName, $typeName, $transformerId)
420
    {
421 9
        if (isset($typeConfig['persister']['service'])) {
422 1
            return $typeConfig['persister']['service'];
423
        }
424
425
        $arguments = array(
426 8
            $typeRef,
427 8
            new Reference($transformerId),
428 8
            $typeConfig['model'],
429
        );
430
431 8
        if ($container->hasDefinition('fos_elastica.serializer_callback_prototype')) {
432 2
            $abstractId = 'fos_elastica.object_serializer_persister';
433 2
            $callbackId = sprintf('%s.%s.serializer.callback', $this->indexConfigs[$indexName]['reference'], $typeName);
434 2
            $arguments[] = array(new Reference($callbackId), 'serialize');
435
        } else {
436 6
            $abstractId = 'fos_elastica.object_persister';
437 6
            $mapping = $this->indexConfigs[$indexName]['types'][$typeName]['mapping'];
438 6
            $argument = $mapping['properties'];
439 6
            if (isset($mapping['_parent'])) {
440 1
                $argument['_parent'] = $mapping['_parent'];
441
            }
442 6
            $arguments[] = $argument;
443
        }
444
445 8
        $serviceId = sprintf('fos_elastica.object_persister.%s.%s', $indexName, $typeName);
446 8
        $serviceDef = new DefinitionDecorator($abstractId);
447 8
        foreach ($arguments as $i => $argument) {
448 8
            $serviceDef->replaceArgument($i, $argument);
449
        }
450
451 8
        $container->setDefinition($serviceId, $serviceDef);
452
453 8
        return $serviceId;
454
    }
455
456
    /**
457
     * Loads a provider for a type.
458
     *
459
     * @param array            $typeConfig
460
     * @param ContainerBuilder $container
461
     * @param string           $objectPersisterId
462
     * @param string           $indexName
463
     * @param string           $typeName
464
     *
465
     * @return string
466
     */
467 5
    private function loadTypeProvider(array $typeConfig, ContainerBuilder $container, $objectPersisterId, $indexName, $typeName)
468
    {
469 5
        if (isset($typeConfig['provider']['service'])) {
470
            return $typeConfig['provider']['service'];
471
        }
472
473
        /* Note: provider services may conflict with "prototype.driver", if the
474
         * index and type names were "prototype" and a driver, respectively.
475
         */
476 5
        $providerId = sprintf('fos_elastica.provider.%s.%s', $indexName, $typeName);
477 5
        $providerDef = new DefinitionDecorator('fos_elastica.provider.prototype.'.$typeConfig['driver']);
478 5
        $providerDef->addTag('fos_elastica.provider', array('index' => $indexName, 'type' => $typeName));
479 5
        $providerDef->replaceArgument(0, new Reference($objectPersisterId));
480 5
        $providerDef->replaceArgument(2, $typeConfig['model']);
481
        // Propel provider can simply ignore Doctrine-specific options
482 5
        $providerDef->replaceArgument(3, array_merge(array_diff_key($typeConfig['provider'], array('service' => 1)), array(
483 5
            'indexName' => $indexName,
484 5
            'typeName' => $typeName,
485
        )));
486 5
        $container->setDefinition($providerId, $providerDef);
487
488 5
        return $providerId;
489
    }
490
491
    /**
492
     * Loads doctrine listeners to handle indexing of new or updated objects.
493
     *
494
     * @param array            $typeConfig
495
     * @param ContainerBuilder $container
496
     * @param string           $objectPersisterId
497
     * @param string           $indexName
498
     * @param string           $typeName
499
     *
500
     * @return string
501
     */
502 5
    private function loadTypeListener(array $typeConfig, ContainerBuilder $container, $objectPersisterId, $indexName, $typeName)
503
    {
504 5
        if (isset($typeConfig['listener']['service'])) {
505
            return $typeConfig['listener']['service'];
506
        }
507
508
        /* Note: listener services may conflict with "prototype.driver", if the
509
         * index and type names were "prototype" and a driver, respectively.
510
         */
511 5
        $abstractListenerId = sprintf('fos_elastica.listener.prototype.%s', $typeConfig['driver']);
512 5
        $listenerId = sprintf('fos_elastica.listener.%s.%s', $indexName, $typeName);
513 5
        $listenerDef = new DefinitionDecorator($abstractListenerId);
514 5
        $listenerDef->replaceArgument(0, new Reference($objectPersisterId));
515 5
        $listenerDef->replaceArgument(2, array(
516 5
            'identifier' => $typeConfig['identifier'],
517 5
            'indexName' => $indexName,
518 5
            'typeName' => $typeName,
519
        ));
520 5
        $listenerDef->replaceArgument(3, $typeConfig['listener']['logger'] ?
521
            new Reference($typeConfig['listener']['logger']) :
522 5
            null
523
        );
524
525 5
        $tagName = null;
526 5
        switch ($typeConfig['driver']) {
527
            case 'orm':
528 5
                $tagName = 'doctrine.event_listener';
529 5
                break;
530
            case 'phpcr':
531
                $tagName = 'doctrine_phpcr.event_listener';
532
                break;
533
            case 'mongodb':
534
                $tagName = 'doctrine_mongodb.odm.event_listener';
535
                break;
536
        }
537
538 5
        if (null !== $tagName) {
539 5
            foreach ($this->getDoctrineEvents($typeConfig) as $event) {
540 5
                $listenerDef->addTag($tagName, array('event' => $event));
541
            }
542
        }
543
544 5
        $container->setDefinition($listenerId, $listenerDef);
545
546 5
        return $listenerId;
547
    }
548
549
    /**
550
     * Map Elastica to Doctrine events for the current driver.
551
     */
552 5
    private function getDoctrineEvents(array $typeConfig)
553
    {
554 5
        switch ($typeConfig['driver']) {
555
            case 'orm':
556 5
                $eventsClass = '\Doctrine\ORM\Events';
557 5
                break;
558
            case 'phpcr':
559
                $eventsClass = '\Doctrine\ODM\PHPCR\Event';
560
                break;
561
            case 'mongodb':
562
                $eventsClass = '\Doctrine\ODM\MongoDB\Events';
563
                break;
564
            default:
565
                throw new InvalidArgumentException(sprintf('Cannot determine events for driver "%s"', $typeConfig['driver']));
566
        }
567
568 5
        $events = array();
569
        $eventMapping = array(
570 5
            'insert' => array(constant($eventsClass.'::postPersist')),
571 5
            'update' => array(constant($eventsClass.'::postUpdate')),
572 5
            'delete' => array(constant($eventsClass.'::preRemove')),
573 5
            'flush' => array($typeConfig['listener']['immediate'] ? constant($eventsClass.'::preFlush') : constant($eventsClass.'::postFlush')),
574
        );
575
576 5
        foreach ($eventMapping as $event => $doctrineEvents) {
577 5
            if (isset($typeConfig['listener'][$event]) && $typeConfig['listener'][$event]) {
578 5
                $events = array_merge($events, $doctrineEvents);
579
            }
580
        }
581
582 5
        return $events;
583
    }
584
585
    /**
586
     * Loads a Type specific Finder.
587
     *
588
     * @param array            $typeConfig
589
     * @param ContainerBuilder $container
590
     * @param string           $elasticaToModelId
591
     * @param Reference        $typeRef
592
     * @param string           $indexName
593
     * @param string           $typeName
594
     *
595
     * @return string
596
     */
597 5
    private function loadTypeFinder(array $typeConfig, ContainerBuilder $container, $elasticaToModelId, Reference $typeRef, $indexName, $typeName)
598
    {
599 5
        if (isset($typeConfig['finder']['service'])) {
600
            $finderId = $typeConfig['finder']['service'];
601
        } else {
602 5
            $finderId = sprintf('fos_elastica.finder.%s.%s', $indexName, $typeName);
603 5
            $finderDef = new DefinitionDecorator('fos_elastica.finder');
604 5
            $finderDef->replaceArgument(0, $typeRef);
605 5
            $finderDef->replaceArgument(1, new Reference($elasticaToModelId));
606 5
            $container->setDefinition($finderId, $finderDef);
607
        }
608
609 5
        $managerId = sprintf('fos_elastica.manager.%s', $typeConfig['driver']);
610 5
        $managerDef = $container->getDefinition($managerId);
611 5
        $arguments = array( $typeConfig['model'], new Reference($finderId));
612 5
        if (isset($typeConfig['repository'])) {
613
            $arguments[] = $typeConfig['repository'];
614
        }
615 5
        $managerDef->addMethodCall('addEntity', $arguments);
616
617 5
        return $finderId;
618
    }
619
620
    /**
621
     * Loads the index manager.
622
     *
623
     * @param ContainerBuilder $container
624
     **/
625
    private function loadIndexManager(ContainerBuilder $container)
626
    {
627 13
        $indexRefs = array_map(function ($index) {
628 13
            return $index['reference'];
629 13
        }, $this->indexConfigs);
630
631 13
        $managerDef = $container->getDefinition('fos_elastica.index_manager');
632 13
        $managerDef->replaceArgument(0, $indexRefs);
633 13
    }
634
635
    /**
636
     * Makes sure a specific driver has been loaded.
637
     *
638
     * @param ContainerBuilder $container
639
     * @param string           $driver
640
     */
641 8
    private function loadDriver(ContainerBuilder $container, $driver)
642
    {
643 8
        if (in_array($driver, $this->loadedDrivers)) {
644 8
            return;
645
        }
646
647 8
        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
648 8
        $loader->load($driver.'.xml');
649 8
        $this->loadedDrivers[] = $driver;
650 8
    }
651
652
    /**
653
     * Loads and configures the serializer prototype.
654
     *
655
     * @param array            $config
656
     * @param ContainerBuilder $container
657
     */
658 2
    private function loadSerializer($config, ContainerBuilder $container)
659
    {
660 2
        $container->setAlias('fos_elastica.serializer', $config['serializer']);
661
662 2
        $serializer = $container->getDefinition('fos_elastica.serializer_callback_prototype');
663 2
        $serializer->setClass($config['callback_class']);
664
665 2
        $callbackClassImplementedInterfaces = class_implements($config['callback_class']);
666 2
        if (isset($callbackClassImplementedInterfaces['Symfony\Component\DependencyInjection\ContainerAwareInterface'])) {
667
            $serializer->addMethodCall('setContainer', array(new Reference('service_container')));
668
        }
669 2
    }
670
671
    /**
672
     * Creates a default manager alias for defined default manager or the first loaded driver.
673
     *
674
     * @param string           $defaultManager
675
     * @param ContainerBuilder $container
676
     */
677 13
    private function createDefaultManagerAlias($defaultManager, ContainerBuilder $container)
678
    {
679 13
        if (0 == count($this->loadedDrivers)) {
680 5
            return;
681
        }
682
683 8
        if (count($this->loadedDrivers) > 1
684 8
            && in_array($defaultManager, $this->loadedDrivers)
685
        ) {
686
            $defaultManagerService = $defaultManager;
687
        } else {
688 8
            $defaultManagerService = $this->loadedDrivers[0];
689
        }
690
691 8
        $container->setAlias('fos_elastica.manager', sprintf('fos_elastica.manager.%s', $defaultManagerService));
692 8
    }
693
694
    /**
695
     * Returns a reference to a client given its configured name.
696
     *
697
     * @param string $clientName
698
     *
699
     * @return Reference
700
     *
701
     * @throws \InvalidArgumentException
702
     */
703 2
    private function getClient($clientName)
704
    {
705 2
        if (!array_key_exists($clientName, $this->clients)) {
706
            throw new InvalidArgumentException(sprintf('The elastica client with name "%s" is not defined', $clientName));
707
        }
708
709 2
        return $this->clients[$clientName]['reference'];
710
    }
711
}
712