mirror of https://github.com/rectorphp/rector.git
112 lines
3.7 KiB
PHP
112 lines
3.7 KiB
PHP
<?php
|
|
|
|
declare (strict_types=1);
|
|
namespace Rector\NodeTypeResolver\NodeTypeResolver;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\Node\Name;
|
|
use PhpParser\Node\Name\FullyQualified;
|
|
use PhpParser\Node\Stmt\Class_;
|
|
use PhpParser\Node\Stmt\ClassLike;
|
|
use PHPStan\Reflection\ReflectionProvider;
|
|
use PHPStan\Type\ArrayType;
|
|
use PHPStan\Type\MixedType;
|
|
use PHPStan\Type\ObjectType;
|
|
use PHPStan\Type\Type;
|
|
use PHPStan\Type\UnionType;
|
|
use Rector\Core\Enum\ObjectReference;
|
|
use Rector\Core\PhpParser\Node\BetterNodeFinder;
|
|
use Rector\NodeNameResolver\NodeNameResolver;
|
|
use Rector\NodeTypeResolver\Contract\NodeTypeResolverInterface;
|
|
/**
|
|
* @see \Rector\Tests\NodeTypeResolver\PerNodeTypeResolver\NameTypeResolver\NameTypeResolverTest
|
|
*
|
|
* @implements NodeTypeResolverInterface<Name|FullyQualified>
|
|
*/
|
|
final class NameTypeResolver implements NodeTypeResolverInterface
|
|
{
|
|
/**
|
|
* @readonly
|
|
* @var \PHPStan\Reflection\ReflectionProvider
|
|
*/
|
|
private $reflectionProvider;
|
|
/**
|
|
* @readonly
|
|
* @var \Rector\Core\PhpParser\Node\BetterNodeFinder
|
|
*/
|
|
private $betterNodeFinder;
|
|
/**
|
|
* @readonly
|
|
* @var \Rector\NodeNameResolver\NodeNameResolver
|
|
*/
|
|
private $nodeNameResolver;
|
|
public function __construct(ReflectionProvider $reflectionProvider, BetterNodeFinder $betterNodeFinder, NodeNameResolver $nodeNameResolver)
|
|
{
|
|
$this->reflectionProvider = $reflectionProvider;
|
|
$this->betterNodeFinder = $betterNodeFinder;
|
|
$this->nodeNameResolver = $nodeNameResolver;
|
|
}
|
|
/**
|
|
* @return array<class-string<Node>>
|
|
*/
|
|
public function getNodeClasses() : array
|
|
{
|
|
return [Name::class, FullyQualified::class];
|
|
}
|
|
/**
|
|
* @param Name $node
|
|
*/
|
|
public function resolve(Node $node) : Type
|
|
{
|
|
if ($node->toString() === ObjectReference::PARENT) {
|
|
return $this->resolveParent($node);
|
|
}
|
|
$fullyQualifiedName = $this->resolveFullyQualifiedName($node);
|
|
if ($node->toString() === 'array') {
|
|
return new ArrayType(new MixedType(), new MixedType());
|
|
}
|
|
return new ObjectType($fullyQualifiedName);
|
|
}
|
|
/**
|
|
* @return \PHPStan\Type\MixedType|\PHPStan\Type\ObjectType|\PHPStan\Type\UnionType
|
|
*/
|
|
private function resolveParent(Name $name)
|
|
{
|
|
$class = $this->betterNodeFinder->findParentType($name, Class_::class);
|
|
if (!$class instanceof Class_) {
|
|
return new MixedType();
|
|
}
|
|
$className = $this->nodeNameResolver->getName($class);
|
|
if (!\is_string($className)) {
|
|
return new MixedType();
|
|
}
|
|
if (!$this->reflectionProvider->hasClass($className)) {
|
|
return new MixedType();
|
|
}
|
|
$classReflection = $this->reflectionProvider->getClass($className);
|
|
$parentClassObjectTypes = [];
|
|
foreach ($classReflection->getParents() as $parentClassReflection) {
|
|
$parentClassObjectTypes[] = new ObjectType($parentClassReflection->getName());
|
|
}
|
|
if ($parentClassObjectTypes === []) {
|
|
return new MixedType();
|
|
}
|
|
if (\count($parentClassObjectTypes) === 1) {
|
|
return $parentClassObjectTypes[0];
|
|
}
|
|
return new UnionType($parentClassObjectTypes);
|
|
}
|
|
private function resolveFullyQualifiedName(Name $name) : string
|
|
{
|
|
$nameValue = $name->toString();
|
|
if (\in_array($nameValue, [ObjectReference::SELF, ObjectReference::STATIC, 'this'], \true)) {
|
|
$classLike = $this->betterNodeFinder->findParentType($name, ClassLike::class);
|
|
if (!$classLike instanceof ClassLike) {
|
|
return $name->toString();
|
|
}
|
|
return (string) $this->nodeNameResolver->getName($classLike);
|
|
}
|
|
return $nameValue;
|
|
}
|
|
}
|