rector/rules/coding-style/src/Node/DocAliasResolver.php

<?php

declare(strict_types=1);

namespace Rector\CodingStyle\Node;

use Nette\Utils\Strings;
use PhpParser\Node;
use PHPStan\Type\Type;
use PHPStan\Type\UnionType;
use Rector\BetterPhpDocParser\PhpDocInfo\PhpDocInfo;
use Rector\Core\PhpParser\NodeTraverser\CallableNodeTraverser;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Rector\PHPStan\Type\AliasedObjectType;

final class DocAliasResolver
{
    /**
     * @var CallableNodeTraverser
     */
    private $callableNodeTraverser;

    public function __construct(CallableNodeTraverser $callableNodeTraverser)
    {
        $this->callableNodeTraverser = $callableNodeTraverser;
    }

    /**
     * @return string[]
     */
    public function resolve(Node $node): array
    {
        $possibleDocAliases = [];

        $this->callableNodeTraverser->traverseNodesWithCallable($node, function (Node $node) use (
            &$possibleDocAliases
        ): void {
            if ($node->getDocComment() === null) {
                return;
            }

            /** @var PhpDocInfo $phpDocInfo */
            $phpDocInfo = $node->getAttribute(AttributeKey::PHP_DOC_INFO);

            if ($phpDocInfo->getVarType()) {
                $possibleDocAliases = $this->appendPossibleAliases($phpDocInfo->getVarType(), $possibleDocAliases);
                $possibleDocAliases = $this->appendPossibleAliases($phpDocInfo->getReturnType(), $possibleDocAliases);

                foreach ($phpDocInfo->getParamTypes() as $paramType) {
                    $possibleDocAliases = $this->appendPossibleAliases($paramType, $possibleDocAliases);
                }
            }

            // e.g. "use Dotrine\ORM\Mapping as ORM" etc.
            $matches = Strings::matchAll($node->getDocComment()->getText(), '#\@(?<possible_alias>\w+)(\\\\)?#s');
            foreach ($matches as $match) {
                $possibleDocAliases[] = $match['possible_alias'];
            }
        });

        return array_unique($possibleDocAliases);
    }

    /**
     * @return string[]
     */
    private function appendPossibleAliases(Type $varType, array $possibleDocAliases): array
    {
        if ($varType instanceof AliasedObjectType) {
            $possibleDocAliases[] = $varType->getClassName();
        }

        if ($varType instanceof UnionType) {
            foreach ($varType->getTypes() as $type) {
                $possibleDocAliases = $this->appendPossibleAliases($type, $possibleDocAliases);
            }
        }

        return $possibleDocAliases;
    }
}