wrap Expression

rectorphp · Aug 24, 2023 · f54fa71 · f54fa71
1 parent 0012df8
commit f54fa71
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/Rector/AbstractRector.php b/src/Rector/AbstractRector.php
@@ -5,6 +5,9 @@
 namespace Rector\Core\Rector;
 
 use PhpParser\Node;
+use PhpParser\Node\Expr;
+use PhpParser\Node\Stmt;
+use PhpParser\Node\Stmt\Expression;
 use PhpParser\Node\Stmt\InlineHTML;
 use PhpParser\Node\Stmt\Nop;
 use PhpParser\NodeTraverser;
@@ -320,6 +323,10 @@ private function postRefactorProcess(
             return $originalNode;
         }
 
+        if ($refactoredNode instanceof Expr && $originalNode instanceof Stmt) {
+            $refactoredNode = new Expression($refactoredNode);
+        }
+
         $this->refreshScopeNodes($refactoredNode, $filePath, $currentScope);
 
         $this->nodesToReturn[$originalNodeHash] = $refactoredNode;