Completed
Pull Request — master (#1569)
by
unknown
04:46
created

FOSElasticaExtension::loadSerializer()   A

Complexity

Conditions 2
Paths 2

Size

Total Lines 11

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 6
CRAP Score 2.0116

Importance

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