Humbug\PhpScoper\PhpParser\NodeVisitor\Collection\UseStmtCollection

Complexity

Total Complexity

14

Size/Duplication

Total Lines	78
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Test Coverage

Coverage

91.67%

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	add()	0	4	1
C	findStatementForNode()	0	39	12
A	getIterator()	0	4	1

<?php

declare(strict_types=1);

/*
 * This file is part of the humbug/php-scoper package.
 *
 * Copyright (c) 2017 Théo FIDRY <theo.fidry@gmail.com>,
 *                    Pádraic Brady <padraic.brady@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Humbug\PhpScoper\PhpParser\NodeVisitor\Collection;

use ArrayIterator;
use Humbug\PhpScoper\PhpParser\NodeVisitor\AppendParentNode;
use IteratorAggregate;
use PhpParser\Node\Expr\ConstFetch;
use PhpParser\Node\Expr\FuncCall;
use PhpParser\Node\Name;
use PhpParser\Node\Stmt\Use_;
use PhpParser\Node\Stmt\UseUse;
use function Humbug\PhpScoper\clone_node;

/**
 * Utility class collecting all the use statements for the scoped files allowing to easily find the use which a node
 * may use.
 *
 * @private
 */
final class UseStmtCollection implements IteratorAggregate
{
    /**
     * @var Use_[][]
     */
    private $nodes = [
        null => [],
    ];

    public function add(?Name $namespaceName, Use_ $node): void
    {
        $this->nodes[(string) $namespaceName][] = clone_node($node);
    }

    /**
     * Finds the statements matching the given name.
     *
     * $name = 'Foo';
     *
     * use X;
     * use Bar\Foo;
     * use Y;
     *
     * will return the use statement for `Bar\Foo`.
     *
     * @param Name|null $namespaceName
     * @param Name      $node
     *
     * @return null|Name
     */
    public function findStatementForNode(?Name $namespaceName, Name $node): ?Name
    {
        $name = strtolower($node->getFirst());

        $parentNode = AppendParentNode::findParent($node);

        $useStatements = $this->nodes[(string) $namespaceName] ?? [];

        foreach ($useStatements as $use_) {
            foreach ($use_->uses as $useStatement) {
                if ($useStatement instanceof UseUse) {
                    if ($name === strtolower($useStatement->alias)) {
                        if ($parentNode instanceof FuncCall && 1 === count($node->parts)) {
                            if (Use_::TYPE_FUNCTION === $use_->type) {
                                return $useStatement->name;
                            }

                            continue;
                        }

                        if ($parentNode instanceof ConstFetch && 1 === count($node->parts)) {
                            if (Use_::TYPE_CONSTANT === $use_->type) {
                                return $useStatement->name;
                            }

                            continue;
                        }

                        // Match the alias
                        return $useStatement->name;
                    } elseif (null !== $useStatement->alias) {
                        continue;
                    }
                }
            }
        }

        return null;
    }

    /**
     * @inheritdoc
     */
    public function getIterator(): iterable
    {
        return new ArrayIterator($this->nodes);
    }
}