use existing loop check

Author: TomasVotruba

rules/Php70/Rector/Break_/BreakNotInLoopOrSwitchToReturnRector.php

@@ -5,12 +5,8 @@
 namespace Rector\Php70\Rector\Break_;
 
 use PhpParser\Node;
-use PhpParser\Node\Expr\ArrowFunction;
-use PhpParser\Node\FunctionLike;
 use PhpParser\Node\Stmt\Break_;
-use PhpParser\Node\Stmt\Class_;
 use PhpParser\Node\Stmt\Return_;
-use PhpParser\Node\Stmt\Switch_;
 use PhpParser\NodeVisitor;
 use Rector\NodeNestingScope\ContextAnalyzer;
 use Rector\Rector\AbstractRector;
@@ -24,11 +20,6 @@
  */
 final class BreakNotInLoopOrSwitchToReturnRector extends AbstractRector implements MinPhpVersionInterface
 {
-    /**
-     * @var string
-     */
-    private const IS_BREAK_IN_SWITCH = 'is_break_in_switch';
-
     public function __construct(
         private readonly ContextAnalyzer $contextAnalyzer
     ) {
@@ -82,43 +73,19 @@ public function run()
      */
     public function getNodeTypes(): array
     {
-        return [Switch_::class, Break_::class];
+        return [Break_::class];
     }
 
     /**
-     * @param Switch_|Break_ $node
+     * @param Break_ $node
      * @return Return_|null|NodeVisitor::REMOVE_NODE
      */
     public function refactor(Node $node): Return_|null|int
     {
-        if ($node instanceof Switch_) {
-            $this->traverseNodesWithCallable(
-                $node->cases,
-                static function (Node $subNode): ?int {
-                    if ($subNode instanceof Class_ || ($subNode instanceof FunctionLike && ! $subNode instanceof ArrowFunction)) {
-                        return NodeVisitor::DONT_TRAVERSE_CURRENT_AND_CHILDREN;
-                    }
-
-                    if (! $subNode instanceof Break_) {
-                        return null;
-                    }
-
-                    $subNode->setAttribute(self::IS_BREAK_IN_SWITCH, true);
-                    return null;
-                }
-            );
-
-            return null;
-        }
-
         if ($this->contextAnalyzer->isInLoop($node)) {
             return null;
         }
 
-        if ($node->getAttribute(self::IS_BREAK_IN_SWITCH) === true) {
-            return null;
-        }
-
         if ($this->contextAnalyzer->isInIf($node)) {
             return new Return_();
         }

src/NodeNestingScope/ContextAnalyzer.php

@@ -17,7 +17,7 @@ final class ContextAnalyzer
      */
     public function isInLoop(Node $node): bool
     {
-        return $node->getAttribute(AttributeKey::IS_IN_LOOP) === true;
+        return $node->getAttribute(AttributeKey::IS_IN_LOOP_OR_SWITCH) === true;
     }
 
     /**

src/NodeTypeResolver/Node/AttributeKey.php

@@ -172,7 +172,7 @@ final class AttributeKey
     /**
      * @var string
      */
-    public const IS_IN_LOOP = 'is_in_loop';
+    public const IS_IN_LOOP_OR_SWITCH = 'is_in_loop';
 
     /**
      * @var string

src/NodeTypeResolver/PHPStan/Scope/NodeVisitor/ContextNodeVisitor.php

@@ -162,6 +162,7 @@ private function processContextInLoop(For_|Foreach_|While_|Do_|Switch_ $node): v
         }
 
         $stmts = $node instanceof Switch_ ? $node->cases : $node->stmts;
+
         $this->simpleCallableNodeTraverser->traverseNodesWithCallable(
             $stmts,
             static function (Node $subNode): ?int {
@@ -170,7 +171,7 @@ static function (Node $subNode): ?int {
                 }
 
                 if ($subNode instanceof If_ || $subNode instanceof Break_) {
-                    $subNode->setAttribute(AttributeKey::IS_IN_LOOP, true);
+                    $subNode->setAttribute(AttributeKey::IS_IN_LOOP_OR_SWITCH, true);
                 }
 
                 return null;