2021-01-18 12:22:26 +00:00
|
|
|
<?php
|
|
|
|
|
|
|
|
declare(strict_types=1);
|
|
|
|
|
|
|
|
namespace Rector\TypeDeclaration\Rector\Identical;
|
|
|
|
|
|
|
|
use PhpParser\Node;
|
|
|
|
use PhpParser\Node\Expr;
|
|
|
|
use PhpParser\Node\Expr\Assign;
|
|
|
|
use PhpParser\Node\Expr\BinaryOp\Identical;
|
|
|
|
use PhpParser\Node\Expr\BooleanNot;
|
|
|
|
use PhpParser\Node\Expr\Instanceof_;
|
|
|
|
use PhpParser\Node\Name\FullyQualified;
|
|
|
|
use PhpParser\Node\Stmt\Expression;
|
|
|
|
use PHPStan\PhpDocParser\Ast\PhpDoc\VarTagValueNode;
|
|
|
|
use PHPStan\PhpDocParser\Ast\Type\TypeNode;
|
|
|
|
use Rector\AttributeAwarePhpDoc\Ast\Type\AttributeAwareIdentifierTypeNode;
|
|
|
|
use Rector\AttributeAwarePhpDoc\Ast\Type\AttributeAwareUnionTypeNode;
|
|
|
|
use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfo;
|
|
|
|
use Rector\Core\Rector\AbstractRector;
|
|
|
|
use Rector\NodeTypeResolver\Node\AttributeKey;
|
|
|
|
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
|
|
|
|
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @see \Rector\TypeDeclaration\Tests\Rector\Identical\FlipTypeControlToUseExclusiveTypeRector\FlipTypeControlToUseExclusiveTypeRectorTest
|
|
|
|
*/
|
|
|
|
final class FlipTypeControlToUseExclusiveTypeRector extends AbstractRector
|
|
|
|
{
|
|
|
|
public function getRuleDefinition(): RuleDefinition
|
|
|
|
{
|
|
|
|
return new RuleDefinition(
|
|
|
|
'Flip type control to use exclusive type',
|
|
|
|
[
|
|
|
|
new CodeSample(
|
|
|
|
<<<'CODE_SAMPLE'
|
|
|
|
class SomeClass
|
|
|
|
{
|
|
|
|
public function __construct(array $values)
|
|
|
|
{
|
|
|
|
/** @var PhpDocInfo|null $phpDocInfo */
|
|
|
|
$phpDocInfo = $functionLike->getAttribute(AttributeKey::PHP_DOC_INFO);
|
|
|
|
if ($phpDocInfo === null) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
CODE_SAMPLE
|
|
|
|
,
|
|
|
|
<<<'CODE_SAMPLE'
|
|
|
|
class SomeClass
|
|
|
|
{
|
|
|
|
public function __construct(array $values)
|
|
|
|
{
|
|
|
|
$phpDocInfo = $functionLike->getAttribute(AttributeKey::PHP_DOC_INFO);
|
|
|
|
if (! $phpDocInfo instanceof PhpDocInfo) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
CODE_SAMPLE
|
|
|
|
),
|
|
|
|
|
|
|
|
]);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return string[]
|
|
|
|
*/
|
|
|
|
public function getNodeTypes(): array
|
|
|
|
{
|
|
|
|
return [Identical::class];
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param Identical $node
|
|
|
|
*/
|
|
|
|
public function refactor(Node $node): ?Node
|
|
|
|
{
|
|
|
|
if (! $this->isNull($node->left) && ! $this->isNull($node->right)) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
$variable = $this->isNull($node->left)
|
|
|
|
? $node->right
|
|
|
|
: $node->left;
|
|
|
|
|
|
|
|
$assign = $this->getVariableAssign($node, $variable);
|
|
|
|
if (! $assign instanceof Assign) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
$expression = $assign->getAttribute(AttributeKey::PARENT_NODE);
|
|
|
|
if (! $expression instanceof Expression) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2021-01-18 21:45:13 +00:00
|
|
|
$phpDocInfo = $this->phpDocInfoFactory->createFromNodeOrEmpty($expression);
|
2021-01-18 12:22:26 +00:00
|
|
|
|
|
|
|
/** @var AttributeAwareIdentifierTypeNode[] $types */
|
|
|
|
$types = $this->getTypes($phpDocInfo);
|
|
|
|
if ($this->skipNotNullOneOf($types)) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
return $this->processConvertToExclusiveType($types, $variable, $phpDocInfo);
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param AttributeAwareIdentifierTypeNode[] $types
|
|
|
|
*/
|
|
|
|
private function processConvertToExclusiveType(array $types, Expr $expr, PhpDocInfo $phpDocInfo): ?BooleanNot
|
|
|
|
{
|
|
|
|
$type = $types[0]->name === 'null'
|
|
|
|
? $types[1]->name
|
|
|
|
: $types[0]->name;
|
|
|
|
|
|
|
|
if (! class_exists($type) && ! interface_exists($type)) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** @var VarTagValueNode $tagValueNode */
|
|
|
|
$tagValueNode = $phpDocInfo->getVarTagValueNode();
|
|
|
|
$phpDocInfo->removeTagValueNodeFromNode($tagValueNode);
|
|
|
|
return new BooleanNot(new Instanceof_($expr, new FullyQualified($type)));
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return TypeNode[]
|
|
|
|
*/
|
|
|
|
private function getTypes(PhpDocInfo $phpDocInfo): array
|
|
|
|
{
|
|
|
|
$tagValueNode = $phpDocInfo->getVarTagValueNode();
|
|
|
|
if (! $tagValueNode instanceof VarTagValueNode) {
|
|
|
|
return [];
|
|
|
|
}
|
|
|
|
|
|
|
|
if (! $tagValueNode->type instanceof AttributeAwareUnionTypeNode) {
|
|
|
|
return [];
|
|
|
|
}
|
|
|
|
|
|
|
|
if (count($tagValueNode->type->types) > 2) {
|
|
|
|
return [];
|
|
|
|
}
|
|
|
|
|
|
|
|
return $tagValueNode->type->types;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function getVariableAssign(Identical $identical, Expr $expr): ?Node
|
|
|
|
{
|
|
|
|
return $this->betterNodeFinder->findFirstPrevious($identical, function (Node $node) use ($expr): bool {
|
|
|
|
if (! $node instanceof Assign) {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
return $this->areNodesEqual($node->var, $expr);
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param AttributeAwareIdentifierTypeNode[] $types
|
|
|
|
*/
|
|
|
|
private function skipNotNullOneOf(array $types): bool
|
|
|
|
{
|
|
|
|
if ($types === []) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
foreach ($types as $type) {
|
|
|
|
if (! $type instanceof AttributeAwareIdentifierTypeNode) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($types[0]->name === $types[1]->name) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
if ($types[0]->name === 'null') {
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
return $types[1]->name !== 'null';
|
|
|
|
}
|
|
|
|
}
|