4 * This file is part of the Behat.
5 * (c) Konstantin Kudryashov <ever.zet@gmail.com>
7 * For the full copyright and license information, please view the LICENSE
8 * file that was distributed with this source code.
11 namespace Behat\Behat\Snippet\ServiceContainer;
13 use Behat\Testwork\Cli\ServiceContainer\CliExtension;
14 use Behat\Testwork\EventDispatcher\ServiceContainer\EventDispatcherExtension;
15 use Behat\Testwork\ServiceContainer\Extension;
16 use Behat\Testwork\ServiceContainer\ExtensionManager;
17 use Behat\Testwork\ServiceContainer\ServiceProcessor;
18 use Behat\Testwork\Translator\ServiceContainer\TranslatorExtension;
19 use Symfony\Component\Config\Definition\Builder\ArrayNodeDefinition;
20 use Symfony\Component\DependencyInjection\ContainerBuilder;
21 use Symfony\Component\DependencyInjection\Definition;
22 use Symfony\Component\DependencyInjection\Reference;
25 * Provides snippet generation, printing and appending functionality.
27 * @author Konstantin Kudryashov <ever.zet@gmail.com>
29 class SnippetExtension implements Extension
34 const REGISTRY_ID = 'snippet.registry';
35 const WRITER_ID = 'snippet.writer';
38 * Available extension points
40 const GENERATOR_TAG = 'snippet.generator';
41 const APPENDER_TAG = 'snippet.appender';
44 * @var ServiceProcessor
49 * Initializes extension.
51 * @param null|ServiceProcessor $processor
53 public function __construct(ServiceProcessor $processor = null)
55 $this->processor = $processor ? : new ServiceProcessor();
61 public function getConfigKey()
69 public function initialize(ExtensionManager $extensionManager)
76 public function configure(ArrayNodeDefinition $builder)
83 public function load(ContainerBuilder $container, array $config)
85 $this->loadController($container);
86 $this->loadRegistry($container);
87 $this->loadWriter($container);
93 public function process(ContainerBuilder $container)
95 $this->processGenerators($container);
96 $this->processAppenders($container);
100 * @param ContainerBuilder $container
102 protected function loadController(ContainerBuilder $container)
104 $definition = new Definition('Behat\Behat\Snippet\Printer\ConsoleSnippetPrinter', array(
105 new Reference(CliExtension::OUTPUT_ID),
106 new Reference(TranslatorExtension::TRANSLATOR_ID)
108 $container->setDefinition('snippet.printer', $definition);
110 $definition = new Definition('Behat\Behat\Snippet\Cli\SnippetsController', array(
111 new Reference(self::REGISTRY_ID),
112 new Reference(self::WRITER_ID),
113 new Reference('snippet.printer'),
114 new Reference(EventDispatcherExtension::DISPATCHER_ID)
116 $definition->addTag(CliExtension::CONTROLLER_TAG, array('priority' => 400));
117 $container->setDefinition(CliExtension::CONTROLLER_TAG . '.snippet', $definition);
121 * @param ContainerBuilder $container
123 protected function loadRegistry(ContainerBuilder $container)
125 $definition = new Definition('Behat\Behat\Snippet\SnippetRegistry');
126 $container->setDefinition(self::REGISTRY_ID, $definition);
130 * @param ContainerBuilder $container
132 protected function loadWriter(ContainerBuilder $container)
134 $definition = new Definition('Behat\Behat\Snippet\SnippetWriter');
135 $container->setDefinition(self::WRITER_ID, $definition);
139 * @param ContainerBuilder $container
141 protected function processGenerators(ContainerBuilder $container)
143 $references = $this->processor->findAndSortTaggedServices($container, self::GENERATOR_TAG);
144 $definition = $container->getDefinition(self::REGISTRY_ID);
146 foreach ($references as $reference) {
147 $definition->addMethodCall('registerSnippetGenerator', array($reference));
152 * @param ContainerBuilder $container
154 protected function processAppenders(ContainerBuilder $container)
156 $references = $this->processor->findAndSortTaggedServices($container, self::APPENDER_TAG);
157 $definition = $container->getDefinition(self::WRITER_ID);
159 foreach ($references as $reference) {
160 $definition->addMethodCall('registerSnippetAppender', array($reference));