mirror of
https://github.com/rectorphp/rector.git
synced 2024-05-28 23:10:51 +00:00
8b8c4dc016
3cf3e1fc25
[TypeDeclaration] Add ParamTypeByParentCallTypeRector (#519)
274 lines
12 KiB
PHP
274 lines
12 KiB
PHP
<?php
|
|
|
|
declare (strict_types=1);
|
|
namespace Rector\Php80\Rector\If_;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\Node\Expr;
|
|
use PhpParser\Node\Expr\Assign;
|
|
use PhpParser\Node\Expr\NullsafeMethodCall;
|
|
use PhpParser\Node\Expr\NullsafePropertyFetch;
|
|
use PhpParser\Node\Expr\PropertyFetch;
|
|
use PhpParser\Node\Identifier;
|
|
use PhpParser\Node\Stmt\Expression;
|
|
use PhpParser\Node\Stmt\If_;
|
|
use PhpParser\Node\Stmt\Return_;
|
|
use Rector\Core\NodeManipulator\IfManipulator;
|
|
use Rector\Core\NodeManipulator\NullsafeManipulator;
|
|
use Rector\Core\Rector\AbstractRector;
|
|
use Rector\NodeTypeResolver\Node\AttributeKey;
|
|
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
|
|
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
|
|
/**
|
|
* @changelog https://wiki.php.net/rfc/nullsafe_operator
|
|
* @see \Rector\Tests\Php80\Rector\If_\NullsafeOperatorRector\NullsafeOperatorRectorTest
|
|
*/
|
|
final class NullsafeOperatorRector extends \Rector\Core\Rector\AbstractRector
|
|
{
|
|
/**
|
|
* @var string
|
|
*/
|
|
private const NAME = 'name';
|
|
/**
|
|
* @var \Rector\Core\NodeManipulator\IfManipulator
|
|
*/
|
|
private $ifManipulator;
|
|
/**
|
|
* @var \Rector\Core\NodeManipulator\NullsafeManipulator
|
|
*/
|
|
private $nullsafeManipulator;
|
|
public function __construct(\Rector\Core\NodeManipulator\IfManipulator $ifManipulator, \Rector\Core\NodeManipulator\NullsafeManipulator $nullsafeManipulator)
|
|
{
|
|
$this->ifManipulator = $ifManipulator;
|
|
$this->nullsafeManipulator = $nullsafeManipulator;
|
|
}
|
|
public function getRuleDefinition() : \Symplify\RuleDocGenerator\ValueObject\RuleDefinition
|
|
{
|
|
return new \Symplify\RuleDocGenerator\ValueObject\RuleDefinition('Change if null check with nullsafe operator ?-> with full short circuiting', [new \Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample(<<<'CODE_SAMPLE'
|
|
class SomeClass
|
|
{
|
|
public function run($someObject)
|
|
{
|
|
$someObject2 = $someObject->mayFail1();
|
|
if ($someObject2 === null) {
|
|
return null;
|
|
}
|
|
|
|
return $someObject2->mayFail2();
|
|
}
|
|
}
|
|
CODE_SAMPLE
|
|
, <<<'CODE_SAMPLE'
|
|
class SomeClass
|
|
{
|
|
public function run($someObject)
|
|
{
|
|
return $someObject->mayFail1()?->mayFail2();
|
|
}
|
|
}
|
|
CODE_SAMPLE
|
|
)]);
|
|
}
|
|
/**
|
|
* @return array<class-string<Node>>
|
|
*/
|
|
public function getNodeTypes() : array
|
|
{
|
|
return [\PhpParser\Node\Stmt\If_::class];
|
|
}
|
|
/**
|
|
* @param If_ $node
|
|
*/
|
|
public function refactor(\PhpParser\Node $node) : ?\PhpParser\Node
|
|
{
|
|
$processNullSafeOperator = $this->processNullSafeOperatorIdentical($node);
|
|
if ($processNullSafeOperator !== null) {
|
|
/** @var Expression $prevNode */
|
|
$prevNode = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
$this->removeNode($prevNode);
|
|
return $processNullSafeOperator;
|
|
}
|
|
return $this->processNullSafeOperatorNotIdentical($node);
|
|
}
|
|
private function processNullSafeOperatorIdentical(\PhpParser\Node\Stmt\If_ $if, bool $isStartIf = \true) : ?\PhpParser\Node
|
|
{
|
|
$comparedNode = $this->ifManipulator->matchIfValueReturnValue($if);
|
|
if (!$comparedNode instanceof \PhpParser\Node\Expr) {
|
|
return null;
|
|
}
|
|
$prevNode = $if->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
$nextNode = $if->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
if (!$prevNode instanceof \PhpParser\Node) {
|
|
return null;
|
|
}
|
|
if (!$nextNode instanceof \PhpParser\Node) {
|
|
return null;
|
|
}
|
|
if (!$prevNode instanceof \PhpParser\Node\Stmt\Expression) {
|
|
return null;
|
|
}
|
|
if (!$this->ifManipulator->isIfCondUsingAssignIdenticalVariable($if, $prevNode->expr)) {
|
|
return null;
|
|
}
|
|
$prevExpr = $prevNode->expr;
|
|
if (!$prevExpr instanceof \PhpParser\Node\Expr\Assign) {
|
|
return null;
|
|
}
|
|
return $this->processAssign($prevExpr, $prevNode, $nextNode, $isStartIf);
|
|
}
|
|
private function processNullSafeOperatorNotIdentical(\PhpParser\Node\Stmt\If_ $if, ?\PhpParser\Node\Expr $expr = null) : ?\PhpParser\Node
|
|
{
|
|
$assign = $this->ifManipulator->matchIfNotNullNextAssignment($if);
|
|
if (!$assign instanceof \PhpParser\Node\Expr\Assign) {
|
|
return null;
|
|
}
|
|
$assignExpr = $assign->expr;
|
|
if ($this->ifManipulator->isIfCondUsingAssignNotIdenticalVariable($if, $assignExpr)) {
|
|
return null;
|
|
}
|
|
$expression = $assign->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PARENT_NODE);
|
|
if (!$expression instanceof \PhpParser\Node\Stmt\Expression) {
|
|
return null;
|
|
}
|
|
$nextNode = $expression->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
$nullSafe = $this->nullsafeManipulator->processNullSafeExpr($assignExpr);
|
|
if ($nullSafe === null) {
|
|
return null;
|
|
}
|
|
if ($expr !== null) {
|
|
/** @var Identifier $nullSafeIdentifier */
|
|
$nullSafeIdentifier = $nullSafe->name;
|
|
/** @var NullsafeMethodCall|NullsafePropertyFetch $nullSafe */
|
|
$nullSafe = $this->nullsafeManipulator->processNullSafeExprResult($expr, $nullSafeIdentifier);
|
|
}
|
|
$nextOfNextNode = $this->processIfMayInNextNode($nextNode);
|
|
if ($nextOfNextNode !== null) {
|
|
return $nextOfNextNode;
|
|
}
|
|
if (!$nextNode instanceof \PhpParser\Node\Stmt\If_) {
|
|
return new \PhpParser\Node\Expr\Assign($assign->var, $nullSafe);
|
|
}
|
|
return $this->processNullSafeOperatorNotIdentical($nextNode, $nullSafe);
|
|
}
|
|
private function processAssign(\PhpParser\Node\Expr\Assign $assign, \PhpParser\Node\Stmt\Expression $prevExpression, \PhpParser\Node $nextNode, bool $isStartIf) : ?\PhpParser\Node
|
|
{
|
|
if ($assign instanceof \PhpParser\Node\Expr\Assign && \property_exists($assign->expr, self::NAME) && \property_exists($nextNode, 'expr') && \property_exists($nextNode->expr, self::NAME)) {
|
|
return $this->processAssignInCurrentNode($assign, $prevExpression, $nextNode, $isStartIf);
|
|
}
|
|
return $this->processAssignMayInNextNode($nextNode);
|
|
}
|
|
private function processIfMayInNextNode(?\PhpParser\Node $nextNode = null) : ?\PhpParser\Node
|
|
{
|
|
if (!$nextNode instanceof \PhpParser\Node) {
|
|
return null;
|
|
}
|
|
$nextOfNextNode = $nextNode->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
while ($nextOfNextNode) {
|
|
if ($nextOfNextNode instanceof \PhpParser\Node\Stmt\If_) {
|
|
/** @var If_ $beforeIf */
|
|
$beforeIf = $nextOfNextNode->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PARENT_NODE);
|
|
$nullSafe = $this->processNullSafeOperatorNotIdentical($nextOfNextNode);
|
|
if (!$nullSafe instanceof \PhpParser\Node\Expr\NullsafeMethodCall && !$nullSafe instanceof \PhpParser\Node\Expr\PropertyFetch) {
|
|
return $beforeIf;
|
|
}
|
|
$beforeIf->stmts[\count($beforeIf->stmts) - 1] = new \PhpParser\Node\Stmt\Expression($nullSafe);
|
|
return $beforeIf;
|
|
}
|
|
$nextOfNextNode = $nextOfNextNode->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
}
|
|
return null;
|
|
}
|
|
private function processAssignInCurrentNode(\PhpParser\Node\Expr\Assign $assign, \PhpParser\Node\Stmt\Expression $expression, \PhpParser\Node $nextNode, bool $isStartIf) : ?\PhpParser\Node
|
|
{
|
|
$assignNullSafe = $isStartIf ? $assign->expr : $this->nullsafeManipulator->processNullSafeExpr($assign->expr);
|
|
$nullSafe = $this->nullsafeManipulator->processNullSafeExprResult($assignNullSafe, $nextNode->expr->name);
|
|
$prevAssign = $expression->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
if ($prevAssign instanceof \PhpParser\Node\Stmt\If_) {
|
|
$nullSafe = $this->getNullSafeOnPrevAssignIsIf($prevAssign, $nextNode, $nullSafe);
|
|
}
|
|
$this->removeNode($nextNode);
|
|
if ($nextNode instanceof \PhpParser\Node\Stmt\Return_) {
|
|
$nextNode->expr = $nullSafe;
|
|
return $nextNode;
|
|
}
|
|
return $nullSafe;
|
|
}
|
|
private function processAssignMayInNextNode(\PhpParser\Node $nextNode) : ?\PhpParser\Node
|
|
{
|
|
if (!$nextNode instanceof \PhpParser\Node\Stmt\Expression) {
|
|
return null;
|
|
}
|
|
if (!$nextNode->expr instanceof \PhpParser\Node\Expr\Assign) {
|
|
return null;
|
|
}
|
|
$mayNextIf = $nextNode->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
if (!$mayNextIf instanceof \PhpParser\Node\Stmt\If_) {
|
|
return null;
|
|
}
|
|
if ($this->ifManipulator->isIfCondUsingAssignIdenticalVariable($mayNextIf, $nextNode->expr)) {
|
|
return $this->processNullSafeOperatorIdentical($mayNextIf, \false);
|
|
}
|
|
return null;
|
|
}
|
|
private function getNullSafeOnPrevAssignIsIf(\PhpParser\Node\Stmt\If_ $if, \PhpParser\Node $nextNode, ?\PhpParser\Node\Expr $expr) : ?\PhpParser\Node\Expr
|
|
{
|
|
$prevIf = $if->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
if (!$prevIf instanceof \PhpParser\Node\Stmt\Expression) {
|
|
return $expr;
|
|
}
|
|
if (!$this->ifManipulator->isIfCondUsingAssignIdenticalVariable($if, $prevIf->expr)) {
|
|
return $expr;
|
|
}
|
|
$start = $prevIf;
|
|
while ($prevIf instanceof \PhpParser\Node\Stmt\Expression) {
|
|
$expressionNode = $prevIf->expr;
|
|
if (!$expressionNode instanceof \PhpParser\Node\Expr\Assign) {
|
|
return null;
|
|
}
|
|
$expr = $this->nullsafeManipulator->processNullSafeExpr($expressionNode->expr);
|
|
/** @var Node $prevPrevIf */
|
|
$prevPrevIf = $prevIf->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
/** @var Node $prevPrevPrevIf */
|
|
$prevPrevPrevIf = $prevPrevIf->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
|
|
if (!$prevPrevPrevIf instanceof \PhpParser\Node\Stmt\Expression && $prevPrevPrevIf !== null) {
|
|
$start = $this->getPreviousIf($prevPrevPrevIf);
|
|
break;
|
|
}
|
|
$prevIf = $prevPrevPrevIf;
|
|
}
|
|
if (!$expr instanceof \PhpParser\Node\Expr\NullsafeMethodCall && !$expr instanceof \PhpParser\Node\Expr\NullsafePropertyFetch) {
|
|
return $expr;
|
|
}
|
|
/** @var Expr $expr */
|
|
$expr = $expr->var->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PARENT_NODE);
|
|
$expr = $this->getNullSafeAfterStartUntilBeforeEnd($start, $expr);
|
|
return $this->nullsafeManipulator->processNullSafeExprResult($expr, $nextNode->expr->name);
|
|
}
|
|
private function getPreviousIf(\PhpParser\Node $node) : ?\PhpParser\Node
|
|
{
|
|
/** @var If_ $if */
|
|
$if = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
/** @var Expression $expression */
|
|
$expression = $if->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
/** @var Expression $nextExpression */
|
|
$nextExpression = $expression->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
return $nextExpression->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
}
|
|
private function getNullSafeAfterStartUntilBeforeEnd(?\PhpParser\Node $node, ?\PhpParser\Node\Expr $expr) : ?\PhpParser\Node\Expr
|
|
{
|
|
while ($node) {
|
|
$expr = $this->nullsafeManipulator->processNullSafeExprResult($expr, $node->expr->expr->name);
|
|
$node = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
while ($node) {
|
|
/** @var If_ $if */
|
|
$if = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
if ($node instanceof \PhpParser\Node\Stmt\Expression && $this->ifManipulator->isIfCondUsingAssignIdenticalVariable($if, $node->expr)) {
|
|
break;
|
|
}
|
|
$node = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::NEXT_NODE);
|
|
}
|
|
}
|
|
return $expr;
|
|
}
|
|
}
|