Version 1

[yaffs-website] / vendor / psy / psysh / src / Psy / Formatter / CodeFormatter.php

diff --git a/vendor/psy/psysh/src/Psy/Formatter/CodeFormatter.php b/vendor/psy/psysh/src/Psy/Formatter/CodeFormatter.php
new file mode 100644 (file)
index 0000000..bab5b82
--- /dev/null
+++ b/vendor/psy/psysh/src/Psy/Formatter/CodeFormatter.php
@@ -0,0 +1,76 @@
+<?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\Formatter;
+
+use JakubOnderka\PhpConsoleHighlighter\Highlighter;
+use Psy\Configuration;
+use Psy\ConsoleColorFactory;
+use Psy\Exception\RuntimeException;
+
+/**
+ * A pretty-printer for code.
+ */
+class CodeFormatter implements Formatter
+{
+    /**
+     * Format the code represented by $reflector.
+     *
+     * @param \Reflector  $reflector
+     * @param null|string $colorMode (default: null)
+     *
+     * @return string formatted code
+     */
+    public static function format(\Reflector $reflector, $colorMode = null)
+    {
+        if (!self::isReflectable($reflector)) {
+            throw new RuntimeException('Source code unavailable.');
+        }
+
+        $colorMode = $colorMode ?: Configuration::COLOR_MODE_AUTO;
+
+        if ($reflector instanceof \ReflectionGenerator) {
+            $reflector = $reflector->getFunction();
+        }
+
+        if ($fileName = $reflector->getFileName()) {
+            if (!is_file($fileName)) {
+                throw new RuntimeException('Source code unavailable.');
+            }
+
+            $file  = file_get_contents($fileName);
+            $start = $reflector->getStartLine();
+            $end   = $reflector->getEndLine() - $start;
+
+            $factory = new ConsoleColorFactory($colorMode);
+            $colors = $factory->getConsoleColor();
+            $highlighter = new Highlighter($colors);
+
+            return $highlighter->getCodeSnippet($file, $start, 0, $end);
+        } else {
+            throw new RuntimeException('Source code unavailable.');
+        }
+    }
+
+    /**
+     * Check whether a Reflector instance is reflectable by this formatter.
+     *
+     * @param \Reflector $reflector
+     *
+     * @return bool
+     */
+    private static function isReflectable(\Reflector $reflector)
+    {
+        return $reflector instanceof \ReflectionClass ||
+            $reflector instanceof \ReflectionFunctionAbstract ||
+            $reflector instanceof \ReflectionGenerator;
+    }
+}