rector/rules/CodeQuality/Rector/Identical/SimplifyArraySearchRector.php

74 lines
2.6 KiB
PHP
Raw Normal View History

2019-10-13 05:59:52 +00:00
<?php
declare (strict_types=1);
namespace Rector\CodeQuality\Rector\Identical;
use PhpParser\Node;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\BinaryOp\Identical;
use PhpParser\Node\Expr\BinaryOp\NotIdentical;
use PhpParser\Node\Expr\BooleanNot;
use PhpParser\Node\Expr\FuncCall;
use Rector\NodeManipulator\BinaryOpManipulator;
use Rector\Php71\ValueObject\TwoNodeMatch;
use Rector\PhpParser\Node\Value\ValueResolver;
use Rector\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
2019-09-03 09:11:45 +00:00
/**
* @see \Rector\Tests\CodeQuality\Rector\Identical\SimplifyArraySearchRector\SimplifyArraySearchRectorTest
2019-09-03 09:11:45 +00:00
*/
final class SimplifyArraySearchRector extends AbstractRector
{
2018-11-07 17:04:38 +00:00
/**
* @readonly
* @var \Rector\NodeManipulator\BinaryOpManipulator
2018-11-07 17:04:38 +00:00
*/
private $binaryOpManipulator;
/**
* @readonly
* @var \Rector\PhpParser\Node\Value\ValueResolver
*/
private $valueResolver;
public function __construct(BinaryOpManipulator $binaryOpManipulator, ValueResolver $valueResolver)
2018-11-07 17:04:38 +00:00
{
$this->binaryOpManipulator = $binaryOpManipulator;
$this->valueResolver = $valueResolver;
2018-11-07 17:04:38 +00:00
}
public function getRuleDefinition() : RuleDefinition
{
return new RuleDefinition('Simplify array_search to in_array', [new CodeSample('array_search("searching", $array) !== false;', 'in_array("searching", $array);'), new CodeSample('array_search("searching", $array, true) !== false;', 'in_array("searching", $array, true);')]);
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [Identical::class, NotIdentical::class];
}
/**
* @param Identical|NotIdentical $node
*/
public function refactor(Node $node) : ?Node
{
$twoNodeMatch = $this->binaryOpManipulator->matchFirstAndSecondConditionNode($node, function (Node $node) : bool {
if (!$node instanceof FuncCall) {
return \false;
2018-11-07 17:04:38 +00:00
}
return $this->nodeNameResolver->isName($node, 'array_search');
}, function (Node $node) : bool {
return $node instanceof Expr && $this->valueResolver->isFalse($node);
});
if (!$twoNodeMatch instanceof TwoNodeMatch) {
return null;
}
/** @var FuncCall $funcCallExpr */
$funcCallExpr = $twoNodeMatch->getFirstExpr();
$inArrayFuncCall = $this->nodeFactory->createFuncCall('in_array', $funcCallExpr->args);
if ($node instanceof Identical) {
return new BooleanNot($inArrayFuncCall);
}
return $inArrayFuncCall;
}
}