src/Core/System/CustomEntity/CustomEntityRegistrar.php line 35

  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\System\CustomEntity;
  5. use Doctrine\DBAL\Connection;
  6. use Doctrine\DBAL\Exception;
  7. use Shopware\Core\Framework\DataAbstractionLayer\DefinitionInstanceRegistry;
  8. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepository;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityLoadedEventFactory;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Read\EntityReaderInterface;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\EntityAggregatorInterface;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\EntitySearcherInterface;
  13. use Shopware\Core\Framework\DataAbstractionLayer\VersionManager;
  14. use Shopware\Core\Framework\Log\Package;
  15. use Shopware\Core\System\CustomEntity\Schema\DynamicEntityDefinition;
  16. use Symfony\Component\DependencyInjection\ContainerInterface;
  18. /**
  19.  * @internal
  20.  */
  21. #[Package('core')]
  22. class CustomEntityRegistrar
  23. {
  24.     public function __construct(private readonly ContainerInterface $container)
  25.     {
  26.     }
  28.     public function register(): void
  29.     {
  30.         if (!$this->container->get(Connection::class)->isConnected()) {
  31.             return;
  32.         }
  34.         try {
  35.             $entities = $this->container->get(Connection::class)->fetchAllAssociative('
  36.                 SELECT custom_entity.name, custom_entity.fields, custom_entity.flags
  37.                 FROM custom_entity
  38.                     LEFT JOIN app ON app.id = custom_entity.app_id
  39.                 WHERE custom_entity.app_id IS NULL OR app.active = 1
  40.             ');
  41.         } catch (Exception) {
  42.             // kernel booted without database connection, or booted for migration and custom entity table not created yet
  43.             return;
  44.         }
  46.         $definitions = [];
  47.         $registry = $this->container->get(DefinitionInstanceRegistry::class);
  49.         foreach ($entities as $entity) {
  50.             $fields = json_decode((string) $entity['fields'], true, 512, \JSON_THROW_ON_ERROR);
  52.             try {
  53.                 $flags = json_decode((string) $entity['flags'], true, 512, \JSON_THROW_ON_ERROR);
  54.             } catch (\JsonException $e) {
  55.                 $flags = [];
  56.             }
  58.             $definition = DynamicEntityDefinition::create($entity['name'], $fields, $flags, $this->container);
  60.             $definitions[] = $definition;
  62.             $this->container->set($definition->getEntityName(), $definition);
  63.             $this->container->set($definition->getEntityName() . '.repository', $this->createRepository($definition));
  64.             $registry->register($definition, $definition->getEntityName());
  65.         }
  67.         foreach ($definitions as $definition) {
  68.             // triggers field generation to generate reverse foreign keys, translation definitions and mapping definitions
  69.             $definition->getFields();
  70.         }
  71.     }
  73.     private function createRepository(DynamicEntityDefinition $definition): EntityRepository
  74.     {
  75.         return new EntityRepository(
  76.             $definition,
  77.             $this->container->get(EntityReaderInterface::class),
  78.             $this->container->get(VersionManager::class),
  79.             $this->container->get(EntitySearcherInterface::class),
  80.             $this->container->get(EntityAggregatorInterface::class),
  81.             $this->container->get('event_dispatcher'),
  82.             $this->container->get(EntityLoadedEventFactory::class)
  83.         );
  84.     }
  85. }