rector/src/PhpParser/Node/Value/ValueResolver.php
Tomas Votruba e7097335cd Updated Rector to commit b93e285a4a72ef12d4892852118d5ab7b1b48bdd
b93e285a4a [DX] Remove AddDefaultValueForUndefinedVariableRector as can generate broken code and depends on context (#4729)
2023-08-09 10:56:45 +00:00

327 lines
12 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\Core\PhpParser\Node\Value;
use PhpParser\ConstExprEvaluationException;
use PhpParser\ConstExprEvaluator;
use PhpParser\Node\Expr;
use PhpParser\Node\Expr\BinaryOp\Concat;
use PhpParser\Node\Expr\ClassConstFetch;
use PhpParser\Node\Expr\ConstFetch;
use PhpParser\Node\Name;
use PhpParser\Node\Scalar\MagicConst\Dir;
use PhpParser\Node\Scalar\MagicConst\File;
use PHPStan\Analyser\Scope;
use PHPStan\BetterReflection\Reflection\ReflectionClass;
use PHPStan\Reflection\ClassReflection;
use PHPStan\Reflection\ReflectionProvider;
use PHPStan\Type\Constant\ConstantArrayType;
use PHPStan\Type\ConstantScalarType;
use PHPStan\Type\TypeWithClassName;
use Rector\Core\Enum\ObjectReference;
use Rector\Core\Exception\ShouldNotHappenException;
use Rector\Core\NodeAnalyzer\ConstFetchAnalyzer;
use Rector\Core\Provider\CurrentFileProvider;
use Rector\Core\Reflection\ReflectionResolver;
use Rector\Core\Util\Reflection\PrivatesAccessor;
use Rector\NodeNameResolver\NodeNameResolver;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Rector\NodeTypeResolver\NodeTypeResolver;
use TypeError;
/**
* @see \Rector\Core\Tests\PhpParser\Node\Value\ValueResolverTest
* @todo make use of constant type of $scope->getType()
*/
final class ValueResolver
{
/**
* @readonly
* @var \Rector\NodeNameResolver\NodeNameResolver
*/
private $nodeNameResolver;
/**
* @readonly
* @var \Rector\NodeTypeResolver\NodeTypeResolver
*/
private $nodeTypeResolver;
/**
* @readonly
* @var \Rector\Core\NodeAnalyzer\ConstFetchAnalyzer
*/
private $constFetchAnalyzer;
/**
* @readonly
* @var \PHPStan\Reflection\ReflectionProvider
*/
private $reflectionProvider;
/**
* @readonly
* @var \Rector\Core\Provider\CurrentFileProvider
*/
private $currentFileProvider;
/**
* @readonly
* @var \Rector\Core\Reflection\ReflectionResolver
*/
private $reflectionResolver;
/**
* @readonly
* @var \Rector\Core\Util\Reflection\PrivatesAccessor
*/
private $privatesAccessor;
/**
* @var \PhpParser\ConstExprEvaluator|null
*/
private $constExprEvaluator;
public function __construct(NodeNameResolver $nodeNameResolver, NodeTypeResolver $nodeTypeResolver, ConstFetchAnalyzer $constFetchAnalyzer, ReflectionProvider $reflectionProvider, CurrentFileProvider $currentFileProvider, ReflectionResolver $reflectionResolver, PrivatesAccessor $privatesAccessor)
{
$this->nodeNameResolver = $nodeNameResolver;
$this->nodeTypeResolver = $nodeTypeResolver;
$this->constFetchAnalyzer = $constFetchAnalyzer;
$this->reflectionProvider = $reflectionProvider;
$this->currentFileProvider = $currentFileProvider;
$this->reflectionResolver = $reflectionResolver;
$this->privatesAccessor = $privatesAccessor;
}
/**
* @param mixed $value
*/
public function isValue(Expr $expr, $value) : bool
{
return $this->getValue($expr) === $value;
}
/**
* @return mixed
*/
public function getValue(Expr $expr, bool $resolvedClassReference = \false)
{
if ($expr instanceof Concat) {
return $this->processConcat($expr, $resolvedClassReference);
}
if ($expr instanceof ClassConstFetch && $resolvedClassReference) {
$class = $this->nodeNameResolver->getName($expr->class);
if (\in_array($class, [ObjectReference::SELF, ObjectReference::STATIC], \true)) {
$classReflection = $this->reflectionResolver->resolveClassReflection($expr);
if ($classReflection instanceof ClassReflection) {
return $classReflection->getName();
}
}
if ($this->nodeNameResolver->isName($expr->name, 'class')) {
return $class;
}
}
$value = $this->resolveExprValueForConst($expr);
if ($value !== null) {
return $value;
}
if ($expr instanceof ConstFetch) {
return $this->nodeNameResolver->getName($expr);
}
$nodeStaticType = $this->nodeTypeResolver->getType($expr);
if ($nodeStaticType instanceof ConstantArrayType) {
return $this->extractConstantArrayTypeValue($nodeStaticType);
}
if ($nodeStaticType instanceof ConstantScalarType) {
return $nodeStaticType->getValue();
}
return null;
}
/**
* @api symfony
* @param mixed[] $expectedValues
*/
public function isValues(Expr $expr, array $expectedValues) : bool
{
foreach ($expectedValues as $expectedValue) {
if ($this->isValue($expr, $expectedValue)) {
return \true;
}
}
return \false;
}
public function isFalse(Expr $expr) : bool
{
return $this->constFetchAnalyzer->isFalse($expr);
}
public function isTrueOrFalse(Expr $expr) : bool
{
return $this->constFetchAnalyzer->isTrueOrFalse($expr);
}
public function isTrue(Expr $expr) : bool
{
return $this->constFetchAnalyzer->isTrue($expr);
}
public function isNull(Expr $expr) : bool
{
return $this->constFetchAnalyzer->isNull($expr);
}
/**
* @param Expr[]|null[] $nodes
* @param mixed[] $expectedValues
*/
public function areValuesEqual(array $nodes, array $expectedValues) : bool
{
foreach ($nodes as $i => $node) {
if (!$node instanceof Expr) {
return \false;
}
if (!$this->isValue($node, $expectedValues[$i])) {
return \false;
}
}
return \true;
}
/**
* @return mixed
*/
private function resolveExprValueForConst(Expr $expr)
{
try {
$constExprEvaluator = $this->getConstExprEvaluator();
return $constExprEvaluator->evaluateDirectly($expr);
} catch (ConstExprEvaluationException|TypeError $exception) {
}
return null;
}
private function processConcat(Concat $concat, bool $resolvedClassReference) : string
{
return $this->getValue($concat->left, $resolvedClassReference) . $this->getValue($concat->right, $resolvedClassReference);
}
private function getConstExprEvaluator() : ConstExprEvaluator
{
if ($this->constExprEvaluator instanceof ConstExprEvaluator) {
return $this->constExprEvaluator;
}
$this->constExprEvaluator = new ConstExprEvaluator(function (Expr $expr) {
if ($expr instanceof Dir) {
// __DIR__
return $this->resolveDirConstant();
}
if ($expr instanceof File) {
// __FILE__
return $this->resolveFileConstant($expr);
}
// resolve "SomeClass::SOME_CONST"
if ($expr instanceof ClassConstFetch && $expr->class instanceof Name) {
return $this->resolveClassConstFetch($expr);
}
throw new ConstExprEvaluationException(\sprintf('Expression of type "%s" cannot be evaluated', $expr->getType()));
});
return $this->constExprEvaluator;
}
/**
* @return mixed[]|null
*/
private function extractConstantArrayTypeValue(ConstantArrayType $constantArrayType) : ?array
{
$keys = [];
foreach ($constantArrayType->getKeyTypes() as $i => $keyType) {
/** @var ConstantScalarType $keyType */
$keys[$i] = $keyType->getValue();
}
$values = [];
foreach ($constantArrayType->getValueTypes() as $i => $valueType) {
if ($valueType instanceof ConstantArrayType) {
$value = $this->extractConstantArrayTypeValue($valueType);
} elseif ($valueType instanceof ConstantScalarType) {
$value = $valueType->getValue();
} elseif ($valueType instanceof TypeWithClassName) {
continue;
} else {
return null;
}
$values[$keys[$i]] = $value;
}
return $values;
}
private function resolveDirConstant() : string
{
$file = $this->currentFileProvider->getFile();
if (!$file instanceof \Rector\Core\ValueObject\Application\File) {
throw new ShouldNotHappenException();
}
return \dirname($file->getFilePath());
}
private function resolveFileConstant(File $file) : string
{
$file = $this->currentFileProvider->getFile();
if (!$file instanceof \Rector\Core\ValueObject\Application\File) {
throw new ShouldNotHappenException();
}
return $file->getFilePath();
}
/**
* @return string|mixed
*/
private function resolveClassConstFetch(ClassConstFetch $classConstFetch)
{
$class = $this->nodeNameResolver->getName($classConstFetch->class);
$constant = $this->nodeNameResolver->getName($classConstFetch->name);
if ($class === null) {
throw new ShouldNotHappenException();
}
if ($constant === null) {
throw new ShouldNotHappenException();
}
if (\in_array($class, [ObjectReference::SELF, ObjectReference::STATIC, ObjectReference::PARENT], \true)) {
$class = $this->resolveClassFromSelfStaticParent($classConstFetch, $class);
}
if ($constant === 'class') {
return $class;
}
$classConstantReference = $class . '::' . $constant;
if (\defined($classConstantReference)) {
return \constant($classConstantReference);
}
if (!$this->reflectionProvider->hasClass($class)) {
// fallback to constant reference itself, to avoid fatal error
return $classConstantReference;
}
$classReflection = $this->reflectionProvider->getClass($class);
if (!$classReflection->hasConstant($constant)) {
// fallback to constant reference itself, to avoid fatal error
return $classConstantReference;
}
if ($classReflection->isEnum()) {
// fallback to constant reference itself, to avoid fatal error
return $classConstantReference;
}
$constantReflection = $classReflection->getConstant($constant);
$valueExpr = $constantReflection->getValueExpr();
if ($valueExpr instanceof ConstFetch) {
return $this->resolveExprValueForConst($valueExpr);
}
return $this->getValue($valueExpr);
}
private function resolveClassFromSelfStaticParent(ClassConstFetch $classConstFetch, string $class) : string
{
// Scope may be loaded too late, so return empty string early
// it will be resolved on next traverse
$scope = $classConstFetch->getAttribute(AttributeKey::SCOPE);
if (!$scope instanceof Scope) {
return '';
}
$classReflection = $this->reflectionResolver->resolveClassReflection($classConstFetch);
if (!$classReflection instanceof ClassReflection) {
throw new ShouldNotHappenException('Complete class parent node for to class const fetch, so "self" or "static" references is resolvable to a class name');
}
if ($class !== ObjectReference::PARENT) {
return $classReflection->getName();
}
if (!$classReflection->isClass()) {
throw new ShouldNotHappenException('Complete class parent node for to class const fetch, so "parent" references is resolvable to lookup parent class');
}
// XXX rework this hack, after https://github.com/phpstan/phpstan-src/pull/2563 landed
// ensure parent class name still resolved even not autoloaded
$nativeReflection = $classReflection->getNativeReflection();
$betterReflectionClass = $this->privatesAccessor->getPrivateProperty($nativeReflection, 'betterReflectionClass');
/** @var ReflectionClass $betterReflectionClass */
$parentClassName = $betterReflectionClass->getParentClassName();
if ($parentClassName === null) {
throw new ShouldNotHappenException();
}
return $parentClassName;
}
}