Chris@13
|
1 <?php
|
Chris@13
|
2
|
Chris@13
|
3 /*
|
Chris@13
|
4 * This file is part of Psy Shell.
|
Chris@13
|
5 *
|
Chris@13
|
6 * (c) 2012-2018 Justin Hileman
|
Chris@13
|
7 *
|
Chris@13
|
8 * For the full copyright and license information, please view the LICENSE
|
Chris@13
|
9 * file that was distributed with this source code.
|
Chris@13
|
10 */
|
Chris@13
|
11
|
Chris@13
|
12 namespace Psy\Formatter;
|
Chris@13
|
13
|
Chris@13
|
14 use JakubOnderka\PhpConsoleHighlighter\Highlighter;
|
Chris@13
|
15 use Psy\Configuration;
|
Chris@13
|
16 use Psy\ConsoleColorFactory;
|
Chris@13
|
17 use Psy\Exception\RuntimeException;
|
Chris@13
|
18
|
Chris@13
|
19 /**
|
Chris@13
|
20 * A pretty-printer for code.
|
Chris@13
|
21 */
|
Chris@13
|
22 class CodeFormatter implements Formatter
|
Chris@13
|
23 {
|
Chris@13
|
24 /**
|
Chris@13
|
25 * Format the code represented by $reflector.
|
Chris@13
|
26 *
|
Chris@13
|
27 * @param \Reflector $reflector
|
Chris@13
|
28 * @param null|string $colorMode (default: null)
|
Chris@13
|
29 *
|
Chris@13
|
30 * @return string formatted code
|
Chris@13
|
31 */
|
Chris@13
|
32 public static function format(\Reflector $reflector, $colorMode = null)
|
Chris@13
|
33 {
|
Chris@13
|
34 if (!self::isReflectable($reflector)) {
|
Chris@13
|
35 throw new RuntimeException('Source code unavailable');
|
Chris@13
|
36 }
|
Chris@13
|
37
|
Chris@13
|
38 $colorMode = $colorMode ?: Configuration::COLOR_MODE_AUTO;
|
Chris@13
|
39
|
Chris@13
|
40 if ($fileName = $reflector->getFileName()) {
|
Chris@17
|
41 if (!\is_file($fileName)) {
|
Chris@13
|
42 throw new RuntimeException('Source code unavailable');
|
Chris@13
|
43 }
|
Chris@13
|
44
|
Chris@17
|
45 $file = \file_get_contents($fileName);
|
Chris@13
|
46 $start = $reflector->getStartLine();
|
Chris@13
|
47 $end = $reflector->getEndLine() - $start;
|
Chris@13
|
48
|
Chris@13
|
49 $factory = new ConsoleColorFactory($colorMode);
|
Chris@13
|
50 $colors = $factory->getConsoleColor();
|
Chris@13
|
51 $highlighter = new Highlighter($colors);
|
Chris@13
|
52
|
Chris@13
|
53 return $highlighter->getCodeSnippet($file, $start, 0, $end);
|
Chris@13
|
54 } else {
|
Chris@13
|
55 throw new RuntimeException('Source code unavailable');
|
Chris@13
|
56 }
|
Chris@13
|
57 }
|
Chris@13
|
58
|
Chris@13
|
59 /**
|
Chris@13
|
60 * Check whether a Reflector instance is reflectable by this formatter.
|
Chris@13
|
61 *
|
Chris@13
|
62 * @param \Reflector $reflector
|
Chris@13
|
63 *
|
Chris@13
|
64 * @return bool
|
Chris@13
|
65 */
|
Chris@13
|
66 private static function isReflectable(\Reflector $reflector)
|
Chris@13
|
67 {
|
Chris@13
|
68 return $reflector instanceof \ReflectionClass ||
|
Chris@16
|
69 $reflector instanceof \ReflectionFunctionAbstract;
|
Chris@13
|
70 }
|
Chris@13
|
71 }
|