comparison vendor/psy/psysh/src/Psy/CodeCleaner/UseStatementPass.php @ 0:4c8ae668cc8c

Initial import (non-working)

+<?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;
+use PhpParser\Node\Name;
+use PhpParser\Node\Name\FullyQualified as FullyQualifiedName;
+use PhpParser\Node\Stmt\GroupUse;
+use PhpParser\Node\Stmt\Namespace_;
+use PhpParser\Node\Stmt\Use_;
+
+/**
+ * Provide implicit use statements for subsequent execution.
+ *
+ * The use statement pass remembers the last use statement line encountered:
+ *
+ *     use Foo\Bar as Baz;
+ *
+ * ... which it then applies implicitly to all future evaluated code, until the
+ * current namespace is replaced by another namespace.
+ */
+class UseStatementPass extends CodeCleanerPass
+{
+    private $aliases       = array();
+    private $lastAliases   = array();
+    private $lastNamespace = null;
+
+    /**
+     * Re-load the last set of use statements on re-entering a namespace.
+     *
+     * This isn't how namespaces normally work, but because PsySH has to spin
+     * up a new namespace for every line of code, we do this to make things
+     * work like you'd expect.
+     *
+     * @param Node $node
+     */
+    public function enterNode(Node $node)
+    {
+        if ($node instanceof Namespace_) {
+            // If this is the same namespace as last namespace, let's do ourselves
+            // a favor and reload all the aliases...
+            if (strtolower($node->name) === strtolower($this->lastNamespace)) {
+                $this->aliases = $this->lastAliases;
+            }
+        }
+    }
+
+    /**
+     * If this statement is a namespace, forget all the aliases we had.
+     *
+     * If it's a use statement, remember the alias for later. Otherwise, apply
+     * remembered aliases to the code.
+     *
+     * @param Node $node
+     */
+    public function leaveNode(Node $node)
+    {
+        if ($node instanceof Use_) {
+            // Store a reference to every "use" statement, because we'll need
+            // them in a bit.
+            foreach ($node->uses as $use) {
+                $this->aliases[strtolower($use->alias)] = $use->name;
+            }
+
+            return false;
+        } elseif ($node instanceof GroupUse) {
+            // Expand every "use" statement in the group into a full, standalone
+            // "use" and store 'em with the others.
+            foreach ($node->uses as $use) {
+                $this->aliases[strtolower($use->alias)] = Name::concat($node->prefix, $use->name, array(
+                    'startLine' => $node->prefix->getAttribute('startLine'),
+                    'endLine'   => $use->name->getAttribute('endLine'),
+                ));
+            }
+
+            return false;
+        } elseif ($node instanceof Namespace_) {
+            // Start fresh, since we're done with this namespace.
+            $this->lastNamespace = $node->name;
+            $this->lastAliases   = $this->aliases;
+            $this->aliases       = array();
+        } else {
+            foreach ($node as $name => $subNode) {
+                if ($subNode instanceof Name) {
+                    // Implicitly thunk all aliases.
+                    if ($replacement = $this->findAlias($subNode)) {
+                        $node->$name = $replacement;
+                    }
+                }
+            }
+
+            return $node;
+        }
+    }
+
+    /**
+     * Find class/namespace aliases.
+     *
+     * @param Name $name
+     *
+     * @return FullyQualifiedName|null
+     */
+    private function findAlias(Name $name)
+    {
+        $that = strtolower($name);
+        foreach ($this->aliases as $alias => $prefix) {
+            if ($that === $alias) {
+                return new FullyQualifiedName($prefix->toString());
+            } elseif (substr($that, 0, strlen($alias) + 1) === $alias . '\\') {
+                return new FullyQualifiedName($prefix->toString() . substr($name, strlen($alias)));
+            }
+        }
+    }
+}