move reindex

rectorphp · Dec 18, 2024 · db9f844 · db9f844
1 parent ba2ca77
commit db9f844
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 12 deletions.
diff --git a/src/Application/NodeAttributeReIndexer.php b/src/Application/NodeAttributeReIndexer.php
@@ -14,16 +14,35 @@
 use PhpParser\Node\Expr\StaticCall;
 use PhpParser\Node\FunctionLike;
 use PhpParser\Node\MatchArm;
+use PhpParser\Node\Stmt\ClassLike;
 use PhpParser\Node\Stmt\ClassMethod;
+use PhpParser\Node\Stmt\Declare_;
 use PhpParser\Node\Stmt\Function_;
 use PhpParser\Node\Stmt\If_;
 use PhpParser\Node\Stmt\Switch_;
 use PhpParser\Node\Stmt\TryCatch;
+use Rector\Contract\PhpParser\Node\StmtsAwareInterface;
 
 final class NodeAttributeReIndexer
 {
     public static function reIndexNodeAttributes(Node $node): ?Node
     {
+        if (($node instanceof StmtsAwareInterface || $node instanceof ClassLike || $node instanceof Declare_) && $node->stmts !== null) {
+            $node->stmts = array_values($node->stmts);
+
+            if ($node instanceof If_) {
+                $node->elseifs = array_values($node->elseifs);
+                return $node;
+            }
+
+            if ($node instanceof TryCatch) {
+                $node->catches = array_values($node->catches);
+                return $node;
+            }
+
+            return $node;
+        }
+
         if ($node instanceof CallLike) {
             /** @var FuncCall|MethodCall|New_|NullsafeMethodCall|StaticCall $node */
             $node->args = array_values($node->args);
@@ -41,16 +60,6 @@ public static function reIndexNodeAttributes(Node $node): ?Node
             return $node;
         }
 
-        if ($node instanceof If_) {
-            $node->elseifs = array_values($node->elseifs);
-            return $node;
-        }
-
-        if ($node instanceof TryCatch) {
-            $node->catches = array_values($node->catches);
-            return $node;
-        }
-
         if ($node instanceof Switch_) {
             $node->cases = array_values($node->cases);
             return $node;

diff --git a/src/NodeTypeResolver/PHPStan/Scope/NodeVisitor/StmtKeyNodeVisitor.php b/src/NodeTypeResolver/PHPStan/Scope/NodeVisitor/StmtKeyNodeVisitor.php
@@ -24,8 +24,6 @@ public function enterNode(Node $node): ?Node
             return null;
         }
 
-        $node->stmts = array_values($node->stmts);
-
         // re-index stmt key under current node
         foreach ($node->stmts as $key => $childStmt) {
             $childStmt->setAttribute(AttributeKey::STMT_KEY, $key);