Add auto search for annotation extension name

Kdyby · Jan 14, 2018 · f28011a · f28011a
1 parent 6adb732
commit f28011a
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/Kdyby/Doctrine/DI/OrmExtension.php b/src/Kdyby/Doctrine/DI/OrmExtension.php
@@ -622,8 +622,15 @@ protected function processMetadataDriver(Nette\DI\ServiceDefinition $metadataDri
 		}
 
 		if ($impl === self::ANNOTATION_DRIVER) {
+			$annotationExtensionName = self::ANNOTATION_DRIVER;
+			foreach ($this->compiler->getExtensions(AnnotationsExtension::class) as $annotationExtension) {
+				$annotationExtensionName = $annotationExtension->name;
+
+				break;
+			}
+
 			$driver->arguments = [
-				'@' . self::ANNOTATION_DRIVER . '.reader',
+				"@{$annotationExtensionName}.reader",
 				Nette\Utils\Arrays::flatten($driver->arguments)
 			];
 		}