add document repository support (#461)

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,9 +37,11 @@ public function __construct(
 
     public function getRuleDefinition(): RuleDefinition
     {
-        return new RuleDefinition('Add @extends ServiceEntityRepository<T> annotation to repository classes', [
-            new CodeSample(
-                <<<'CODE_SAMPLE'
+        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;
 
 final class SomeRepository extends ServiceEntityRepository
@@ -50,8 +52,8 @@ public function __construct(ManagerRegistry $registry)
     }
 }
 CODE_SAMPLE
-                ,
-                <<<'CODE_SAMPLE'
+                    ,
+                    <<<'CODE_SAMPLE'
 use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
 
 /**
@@ -65,8 +67,10 @@ public function __construct(ManagerRegistry $registry)
     }
 }
 CODE_SAMPLE
-            ),
-        ]);
+                ),
+
+            ]
+        );
     }
 
     /**
@@ -102,7 +106,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