*/ private $shortNamesByFilePath = []; /** * @var \Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfoFactory */ private $phpDocInfoFactory; /** * @readonly * @var \Rector\PhpDocParser\NodeTraverser\SimpleCallableNodeTraverser */ private $simpleCallableNodeTraverser; /** * @readonly * @var \Rector\NodeNameResolver\NodeNameResolver */ private $nodeNameResolver; /** * @readonly * @var \PHPStan\Reflection\ReflectionProvider */ private $reflectionProvider; /** * @readonly * @var \Rector\Core\PhpParser\Node\BetterNodeFinder */ private $betterNodeFinder; /** * @readonly * @var \Rector\CodingStyle\NodeAnalyzer\UseImportNameMatcher */ private $useImportNameMatcher; public function __construct(SimpleCallableNodeTraverser $simpleCallableNodeTraverser, NodeNameResolver $nodeNameResolver, ReflectionProvider $reflectionProvider, BetterNodeFinder $betterNodeFinder, UseImportNameMatcher $useImportNameMatcher) { $this->simpleCallableNodeTraverser = $simpleCallableNodeTraverser; $this->nodeNameResolver = $nodeNameResolver; $this->reflectionProvider = $reflectionProvider; $this->betterNodeFinder = $betterNodeFinder; $this->useImportNameMatcher = $useImportNameMatcher; } // Avoids circular reference /** * @required */ public function autowire(PhpDocInfoFactory $phpDocInfoFactory) : void { $this->phpDocInfoFactory = $phpDocInfoFactory; } /** * @return array */ public function resolveFromFile(File $file) : array { $filePath = $file->getFilePath(); if (isset($this->shortNamesByFilePath[$filePath])) { return $this->shortNamesByFilePath[$filePath]; } $shortNamesToFullyQualifiedNames = $this->resolveForStmts($file->getNewStmts()); $this->shortNamesByFilePath[$filePath] = $shortNamesToFullyQualifiedNames; return $shortNamesToFullyQualifiedNames; } /** * Collects all "class ", "trait " and "interface " * @return string[] */ public function resolveShortClassLikeNamesForNode(Node $node) : array { $namespace = $this->betterNodeFinder->findParentType($node, Namespace_::class); if (!$namespace instanceof Namespace_) { // only handle namespace nodes return []; } /** @var ClassLike[] $classLikes */ $classLikes = $this->betterNodeFinder->findInstanceOf($namespace, ClassLike::class); $shortClassLikeNames = []; foreach ($classLikes as $classLike) { $shortClassLikeNames[] = $this->nodeNameResolver->getShortName($classLike); } return \array_unique($shortClassLikeNames); } /** * @param Stmt[] $stmts * @return array */ private function resolveForStmts(array $stmts) : array { $shortNamesToFullyQualifiedNames = []; $this->simpleCallableNodeTraverser->traverseNodesWithCallable($stmts, function (Node $node) use(&$shortNamesToFullyQualifiedNames) { // class name is used! if ($node instanceof ClassLike && $node->name instanceof Identifier) { $fullyQualifiedName = $this->nodeNameResolver->getName($node); if ($fullyQualifiedName === null) { return null; } $shortNamesToFullyQualifiedNames[$node->name->toString()] = $fullyQualifiedName; return null; } if (!$node instanceof Name) { return null; } $originalName = $node->getAttribute(AttributeKey::ORIGINAL_NAME); if (!$originalName instanceof Name) { return null; } // already short if (\strpos($originalName->toString(), '\\') !== \false) { return null; } $shortNamesToFullyQualifiedNames[$originalName->toString()] = $this->nodeNameResolver->getName($node); return null; }); $docBlockShortNamesToFullyQualifiedNames = $this->resolveFromStmtsDocBlocks($stmts); /** @var array $result */ $result = \array_merge($shortNamesToFullyQualifiedNames, $docBlockShortNamesToFullyQualifiedNames); return $result; } /** * @param Stmt[] $stmts * @return array */ private function resolveFromStmtsDocBlocks(array $stmts) : array { $classReflection = $this->resolveClassReflection($stmts); $shortNames = []; $this->simpleCallableNodeTraverser->traverseNodesWithCallable($stmts, function (Node $node) use(&$shortNames) { // speed up for nodes that are $phpDocInfo = $this->phpDocInfoFactory->createFromNode($node); if (!$phpDocInfo instanceof PhpDocInfo) { return null; } $phpDocNodeTraverser = new PhpDocNodeTraverser(); $phpDocNodeTraverser->traverseWithCallable($phpDocInfo->getPhpDocNode(), '', static function ($node) use(&$shortNames) { if ($node instanceof PhpDocTagNode) { $shortName = \trim($node->name, '@'); if (StringUtils::isMatch($shortName, self::BIG_LETTER_START_REGEX)) { $shortNames[] = $shortName; } return null; } if ($node instanceof IdentifierTypeNode) { $shortNames[] = $node->name; } return null; }); return null; }); return $this->fqnizeShortNames($shortNames, $classReflection, $stmts); } /** * @param Node[] $stmts */ private function resolveClassReflection(array $stmts) : ?ClassReflection { $firstClassLike = $this->betterNodeFinder->findFirstInstanceOf($stmts, ClassLike::class); if (!$firstClassLike instanceof ClassLike) { return null; } $className = (string) $this->nodeNameResolver->getName($firstClassLike); if (!$this->reflectionProvider->hasClass($className)) { return null; } return $this->reflectionProvider->getClass($className); } /** * @param string[] $shortNames * @param Stmt[] $stmts * @return array */ private function fqnizeShortNames(array $shortNames, ?ClassReflection $classReflection, array $stmts) : array { $shortNamesToFullyQualifiedNames = []; $nativeReflectionClass = $classReflection instanceof ClassReflection && !$classReflection->isAnonymous() ? $classReflection->getNativeReflection() : null; foreach ($shortNames as $shortName) { $stmtsMatchedName = $this->useImportNameMatcher->matchNameWithStmts($shortName, $stmts); if ($nativeReflectionClass instanceof ReflectionClass) { $fullyQualifiedName = Reflection::expandClassName($shortName, $nativeReflectionClass); } elseif (\is_string($stmtsMatchedName)) { $fullyQualifiedName = $stmtsMatchedName; } else { $fullyQualifiedName = $shortName; } $shortNamesToFullyQualifiedNames[$shortName] = $fullyQualifiedName; } return $shortNamesToFullyQualifiedNames; } }