add document repository support

Author: TomasVotruba

config/sets/doctrine-code-quality.php

@@ -25,7 +25,7 @@
         TypedPropertyFromToOneRelationTypeRector::class,
         CompleteReturnDocblockFromToManyRector::class,
 
-        // annotations generics
+        // @extends annotations service repository generics
         AddAnnotationToRepositoryRector::class,
     ]);
 

rules-tests/Bundle230/Rector/Class_/AddAnnotationToRepositoryRector/Fixture/some_service_document_repository.php.inc

@@ -0,0 +1,38 @@
+<?php
+
+namespace Rector\Doctrine\Tests\Bundle230\Rector\Class_\AddAnnotationToRepositoryRector\Fixture;
+
+use Doctrine\Bundle\MongoDBBundle\Repository\ServiceDocumentRepository;
+use Doctrine\Persistence\ManagerRegistry;
+use App\Entity\SomeEntity;
+
+final class SomeServiceDocumentRepository extends ServiceDocumentRepository
+{
+    public function __construct(ManagerRegistry $registry)
+    {
+        parent::__construct($registry, SomeEntity::class);
+    }
+}
+
+?>
+-----
+<?php
+
+namespace Rector\Doctrine\Tests\Bundle230\Rector\Class_\AddAnnotationToRepositoryRector\Fixture;
+
+use Doctrine\Bundle\MongoDBBundle\Repository\ServiceDocumentRepository;
+use Doctrine\Persistence\ManagerRegistry;
+use App\Entity\SomeEntity;
+
+/**
+ * @extends \Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository<\App\Entity\SomeEntity>
+ */
+final class SomeServiceDocumentRepository extends ServiceDocumentRepository
+{
+    public function __construct(ManagerRegistry $registry)
+    {
+        parent::__construct($registry, SomeEntity::class);
+    }
+}
+
+?>

rules/Bundle230/Rector/Class_/AddAnnotationToRepositoryRector.php

@@ -37,7 +37,9 @@ public function __construct(
 
     public function getRuleDefinition(): RuleDefinition
     {
-        return new RuleDefinition('Add @extends ServiceEntityRepository<T> annotation to repository classes', [
+        return new RuleDefinition(
+            'Add @extends ServiceEntityRepository<T> annotation to repository classes that extends ServiceEntityRepository or ServiceDocumentRepository',
+            [
             new CodeSample(
                 <<<'CODE_SAMPLE'
 use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
@@ -66,6 +68,7 @@ public function __construct(ManagerRegistry $registry)
 }
 CODE_SAMPLE
             ),
+        
         ]);
     }
 
@@ -102,7 +105,11 @@ private function isRepositoryClass(Class_ $class): bool
             return false;
         }
 
-        return $this->isName($class->extends, DoctrineClass::SERVICE_ENTITY_REPOSITORY);
+        if ($this->isName($class->extends, DoctrineClass::SERVICE_ENTITY_REPOSITORY)) {
+            return true;
+        }
+
+        return $this->isName($class->extends, DoctrineClass::SERVICE_DOCUMENT_REPOSITORY);
     }
 
     private function getEntityClassFromConstructor(Class_ $class): ?string