Completed
Pull Request — master (#992)
by David
05:34
created

loadModelToElasticaTransformer()   A

Complexity

Conditions 3
Paths 3

Size

Total Lines 19
Code Lines 12

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 11
CRAP Score 3.0052

Importance

Changes 1
Bugs 1 Features 0
Metric Value
dl 0
loc 19
ccs 11
cts 12
cp 0.9167
rs 9.4286
c 1
b 1
f 0
cc 3
eloc 12
nc 3
nop 4
crap 3.0052
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 8
    public function load(array $configs, ContainerBuilder $container)
38
    {
39 8
        $configuration = $this->getConfiguration($configs, $container);
40 8
        $config = $this->processConfiguration($configuration, $configs);
41
42 8
        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
43
44 8
        if (empty($config['clients']) || empty($config['indexes'])) {
45
            // No Clients or indexes are defined
46
            return;
47
        }
48
49 8
        foreach (array('config', 'index', 'persister', 'provider', 'source', 'transformer') as $basename) {
50 8
            $loader->load(sprintf('%s.xml', $basename));
51
        }
52
53 8
        if (empty($config['default_client'])) {
54 8
            $keys = array_keys($config['clients']);
55 8
            $config['default_client'] = reset($keys);
56
        }
57
58 8
        if (empty($config['default_index'])) {
59 8
            $keys = array_keys($config['indexes']);
60 8
            $config['default_index'] = reset($keys);
61
        }
62
63 8
        if (isset($config['serializer'])) {
64 2
            $loader->load('serializer.xml');
65
66 2
            $this->loadSerializer($config['serializer'], $container);
67
        }
68
69 8
        $this->loadClients($config['clients'], $container);
70 8
        $container->setAlias('fos_elastica.client', sprintf('fos_elastica.client.%s', $config['default_client']));
71
72 8
        $this->loadIndexes($config['indexes'], $container);
73 8
        $container->setAlias('fos_elastica.index', sprintf('fos_elastica.index.%s', $config['default_index']));
74
75 8
        $container->getDefinition('fos_elastica.config_source.container')->replaceArgument(0, $this->indexConfigs);
76
77 8
        $this->loadIndexManager($container);
78
79 8
        $this->createDefaultManagerAlias($config['default_manager'], $container);
80 8
    }
81
82
    /**
83
     * @param array            $config
84
     * @param ContainerBuilder $container
85
     *
86
     * @return Configuration
87
     */
88 8
    public function getConfiguration(array $config, ContainerBuilder $container)
89
    {
90 8
        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 8
    private function loadClients(array $clients, ContainerBuilder $container)
102
    {
103 8
        foreach ($clients as $name => $clientConfig) {
104 8
            $clientId = sprintf('fos_elastica.client.%s', $name);
105
106 8
            $clientDef = new DefinitionDecorator('fos_elastica.client_prototype');
107 8
            $clientDef->replaceArgument(0, $clientConfig);
108
109 8
            $logger = $clientConfig['connections'][0]['logger'];
110 8
            if (false !== $logger) {
111 8
                $clientDef->addMethodCall('setLogger', array(new Reference($logger)));
112
            }
113 8
            $clientDef->addTag('fos_elastica.client');
114
115 8
            $container->setDefinition($clientId, $clientDef);
116
117 8
            $this->clients[$name] = array(
118 8
                'id' => $clientId,
119 8
                'reference' => new Reference($clientId),
120
            );
121
        }
122 8
    }
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 8
    private function loadIndexes(array $indexes, ContainerBuilder $container)
135
    {
136 8
        $indexableCallbacks = array();
137
138 8
        foreach ($indexes as $name => $index) {
139 8
            $indexId = sprintf('fos_elastica.index.%s', $name);
140 8
            $indexName = isset($index['index_name']) ? $index['index_name'] : $name;
141
142 8
            $indexDef = new DefinitionDecorator('fos_elastica.index_prototype');
143 8
            $indexDef->replaceArgument(0, $indexName);
144 8
            $indexDef->addTag('fos_elastica.index', array(
145 8
                'name' => $name,
146
            ));
147
148 8
            if (method_exists($indexDef, 'setFactory')) {
149 8
                $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');
153
                $indexDef->setFactoryMethod('getIndex');
154
            }
155
156 8
            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);
164
                    $indexDef->setFactoryMethod('getIndex');
165
                }
166
            }
167
168 8
            $container->setDefinition($indexId, $indexDef);
169 8
            $reference = new Reference($indexId);
170
171 8
            $this->indexConfigs[$name] = array(
172 8
                'elasticsearch_name' => $indexName,
173 8
                'reference' => $reference,
174 8
                'name' => $name,
175 8
                'settings' => $index['settings'],
176 8
                'type_prototype' => isset($index['type_prototype']) ? $index['type_prototype'] : array(),
177 8
                'use_alias' => $index['use_alias'],
178
            );
179
180 8
            if ($index['finder']) {
181
                $this->loadIndexFinder($container, $name, $reference);
182
            }
183
184 8
            $this->loadTypes((array) $index['types'], $container, $this->indexConfigs[$name], $indexableCallbacks);
185
        }
186
187 8
        $indexable = $container->getDefinition('fos_elastica.indexable');
188 8
        $indexable->replaceArgument(0, $indexableCallbacks);
189 8
    }
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 8
    private function loadTypes(array $types, ContainerBuilder $container, array $indexConfig, array &$indexableCallbacks)
226
    {
227 8
        foreach ($types as $name => $type) {
228 8
            $indexName = $indexConfig['name'];
229
230 8
            $typeId = sprintf('%s.%s', $indexConfig['reference'], $name);
231 8
            $typeDef = new DefinitionDecorator('fos_elastica.type_prototype');
232 8
            $typeDef->replaceArgument(0, $name);
233
234 8
            if (method_exists($typeDef, 'setFactory')) {
235 8
                $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']);
239
                $typeDef->setFactoryMethod('getType');
240
            }
241
242 8
            $container->setDefinition($typeId, $typeDef);
243
244
            $typeConfig = array(
245 8
                'name' => $name,
246
                'mapping' => array(), // An array containing anything that gets sent directly to ElasticSearch
247
                'config' => array(),
248
            );
249
250
            foreach (array(
251 8
                'dynamic_templates',
252
                'properties',
253
                '_all',
254
                '_boost',
255
                '_id',
256
                '_parent',
257
                '_routing',
258
                '_source',
259
                '_timestamp',
260
                '_ttl',
261
            ) as $field) {
262 8
                if (isset($type[$field])) {
263 8
                    $typeConfig['mapping'][$field] = $type[$field];
264
                }
265
            }
266
267
            foreach (array(
268 8
                'persistence',
269
                'serializer',
270
                'index_analyzer',
271
                'search_analyzer',
272
                'date_detection',
273
                'dynamic_date_formats',
274
                'numeric_detection',
275
            ) as $field) {
276 8
                $typeConfig['config'][$field] = array_key_exists($field, $type) ?
277 8
                    $type[$field] :
278 8
                    null;
279
            }
280
281 8
            $this->indexConfigs[$indexName]['types'][$name] = $typeConfig;
282
283 8
            if (isset($type['persistence'])) {
284 4
                $this->loadTypePersistenceIntegration($type['persistence'], $container, new Reference($typeId), $indexName, $name);
285
286 4
                $typeConfig['persistence'] = $type['persistence'];
287
            }
288
289 8
            if (isset($type['indexable_callback'])) {
290
                $indexableCallbacks[sprintf('%s/%s', $indexName, $name)] = $type['indexable_callback'];
291
            }
292
293 8
            if ($container->hasDefinition('fos_elastica.serializer_callback_prototype')) {
294 2
                $typeSerializerId = sprintf('%s.serializer.callback', $typeId);
295 2
                $typeSerializerDef = new DefinitionDecorator('fos_elastica.serializer_callback_prototype');
296
297 2
                if (isset($type['serializer']['groups'])) {
298 2
                    $typeSerializerDef->addMethodCall('setGroups', array($type['serializer']['groups']));
299
                }
300 2
                if (isset($type['serializer']['version'])) {
301 2
                    $typeSerializerDef->addMethodCall('setVersion', array($type['serializer']['version']));
302
                }
303
304 2
                $typeDef->addMethodCall('setSerializer', array(array(new Reference($typeSerializerId), 'serialize')));
305 8
                $container->setDefinition($typeSerializerId, $typeSerializerDef);
306
            }
307
        }
308 8
    }
309
310
    /**
311
     * Loads the optional provider and finder for a type.
312
     *
313
     * @param array            $typeConfig
314
     * @param ContainerBuilder $container
315
     * @param Reference        $typeRef
316
     * @param string           $indexName
317
     * @param string           $typeName
318
     */
319 4
    private function loadTypePersistenceIntegration(array $typeConfig, ContainerBuilder $container, Reference $typeRef, $indexName, $typeName)
320
    {
321 4
        if (isset($typeConfig['driver'])) {
322 3
            $this->loadDriver($container, $typeConfig['driver']);
323
        }
324
325 4
        $elasticaToModelTransformerId = $this->loadElasticaToModelTransformer($typeConfig, $container, $indexName, $typeName);
326 4
        $modelToElasticaTransformerId = $this->loadModelToElasticaTransformer($typeConfig, $container, $indexName, $typeName);
327 4
        $objectPersisterId = $this->loadObjectPersister($typeConfig, $typeRef, $container, $indexName, $typeName, $modelToElasticaTransformerId);
328
329 4
        if (isset($typeConfig['provider'])) {
330
            $this->loadTypeProvider($typeConfig, $container, $objectPersisterId, $indexName, $typeName);
331
        }
332 4
        if (isset($typeConfig['finder'])) {
333
            $this->loadTypeFinder($typeConfig, $container, $elasticaToModelTransformerId, $typeRef, $indexName, $typeName);
334
        }
335 4
        if (isset($typeConfig['listener'])) {
336
            $this->loadTypeListener($typeConfig, $container, $objectPersisterId, $indexName, $typeName);
337
        }
338 4
    }
339
340
    /**
341
     * Creates and loads an ElasticaToModelTransformer.
342
     *
343
     * @param array            $typeConfig
344
     * @param ContainerBuilder $container
345
     * @param string           $indexName
346
     * @param string           $typeName
347
     *
348
     * @return string
349
     */
350 4
    private function loadElasticaToModelTransformer(array $typeConfig, ContainerBuilder $container, $indexName, $typeName)
351
    {
352 4
        if (isset($typeConfig['elastica_to_model_transformer']['service'])) {
353 1
            return $typeConfig['elastica_to_model_transformer']['service'];
354
        }
355
356
        /* Note: transformer services may conflict with "prototype.driver", if
357
         * the index and type names were "prototype" and a driver, respectively.
358
         */
359 3
        $abstractId = sprintf('fos_elastica.elastica_to_model_transformer.prototype.%s', $typeConfig['driver']);
360 3
        $serviceId = sprintf('fos_elastica.elastica_to_model_transformer.%s.%s', $indexName, $typeName);
361 3
        $serviceDef = new DefinitionDecorator($abstractId);
362 3
        $serviceDef->addTag('fos_elastica.elastica_to_model_transformer', array('type' => $typeName, 'index' => $indexName));
363
364
        // Doctrine has a mandatory service as first argument
365 3
        $argPos = ('propel' === $typeConfig['driver']) ? 0 : 1;
366
367 3
        $serviceDef->replaceArgument($argPos, $typeConfig['model']);
368 3
        $serviceDef->replaceArgument($argPos + 1, array_merge($typeConfig['elastica_to_model_transformer'], array(
369 3
            'identifier' => $typeConfig['identifier'],
370
        )));
371 3
        $container->setDefinition($serviceId, $serviceDef);
372
373 3
        return $serviceId;
374
    }
375
376
    /**
377
     * Creates and loads a ModelToElasticaTransformer for an index/type.
378
     *
379
     * @param array            $typeConfig
380
     * @param ContainerBuilder $container
381
     * @param string           $indexName
382
     * @param string           $typeName
383
     *
384
     * @return string
385
     */
386 4
    private function loadModelToElasticaTransformer(array $typeConfig, ContainerBuilder $container, $indexName, $typeName)
387
    {
388 4
        if (isset($typeConfig['model_to_elastica_transformer']['service'])) {
389
            return $typeConfig['model_to_elastica_transformer']['service'];
390
        }
391
392 4
        $abstractId = $container->hasDefinition('fos_elastica.serializer_callback_prototype') ?
393 2
            'fos_elastica.model_to_elastica_identifier_transformer' :
394 4
            'fos_elastica.model_to_elastica_transformer';
395
396 4
        $serviceId = sprintf('fos_elastica.model_to_elastica_transformer.%s.%s', $indexName, $typeName);
397 4
        $serviceDef = new DefinitionDecorator($abstractId);
398 4
        $serviceDef->replaceArgument(0, array(
399 4
            'identifier' => $typeConfig['identifier'],
400
        ));
401 4
        $container->setDefinition($serviceId, $serviceDef);
402
403 4
        return $serviceId;
404
    }
405
406
    /**
407
     * Creates and loads an object persister for a type.
408
     *
409
     * @param array            $typeConfig
410
     * @param Reference        $typeRef
411
     * @param ContainerBuilder $container
412
     * @param string           $indexName
413
     * @param string           $typeName
414
     * @param string           $transformerId
415
     *
416
     * @return string
417
     */
418 4
    private function loadObjectPersister(array $typeConfig, Reference $typeRef, ContainerBuilder $container, $indexName, $typeName, $transformerId)
419
    {
420 4
        if (isset($typeConfig['persister']['service'])) {
421 1
            return $typeConfig['persister']['service'];
422
        }
423
424
        $arguments = array(
425 3
            $typeRef,
426 3
            new Reference($transformerId),
427 3
            $typeConfig['model'],
428
        );
429
430 3
        if ($container->hasDefinition('fos_elastica.serializer_callback_prototype')) {
431 2
            $abstractId = 'fos_elastica.object_serializer_persister';
432 2
            $callbackId = sprintf('%s.%s.serializer.callback', $this->indexConfigs[$indexName]['reference'], $typeName);
433 2
            $arguments[] = array(new Reference($callbackId), 'serialize');
434
        } else {
435 1
            $abstractId = 'fos_elastica.object_persister';
436 1
            $mapping = $this->indexConfigs[$indexName]['types'][$typeName]['mapping'];
437 1
            $argument = $mapping['properties'];
438 1
            if (isset($mapping['_parent'])) {
439 1
                $argument['_parent'] = $mapping['_parent'];
440
            }
441 1
            $arguments[] = $argument;
442
        }
443
444 3
        $serviceId = sprintf('fos_elastica.object_persister.%s.%s', $indexName, $typeName);
445 3
        $serviceDef = new DefinitionDecorator($abstractId);
446 3
        foreach ($arguments as $i => $argument) {
447 3
            $serviceDef->replaceArgument($i, $argument);
448
        }
449
450 3
        $container->setDefinition($serviceId, $serviceDef);
451
452 3
        return $serviceId;
453
    }
454
455
    /**
456
     * Loads a provider for a type.
457
     *
458
     * @param array            $typeConfig
459
     * @param ContainerBuilder $container
460
     * @param string           $objectPersisterId
461
     * @param string           $indexName
462
     * @param string           $typeName
463
     *
464
     * @return string
465
     */
466
    private function loadTypeProvider(array $typeConfig, ContainerBuilder $container, $objectPersisterId, $indexName, $typeName)
467
    {
468
        if (isset($typeConfig['provider']['service'])) {
469
            return $typeConfig['provider']['service'];
470
        }
471
472
        /* Note: provider services may conflict with "prototype.driver", if the
473
         * index and type names were "prototype" and a driver, respectively.
474
         */
475
        $providerId = sprintf('fos_elastica.provider.%s.%s', $indexName, $typeName);
476
        $providerDef = new DefinitionDecorator('fos_elastica.provider.prototype.'.$typeConfig['driver']);
477
        $providerDef->addTag('fos_elastica.provider', array('index' => $indexName, 'type' => $typeName));
478
        $providerDef->replaceArgument(0, new Reference($objectPersisterId));
479
        $providerDef->replaceArgument(2, $typeConfig['model']);
480
        // Propel provider can simply ignore Doctrine-specific options
481
        $providerDef->replaceArgument(3, array_merge(array_diff_key($typeConfig['provider'], array('service' => 1)), array(
482
            'indexName' => $indexName,
483
            'typeName' => $typeName,
484
        )));
485
        $container->setDefinition($providerId, $providerDef);
486
487
        return $providerId;
488
    }
489
490
    /**
491
     * Loads doctrine listeners to handle indexing of new or updated objects.
492
     *
493
     * @param array            $typeConfig
494
     * @param ContainerBuilder $container
495
     * @param string           $objectPersisterId
496
     * @param string           $indexName
497
     * @param string           $typeName
498
     *
499
     * @return string
500
     */
501
    private function loadTypeListener(array $typeConfig, ContainerBuilder $container, $objectPersisterId, $indexName, $typeName)
502
    {
503
        if (isset($typeConfig['listener']['service'])) {
504
            return $typeConfig['listener']['service'];
505
        }
506
507
        /* Note: listener services may conflict with "prototype.driver", if the
508
         * index and type names were "prototype" and a driver, respectively.
509
         */
510
        $abstractListenerId = sprintf('fos_elastica.listener.prototype.%s', $typeConfig['driver']);
511
        $listenerId = sprintf('fos_elastica.listener.%s.%s', $indexName, $typeName);
512
        $listenerDef = new DefinitionDecorator($abstractListenerId);
513
        $listenerDef->replaceArgument(0, new Reference($objectPersisterId));
514
        $listenerDef->replaceArgument(2, array(
515
            'identifier' => $typeConfig['identifier'],
516
            'indexName' => $indexName,
517
            'typeName' => $typeName,
518
        ));
519
        $listenerDef->replaceArgument(3, $typeConfig['listener']['logger'] ?
520
            new Reference($typeConfig['listener']['logger']) :
521
            null
522
        );
523
524
        $tagName = null;
525
        switch ($typeConfig['driver']) {
526
            case 'orm':
527
                $tagName = 'doctrine.event_listener';
528
                break;
529
            case 'phpcr':
530
                $tagName = 'doctrine_phpcr.event_listener';
531
                break;
532
            case 'mongodb':
533
                $tagName = 'doctrine_mongodb.odm.event_listener';
534
                break;
535
        }
536
537
        if (null !== $tagName) {
538
            foreach ($this->getDoctrineEvents($typeConfig) as $event) {
539
                $listenerDef->addTag($tagName, array('event' => $event));
540
            }
541
        }
542
543
        $container->setDefinition($listenerId, $listenerDef);
544
545
        return $listenerId;
546
    }
547
548
    /**
549
     * Map Elastica to Doctrine events for the current driver.
550
     */
551
    private function getDoctrineEvents(array $typeConfig)
552
    {
553
        switch ($typeConfig['driver']) {
554
            case 'orm':
555
                $eventsClass = '\Doctrine\ORM\Events';
556
                break;
557
            case 'phpcr':
558
                $eventsClass = '\Doctrine\ODM\PHPCR\Event';
559
                break;
560
            case 'mongodb':
561
                $eventsClass = '\Doctrine\ODM\MongoDB\Events';
562
                break;
563
            default:
564
                throw new InvalidArgumentException(sprintf('Cannot determine events for driver "%s"', $typeConfig['driver']));
565
        }
566
567
        $events = array();
568
        $eventMapping = array(
569
            'insert' => array(constant($eventsClass.'::postPersist')),
570
            'update' => array(constant($eventsClass.'::postUpdate')),
571
            'delete' => array(constant($eventsClass.'::preRemove')),
572
            'flush' => array($typeConfig['listener']['immediate'] ? constant($eventsClass.'::preFlush') : constant($eventsClass.'::postFlush')),
573
        );
574
575
        foreach ($eventMapping as $event => $doctrineEvents) {
576
            if (isset($typeConfig['listener'][$event]) && $typeConfig['listener'][$event]) {
577
                $events = array_merge($events, $doctrineEvents);
578
            }
579
        }
580
581
        return $events;
582
    }
583
584
    /**
585
     * Loads a Type specific Finder.
586
     *
587
     * @param array            $typeConfig
588
     * @param ContainerBuilder $container
589
     * @param string           $elasticaToModelId
590
     * @param Reference        $typeRef
591
     * @param string           $indexName
592
     * @param string           $typeName
593
     *
594
     * @return string
595
     */
596
    private function loadTypeFinder(array $typeConfig, ContainerBuilder $container, $elasticaToModelId, Reference $typeRef, $indexName, $typeName)
597
    {
598
        if (isset($typeConfig['finder']['service'])) {
599
            $finderId = $typeConfig['finder']['service'];
600
        } else {
601
            $finderId = sprintf('fos_elastica.finder.%s.%s', $indexName, $typeName);
602
            $finderDef = new DefinitionDecorator('fos_elastica.finder');
603
            $finderDef->replaceArgument(0, $typeRef);
604
            $finderDef->replaceArgument(1, new Reference($elasticaToModelId));
605
            $container->setDefinition($finderId, $finderDef);
606
        }
607
608
        $managerId = sprintf('fos_elastica.manager.%s', $typeConfig['driver']);
609
        $managerDef = $container->getDefinition($managerId);
610
        $arguments = array( $typeConfig['model'], new Reference($finderId));
611
        if (isset($typeConfig['repository'])) {
612
            $arguments[] = $typeConfig['repository'];
613
        }
614
        $managerDef->addMethodCall('addEntity', $arguments);
615
616
        return $finderId;
617
    }
618
619
    /**
620
     * Loads the index manager.
621
     *
622
     * @param ContainerBuilder $container
623
     **/
624
    private function loadIndexManager(ContainerBuilder $container)
625
    {
626 8
        $indexRefs = array_map(function ($index) {
627 8
            return $index['reference'];
628 8
        }, $this->indexConfigs);
629
630 8
        $managerDef = $container->getDefinition('fos_elastica.index_manager');
631 8
        $managerDef->replaceArgument(0, $indexRefs);
632 8
    }
633
634
    /**
635
     * Makes sure a specific driver has been loaded.
636
     *
637
     * @param ContainerBuilder $container
638
     * @param string           $driver
639
     */
640 3
    private function loadDriver(ContainerBuilder $container, $driver)
641
    {
642 3
        if (in_array($driver, $this->loadedDrivers)) {
643 3
            return;
644
        }
645
646 3
        $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
647 3
        $loader->load($driver.'.xml');
648 3
        $this->loadedDrivers[] = $driver;
649 3
    }
650
651
    /**
652
     * Loads and configures the serializer prototype.
653
     *
654
     * @param array            $config
655
     * @param ContainerBuilder $container
656
     */
657 2
    private function loadSerializer($config, ContainerBuilder $container)
658
    {
659 2
        $container->setAlias('fos_elastica.serializer', $config['serializer']);
660
661 2
        $serializer = $container->getDefinition('fos_elastica.serializer_callback_prototype');
662 2
        $serializer->setClass($config['callback_class']);
663
664 2
        $callbackClassImplementedInterfaces = class_implements($config['callback_class']);
665 2
        if (isset($callbackClassImplementedInterfaces['Symfony\Component\DependencyInjection\ContainerAwareInterface'])) {
666
            $serializer->addMethodCall('setContainer', array(new Reference('service_container')));
667
        }
668 2
    }
669
670
    /**
671
     * Creates a default manager alias for defined default manager or the first loaded driver.
672
     *
673
     * @param string           $defaultManager
674
     * @param ContainerBuilder $container
675
     */
676 8
    private function createDefaultManagerAlias($defaultManager, ContainerBuilder $container)
677
    {
678 8
        if (0 == count($this->loadedDrivers)) {
679 5
            return;
680
        }
681
682 3
        if (count($this->loadedDrivers) > 1
683 3
            && in_array($defaultManager, $this->loadedDrivers)
684
        ) {
685
            $defaultManagerService = $defaultManager;
686
        } else {
687 3
            $defaultManagerService = $this->loadedDrivers[0];
688
        }
689
690 3
        $container->setAlias('fos_elastica.manager', sprintf('fos_elastica.manager.%s', $defaultManagerService));
691 3
    }
692
693
    /**
694
     * Returns a reference to a client given its configured name.
695
     *
696
     * @param string $clientName
697
     *
698
     * @return Reference
699
     *
700
     * @throws \InvalidArgumentException
701
     */
702 2
    private function getClient($clientName)
703
    {
704 2
        if (!array_key_exists($clientName, $this->clients)) {
705
            throw new InvalidArgumentException(sprintf('The elastica client with name "%s" is not defined', $clientName));
706
        }
707
708 2
        return $this->clients[$clientName]['reference'];
709
    }
710
}
711