rector/src/NodeManipulator/ClassConstManipulator.php

71 lines
2.6 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\NodeManipulator;
use PhpParser\Node;
use PhpParser\Node\Expr\ClassConstFetch;
use PhpParser\Node\Stmt\ClassConst;
use PhpParser\Node\Stmt\ClassLike;
use PHPStan\Reflection\ClassReflection;
use Rector\NodeNameResolver\NodeNameResolver;
use Rector\PhpParser\AstResolver;
use Rector\PhpParser\Node\BetterNodeFinder;
final class ClassConstManipulator
{
/**
* @readonly
* @var \Rector\PhpParser\Node\BetterNodeFinder
*/
private $betterNodeFinder;
/**
* @readonly
* @var \Rector\NodeNameResolver\NodeNameResolver
*/
private $nodeNameResolver;
/**
* @readonly
* @var \Rector\PhpParser\AstResolver
*/
private $astResolver;
public function __construct(BetterNodeFinder $betterNodeFinder, NodeNameResolver $nodeNameResolver, AstResolver $astResolver)
{
$this->betterNodeFinder = $betterNodeFinder;
$this->nodeNameResolver = $nodeNameResolver;
$this->astResolver = $astResolver;
}
public function hasClassConstFetch(ClassConst $classConst, ClassReflection $classReflection) : bool
{
if (!$classReflection->isClass() && !$classReflection->isEnum()) {
return \true;
}
$className = $classReflection->getName();
foreach ($classReflection->getAncestors() as $ancestorClassReflection) {
$ancestorClass = $this->astResolver->resolveClassFromClassReflection($ancestorClassReflection);
if (!$ancestorClass instanceof ClassLike) {
continue;
}
// has in class?
$isClassConstFetchFound = (bool) $this->betterNodeFinder->findFirst($ancestorClass, function (Node $node) use($classConst, $className) : bool {
// property + static fetch
if (!$node instanceof ClassConstFetch) {
return \false;
}
return $this->isNameMatch($node, $classConst, $className);
});
if ($isClassConstFetchFound) {
return \true;
}
}
return \false;
}
private function isNameMatch(ClassConstFetch $classConstFetch, ClassConst $classConst, string $className) : bool
{
$classConstName = (string) $this->nodeNameResolver->getName($classConst);
$selfConstantName = 'self::' . $classConstName;
$staticConstantName = 'static::' . $classConstName;
$classNameConstantName = $className . '::' . $classConstName;
return $this->nodeNameResolver->isNames($classConstFetch, [$selfConstantName, $staticConstantName, $classNameConstantName]);
}
}