Support DoctrineBundle 2.8.1+

Author: ondrejmirtes

extension.neon

@@ -24,7 +24,6 @@ parameters:
 		- stubs/EntityManager.stub
 		- stubs/EntityManagerDecorator.stub
 		- stubs/EntityManagerInterface.stub
-		- stubs/ServiceEntityRepository.stub
 		- stubs/MongoClassMetadataInfo.stub
 
 		- stubs/Persistence/ManagerRegistry.stub

src/Stubs/Doctrine/StubFilesExtensionLoader.php

@@ -3,6 +3,7 @@
 namespace PHPStan\Stubs\Doctrine;
 
 use PHPStan\PhpDoc\StubFilesExtension;
+use function class_exists;
 use function dirname;
 
 class StubFilesExtensionLoader implements StubFilesExtension
@@ -20,16 +21,25 @@ public function __construct(
 
 	public function getFiles(): array
 	{
-		$path = dirname(dirname(dirname(__DIR__))) . '/stubs';
+		$stubsDir = dirname(dirname(dirname(__DIR__))) . '/stubs';
+		$path = $stubsDir;
 
 		if ($this->bleedingEdge === true) {
 			$path .= '/bleedingEdge';
 		}
 
-		return [
+		$files = [
 			$path . '/ORM/QueryBuilder.stub',
 			$path . '/EntityRepository.stub',
 		];
+
+		if (class_exists('Doctrine\Bundle\DoctrineBundle\Repository\LazyServiceEntityRepository')) {
+			$files[] = $stubsDir . '/LazyServiceEntityRepository.stub';
+		} else {
+			$files[] = $stubsDir . '/ServiceEntityRepository.stub';
+		}
+
+		return $files;
 	}
 
 }

stubs/LazyServiceEntityRepository.stub

@@ -0,0 +1,18 @@
+<?php
+namespace Doctrine\Bundle\DoctrineBundle\Repository;
+
+use Doctrine\ORM\EntityRepository;
+
+/**
+ * @template TEntityClass of object
+ * @template-extends LazyServiceEntityRepository<TEntityClass>
+ */
+class ServiceEntityRepository extends LazyServiceEntityRepository {
+}
+
+/**
+ * @template TEntityClass of object
+ * @template-extends EntityRepository<TEntityClass>
+ */
+class LazyServiceEntityRepository extends EntityRepository {
+}