Version 1
[yaffs-website] / vendor / psy / psysh / src / Psy / TabCompletion / Matcher / MongoClientMatcher.php
diff --git a/vendor/psy/psysh/src/Psy/TabCompletion/Matcher/MongoClientMatcher.php b/vendor/psy/psysh/src/Psy/TabCompletion/Matcher/MongoClientMatcher.php
new file mode 100644 (file)
index 0000000..3573342
--- /dev/null
@@ -0,0 +1,71 @@
+<?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\TabCompletion\Matcher;
+
+/**
+ * A MongoDB Client tab completion Matcher.
+ *
+ * This matcher provides completion for MongoClient database names.
+ *
+ * @author Marc Garcia <markcial@gmail.com>
+ */
+class MongoClientMatcher extends AbstractContextAwareMatcher
+{
+    /**
+     * {@inheritdoc}
+     */
+    public function getMatches(array $tokens, array $info = array())
+    {
+        $input = $this->getInput($tokens);
+
+        $firstToken = array_pop($tokens);
+        if (self::tokenIs($firstToken, self::T_STRING)) {
+            // second token is the object operator
+            array_pop($tokens);
+        }
+        $objectToken = array_pop($tokens);
+        $objectName = str_replace('$', '', $objectToken[1]);
+        $object = $this->getVariable($objectName);
+
+        if (!$object instanceof \MongoClient) {
+            return array();
+        }
+
+        $list = $object->listDBs();
+
+        return array_filter(
+            array_map(function ($info) {
+                return $info['name'];
+            }, $list['databases']),
+            function ($var) use ($input) {
+                return AbstractMatcher::startsWith($input, $var);
+            }
+        );
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function hasMatched(array $tokens)
+    {
+        $token = array_pop($tokens);
+        $prevToken = array_pop($tokens);
+
+        switch (true) {
+            case self::tokenIs($token, self::T_OBJECT_OPERATOR):
+            case self::tokenIs($prevToken, self::T_OBJECT_OPERATOR):
+                return true;
+        }
+
+        return false;
+    }
+}