Completed
Push — master ( 795e6d...9bc2f5 )
by Karel
11s
created

FOSElasticaExtension::loadDriver()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 10
Code Lines 6

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 7
CRAP Score 2

Importance

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