rector/packages/PHPStanStaticTypeMapper/TypeMapper/ClassStringTypeMapper.php
Tomas Votruba d66e622e1b Updated Rector to commit dda87fc6d2
dda87fc6d2 [VendorLocker] Allow both in vendor for child and parent on ParentClassMethodTypeOverrideGuard (#1399)
2021-12-06 06:57:16 +00:00

80 lines
2.9 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\PHPStanStaticTypeMapper\TypeMapper;
use PhpParser\Node;
use PhpParser\Node\Expr;
use PHPStan\PhpDocParser\Ast\Type\GenericTypeNode;
use PHPStan\PhpDocParser\Ast\Type\IdentifierTypeNode;
use PHPStan\PhpDocParser\Ast\Type\TypeNode;
use PHPStan\Type\ClassStringType;
use PHPStan\Type\Generic\GenericClassStringType;
use PHPStan\Type\ObjectType;
use PHPStan\Type\Type;
use Rector\PHPStanStaticTypeMapper\Contract\TypeMapperInterface;
use Rector\PHPStanStaticTypeMapper\Enum\TypeKind;
use Rector\PHPStanStaticTypeMapper\PHPStanStaticTypeMapper;
use RectorPrefix20211206\Symfony\Contracts\Service\Attribute\Required;
/**
* @implements TypeMapperInterface<ClassStringType>
*/
final class ClassStringTypeMapper implements \Rector\PHPStanStaticTypeMapper\Contract\TypeMapperInterface
{
/**
* @var \Rector\PHPStanStaticTypeMapper\PHPStanStaticTypeMapper
*/
private $phpStanStaticTypeMapper;
/**
* @return class-string<Type>
*/
public function getNodeClass() : string
{
return \PHPStan\Type\ClassStringType::class;
}
/**
* @param \PHPStan\Type\Type $type
* @param \Rector\PHPStanStaticTypeMapper\Enum\TypeKind $typeKind
*/
public function mapToPHPStanPhpDocTypeNode($type, $typeKind) : \PHPStan\PhpDocParser\Ast\Type\TypeNode
{
$attributeAwareIdentifierTypeNode = new \PHPStan\PhpDocParser\Ast\Type\IdentifierTypeNode('class-string');
if ($type instanceof \PHPStan\Type\Generic\GenericClassStringType) {
$genericType = $type->getGenericType();
if ($genericType instanceof \PHPStan\Type\ObjectType) {
$className = $genericType->getClassName();
$className = $this->normalizeType($className);
$genericType = new \PHPStan\Type\ObjectType($className);
}
$genericTypeNode = $this->phpStanStaticTypeMapper->mapToPHPStanPhpDocTypeNode($genericType, $typeKind);
return new \PHPStan\PhpDocParser\Ast\Type\GenericTypeNode($attributeAwareIdentifierTypeNode, [$genericTypeNode]);
}
return $attributeAwareIdentifierTypeNode;
}
/**
* @param \PHPStan\Type\Type $type
* @param \Rector\PHPStanStaticTypeMapper\Enum\TypeKind $typeKind
*/
public function mapToPhpParserNode($type, $typeKind) : ?\PhpParser\Node
{
return null;
}
/**
* @required
*/
public function autowire(\Rector\PHPStanStaticTypeMapper\PHPStanStaticTypeMapper $phpStanStaticTypeMapper) : void
{
$this->phpStanStaticTypeMapper = $phpStanStaticTypeMapper;
}
private function normalizeType(string $classType) : string
{
if (\is_a($classType, \PhpParser\Node\Expr::class, \true)) {
return \PhpParser\Node\Expr::class;
}
if (\is_a($classType, \PhpParser\Node::class, \true)) {
return \PhpParser\Node::class;
}
return $classType;
}
}