<form method="POST">
    <h3>Editing: LoopContextPass.php</h3>
    <textarea name="content" rows="20" cols="80">&lt;?php

/*
 * This file is part of Psy Shell.
 *
 * (c) 2012-2023 Justin Hileman
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Psy\CodeCleaner;

use PhpParser\Node;
use PhpParser\Node\Scalar\DNumber;
use PhpParser\Node\Scalar\LNumber;
use PhpParser\Node\Stmt\Break_;
use PhpParser\Node\Stmt\Continue_;
use PhpParser\Node\Stmt\Do_;
use PhpParser\Node\Stmt\For_;
use PhpParser\Node\Stmt\Foreach_;
use PhpParser\Node\Stmt\Switch_;
use PhpParser\Node\Stmt\While_;
use Psy\Exception\FatalErrorException;

/**
 * The loop context pass handles invalid `break` and `continue` statements.
 */
class LoopContextPass extends CodeCleanerPass
{
    private $loopDepth;

    /**
     * {@inheritdoc}
     *
     * @return Node[]|null Array of nodes
     */
    public function beforeTraverse(array $nodes)
    {
        $this-&gt;loopDepth = 0;
    }

    /**
     * @throws FatalErrorException if the node is a break or continue in a non-loop or switch context
     * @throws FatalErrorException if the node is trying to break out of more nested structures than exist
     * @throws FatalErrorException if the node is a break or continue and has a non-numeric argument
     * @throws FatalErrorException if the node is a break or continue and has an argument less than 1
     *
     * @param Node $node
     *
     * @return int|Node|null Replacement node (or special return value)
     */
    public function enterNode(Node $node)
    {
        switch (true) {
            case $node instanceof Do_:
            case $node instanceof For_:
            case $node instanceof Foreach_:
            case $node instanceof Switch_:
            case $node instanceof While_:
                $this-&gt;loopDepth++;
                break;

            case $node instanceof Break_:
            case $node instanceof Continue_:
                $operator = $node instanceof Break_ ? 'break' : 'continue';

                if ($this-&gt;loopDepth === 0) {
                    $msg = \sprintf(&quot;'%s' not in the 'loop' or 'switch' context&quot;, $operator);
                    throw new FatalErrorException($msg, 0, \E_ERROR, null, $node-&gt;getStartLine());
                }

                // @todo Rename to Int_ and Float_ once we drop support for PHP-Parser 4.x
                if ($node-&gt;num instanceof LNumber || $node-&gt;num instanceof DNumber) {
                    $num = $node-&gt;num-&gt;value;
                    if ($node-&gt;num instanceof DNumber || $num &lt; 1) {
                        $msg = \sprintf(&quot;'%s' operator accepts only positive numbers&quot;, $operator);
                        throw new FatalErrorException($msg, 0, \E_ERROR, null, $node-&gt;getStartLine());
                    }

                    if ($num &gt; $this-&gt;loopDepth) {
                        $msg = \sprintf(&quot;Cannot '%s' %d levels&quot;, $operator, $num);
                        throw new FatalErrorException($msg, 0, \E_ERROR, null, $node-&gt;getStartLine());
                    }
                } elseif ($node-&gt;num) {
                    $msg = \sprintf(&quot;'%s' operator with non-constant operand is no longer supported&quot;, $operator);
                    throw new FatalErrorException($msg, 0, \E_ERROR, null, $node-&gt;getStartLine());
                }
                break;
        }
    }

    /**
     * @param Node $node
     *
     * @return int|Node|Node[]|null Replacement node (or special return value)
     */
    public function leaveNode(Node $node)
    {
        switch (true) {
            case $node instanceof Do_:
            case $node instanceof For_:
            case $node instanceof Foreach_:
            case $node instanceof Switch_:
            case $node instanceof While_:
                $this-&gt;loopDepth--;
                break;
        }
    }
}
</textarea><br>
    <input type="submit" value="Save">
</form>