Apply docblock changes using DocBlockUpdater (#135)

src/Rector/Assign/CallOnAppArrayAccessToStandaloneAssignRector.php

@@ -12,6 +12,7 @@
 use PhpParser\Node\Expr\Variable;
 use PhpParser\Node\Stmt\Expression;
 use PHPStan\Type\ObjectType;
+use Rector\Comments\NodeDocBlock\DocBlockUpdater;
 use Rector\Core\Rector\AbstractRector;
 use RectorLaravel\NodeFactory\AppAssignFactory;
 use RectorLaravel\ValueObject\ServiceNameTypeAndVariableName;
@@ -30,6 +31,7 @@ final class CallOnAppArrayAccessToStandaloneAssignRector extends AbstractRector
 
     public function __construct(
         private readonly AppAssignFactory $appAssignFactory,
+        private readonly DocBlockUpdater $docBlockUpdater,
     ) {
         $this->serviceNameTypeAndVariableNames[] = new ServiceNameTypeAndVariableName(
             'validator',
@@ -88,6 +90,8 @@ public function refactor(Node $node): Node|array|int|null
                 $methodCall->var
             );
 
+            $this->docBlockUpdater->updateRefactoredNodeWithPhpDocInfo($assignExpression);
+
             $methodCall->var = new Variable($serviceNameTypeAndVariableName->getVariableName());
 
             // the nop is a workaround because the docs of the first node are somehow stripped away

src/Rector/ClassMethod/AddGenericReturnTypeToRelationsRector.php

@@ -18,6 +18,7 @@
 use PHPStan\Type\Generic\GenericObjectType;
 use PHPStan\Type\ObjectType;
 use Rector\BetterPhpDocParser\ValueObject\Type\FullyQualifiedIdentifierTypeNode;
+use Rector\Comments\NodeDocBlock\DocBlockUpdater;
 use Rector\Core\Rector\AbstractScopeAwareRector;
 use Rector\NodeTypeResolver\TypeComparator\TypeComparator;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
@@ -38,7 +39,8 @@ class AddGenericReturnTypeToRelationsRector extends AbstractScopeAwareRector
     private const RELATION_WITH_CHILD_METHODS = ['belongsTo', 'morphTo'];
 
     public function __construct(
-        private readonly TypeComparator $typeComparator
+        private readonly TypeComparator $typeComparator,
+        private readonly DocBlockUpdater $docBlockUpdater,
     ) {
     }
 
@@ -174,6 +176,8 @@ public function refactorWithScope(Node $node, Scope $scope): ?Node
             $phpDocInfo->addTagValueNode(new ReturnTagValueNode($genericTypeNode, ''));
         }
 
+        $this->docBlockUpdater->updateRefactoredNodeWithPhpDocInfo($node);
+
         return $node;
     }
 

src/Rector/Class_/AddExtendsAnnotationToModelFactoriesRector.php

@@ -15,6 +15,7 @@
 use PHPStan\PhpDocParser\Ast\Type\GenericTypeNode;
 use PHPStan\Type\ObjectType;
 use Rector\BetterPhpDocParser\ValueObject\Type\FullyQualifiedIdentifierTypeNode;
+use Rector\Comments\NodeDocBlock\DocBlockUpdater;
 use Rector\Core\Rector\AbstractRector;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
 use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
@@ -30,6 +31,11 @@ final class AddExtendsAnnotationToModelFactoriesRector extends AbstractRector
 
     private const FACTORY_CLASS_NAME = 'Illuminate\Database\Eloquent\Factories\Factory';
 
+    public function __construct(
+        private readonly DocBlockUpdater $docBlockUpdater,
+    ) {
+    }
+
     public function getRuleDefinition(): RuleDefinition
     {
         return new RuleDefinition('Adds the @extends annotation to Factories.', [
@@ -87,6 +93,8 @@ public function refactor(Node $node): ?Node
 
             $this->addExtendsPhpDocTag($node, $stmt);
 
+            $this->docBlockUpdater->updateRefactoredNodeWithPhpDocInfo($node);
+
             break;
         }