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\Name;
Chris@0: use PhpParser\Node\Stmt\Namespace_;
Chris@0: use Psy\CodeCleaner;
Chris@0: 
Chris@0: /**
Chris@0:  * Provide implicit namespaces for subsequent execution.
Chris@0:  *
Chris@0:  * The namespace pass remembers the last standalone namespace line encountered:
Chris@0:  *
Chris@0:  *     namespace Foo\Bar;
Chris@0:  *
Chris@0:  * ... which it then applies implicitly to all future evaluated code, until the
Chris@0:  * namespace is replaced by another namespace. To reset to the top level
Chris@0:  * namespace, enter `namespace {}`. This is a bit ugly, but it does the trick :)
Chris@0:  */
Chris@0: class NamespacePass extends CodeCleanerPass
Chris@0: {
Chris@0:     private $namespace = null;
Chris@0:     private $cleaner;
Chris@0: 
Chris@0:     /**
Chris@0:      * @param CodeCleaner $cleaner
Chris@0:      */
Chris@0:     public function __construct(CodeCleaner $cleaner)
Chris@0:     {
Chris@0:         $this->cleaner = $cleaner;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * If this is a standalone namespace line, remember it for later.
Chris@0:      *
Chris@0:      * Otherwise, apply remembered namespaces to the code until a new namespace
Chris@0:      * is encountered.
Chris@0:      *
Chris@0:      * @param array $nodes
Chris@0:      */
Chris@0:     public function beforeTraverse(array $nodes)
Chris@0:     {
Chris@0:         if (empty($nodes)) {
Chris@0:             return $nodes;
Chris@0:         }
Chris@0: 
Chris@0:         $last = end($nodes);
Chris@0:         if (!$last instanceof Namespace_) {
Chris@0:             return $this->namespace ? array(new Namespace_($this->namespace, $nodes)) : $nodes;
Chris@0:         }
Chris@0: 
Chris@0:         $this->setNamespace($last->name);
Chris@0: 
Chris@0:         return $nodes;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Remember the namespace and (re)set the namespace on the CodeCleaner as
Chris@0:      * well.
Chris@0:      *
Chris@0:      * @param null|Name $namespace
Chris@0:      */
Chris@0:     private function setNamespace($namespace)
Chris@0:     {
Chris@0:         $this->namespace = $namespace;
Chris@0:         $this->cleaner->setNamespace($namespace === null ? null : $namespace->parts);
Chris@0:     }
Chris@0: }