Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of Psy Shell.
Chris@0:  *
Chris@0:  * (c) 2012-2017 Justin Hileman
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Psy\CodeCleaner;
Chris@0: 
Chris@0: use PhpParser\Node;
Chris@0: use PhpParser\Node\Name;
Chris@0: use PhpParser\Node\Name\FullyQualified as FullyQualifiedName;
Chris@0: use PhpParser\Node\Stmt\Namespace_;
Chris@0: 
Chris@0: /**
Chris@0:  * Abstract namespace-aware code cleaner pass.
Chris@0:  */
Chris@0: abstract class NamespaceAwarePass extends CodeCleanerPass
Chris@0: {
Chris@0:     protected $namespace;
Chris@0:     protected $currentScope;
Chris@0: 
Chris@0:     /**
Chris@0:      * @todo should this be final? Extending classes should be sure to either
Chris@0:      * use afterTraverse or call parent::beforeTraverse() when overloading.
Chris@0:      *
Chris@0:      * Reset the namespace and the current scope before beginning analysis
Chris@0:      */
Chris@0:     public function beforeTraverse(array $nodes)
Chris@0:     {
Chris@0:         $this->namespace    = array();
Chris@0:         $this->currentScope = array();
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * @todo should this be final? Extending classes should be sure to either use
Chris@0:      * leaveNode or call parent::enterNode() when overloading
Chris@0:      *
Chris@0:      * @param Node $node
Chris@0:      */
Chris@0:     public function enterNode(Node $node)
Chris@0:     {
Chris@0:         if ($node instanceof Namespace_) {
Chris@0:             $this->namespace = isset($node->name) ? $node->name->parts : array();
Chris@0:         }
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Get a fully-qualified name (class, function, interface, etc).
Chris@0:      *
Chris@0:      * @param mixed $name
Chris@0:      *
Chris@0:      * @return string
Chris@0:      */
Chris@0:     protected function getFullyQualifiedName($name)
Chris@0:     {
Chris@0:         if ($name instanceof FullyQualifiedName) {
Chris@0:             return implode('\\', $name->parts);
Chris@0:         } elseif ($name instanceof Name) {
Chris@0:             $name = $name->parts;
Chris@0:         } elseif (!is_array($name)) {
Chris@0:             $name = array($name);
Chris@0:         }
Chris@0: 
Chris@0:         return implode('\\', array_merge($this->namespace, $name));
Chris@0:     }
Chris@0: }