Support for a custom repository class

Author: ondrejmirtes

extension.neon

@@ -1,3 +1,7 @@
+parameters:
+	doctrine:
+		repositoryClass: Doctrine\ORM\EntityRepository
+
 services:
 	-
 		class: PHPStan\Reflection\Doctrine\DoctrineSelectableClassReflectionExtension
@@ -13,6 +17,8 @@ services:
 			- phpstan.broker.dynamicMethodReturnTypeExtension
 	-
 		class: PHPStan\Type\Doctrine\EntityManagerGetRepositoryDynamicReturnTypeExtension
+		arguments:
+			repositoryClass: %doctrine.repositoryClass%
 		tags:
 			- phpstan.broker.dynamicMethodReturnTypeExtension
 	-

src/Type/Doctrine/EntityManagerGetRepositoryDynamicReturnTypeExtension.php

@@ -10,6 +10,16 @@
 class EntityManagerGetRepositoryDynamicReturnTypeExtension implements \PHPStan\Type\DynamicMethodReturnTypeExtension
 {
 
+	/**
+	 * @var string
+	 */
+	private $repositoryClass;
+
+	public function __construct(string $repositoryClass)
+	{
+		$this->repositoryClass = $repositoryClass;
+	}
+
 	public static function getClass(): string
 	{
 		return \Doctrine\ORM\EntityManager::class;
@@ -48,7 +58,7 @@ public function getTypeFromMethodCall(
 			$class = $scope->getClassReflection()->getName();
 		}
 
-		return new EntityRepositoryType($class);
+		return new EntityRepositoryType($class, $this->repositoryClass);
 	}
 
 }

src/Type/Doctrine/EntityRepositoryType.php

@@ -12,9 +12,9 @@ class EntityRepositoryType extends ObjectType
 	 */
 	private $entityClass;
 
-	public function __construct(string $entityClass)
+	public function __construct(string $entityClass, string $repositoryClass)
 	{
-		parent::__construct(\Doctrine\ORM\EntityRepository::class);
+		parent::__construct($repositoryClass);
 		$this->entityClass = $entityClass;
 	}