Version 1
[yaffs-website] / vendor / psy / psysh / src / Psy / CodeCleaner / NamespacePass.php
diff --git a/vendor/psy/psysh/src/Psy/CodeCleaner/NamespacePass.php b/vendor/psy/psysh/src/Psy/CodeCleaner/NamespacePass.php
new file mode 100644 (file)
index 0000000..40cc6bb
--- /dev/null
@@ -0,0 +1,79 @@
+<?php
+
+/*
+ * This file is part of Psy Shell.
+ *
+ * (c) 2012-2017 Justin Hileman
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Psy\CodeCleaner;
+
+use PhpParser\Node\Name;
+use PhpParser\Node\Stmt\Namespace_ as NamespaceStmt;
+use Psy\CodeCleaner;
+
+/**
+ * Provide implicit namespaces for subsequent execution.
+ *
+ * The namespace pass remembers the last standalone namespace line encountered:
+ *
+ *     namespace Foo\Bar;
+ *
+ * ... which it then applies implicitly to all future evaluated code, until the
+ * namespace is replaced by another namespace. To reset to the top level
+ * namespace, enter `namespace {}`. This is a bit ugly, but it does the trick :)
+ */
+class NamespacePass extends CodeCleanerPass
+{
+    private $namespace = null;
+    private $cleaner;
+
+    /**
+     * @param CodeCleaner $cleaner
+     */
+    public function __construct(CodeCleaner $cleaner)
+    {
+        $this->cleaner = $cleaner;
+    }
+
+    /**
+     * If this is a standalone namespace line, remember it for later.
+     *
+     * Otherwise, apply remembered namespaces to the code until a new namespace
+     * is encountered.
+     *
+     * @param array $nodes
+     */
+    public function beforeTraverse(array $nodes)
+    {
+        $first = reset($nodes);
+        if (count($nodes) === 1 && $first instanceof NamespaceStmt && empty($first->stmts)) {
+            $this->setNamespace($first->name);
+        } else {
+            foreach ($nodes as $key => $node) {
+                if ($node instanceof NamespaceStmt) {
+                    $this->setNamespace(null);
+                } elseif ($this->namespace !== null) {
+                    $nodes[$key] = new NamespaceStmt($this->namespace, array($node));
+                }
+            }
+        }
+
+        return $nodes;
+    }
+
+    /**
+     * Remember the namespace and (re)set the namespace on the CodeCleaner as
+     * well.
+     *
+     * @param null|Name $namespace
+     */
+    private function setNamespace($namespace)
+    {
+        $this->namespace = $namespace;
+        $this->cleaner->setNamespace($namespace === null ? null : $namespace->parts);
+    }
+}