RepeatedAndNotEqualToNotInArrayRector.php

<?php

declare(strict_types=1);

namespace Rector\CodeQuality\Rector\BooleanAnd;

use PhpParser\Node;
use PhpParser\Node\Arg;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\BinaryOp\BooleanAnd;
use PhpParser\Node\Expr\BinaryOp\NotEqual;
use PhpParser\Node\Expr\BinaryOp\NotIdentical;
use PhpParser\Node\Expr\BooleanNot;
use PhpParser\Node\Expr\ConstFetch;
use PhpParser\Node\Expr\FuncCall;
use PhpParser\Node\Name;
use Rector\CodeQuality\NodeFactory\InArrayFromRepeatedCompareFactory;
use Rector\CodeQuality\ValueObject\ComparedExprAndValueExpr;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Rector\PhpParser\Node\BetterNodeFinder;
use Rector\Rector\AbstractRector;
use Rector\RuleDoc\CodeSample\CodeSample;
use Rector\RuleDoc\RuleDefinition;

/**
 * @see \Rector\Tests\CodeQuality\Rector\BooleanAnd\RepeatedAndNotEqualToNotInArrayRector\RepeatedAndNotEqualToNotInArrayRectorTest
 */
final class RepeatedAndNotEqualToNotInArrayRector extends AbstractRector
{
    public function __construct(
        private readonly BetterNodeFinder $betterNodeFinder,
        private readonly InArrayFromRepeatedCompareFactory $inArrayFromRepeatedCompareFactory,
    ) {
    }

    public function getRuleDefinition(): RuleDefinition
    {
        return new RuleDefinition(
            'Simplify repeated && compare of same value, to ! in_array() call',
            [
                new CodeSample(
                    <<<'CODE_SAMPLE'
if ($value !== 10 && $value !== 20 && $value !== 30) {
    // ...
}

CODE_SAMPLE
                    ,
                    <<<'CODE_SAMPLE'
if (! in_array($value, [10, 20, 30], true)) {
    // ...
}
CODE_SAMPLE
                ),
            ]
        );
    }

    /**
     * @return array<class-string<Node>>
     */
    public function getNodeTypes(): array
    {
        return [BooleanAnd::class];
    }

    /**
     * @param BooleanAnd $node
     */
    public function refactor(Node $node): ?BooleanNot
    {
        // in_array() func call is not allowed in a constant expression
        if ($this->isInConstantExpression($node)) {
            return null;
        }

        if (! $this->isNotEqualOrNotIdentical($node->right)) {
            return null;
        }

        // match compared variable and expr
        if (! $node->left instanceof BooleanAnd && ! $this->isNotEqualOrNotIdentical($node->left)) {
            return null;
        }

        $comparedExprAndValueExprs = $this->matchComparedAndDesiredValues($node);
        if ($comparedExprAndValueExprs === null) {
            return null;
        }

        $args = $this->inArrayFromRepeatedCompareFactory->createInArrayArgs($comparedExprAndValueExprs);
        if ($args === null) {
            return null;
        }

        if ($this->isStrictComparison($node)) {
            $args[] = new Arg(new ConstFetch(new Name('true')));
        }

        $inArrayFuncCall = new FuncCall(new Name('in_array'), $args);

        return new BooleanNot($inArrayFuncCall);
    }

    private function isInConstantExpression(BooleanAnd $booleanAnd): bool
    {
        if ($booleanAnd->getAttribute(AttributeKey::IS_DEFAULT_PROPERTY_VALUE) === true) {
            return true;
        }

        if ($booleanAnd->getAttribute(AttributeKey::IS_CLASS_CONST_VALUE) === true) {
            return true;
        }

        return $booleanAnd->getAttribute(AttributeKey::IS_PARAM_DEFAULT) === true;
    }

    private function isNotEqualOrNotIdentical(Expr $expr): bool
    {
        if ($expr instanceof NotIdentical) {
            return true;
        }

        return $expr instanceof NotEqual;
    }

    private function matchComparedExprAndValueExpr(NotIdentical|NotEqual $expr): ComparedExprAndValueExpr
    {
        return new ComparedExprAndValueExpr($expr->left, $expr->right);
    }

    /**
     * @return null|ComparedExprAndValueExpr[]
     */
    private function matchComparedAndDesiredValues(BooleanAnd $booleanAnd): ?array
    {
        /** @var NotIdentical|NotEqual $rightCompare */
        $rightCompare = $booleanAnd->right;

        // match compared expr and desired value
        $comparedExprAndValueExprs = [$this->matchComparedExprAndValueExpr($rightCompare)];

        $currentBooleanAnd = $booleanAnd;

        while ($currentBooleanAnd->left instanceof BooleanAnd) {
            if (! $this->isNotEqualOrNotIdentical($currentBooleanAnd->left->right)) {
                return null;
            }

            /** @var NotIdentical|NotEqual $leftRight */
            $leftRight = $currentBooleanAnd->left->right;
            $comparedExprAndValueExprs[] = $this->matchComparedExprAndValueExpr($leftRight);
            $currentBooleanAnd = $currentBooleanAnd->left;
        }

        if (! $this->isNotEqualOrNotIdentical($currentBooleanAnd->left)) {
            return null;
        }

        /** @var NotIdentical|NotEqual $leftCompare */
        $leftCompare = $currentBooleanAnd->left;

        $comparedExprAndValueExprs[] = $this->matchComparedExprAndValueExpr($leftCompare);

        // keep original natural order, as left/right goes from bottom up
        return array_reverse($comparedExprAndValueExprs);
    }

    private function isStrictComparison(BooleanAnd $booleanAnd): bool
    {
        $notIdenticals = $this->betterNodeFinder->findInstanceOf($booleanAnd, NotIdentical::class);
        if ($notIdenticals === []) {
            return false;
        }

        $notEquals = $this->betterNodeFinder->findInstanceOf($booleanAnd, NotEqual::class);

        // mix not identical and not equals, keep as is
        // @see https://3v4l.org/2SoHZ
        return $notEquals === [];
    }
}