72 lines
1.9 KiB
PHP
72 lines
1.9 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of Psy Shell.
|
|
*
|
|
* (c) 2012-2018 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 ($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;
|
|
}
|
|
}
|