Merge pull request #43 from kbond/auto-register

[RFC] Add auto-register compiler pass

Author: cmodijk

doc/command_bus_bundle.md

@@ -64,6 +64,14 @@ services:
             - { name: command_handler, handles: Fully\Qualified\Class\Name\Of\RegisterUser }
 ```
 
+### Auto-Register command handlers
+
+You can omit the `handles` tag attribute if your handler meets the following conditions:
+
+1. Uses the "class_based" command name resolving strategy
+2. Handles a single command using the `__invoke` method
+3. Has a single, non optional class type hinted `__invoke` method parameter
+
 > #### Command handlers are lazy-loaded
 >
 > Since only one of the command handlers is going to handle any particular command, command handlers are lazy-loaded.

doc/event_bus_bundle.md

@@ -65,6 +65,14 @@ services:
             - { name: event_subscriber, subscribes_to: Fully\Qualified\Class\Name\Of\UserRegistered }
 ```
 
+### Auto-Register event subscribers
+
+You can omit the `subscribes_to` tag attribute if your subscriber meets the following conditions:
+
+1. Uses the "class_based" event name resolving strategy
+2. Subscribers to single event using the `__invoke` method
+3. Has a single, non optional class type hinted `__invoke` method parameter
+
 > #### Event subscribers are lazy-loaded
 >
 > Since only some of the event subscribers are going to handle any particular event, event subscribers are lazy-loaded.

src/DependencyInjection/Compiler/AutoRegister.php

@@ -0,0 +1,58 @@
+<?php
+
+namespace SimpleBus\SymfonyBridge\DependencyInjection\Compiler;
+
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+
+/**
+ * @author Kevin Bond <[email protected]>
+ */
+final class AutoRegister implements CompilerPassInterface
+{
+    private $tagName;
+    private $tagAttribute;
+
+    public function __construct($tagName, $tagAttribute)
+    {
+        $this->tagName = $tagName;
+        $this->tagAttribute = $tagAttribute;
+    }
+
+    public function process(ContainerBuilder $container)
+    {
+        foreach ($container->findTaggedServiceIds($this->tagName) as $serviceId => $tags) {
+            foreach ($tags as $tagAttributes) {
+
+                // if tag attributes are set, skip
+                if (isset($tagAttributes[$this->tagAttribute])) {
+                    continue;
+                }
+
+                $definition = $container->getDefinition($serviceId);
+
+                // check if service id is class name
+                $reflectionClass = new \ReflectionClass($definition->getClass() ?: $serviceId);
+
+                // if no __invoke method, skip
+                if (!$reflectionClass->hasMethod('__invoke')) {
+                    continue;
+                }
+
+                $invokeParameters = $reflectionClass->getMethod('__invoke')->getParameters();
+
+                // if no param or optional param, skip
+                if (count($invokeParameters) !== 1 || $invokeParameters[0]->isOptional()) {
+                    return;
+                }
+
+                // get the class name
+                $handles = $invokeParameters[0]->getClass()->getName();
+
+                // auto handle
+                $definition->clearTag($this->tagName);
+                $definition->addTag($this->tagName, [$this->tagAttribute => $handles]);
+            }
+        }
+    }
+}

src/SimpleBusCommandBusBundle.php

@@ -3,8 +3,10 @@
 namespace SimpleBus\SymfonyBridge;
 
 use SimpleBus\SymfonyBridge\DependencyInjection\CommandBusExtension;
+use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\AutoRegister;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\ConfigureMiddlewares;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\RegisterHandlers;
+use Symfony\Component\DependencyInjection\Compiler\PassConfig;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
 
@@ -19,6 +21,12 @@ public function __construct($alias = 'command_bus')
 
     public function build(ContainerBuilder $container)
     {
+        $container->addCompilerPass(
+            new AutoRegister('command_handler', 'handles'),
+            PassConfig::TYPE_BEFORE_OPTIMIZATION,
+            10
+        );
+
         $container->addCompilerPass(
             new ConfigureMiddlewares(
                 'command_bus',

src/SimpleBusEventBusBundle.php

@@ -3,11 +3,13 @@
 namespace SimpleBus\SymfonyBridge;
 
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\AddMiddlewareTags;
+use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\AutoRegister;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\CompilerPassUtil;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\ConfigureMiddlewares;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\RegisterMessageRecorders;
 use SimpleBus\SymfonyBridge\DependencyInjection\Compiler\RegisterSubscribers;
 use SimpleBus\SymfonyBridge\DependencyInjection\EventBusExtension;
+use Symfony\Component\DependencyInjection\Compiler\PassConfig;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
 
@@ -26,6 +28,12 @@ public function build(ContainerBuilder $container)
     {
         $this->checkRequirements(array('SimpleBusCommandBusBundle'), $container);
 
+        $container->addCompilerPass(
+            new AutoRegister('event_subscriber', 'subscribes_to'),
+            PassConfig::TYPE_BEFORE_OPTIMIZATION,
+            10
+        );
+
         $container->addCompilerPass(
             new ConfigureMiddlewares(
                 'event_bus',

tests/Functional/SmokeTest.php

@@ -4,20 +4,27 @@
 
 use Doctrine\ORM\EntityManager;
 use Doctrine\ORM\Tools\SchemaTool;
+use SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto\AutoCommand;
+use SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto\AutoEvent;
 use SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\TestCommand;
 use SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\TestKernel;
+use Symfony\Bundle\FrameworkBundle\Test\KernelTestCase;
 use Symfony\Component\DependencyInjection\ContainerInterface;
 
-class SmokeTest extends \PHPUnit_Framework_TestCase
+class SmokeTest extends KernelTestCase
 {
+    protected static function getKernelClass()
+    {
+        return TestKernel::class;
+    }
+
     /**
      * @test
      */
     public function it_handles_a_command_then_dispatches_events_for_all_modified_entities()
     {
-        $kernel = new TestKernel('test', true);
-        $kernel->boot();
-        $container = $kernel->getContainer();
+        self::bootKernel(['environment' => 'config1']);
+        $container = self::$kernel->getContainer();
 
         $this->createSchema($container);
 
@@ -42,6 +49,42 @@ public function it_handles_a_command_then_dispatches_events_for_all_modified_ent
         $this->assertContains('event_bus.DEBUG: Finished notifying a subscriber', $loggedMessages);
     }
 
+    /**
+     * @test
+     */
+    public function it_can_auto_register_event_subscribers()
+    {
+        self::bootKernel(['environment' => 'config2']);
+        $container = self::$kernel->getContainer();
+
+        $subscriber = $container->get('auto_event_subscriber');
+        $event = new AutoEvent();
+
+        $this->assertNull($subscriber->handled);
+
+        $container->get('event_bus')->handle($event);
+
+        $this->assertSame($event, $subscriber->handled);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_auto_register_command_handlers()
+    {
+        self::bootKernel(['environment' => 'config2']);
+        $container = self::$kernel->getContainer();
+
+        $handler = $container->get('auto_command_handler');
+        $command = new AutoCommand();
+
+        $this->assertNull($handler->handled);
+
+        $container->get('command_bus')->handle($command);
+
+        $this->assertSame($command, $handler->handled);
+    }
+
     private function createSchema(ContainerInterface $container)
     {
         $entityManager = $container->get('doctrine.orm.entity_manager');

tests/Functional/SmokeTest/Auto/AutoCommand.php

@@ -0,0 +1,7 @@
+<?php
+
+namespace SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto;
+
+final class AutoCommand
+{
+}

tests/Functional/SmokeTest/Auto/AutoCommandHandler.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto;
+
+final class AutoCommandHandler
+{
+    public $handled;
+
+    public function __invoke(AutoCommand $command)
+    {
+        $this->handled = $command;
+    }
+}

tests/Functional/SmokeTest/Auto/AutoEvent.php

@@ -0,0 +1,7 @@
+<?php
+
+namespace SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto;
+
+final class AutoEvent
+{
+}

tests/Functional/SmokeTest/Auto/AutoEventSubscriber.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace SimpleBus\SymfonyBridge\Tests\Functional\SmokeTest\Auto;
+
+final class AutoEventSubscriber
+{
+    public $handled;
+
+    public function __invoke(AutoEvent $event)
+    {
+        $this->handled = $event;
+    }
+}