mirror of https://github.com/rectorphp/rector.git
48 lines
1.2 KiB
PHP
48 lines
1.2 KiB
PHP
<?php
|
|
|
|
declare (strict_types=1);
|
|
namespace Rector\NodeTypeResolver\NodeTypeResolver;
|
|
|
|
use PhpParser\Node;
|
|
use PhpParser\Node\Expr;
|
|
use PhpParser\Node\Stmt\Return_;
|
|
use PHPStan\Type\Type;
|
|
use PHPStan\Type\VoidType;
|
|
use Rector\NodeTypeResolver\Contract\NodeTypeResolverInterface;
|
|
use Rector\NodeTypeResolver\NodeTypeResolver;
|
|
use RectorPrefix202305\Symfony\Contracts\Service\Attribute\Required;
|
|
/**
|
|
* @implements NodeTypeResolverInterface<Return_>
|
|
*/
|
|
final class ReturnTypeResolver implements NodeTypeResolverInterface
|
|
{
|
|
/**
|
|
* @var \Rector\NodeTypeResolver\NodeTypeResolver
|
|
*/
|
|
private $nodeTypeResolver;
|
|
/**
|
|
* @required
|
|
*/
|
|
public function autowire(NodeTypeResolver $nodeTypeResolver) : void
|
|
{
|
|
$this->nodeTypeResolver = $nodeTypeResolver;
|
|
}
|
|
/**
|
|
* @return array<class-string<Node>>
|
|
*/
|
|
public function getNodeClasses() : array
|
|
{
|
|
return [Return_::class];
|
|
}
|
|
/**
|
|
* @param Return_ $node
|
|
*/
|
|
public function resolve(Node $node) : Type
|
|
{
|
|
if (!$node->expr instanceof Expr) {
|
|
return new VoidType();
|
|
}
|
|
return $this->nodeTypeResolver->getType($node->expr);
|
|
}
|
|
}
|