Humbug\PhpScoper\PhpParser\NodeVisitor\Collection\NamespaceStmtCollection

Complexity

Total Complexity

11

Size/Duplication

Total Lines	78
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

70.83%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	add()	0	6	1
A	findNamespaceForNode()	0	13	3
A	getCurrentNamespaceName()	0	8	2
A	count()	0	4	1
A	getNodeNamespace()	0	14	3
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 Countable;
use Humbug\PhpScoper\PhpParser\NodeVisitor\ParentNodeAppender;
use IteratorAggregate;
use PhpParser\Node;
use PhpParser\Node\Name;
use PhpParser\Node\Stmt\Namespace_;
use function count;

/**
 * Utility class collecting all the namespaces for the scoped files allowing to easily find the namespace to which
 * belongs a node.
 *
 * @private
 */
final class NamespaceStmtCollection implements IteratorAggregate, Countable
{
    /**
     * @var Namespace_[]
     */
    private $nodes = [];

    /**
     * @var Name|null[] Associative array with the potentially prefixed namespace names as keys and their original name
     *                  as value.
     */
    private $mapping = [];

    /**
     * @param Namespace_ $node         New namespace, may have been prefixed.
     * @param Namespace_ $originalName Original unchanged namespace.
     */
    public function add(Namespace_ $node, Namespace_ $originalName)
    {
        $this->nodes[] = $originalName;

        $this->mapping[(string) $node->name] = $originalName->name;
    }

    public function findNamespaceForNode(Node $node): ?Name
    {
        if (0 === count($this->nodes)) {
            return null;
        }

        // Shortcut if there is only one namespace
        if (1 === count($this->nodes)) {
            return $this->nodes[0]->name;
        }

        return $this->getNodeNamespace($node);
    }

    public function getCurrentNamespaceName(): ?Name
    {
        if (0 === count($this->nodes)) {
            return null;
        }

        return end($this->nodes)->name;
    }

    /**
     * @inheritdoc
     */
    public function count(): int
    {
        return count($this->nodes);
    }

    private function getNodeNamespace(Node $node): ?Name
    {
        if (false === ParentNodeAppender::hasParent($node)) {
            return null;
        }

        $parentNode = ParentNodeAppender::getParent($node);

        if ($parentNode instanceof Namespace_) {
            return $this->mapping[(string) $parentNode->name];
        }

        return $this->getNodeNamespace($parentNode);
    }

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