Merge pull request #590 from mabar/initialize-metadata

[Nette] Initialize metadata on DIC startup - v4.0
nextras · Feb 8, 2023 · b8e87d4 · b8e87d4
2 parents 2abc593 + 4d289ec
commit b8e87d4
Showing 1 changed file with 26 additions and 5 deletions.
diff --git a/src/Bridges/NetteDI/OrmExtension.php b/src/Bridges/NetteDI/OrmExtension.php
@@ -27,16 +27,19 @@ class OrmExtension extends CompilerExtension
 	/** @var string */
 	protected $modelClass;
 
+	/** @var array<mixed> */
+	private $configDefaults = [
+		'model' => Model::class,
+		'repositoryFinder' => PhpDocRepositoryFinder::class,
+		'initializeMetadata' => false,
+	];
+
 
 	public function loadConfiguration()
 	{
 		$this->builder = $this->getContainerBuilder();
 
-		$configDefaults = [
-			'model' => Model::class,
-			'repositoryFinder' => PhpDocRepositoryFinder::class,
-		];
-		$config = $this->validateConfig($configDefaults); // @phpstan-ignore-line
+		$config = $this->validateConfig($this->configDefaults); // @phpstan-ignore-line
 		$this->modelClass = $config['model'];
 
 		$repositoryFinderClass = $config['repositoryFinder'];
@@ -74,6 +77,12 @@ public function beforeCompile()
 		$this->setupDbalMapperDependencies();
 	}
 
+	public function afterCompile(ClassType $class)
+	{
+		$config = $this->validateConfig($this->configDefaults); // @phpstan-ignore-line
+
+		$this->initializeMetadata($class, $config['initializeMetadata']);
+	}
 
 	protected function setupCache(): void
 	{
@@ -183,4 +192,16 @@ protected function setupModel(string $modelClass, array $repositoriesConfig): vo
 				'metadataStorage' => $this->prefix('@metadataStorage'),
 			]);
 	}
+
+	protected function initializeMetadata(ClassType $classType, bool $init)
+	{
+		if (!$init) {
+			return;
+		}
+
+		$initialize = $classType->getMethod('initialize');
+		$initialize->addBody('$this->getService(?);', [
+			$this->prefix('metadataStorage'),
+		]);
+	}
 }