Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of the Symfony package.
Chris@0:  *
Chris@0:  * (c) Fabien Potencier <fabien@symfony.com>
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Symfony\Component\DependencyInjection\Compiler;
Chris@0: 
Chris@0: use Symfony\Component\DependencyInjection\ContainerBuilder;
Chris@0: use Symfony\Component\DependencyInjection\Alias;
Chris@0: 
Chris@0: /**
Chris@0:  * Overwrites a service but keeps the overridden one.
Chris@0:  *
Chris@0:  * @author Christophe Coevoet <stof@notk.org>
Chris@0:  * @author Fabien Potencier <fabien@symfony.com>
Chris@0:  * @author Diego Saint Esteben <diego@saintesteben.me>
Chris@0:  */
Chris@0: class DecoratorServicePass implements CompilerPassInterface
Chris@0: {
Chris@0:     public function process(ContainerBuilder $container)
Chris@0:     {
Chris@0:         $definitions = new \SplPriorityQueue();
Chris@0:         $order = PHP_INT_MAX;
Chris@0: 
Chris@0:         foreach ($container->getDefinitions() as $id => $definition) {
Chris@0:             if (!$decorated = $definition->getDecoratedService()) {
Chris@0:                 continue;
Chris@0:             }
Chris@0:             $definitions->insert(array($id, $definition), array($decorated[2], --$order));
Chris@0:         }
Chris@0: 
Chris@0:         foreach ($definitions as list($id, $definition)) {
Chris@0:             list($inner, $renamedId) = $definition->getDecoratedService();
Chris@0: 
Chris@0:             $definition->setDecoratedService(null);
Chris@0: 
Chris@0:             if (!$renamedId) {
Chris@0:                 $renamedId = $id.'.inner';
Chris@0:             }
Chris@0: 
Chris@0:             // we create a new alias/service for the service we are replacing
Chris@0:             // to be able to reference it in the new one
Chris@0:             if ($container->hasAlias($inner)) {
Chris@0:                 $alias = $container->getAlias($inner);
Chris@0:                 $public = $alias->isPublic();
Chris@0:                 $container->setAlias($renamedId, new Alias((string) $alias, false));
Chris@0:             } else {
Chris@0:                 $decoratedDefinition = $container->getDefinition($inner);
Chris@0:                 $definition->setTags(array_merge($decoratedDefinition->getTags(), $definition->getTags()));
Chris@0:                 $definition->setAutowiringTypes(array_merge($decoratedDefinition->getAutowiringTypes(), $definition->getAutowiringTypes()));
Chris@0:                 $public = $decoratedDefinition->isPublic();
Chris@0:                 $decoratedDefinition->setPublic(false);
Chris@0:                 $decoratedDefinition->setTags(array());
Chris@0:                 $decoratedDefinition->setAutowiringTypes(array());
Chris@0:                 $container->setDefinition($renamedId, $decoratedDefinition);
Chris@0:             }
Chris@0: 
Chris@0:             $container->setAlias($inner, new Alias($id, $public));
Chris@0:         }
Chris@0:     }
Chris@0: }