hanneskod\classtools\Transformer\Action\NamespaceWrapper

Complexity

Total Complexity

6

Size/Duplication

Total Lines	44
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	beforeTraverse()	0	22	5

<?php
/**
 * This program is free software. It comes without any warranty, to
 * the extent permitted by applicable law. You can redistribute it
 * and/or modify it under the terms of the Do What The Fuck You Want
 * To Public License, Version 2, as published by Sam Hocevar. See
 * http://www.wtfpl.net/ for more details.
 */

declare(strict_types = 1);

namespace hanneskod\classtools\Transformer\Action;

use PhpParser\NodeVisitorAbstract;
use PhpParser\Node\Stmt\Namespace_;
use PhpParser\Node\Name;

/**
 * Wrap code in namespace
 *
 * @author Hannes Forsgård <hannes.forsgard@fripost.org>
 */
class NamespaceWrapper extends NodeVisitorAbstract
{
    /**
     * @var string Name of namespace
     */
    private $namespaceName;

    /**
     * Wrap code in namespace
     */
    public function __construct(string $namespaceName)
    {
        $this->namespaceName = $namespaceName;
    }

    /**
     * {inheritdoc}
     *
     * @param  array $nodes
     * @return Namespace_[]
     */
    public function beforeTraverse(array $nodes): array
    {
        // Merge if code is namespaced
        if (isset($nodes[0]) && $nodes[0] instanceof Namespace_) {
            if ($this->namespaceName) {
                if ((string)$nodes[0]->name == '') {
                    $nodes[0]->name = new Name($this->namespaceName);
                } else {
                    $nodes[0]->name = Name::concat($this->namespaceName, $nodes[0]->name);
                }
            }
            return $nodes;
        }

        // Else create new node
        return [
            new Namespace_(
                new Name($this->namespaceName),
                $nodes
            )
        ];
    }
}