FriendsOfSymfony / FOSElasticaBundle

use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;

use Symfony\Component\DependencyInjection\ContainerBuilder;

final class RegisterPagerProvidersPass implements CompilerPassInterface

{

    /**

     * {@inheritdoc}

     */

    public function process(ContainerBuilder $container)

    {

        if (!$container->hasDefinition('fos_elastica.pager_provider_registry')) {

            return;

        }

        $defaultIndex = $container->getParameter('fos_elastica.default_index');

        $registry = $container->getDefinition('fos_elastica.pager_provider_registry');

        $registeredProviders = [];

        foreach ($container->findTaggedServiceIds('fos_elastica.pager_provider', true) as $id => $attributes) {

            foreach ($attributes as $attribute) {

                if (!isset($attribute['type'])) {

                    throw new \InvalidArgumentException(sprintf('Elastica provider "%s" must specify the "type" attribute.', $id));

                }

                $index = isset($attribute['index']) ? $attribute['index'] : $defaultIndex;

                $type = $attribute['type'];

                if (isset($registeredProviders[$index][$type])) {

                    throw new \InvalidArgumentException(sprintf(

                        'Cannot register provider "%s". The provider "%s" has been registered for same index "%s" and type "%s"',

                        $id,

                        $registeredProviders[$index][$type],

                        $index,

                        $type

                    ));

                }

                $providerDef = $container->getDefinition($id);

                if (!$providerDef->getFactory() && $providerDef->getClass()) {

                    // You are on your own if you use a factory to create a provider.

                    // It would fail in runtime if the factory does not return a proper provider.

                    $this->assertClassImplementsPagerProviderInterface($id, $container->getParameterBag()->resolveValue($providerDef->getClass()));

                }

                if (!$providerDef->isPublic()) {

                    throw new \InvalidArgumentException(sprintf('Elastica persister "%s" must be a public service', $id));

                }

                $registeredProviders[$index][$type] = $id;

            }

        }

        $registry->replaceArgument(0, $registeredProviders);

    }

    /**

     * @param $providerId

     * @param $providerClass

     *

     * @throws \InvalidArgumentException if provider service does not implement PagerProviderInterface

     */

    private function assertClassImplementsPagerProviderInterface($providerId, $providerClass)

    {

        $rc = new \ReflectionClass($providerClass);

        if (!$rc->implementsInterface(PagerProviderInterface::class)) {

            throw new \InvalidArgumentException(sprintf(

                'Elastica provider "%s" with class "%s" must implement "%s".',

                $providerId,

                $providerClass,

                PagerProviderInterface::class

            ));

        }

    }

}

use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;

use Symfony\Component\DependencyInjection\ContainerBuilder;

final class RegisterPersistersPass implements CompilerPassInterface

{

    /**

     * {@inheritdoc}

     */

    public function process(ContainerBuilder $container)

    {

        if (!$container->hasDefinition('fos_elastica.persister_registry')) {

            return;

        }

        $defaultIndex = $container->getParameter('fos_elastica.default_index');

        $registry = $container->getDefinition('fos_elastica.persister_registry');

        $registeredPersisters = [];

        foreach ($container->findTaggedServiceIds('fos_elastica.persister', true) as $id => $attributes) {

            foreach ($attributes as $attribute) {

                if (!isset($attribute['type'])) {

                    throw new \InvalidArgumentException(sprintf('Elastica persister "%s" must specify the "type" attribute.', $id));

                }

                $index = isset($attribute['index']) ? $attribute['index'] : $defaultIndex;

                $type = $attribute['type'];

                if (isset($registeredPersisters[$index][$type])) {

                    throw new \InvalidArgumentException(sprintf(

                        'Cannot register persister "%s". The persister "%s" has been registered for same index "%s" and type "%s"',

                        $id,

                        $registeredPersisters[$index][$type],

                        $index,

                        $type

                    ));

                }

                $persisterDef = $container->getDefinition($id);

                if (!$persisterDef->getFactory() && $persisterDef->getClass()) {

                    // You are on your own if you use a factory to create a persister.

                    // It would fail in runtime if the factory does not return a proper persister.

                    $this->assertClassImplementsPersisterInterface($id, $container->getParameterBag()->resolveValue($persisterDef->getClass()));

                }

                if (!$persisterDef->isPublic()) {

                    throw new \InvalidArgumentException(sprintf('Elastica persister "%s" must be a public service', $id));

                }

                $registeredPersisters[$index][$type] = $id;

            }

        }

        $registry->replaceArgument(0, $registeredPersisters);

    }

    /**

     * @param $persisterId

     * @param $persisterClass

     *

     * @throws \InvalidArgumentException if persister service does not implement ObjectPersisterInterface

     */

    private function assertClassImplementsPersisterInterface($persisterId, $persisterClass)

    {

        $rc = new \ReflectionClass($persisterClass);

        if (!$rc->implementsInterface(ObjectPersisterInterface::class)) {

            throw new \InvalidArgumentException(sprintf(

                'Elastica persister "%s" with class "%s" must implement "%s".',

                $persisterId,

                $persisterClass,

                ObjectPersisterInterface::class

            ));

        }

    }

}