rector/rules/Php80/NodeAnalyzer/SwitchAnalyzer.php

<?php

declare (strict_types=1);
namespace Rector\Php80\NodeAnalyzer;

use PhpParser\Node;
use PhpParser\Node\Expr;
use PhpParser\Node\Stmt\Break_;
use PhpParser\Node\Stmt\Case_;
use PhpParser\Node\Stmt\Return_;
use PhpParser\Node\Stmt\Switch_;
use PHPStan\Type\MixedType;
use Rector\NodeTypeResolver\NodeTypeResolver;
use Rector\NodeTypeResolver\PHPStan\Type\TypeFactory;
final class SwitchAnalyzer
{
    /**
     * @readonly
     * @var \Rector\NodeTypeResolver\NodeTypeResolver
     */
    private $nodeTypeResolver;
    /**
     * @readonly
     * @var \Rector\NodeTypeResolver\PHPStan\Type\TypeFactory
     */
    private $typeFactory;
    public function __construct(NodeTypeResolver $nodeTypeResolver, TypeFactory $typeFactory)
    {
        $this->nodeTypeResolver = $nodeTypeResolver;
        $this->typeFactory = $typeFactory;
    }
    /**
     * @param Case_[] $cases
     */
    public function hasDifferentTypeCases(array $cases, Expr $expr) : bool
    {
        $types = [];
        foreach ($cases as $case) {
            if ($case->cond instanceof Expr) {
                $types[] = $this->nodeTypeResolver->getType($case->cond);
            }
        }
        if ($types === []) {
            return \false;
        }
        $uniqueTypes = $this->typeFactory->uniquateTypes($types);
        $countUniqueTypes = \count($uniqueTypes);
        if ($countUniqueTypes === 1 && $uniqueTypes[0]->isInteger()->yes()) {
            $switchCondType = $this->nodeTypeResolver->getType($expr);
            if (!$switchCondType instanceof MixedType && $switchCondType->isString()->maybe()) {
                return \true;
            }
        }
        return $countUniqueTypes > 1;
    }
    public function hasEachCaseBreak(Switch_ $switch) : bool
    {
        $totalCases = \count($switch->cases);
        if ($totalCases === 1) {
            return $this->containsCaseReturn($switch->cases[0]);
        }
        foreach ($switch->cases as $key => $case) {
            if ($key === $totalCases - 1) {
                return \true;
            }
            if ($this->hasBreakOrReturnOrEmpty($case)) {
                continue;
            }
            return \false;
        }
        return \true;
    }
    public function hasEachCaseSingleStmt(Switch_ $switch) : bool
    {
        foreach ($switch->cases as $case) {
            if (!$case->cond instanceof Expr) {
                continue;
            }
            $stmtsWithoutBreak = \array_filter($case->stmts, static function (Node $node) : bool {
                return !$node instanceof Break_;
            });
            if (\count($stmtsWithoutBreak) !== 1) {
                return \false;
            }
        }
        return \true;
    }
    public function hasDefaultSingleStmt(Switch_ $switch) : bool
    {
        foreach ($switch->cases as $case) {
            if (!$case->cond instanceof Expr) {
                $stmtsWithoutBreak = \array_filter($case->stmts, static function (Node $node) : bool {
                    return !$node instanceof Break_;
                });
                return \count($stmtsWithoutBreak) === 1;
            }
        }
        return \false;
    }
    private function hasBreakOrReturnOrEmpty(Case_ $case) : bool
    {
        if ($case->stmts === []) {
            return \true;
        }
        foreach ($case->stmts as $caseStmt) {
            if ($caseStmt instanceof Break_) {
                return \true;
            }
            if ($caseStmt instanceof Return_) {
                return \true;
            }
        }
        return \false;
    }
    private function containsCaseReturn(Case_ $case) : bool
    {
        foreach ($case->stmts as $stmt) {
            if ($stmt instanceof Return_) {
                return \true;
            }
        }
        return \false;
    }
}