Completed
Pull Request — master (#1134)
by Istvan
14:27
created

FOSElasticaExtension::load()   C

Complexity

Conditions 7
Paths 17

Size

Total Lines 44
Code Lines 24

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 28
CRAP Score 7.002

Importance

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