[DowngradePhp84] Apply on return on DowngradeArrayAllRector

Author: samsonasik

rules/DowngradePhp84/Rector/Expression/DowngradeArrayAllRector.php

@@ -10,12 +10,16 @@
 use PhpParser\Node\Expr\BooleanNot;
 use PhpParser\Node\Expr\ConstFetch;
 use PhpParser\Node\Expr\FuncCall;
+use PhpParser\Node\Expr\Variable;
 use PhpParser\Node\Name;
 use PhpParser\Node\Stmt;
 use PhpParser\Node\Stmt\Break_;
 use PhpParser\Node\Stmt\Expression;
 use PhpParser\Node\Stmt\Foreach_;
 use PhpParser\Node\Stmt\If_;
+use PhpParser\Node\Stmt\Return_;
+use Rector\Naming\Naming\VariableNaming;
+use Rector\PHPStan\ScopeFetcher;
 use Rector\Rector\AbstractRector;
 use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
 use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
@@ -27,9 +31,15 @@
  */
 final class DowngradeArrayAllRector extends AbstractRector
 {
+    public function __construct(
+        private VariableNaming $variableNaming
+    )
+    {
+    }
+
     public function getNodeTypes(): array
     {
-        return [Expression::class];
+        return [Expression::class, Return_::class];
     }
 
     public function getRuleDefinition(): RuleDefinition
@@ -57,28 +67,28 @@ public function getRuleDefinition(): RuleDefinition
     }
 
     /**
-     * @param Expression $node
+     * @param Expression|Return_ $node
      * @return Stmt[]|null
      */
     public function refactor(Node $node): ?array
     {
-        if (! $node->expr instanceof Assign) {
-            return null;
-        }
+        $expr = $node instanceof Expression && $node->expr instanceof Assign
+            ? $node->expr->expr
+            : $node->expr;
 
-        if (! $node->expr->expr instanceof FuncCall) {
+        if (! $expr instanceof FuncCall) {
             return null;
         }
 
-        if (! $this->isName($node->expr->expr, 'array_all')) {
+        if (! $this->isName($expr, 'array_all')) {
             return null;
         }
 
-        if ($node->expr->expr->isFirstClassCallable()) {
+        if ($expr->isFirstClassCallable()) {
             return null;
         }
 
-        $args = $node->expr->expr->getArgs();
+        $args = $expr->getArgs();
         if (count($args) !== 2) {
             return null;
         }
@@ -87,17 +97,22 @@ public function refactor(Node $node): ?array
             return null;
         }
 
+        $scope = ScopeFetcher::fetch($node);
+        $variable = $node instanceof Expression && $node->expr instanceof Assign
+            ? $node->expr->var
+            : new Variable($this->variableNaming->createCountedValueName('found', $scope));
+
         $valueCond = $args[1]->value->expr;
         $if = new If_(new BooleanNot($valueCond), [
             'stmts' => [
-                new Expression(new Assign($node->expr->var, new ConstFetch(new Name('false')))),
+                new Expression(new Assign($variable, new ConstFetch(new Name('false')))),
                 new Break_(),
             ],
         ]);
 
-        return [
+        $result = [
             // init
-            new Expression(new Assign($node->expr->var, new ConstFetch(new Name('true')))),
+            new Expression(new Assign($variable, new ConstFetch(new Name('true')))),
 
             // foreach loop
             new Foreach_(
@@ -113,5 +128,11 @@ public function refactor(Node $node): ?array
                     ],
             ),
         ];
+
+        if ($node instanceof Return_) {
+            $result[] = new Return_($variable);
+        }
+
+        return $result;
     }
 }