vendor/zenstruck/foundry/src/Bundle/DependencyInjection/ZenstruckFoundryExtension.php line 126

  1. <?php
  3. /*
  4.  * This file is part of the zenstruck/foundry package.
  5.  *
  6.  * (c) Kevin Bond <kevinbond@gmail.com>
  7.  *
  8.  * For the full copyright and license information, please view the LICENSE
  9.  * file that was distributed with this source code.
  10.  */
  12. namespace Zenstruck\Foundry\Bundle\DependencyInjection;
  14. use Doctrine\Bundle\DoctrineBundle\DoctrineBundle;
  15. use Doctrine\Bundle\MongoDBBundle\DoctrineMongoDBBundle;
  16. use Symfony\Bundle\MakerBundle\Maker\AbstractMaker;
  17. use Symfony\Bundle\MakerBundle\MakerBundle;
  18. use Symfony\Component\Config\FileLocator;
  19. use Symfony\Component\DependencyInjection\ContainerBuilder;
  20. use Symfony\Component\DependencyInjection\Loader\FileLoader;
  21. use Symfony\Component\DependencyInjection\Loader\XmlFileLoader;
  22. use Symfony\Component\HttpKernel\DependencyInjection\ConfigurableExtension;
  23. use Zenstruck\Foundry\Bundle\Command\StubMakeFactory;
  24. use Zenstruck\Foundry\Bundle\Command\StubMakeStory;
  25. use Zenstruck\Foundry\ModelFactory;
  26. use Zenstruck\Foundry\Story;
  27. use Zenstruck\Foundry\Test\ORMDatabaseResetter;
  29. /**
  30.  * @author Kevin Bond <kevinbond@gmail.com>
  31.  */
  32. final class ZenstruckFoundryExtension extends ConfigurableExtension
  33. {
  34.     protected function loadInternal(array $mergedConfig, ContainerBuilder $container): void
  35.     {
  36.         $loader = new XmlFileLoader($container, new FileLocator(__DIR__.'/../Resources/config'));
  38.         $loader->load('services.xml');
  40.         $container->registerForAutoconfiguration(Story::class)
  41.             ->addTag('foundry.story')
  42.         ;
  44.         $container->registerForAutoconfiguration(ModelFactory::class)
  45.             ->addTag('foundry.factory')
  46.         ;
  48.         $this->configureFaker($mergedConfig['faker'], $container);
  49.         $this->configureDefaultInstantiator($mergedConfig['instantiator'], $container);
  50.         $this->configureDatabaseResetter($mergedConfig['database_resetter'], $container);
  51.         $this->configureMakeFactory($mergedConfig['make_factory'], $container, $loader);
  53.         if (true === $mergedConfig['auto_refresh_proxies']) {
  54.             $container->getDefinition('.zenstruck_foundry.configuration')->addMethodCall('enableDefaultProxyAutoRefresh');
  55.         } elseif (false === $mergedConfig['auto_refresh_proxies']) {
  56.             $container->getDefinition('.zenstruck_foundry.configuration')->addMethodCall('disableDefaultProxyAutoRefresh');
  57.         }
  58.     }
  60.     private function configureFaker(array $config, ContainerBuilder $container): void
  61.     {
  62.         if ($config['service']) {
  63.             $container->setAlias('.zenstruck_foundry.faker', $config['service']);
  65.             return;
  66.         }
  68.         $definition = $container->getDefinition('.zenstruck_foundry.faker');
  70.         if ($config['locale']) {
  71.             $definition->addArgument($config['locale']);
  72.         }
  74.         if ($config['seed']) {
  75.             $definition->addMethodCall('seed', [$config['seed']]);
  76.         }
  77.     }
  79.     private function configureDefaultInstantiator(array $config, ContainerBuilder $container): void
  80.     {
  81.         if ($config['service']) {
  82.             $container->setAlias('.zenstruck_foundry.default_instantiator', $config['service']);
  84.             return;
  85.         }
  87.         $definition = $container->getDefinition('.zenstruck_foundry.default_instantiator');
  89.         if ($config['without_constructor']) {
  90.             $definition->addMethodCall('withoutConstructor');
  91.         }
  93.         if ($config['allow_extra_attributes']) {
  94.             $definition->addMethodCall('allowExtraAttributes');
  95.         }
  97.         if ($config['always_force_properties']) {
  98.             $definition->addMethodCall('alwaysForceProperties');
  99.         }
  100.     }
  102.     private function configureDatabaseResetter(array $config, ContainerBuilder $container): void
  103.     {
  104.         $configurationDefinition = $container->getDefinition('.zenstruck_foundry.configuration');
  106.         if (false === $config['enabled']) {
  107.             $configurationDefinition->addMethodCall('disableDatabaseReset');
  108.         }
  110.         if (isset($config['orm']) && !self::isBundleLoaded($container, DoctrineBundle::class)) {
  111.             throw new \InvalidArgumentException('doctrine/doctrine-bundle should be enabled to use config under "database_resetter.orm".');
  112.         }
  114.         if (isset($config['odm']) && !self::isBundleLoaded($container, DoctrineMongoDBBundle::class)) {
  115.             throw new \InvalidArgumentException('doctrine/mongodb-odm-bundle should be enabled to use config under "database_resetter.odm".');
  116.         }
  118.         $configurationDefinition->setArgument('$ormConnectionsToReset', $config['orm']['connections'] ?? []);
  119.         $configurationDefinition->setArgument('$ormObjectManagersToReset', $config['orm']['object_managers'] ?? []);
  120.         $configurationDefinition->setArgument('$ormResetMode', $config['orm']['reset_mode'] ?? ORMDatabaseResetter::RESET_MODE_SCHEMA);
  121.         $configurationDefinition->setArgument('$odmObjectManagersToReset', $config['odm']['object_managers'] ?? []);
  122.     }
  124.     private function configureMakeFactory(array $makerConfig, ContainerBuilder $container, FileLoader $loader): void
  125.     {
  126.         if (\class_exists(AbstractMaker::class) && self::isBundleLoaded($container, MakerBundle::class)) {
  127.             $loader->load('maker.xml');
  129.             $makeFactoryDefinition = $container->getDefinition('.zenstruck_foundry.maker.factory');
  130.             $makeFactoryDefinition->setArgument('$defaultNamespace', $makerConfig['default_namespace']);
  131.         } else {
  132.             $container->register('.zenstruck_foundry.maker.factory_stub', StubMakeFactory::class)->addTag('console.command');
  133.             $container->register('.zenstruck_foundry.maker.story_stub', StubMakeStory::class)->addTag('console.command');
  134.         }
  135.     }
  137.     private static function isBundleLoaded(ContainerBuilder $container, string $bundleName): bool
  138.     {
  139.         return \in_array(
  140.             $bundleName,
  141.             \is_array($bundles = $container->getParameter('kernel.bundles')) ? $bundles : [],
  142.             true
  143.         );
  144.     }
  145. }