Add service migrations

Author: HypeMC

AbstractServiceMigration.php

@@ -0,0 +1,24 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Doctrine\Bundle\MigrationsBundle;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\Migrations\AbstractMigration;
+use Psr\Container\ContainerInterface;
+use Psr\Log\LoggerInterface;
+use Symfony\Contracts\Service\ServiceSubscriberInterface;
+
+abstract class AbstractServiceMigration extends AbstractMigration implements ServiceSubscriberInterface
+{
+    /** @var ContainerInterface */
+    protected $container;
+
+    final public function __construct(Connection $connection, LoggerInterface $logger, ContainerInterface $container)
+    {
+        parent::__construct($connection, $logger);
+
+        $this->container = $container;
+    }
+}

DependencyInjection/CompilerPass/RegisterMigrationsPass.php

@@ -0,0 +1,44 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Doctrine\Bundle\MigrationsBundle\DependencyInjection\CompilerPass;
+
+use Doctrine\Bundle\MigrationsBundle\AbstractServiceMigration;
+use Symfony\Component\DependencyInjection\Argument\ServiceLocatorArgument;
+use Symfony\Component\DependencyInjection\ChildDefinition;
+use Symfony\Component\DependencyInjection\Compiler\CompilerPassInterface;
+use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\TypedReference;
+
+use function is_subclass_of;
+
+class RegisterMigrationsPass implements CompilerPassInterface
+{
+    public function process(ContainerBuilder $container): void
+    {
+        $migrationRefs = [];
+
+        foreach ($container->findTaggedServiceIds('doctrine_migrations.migration', true) as $id => $attributes) {
+            $class = $container->getDefinition($id)->getClass();
+            if (is_subclass_of($class, AbstractServiceMigration::class)) {
+                $definition = new ChildDefinition('doctrine_migrations.abstract_migration');
+                $definition->setClass($class);
+                $definition->addTag('container.service_subscriber');
+
+                $container->setDefinition($id, $definition);
+
+                $migrationRefs[$id] = new TypedReference($id, $class);
+            } else {
+                $container->removeDefinition($id);
+            }
+        }
+
+        if ($migrationRefs !== []) {
+            $container->getDefinition('doctrine.migrations.service_migrations_factory')
+                ->replaceArgument(1, new ServiceLocatorArgument($migrationRefs));
+        } else {
+            $container->removeDefinition('doctrine.migrations.service_migrations_factory');
+        }
+    }
+}

DependencyInjection/DoctrineMigrationsExtension.php

@@ -6,6 +6,7 @@
 
 use Doctrine\Bundle\MigrationsBundle\Collector\MigrationsCollector;
 use Doctrine\Bundle\MigrationsBundle\Collector\MigrationsFlattener;
+use Doctrine\Migrations\AbstractMigration;
 use Doctrine\Migrations\Metadata\Storage\MetadataStorage;
 use Doctrine\Migrations\Metadata\Storage\TableMetadataStorageConfiguration;
 use Doctrine\Migrations\Version\MigrationFactory;
@@ -52,6 +53,9 @@ public function load(array $configs, ContainerBuilder $container): void
 
         $loader->load('services.xml');
 
+        $container->registerForAutoconfiguration(AbstractMigration::class)
+            ->addTag('doctrine_migrations.migration');
+
         $configurationDefinition = $container->getDefinition('doctrine.migrations.configuration');
 
         foreach ($config['migrations_paths'] as $ns => $path) {

DoctrineMigrationsBundle.php

@@ -1,8 +1,11 @@
 <?php
 
+declare(strict_types=1);
+
 namespace Doctrine\Bundle\MigrationsBundle;
 
 use Doctrine\Bundle\MigrationsBundle\DependencyInjection\CompilerPass\ConfigureDependencyFactoryPass;
+use Doctrine\Bundle\MigrationsBundle\DependencyInjection\CompilerPass\RegisterMigrationsPass;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
 
@@ -14,9 +17,9 @@
  */
 class DoctrineMigrationsBundle extends Bundle
 {
-    /** @return void */
-    public function build(ContainerBuilder $container)
+    public function build(ContainerBuilder $container): void
     {
-      $container->addCompilerPass(new ConfigureDependencyFactoryPass());
+        $container->addCompilerPass(new ConfigureDependencyFactoryPass());
+        $container->addCompilerPass(new RegisterMigrationsPass());
     }
 }

MigrationsFactory/ServiceMigrationFactory.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Doctrine\Bundle\MigrationsBundle\MigrationsFactory;
+
+use Doctrine\Migrations\AbstractMigration;
+use Doctrine\Migrations\Version\MigrationFactory;
+use Psr\Container\ContainerInterface;
+
+class ServiceMigrationFactory implements MigrationFactory
+{
+    /** @var MigrationFactory */
+    private $migrationFactory;
+
+    /** @var ContainerInterface */
+    private $container;
+
+    public function __construct(MigrationFactory $migrationFactory, ContainerInterface $container)
+    {
+        $this->migrationFactory = $migrationFactory;
+        $this->container        = $container;
+    }
+
+    public function createVersion(string $migrationClassName): AbstractMigration
+    {
+        if ($this->container->has($migrationClassName)) {
+            return $this->container->get($migrationClassName);
+        }
+
+        return $this->migrationFactory->createVersion($migrationClassName);
+    }
+}

Resources/config/services.xml

@@ -42,6 +42,14 @@
             <factory service="doctrine.migrations.dependency_factory" method="getMigrationFactory"/>
         </service>
 
+        <service id="doctrine.migrations.service_migrations_factory"
+                 class="Doctrine\Bundle\MigrationsBundle\MigrationsFactory\ServiceMigrationFactory"
+                 decorates="doctrine.migrations.migrations_factory" decoration-priority="-1"
+        >
+            <argument id="doctrine.migrations.service_migrations_factory.inner" type="service" />
+            <argument type="abstract">migrations locator</argument>
+        </service>
+
         <service id="doctrine.migrations.container_aware_migrations_factory"
                  class="Doctrine\Bundle\MigrationsBundle\MigrationsFactory\ContainerAwareMigrationFactory"
                  decorates="doctrine.migrations.migrations_factory"
@@ -146,6 +154,22 @@
             <tag name="console.command" command="doctrine:migrations:version" />
         </service>
 
+        <service id="doctrine_migrations.abstract_migration" class="Doctrine\Bundle\MigrationsBundle\AbstractServiceMigration" abstract="true">
+
+            <argument type="service">
+                <service class="Doctrine\DBAL\Connection">
+                    <factory service="doctrine.migrations.dependency_factory" method="getConnection" />
+                </service>
+            </argument>
+            <argument type="service">
+                <service class="Psr\Log\LoggerInterface">
+                    <factory service="doctrine.migrations.dependency_factory" method="getLogger" />
+                </service>
+            </argument>
+            <argument></argument>
+
+        </service>
+
     </services>
 
 </container>