Chris@0: <?php
Chris@0: 
Chris@0: /*
Chris@0:  * This file is part of Psy Shell.
Chris@0:  *
Chris@0:  * (c) 2012-2017 Justin Hileman
Chris@0:  *
Chris@0:  * For the full copyright and license information, please view the LICENSE
Chris@0:  * file that was distributed with this source code.
Chris@0:  */
Chris@0: 
Chris@0: namespace Psy\Formatter;
Chris@0: 
Chris@0: use JakubOnderka\PhpConsoleHighlighter\Highlighter;
Chris@0: use Psy\Configuration;
Chris@0: use Psy\ConsoleColorFactory;
Chris@0: use Psy\Exception\RuntimeException;
Chris@0: 
Chris@0: /**
Chris@0:  * A pretty-printer for code.
Chris@0:  */
Chris@0: class CodeFormatter implements Formatter
Chris@0: {
Chris@0:     /**
Chris@0:      * Format the code represented by $reflector.
Chris@0:      *
Chris@0:      * @param \Reflector  $reflector
Chris@0:      * @param null|string $colorMode (default: null)
Chris@0:      *
Chris@0:      * @return string formatted code
Chris@0:      */
Chris@0:     public static function format(\Reflector $reflector, $colorMode = null)
Chris@0:     {
Chris@0:         if (!self::isReflectable($reflector)) {
Chris@0:             throw new RuntimeException('Source code unavailable.');
Chris@0:         }
Chris@0: 
Chris@0:         $colorMode = $colorMode ?: Configuration::COLOR_MODE_AUTO;
Chris@0: 
Chris@0:         if ($reflector instanceof \ReflectionGenerator) {
Chris@0:             $reflector = $reflector->getFunction();
Chris@0:         }
Chris@0: 
Chris@0:         if ($fileName = $reflector->getFileName()) {
Chris@0:             if (!is_file($fileName)) {
Chris@0:                 throw new RuntimeException('Source code unavailable.');
Chris@0:             }
Chris@0: 
Chris@0:             $file  = file_get_contents($fileName);
Chris@0:             $start = $reflector->getStartLine();
Chris@0:             $end   = $reflector->getEndLine() - $start;
Chris@0: 
Chris@0:             $factory = new ConsoleColorFactory($colorMode);
Chris@0:             $colors = $factory->getConsoleColor();
Chris@0:             $highlighter = new Highlighter($colors);
Chris@0: 
Chris@0:             return $highlighter->getCodeSnippet($file, $start, 0, $end);
Chris@0:         } else {
Chris@0:             throw new RuntimeException('Source code unavailable.');
Chris@0:         }
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Check whether a Reflector instance is reflectable by this formatter.
Chris@0:      *
Chris@0:      * @param \Reflector $reflector
Chris@0:      *
Chris@0:      * @return bool
Chris@0:      */
Chris@0:     private static function isReflectable(\Reflector $reflector)
Chris@0:     {
Chris@0:         return $reflector instanceof \ReflectionClass ||
Chris@0:             $reflector instanceof \ReflectionFunctionAbstract ||
Chris@0:             $reflector instanceof \ReflectionGenerator;
Chris@0:     }
Chris@0: }