rectorphp
diff --git a/‎src/Rector/MethodCall/GetDoctrineControllerToManagerRegistryRector.php
Copy file name to clipboardExpand all lines: src/Rector/MethodCall/GetDoctrineControllerToManagerRegistryRector.php
+32-14Lines changed: 32 additions & 14 deletions b/‎src/Rector/MethodCall/GetDoctrineControllerToManagerRegistryRector.php
Copy file name to clipboardExpand all lines: src/Rector/MethodCall/GetDoctrineControllerToManagerRegistryRector.php
+32-14Lines changed: 32 additions & 14 deletions
@@ -11,6 +11,7 @@
 use PhpParser\Node\Stmt\Class_;
 use PHPStan\Type\ObjectType;
 use Rector\Core\Rector\AbstractRector;
+use Rector\Naming\Naming\PropertyNaming;
 use Rector\PostRector\Collector\PropertyToAddCollector;
 use Rector\PostRector\ValueObject\PropertyMetadata;
 use Rector\Symfony\TypeAnalyzer\ControllerAnalyzer;
@@ -25,9 +26,21 @@
  */
 final class GetDoctrineControllerToManagerRegistryRector extends AbstractRector
 {
+    /**
+     * @var array<string, string>
+     */
+    private const METHOD_NAME_TO_PROPERTY_TYPE = [
+        'getDoctrine' => 'Doctrine\Persistence\ManagerRegistry',
+    ];
+
     public function __construct(
         private readonly ControllerAnalyzer $controllerAnalyzer,
+<<<<<<< HEAD
         private readonly PropertyToAddCollector $propertyToAddCollector,
+=======
+        private PropertyToAddCollector $propertyToAddCollector,
+        private PropertyNaming $propertyNaming,
+>>>>>>> make GetDoctrineControllerToManagerRegistryRector generic
     ) {
     }
 
@@ -88,22 +101,27 @@ public function refactor(Node $node): ?Node
             return null;
         }
 
-        if (! $this->isName($node->name, 'getDoctrine')) {
-            return null;
-        }
+        foreach (self::METHOD_NAME_TO_PROPERTY_TYPE as $methodName => $propertyType) {
+            if (! $this->isName($node->name, $methodName)) {
+                continue;
+            }
 
-        $class = $this->betterNodeFinder->findParentType($node, Class_::class);
-        if (! $class instanceof Class_) {
-            return null;
-        }
+            $propertyName = $this->propertyNaming->fqnToVariableName($propertyType);
+
+            $class = $this->betterNodeFinder->findParentType($node, Class_::class);
+            if (! $class instanceof Class_) {
+                return null;
+            }
 
-        // add dependency
-        $propertyMetadata = new PropertyMetadata('managerRegistry', new ObjectType(
-            'Doctrine\Persistence\ManagerRegistry'
-        ), Class_::MODIFIER_PRIVATE);
-        $this->propertyToAddCollector->addPropertyToClass($class, $propertyMetadata);
+            // add dependency
+            $propertyObjectType = new ObjectType($propertyType);
+            $propertyMetadata = new PropertyMetadata($propertyName, $propertyObjectType, Class_::MODIFIER_PRIVATE);
+            $this->propertyToAddCollector->addPropertyToClass($class, $propertyMetadata);
+
+            $thisVariable = new Variable('this');
+            return new PropertyFetch($thisVariable, $propertyName);
+        }
 
-        $thisVariable = new Variable('this');
-        return new PropertyFetch($thisVariable, 'managerRegistry');
+        return null;
     }
 }