www.aleph1.co.uk Git - yaffs-website/blob - vendor/psy/psysh/src/CodeCleaner/NamespacePass.php

   1 <?php
   2
   3 /*
   4  * This file is part of Psy Shell.
   5  *
   6  * (c) 2012-2018 Justin Hileman
   7  *
   8  * For the full copyright and license information, please view the LICENSE
   9  * file that was distributed with this source code.
  10  */
  11
  12 namespace Psy\CodeCleaner;
  13
  14 use PhpParser\Node\Name;
  15 use PhpParser\Node\Stmt\Namespace_;
  16 use Psy\CodeCleaner;
  17
  18 /**
  19  * Provide implicit namespaces for subsequent execution.
  20  *
  21  * The namespace pass remembers the last standalone namespace line encountered:
  22  *
  23  *     namespace Foo\Bar;
  24  *
  25  * ... which it then applies implicitly to all future evaluated code, until the
  26  * namespace is replaced by another namespace. To reset to the top level
  27  * namespace, enter `namespace {}`. This is a bit ugly, but it does the trick :)
  28  */
  29 class NamespacePass extends CodeCleanerPass
  30 {
  31     private $namespace = null;
  32     private $cleaner;
  33
  34     /**
  35      * @param CodeCleaner $cleaner
  36      */
  37     public function __construct(CodeCleaner $cleaner)
  38     {
  39         $this->cleaner = $cleaner;
  40     }
  41
  42     /**
  43      * If this is a standalone namespace line, remember it for later.
  44      *
  45      * Otherwise, apply remembered namespaces to the code until a new namespace
  46      * is encountered.
  47      *
  48      * @param array $nodes
  49      */
  50     public function beforeTraverse(array $nodes)
  51     {
  52         if (empty($nodes)) {
  53             return $nodes;
  54         }
  55
  56         $last = \end($nodes);
  57
  58         if ($last instanceof Namespace_) {
  59             $kind = $last->getAttribute('kind');
  60
  61             // Treat all namespace statements pre-PHP-Parser v3.1.2 as "open",
  62             // even though we really have no way of knowing.
  63             if ($kind === null || $kind === Namespace_::KIND_SEMICOLON) {
  64                 // Save the current namespace for open namespaces
  65                 $this->setNamespace($last->name);
  66             } else {
  67                 // Clear the current namespace after a braced namespace
  68                 $this->setNamespace(null);
  69             }
  70
  71             return $nodes;
  72         }
  73
  74         return $this->namespace ? [new Namespace_($this->namespace, $nodes)] : $nodes;
  75     }
  76
  77     /**
  78      * Remember the namespace and (re)set the namespace on the CodeCleaner as
  79      * well.
  80      *
  81      * @param null|Name $namespace
  82      */
  83     private function setNamespace($namespace)
  84     {
  85         $this->namespace = $namespace;
  86         $this->cleaner->setNamespace($namespace === null ? null : $namespace->parts);
  87     }
  88 }