AliasUsesResolver.php

<?php

declare(strict_types=1);

namespace Rector\CodingStyle\ClassNameImport;

use PhpParser\Node;
use PhpParser\Node\Identifier;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Namespace_;
use PhpParser\Node\Stmt\Use_;
use PhpParser\Node\UseItem;
use Rector\PhpParser\Node\FileNode;

final readonly class AliasUsesResolver
{
    public function __construct(
        private UseImportsTraverser $useImportsTraverser
    ) {
    }

    /**
     * @param Stmt[] $stmts
     * @return string[]
     */
    public function resolveFromNode(Node $node, array $stmts): array
    {
        if (! $node instanceof Namespace_ && ! $node instanceof FileNode) {
            /** @var Namespace_[]|FileNode[] $namespaces */
            $namespaces = array_filter(
                $stmts,
                static fn (Stmt $stmt): bool => $stmt instanceof Namespace_ || $stmt instanceof FileNode
            );
            if (count($namespaces) !== 1) {
                return [];
            }

            $node = current($namespaces);
        }

        return $this->resolveFromStmts($node->stmts);
    }

    /**
     * @param Stmt[] $stmts
     * @return string[]
     */
    public function resolveFromStmts(array $stmts): array
    {
        $aliasedUses = [];

        /** @param Use_::TYPE_* $useType */
        $this->useImportsTraverser->traverserStmts($stmts, static function (
            int $useType,
            UseItem $useItem,
            string $name
        ) use (&$aliasedUses): void {
            if ($useType !== Use_::TYPE_NORMAL) {
                return;
            }

            if (! $useItem->alias instanceof Identifier) {
                return;
            }

            $aliasedUses[] = $name;
        });

        return $aliasedUses;
    }
}