rector/packages/PHPStanStaticTypeMapper/TypeMapper/IntersectionTypeMapper.php

97 lines
3.6 KiB
PHP
Raw Normal View History

2020-01-15 02:16:22 +00:00
<?php
declare (strict_types=1);
2022-06-06 16:43:29 +00:00
namespace RectorPrefix20220606\Rector\PHPStanStaticTypeMapper\TypeMapper;
2020-01-15 02:16:22 +00:00
2022-06-06 16:43:29 +00:00
use RectorPrefix20220606\PhpParser\Node;
use RectorPrefix20220606\PhpParser\Node\Name;
use RectorPrefix20220606\PHPStan\PhpDocParser\Ast\Type\TypeNode;
use RectorPrefix20220606\PHPStan\Type\Generic\GenericClassStringType;
use RectorPrefix20220606\PHPStan\Type\IntersectionType;
use RectorPrefix20220606\PHPStan\Type\Type;
use RectorPrefix20220606\Rector\BetterPhpDocParser\ValueObject\Type\BracketsAwareIntersectionTypeNode;
use RectorPrefix20220606\Rector\Core\Exception\ShouldNotHappenException;
use RectorPrefix20220606\Rector\Core\Php\PhpVersionProvider;
use RectorPrefix20220606\Rector\Core\ValueObject\PhpVersionFeature;
use RectorPrefix20220606\Rector\PHPStanStaticTypeMapper\Contract\TypeMapperInterface;
use RectorPrefix20220606\Rector\PHPStanStaticTypeMapper\PHPStanStaticTypeMapper;
use RectorPrefix20220606\Symfony\Contracts\Service\Attribute\Required;
/**
* @implements TypeMapperInterface<IntersectionType>
*/
2022-06-06 16:43:29 +00:00
final class IntersectionTypeMapper implements TypeMapperInterface
2020-01-15 02:16:22 +00:00
{
/**
* @var string
*/
private const STRING = 'string';
2020-01-15 02:16:22 +00:00
/**
* @var \Rector\PHPStanStaticTypeMapper\PHPStanStaticTypeMapper
2020-01-15 02:16:22 +00:00
*/
private $phpStanStaticTypeMapper;
/**
* @readonly
* @var \Rector\Core\Php\PhpVersionProvider
*/
private $phpVersionProvider;
2022-06-06 16:43:29 +00:00
public function __construct(PhpVersionProvider $phpVersionProvider)
{
$this->phpVersionProvider = $phpVersionProvider;
}
2020-01-15 02:16:22 +00:00
/**
* @required
*/
2022-06-06 16:43:29 +00:00
public function autowire(PHPStanStaticTypeMapper $phpStanStaticTypeMapper) : void
2020-01-15 02:16:22 +00:00
{
$this->phpStanStaticTypeMapper = $phpStanStaticTypeMapper;
}
/**
* @return class-string<Type>
*/
public function getNodeClass() : string
2020-01-15 02:16:22 +00:00
{
2022-06-06 16:43:29 +00:00
return IntersectionType::class;
2020-01-15 02:16:22 +00:00
}
/**
* @param IntersectionType $type
2020-01-15 02:16:22 +00:00
*/
2022-06-06 16:43:29 +00:00
public function mapToPHPStanPhpDocTypeNode(Type $type, string $typeKind) : TypeNode
2020-01-15 02:16:22 +00:00
{
$intersectionTypesNodes = [];
foreach ($type->getTypes() as $intersectionedType) {
$intersectionTypesNodes[] = $this->phpStanStaticTypeMapper->mapToPHPStanPhpDocTypeNode($intersectionedType, $typeKind);
2020-01-15 02:16:22 +00:00
}
$intersectionTypesNodes = \array_unique($intersectionTypesNodes);
if (\count($intersectionTypesNodes) === 1) {
return $intersectionTypesNodes[0];
}
2022-06-06 16:43:29 +00:00
return new BracketsAwareIntersectionTypeNode($intersectionTypesNodes);
2020-01-15 02:16:22 +00:00
}
/**
* @param IntersectionType $type
2020-01-15 02:16:22 +00:00
*/
2022-06-06 16:43:29 +00:00
public function mapToPhpParserNode(Type $type, string $typeKind) : ?Node
2020-01-15 02:16:22 +00:00
{
2022-06-06 16:43:29 +00:00
if (!$this->phpVersionProvider->isAtLeastPhpVersion(PhpVersionFeature::INTERSECTION_TYPES)) {
return null;
}
$intersectionedTypeNodes = [];
foreach ($type->getTypes() as $intersectionedType) {
$resolvedType = $this->phpStanStaticTypeMapper->mapToPhpParserNode($intersectionedType, $typeKind);
2022-06-06 16:43:29 +00:00
if ($intersectionedType instanceof GenericClassStringType) {
$resolvedTypeName = self::STRING;
2022-06-06 16:43:29 +00:00
$resolvedType = new Name(self::STRING);
} elseif (!$resolvedType instanceof Name) {
throw new ShouldNotHappenException();
} else {
$resolvedTypeName = (string) $resolvedType;
}
if (\in_array($resolvedTypeName, [self::STRING, 'object'], \true)) {
return $resolvedType;
}
$intersectionedTypeNodes[] = $resolvedType;
}
2022-06-06 16:43:29 +00:00
return new Node\IntersectionType($intersectionedTypeNodes);
2020-01-15 02:16:22 +00:00
}
}