<form method="POST">
    <h3>Editing: ImplicitReturnPass.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\Expr;
use PhpParser\Node\Expr\Exit_;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Break_;
use PhpParser\Node\Stmt\Expression;
use PhpParser\Node\Stmt\If_;
use PhpParser\Node\Stmt\Namespace_;
use PhpParser\Node\Stmt\Return_;
use PhpParser\Node\Stmt\Switch_;

/**
 * Add an implicit &quot;return&quot; to the last statement, provided it can be returned.
 */
class ImplicitReturnPass extends CodeCleanerPass
{
    /**
     * @param array $nodes
     *
     * @return array
     */
    public function beforeTraverse(array $nodes): array
    {
        return $this-&gt;addImplicitReturn($nodes);
    }

    /**
     * @param array $nodes
     *
     * @return array
     */
    private function addImplicitReturn(array $nodes): array
    {
        // If nodes is empty, it can't have a return value.
        if (empty($nodes)) {
            return [new Return_(NoReturnValue::create())];
        }

        $last = \end($nodes);

        // Special case a few types of statements to add an implicit return
        // value (even though they technically don't have any return value)
        // because showing a return value in these instances is useful and not
        // very surprising.
        if ($last instanceof If_) {
            $last-&gt;stmts = $this-&gt;addImplicitReturn($last-&gt;stmts);

            foreach ($last-&gt;elseifs as $elseif) {
                $elseif-&gt;stmts = $this-&gt;addImplicitReturn($elseif-&gt;stmts);
            }

            if ($last-&gt;else) {
                $last-&gt;else-&gt;stmts = $this-&gt;addImplicitReturn($last-&gt;else-&gt;stmts);
            }
        } elseif ($last instanceof Switch_) {
            foreach ($last-&gt;cases as $case) {
                // only add an implicit return to cases which end in break
                $caseLast = \end($case-&gt;stmts);
                if ($caseLast instanceof Break_) {
                    $case-&gt;stmts = $this-&gt;addImplicitReturn(\array_slice($case-&gt;stmts, 0, -1));
                    $case-&gt;stmts[] = $caseLast;
                }
            }
        } elseif ($last instanceof Expr &amp;&amp; !($last instanceof Exit_)) {
            // @codeCoverageIgnoreStart
            $nodes[\count($nodes) - 1] = new Return_($last, [
                'startLine' =&gt; $last-&gt;getStartLine(),
                'endLine'   =&gt; $last-&gt;getEndLine(),
            ]);
        // @codeCoverageIgnoreEnd
        } elseif ($last instanceof Expression &amp;&amp; !($last-&gt;expr instanceof Exit_)) {
            $nodes[\count($nodes) - 1] = new Return_($last-&gt;expr, [
                'startLine' =&gt; $last-&gt;getStartLine(),
                'endLine'   =&gt; $last-&gt;getEndLine(),
            ]);
        } elseif ($last instanceof Namespace_) {
            $last-&gt;stmts = $this-&gt;addImplicitReturn($last-&gt;stmts);
        }

        // Return a &quot;no return value&quot; for all non-expression statements, so that
        // PsySH can suppress the `null` that `eval()` returns otherwise.
        //
        // Note that statements special cased above (if/elseif/else, switch)
        // _might_ implicitly return a value before this catch-all return is
        // reached.
        //
        // We're not adding a fallback return after namespace statements,
        // because code outside namespace statements doesn't really work, and
        // there's already an implicit return in the namespace statement anyway.
        if (self::isNonExpressionStmt($last)) {
            $nodes[] = new Return_(NoReturnValue::create());
        }

        return $nodes;
    }

    /**
     * Check whether a given node is a non-expression statement.
     *
     * As of PHP Parser 4.x, Expressions are now instances of Stmt as well, so
     * we'll exclude them here.
     *
     * @param Node $node
     */
    private static function isNonExpressionStmt(Node $node): bool
    {
        return $node instanceof Stmt &amp;&amp;
            !$node instanceof Expression &amp;&amp;
            !$node instanceof Return_ &amp;&amp;
            !$node instanceof Namespace_;
    }
}
</textarea><br>
    <input type="submit" value="Save">
</form>