2019-10-13 05:59:52 +00:00
|
|
|
<?php
|
|
|
|
|
|
|
|
declare(strict_types=1);
|
2018-08-06 21:57:05 +00:00
|
|
|
|
|
|
|
namespace Rector\NodeTypeResolver\PerNodeTypeResolver;
|
|
|
|
|
|
|
|
use PhpParser\Node;
|
2020-01-27 23:28:16 +00:00
|
|
|
use PhpParser\Node\Expr\Variable;
|
2019-03-10 23:47:43 +00:00
|
|
|
use PhpParser\Node\FunctionLike;
|
2020-01-27 23:45:52 +00:00
|
|
|
use PhpParser\Node\Identifier;
|
2018-08-06 21:57:05 +00:00
|
|
|
use PhpParser\Node\Param;
|
2020-01-27 23:28:16 +00:00
|
|
|
use PhpParser\NodeTraverser;
|
|
|
|
use PHPStan\Type\MixedType;
|
2019-09-04 12:10:29 +00:00
|
|
|
use PHPStan\Type\ObjectType;
|
|
|
|
use PHPStan\Type\Type;
|
2020-01-31 08:18:45 +00:00
|
|
|
use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfo;
|
2018-08-06 21:57:05 +00:00
|
|
|
use Rector\NodeTypeResolver\Contract\PerNodeTypeResolver\PerNodeTypeResolverInterface;
|
2019-04-13 09:20:27 +00:00
|
|
|
use Rector\NodeTypeResolver\Node\AttributeKey;
|
2020-01-27 23:28:16 +00:00
|
|
|
use Rector\NodeTypeResolver\NodeTypeResolver;
|
2019-01-14 18:21:23 +00:00
|
|
|
use Rector\PhpParser\Node\Resolver\NameResolver;
|
2020-01-27 23:28:16 +00:00
|
|
|
use Rector\PhpParser\NodeTraverser\CallableNodeTraverser;
|
2018-08-06 21:57:05 +00:00
|
|
|
|
2019-09-01 20:38:31 +00:00
|
|
|
/**
|
2019-09-03 09:11:45 +00:00
|
|
|
* @see \Rector\NodeTypeResolver\Tests\PerNodeTypeResolver\ParamTypeResolver\ParamTypeResolverTest
|
2019-09-01 20:38:31 +00:00
|
|
|
*/
|
2018-08-06 21:57:05 +00:00
|
|
|
final class ParamTypeResolver implements PerNodeTypeResolverInterface
|
|
|
|
{
|
2019-01-14 18:21:23 +00:00
|
|
|
/**
|
|
|
|
* @var NameResolver
|
|
|
|
*/
|
|
|
|
private $nameResolver;
|
|
|
|
|
2020-01-27 23:28:16 +00:00
|
|
|
/**
|
|
|
|
* @var CallableNodeTraverser
|
|
|
|
*/
|
|
|
|
private $callableNodeTraverser;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @var NodeTypeResolver
|
|
|
|
*/
|
|
|
|
private $nodeTypeResolver;
|
|
|
|
|
2020-01-31 08:18:45 +00:00
|
|
|
public function __construct(NameResolver $nameResolver, CallableNodeTraverser $callableNodeTraverser)
|
|
|
|
{
|
2019-01-14 18:21:23 +00:00
|
|
|
$this->nameResolver = $nameResolver;
|
2020-01-27 23:28:16 +00:00
|
|
|
$this->callableNodeTraverser = $callableNodeTraverser;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @required
|
|
|
|
*/
|
|
|
|
public function autowirePropertyTypeResolver(NodeTypeResolver $nodeTypeResolver): void
|
|
|
|
{
|
|
|
|
$this->nodeTypeResolver = $nodeTypeResolver;
|
2019-01-14 18:21:23 +00:00
|
|
|
}
|
|
|
|
|
2018-08-06 21:57:05 +00:00
|
|
|
/**
|
|
|
|
* @return string[]
|
|
|
|
*/
|
|
|
|
public function getNodeClasses(): array
|
|
|
|
{
|
|
|
|
return [Param::class];
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2020-01-27 23:28:16 +00:00
|
|
|
* @param Param $node
|
2018-08-06 21:57:05 +00:00
|
|
|
*/
|
2020-01-27 23:28:16 +00:00
|
|
|
public function resolve(Node $node): Type
|
2018-08-06 21:57:05 +00:00
|
|
|
{
|
2020-01-27 23:45:52 +00:00
|
|
|
$paramType = $this->resolveFromType($node);
|
|
|
|
if (! $paramType instanceof MixedType) {
|
|
|
|
return $paramType;
|
2019-03-10 23:47:43 +00:00
|
|
|
}
|
|
|
|
|
2020-01-27 23:45:52 +00:00
|
|
|
$firstVariableUseType = $this->resolveFromFirstVariableUse($node);
|
|
|
|
if (! $firstVariableUseType instanceof MixedType) {
|
|
|
|
return $firstVariableUseType;
|
2020-01-27 23:28:16 +00:00
|
|
|
}
|
|
|
|
|
2020-01-27 23:45:52 +00:00
|
|
|
return $this->resolveFromFunctionDocBlock($node);
|
2018-08-06 21:57:05 +00:00
|
|
|
}
|
2020-01-27 23:28:16 +00:00
|
|
|
|
2020-02-01 16:04:38 +00:00
|
|
|
private function resolveFromType(Node $node)
|
|
|
|
{
|
|
|
|
if ($node->type !== null && ! $node->type instanceof Identifier) {
|
|
|
|
$resolveTypeName = $this->nameResolver->getName($node->type);
|
|
|
|
if ($resolveTypeName) {
|
|
|
|
// @todo map the other way every type :)
|
|
|
|
return new ObjectType($resolveTypeName);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return new MixedType();
|
|
|
|
}
|
|
|
|
|
2020-01-27 23:45:52 +00:00
|
|
|
private function resolveFromFirstVariableUse(Param $param): Type
|
2020-01-27 23:28:16 +00:00
|
|
|
{
|
|
|
|
$classMethod = $param->getAttribute(AttributeKey::METHOD_NODE);
|
|
|
|
if ($classMethod === null) {
|
|
|
|
return new MixedType();
|
|
|
|
}
|
|
|
|
|
|
|
|
/** @var string $paramName */
|
|
|
|
$paramName = $this->nameResolver->getName($param);
|
|
|
|
$paramStaticType = new MixedType();
|
|
|
|
|
|
|
|
// special case for param inside method/function
|
|
|
|
$this->callableNodeTraverser->traverseNodesWithCallable(
|
|
|
|
(array) $classMethod->stmts,
|
|
|
|
function (Node $node) use ($paramName, &$paramStaticType): ?int {
|
|
|
|
if (! $node instanceof Variable) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (! $this->nameResolver->isName($node, $paramName)) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
$paramStaticType = $this->nodeTypeResolver->resolve($node);
|
|
|
|
|
|
|
|
return NodeTraverser::STOP_TRAVERSAL;
|
|
|
|
}
|
|
|
|
);
|
|
|
|
|
|
|
|
return $paramStaticType;
|
|
|
|
}
|
2020-01-27 23:45:52 +00:00
|
|
|
|
|
|
|
private function resolveFromFunctionDocBlock(Param $param): Type
|
|
|
|
{
|
|
|
|
/** @var FunctionLike $parentNode */
|
|
|
|
$parentNode = $param->getAttribute(AttributeKey::PARENT_NODE);
|
|
|
|
|
|
|
|
/** @var string $paramName */
|
|
|
|
$paramName = $this->nameResolver->getName($param);
|
|
|
|
|
2020-02-02 18:15:36 +00:00
|
|
|
/** @var PhpDocInfo $phpDocInfo */
|
2020-01-31 08:18:45 +00:00
|
|
|
$phpDocInfo = $parentNode->getAttribute(AttributeKey::PHP_DOC_INFO);
|
|
|
|
|
|
|
|
return $phpDocInfo->getParamType($paramName);
|
2020-01-27 23:45:52 +00:00
|
|
|
}
|
2018-08-06 21:57:05 +00:00
|
|
|
}
|