220 lines
6.6 KiB
PHP
220 lines
6.6 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Symfony\Component\Console\Formatter;
|
|
|
|
use Symfony\Component\Console\Exception\InvalidArgumentException;
|
|
|
|
/**
|
|
* Formatter style class for defining styles.
|
|
*
|
|
* @author Konstantin Kudryashov <ever.zet@gmail.com>
|
|
*/
|
|
class OutputFormatterStyle implements OutputFormatterStyleInterface
|
|
{
|
|
private static $availableForegroundColors = array(
|
|
'black' => array('set' => 30, 'unset' => 39),
|
|
'red' => array('set' => 31, 'unset' => 39),
|
|
'green' => array('set' => 32, 'unset' => 39),
|
|
'yellow' => array('set' => 33, 'unset' => 39),
|
|
'blue' => array('set' => 34, 'unset' => 39),
|
|
'magenta' => array('set' => 35, 'unset' => 39),
|
|
'cyan' => array('set' => 36, 'unset' => 39),
|
|
'white' => array('set' => 37, 'unset' => 39),
|
|
'default' => array('set' => 39, 'unset' => 39),
|
|
);
|
|
private static $availableBackgroundColors = array(
|
|
'black' => array('set' => 40, 'unset' => 49),
|
|
'red' => array('set' => 41, 'unset' => 49),
|
|
'green' => array('set' => 42, 'unset' => 49),
|
|
'yellow' => array('set' => 43, 'unset' => 49),
|
|
'blue' => array('set' => 44, 'unset' => 49),
|
|
'magenta' => array('set' => 45, 'unset' => 49),
|
|
'cyan' => array('set' => 46, 'unset' => 49),
|
|
'white' => array('set' => 47, 'unset' => 49),
|
|
'default' => array('set' => 49, 'unset' => 49),
|
|
);
|
|
private static $availableOptions = array(
|
|
'bold' => array('set' => 1, 'unset' => 22),
|
|
'underscore' => array('set' => 4, 'unset' => 24),
|
|
'blink' => array('set' => 5, 'unset' => 25),
|
|
'reverse' => array('set' => 7, 'unset' => 27),
|
|
'conceal' => array('set' => 8, 'unset' => 28),
|
|
);
|
|
|
|
private $foreground;
|
|
private $background;
|
|
private $options = array();
|
|
|
|
/**
|
|
* Initializes output formatter style.
|
|
*
|
|
* @param string|null $foreground The style foreground color name
|
|
* @param string|null $background The style background color name
|
|
* @param array $options The style options
|
|
*/
|
|
public function __construct($foreground = null, $background = null, array $options = array())
|
|
{
|
|
if (null !== $foreground) {
|
|
$this->setForeground($foreground);
|
|
}
|
|
if (null !== $background) {
|
|
$this->setBackground($background);
|
|
}
|
|
if (count($options)) {
|
|
$this->setOptions($options);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Sets style foreground color.
|
|
*
|
|
* @param string|null $color The color name
|
|
*
|
|
* @throws InvalidArgumentException When the color name isn't defined
|
|
*/
|
|
public function setForeground($color = null)
|
|
{
|
|
if (null === $color) {
|
|
$this->foreground = null;
|
|
|
|
return;
|
|
}
|
|
|
|
if (!isset(static::$availableForegroundColors[$color])) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'Invalid foreground color specified: "%s". Expected one of (%s)',
|
|
$color,
|
|
implode(', ', array_keys(static::$availableForegroundColors))
|
|
));
|
|
}
|
|
|
|
$this->foreground = static::$availableForegroundColors[$color];
|
|
}
|
|
|
|
/**
|
|
* Sets style background color.
|
|
*
|
|
* @param string|null $color The color name
|
|
*
|
|
* @throws InvalidArgumentException When the color name isn't defined
|
|
*/
|
|
public function setBackground($color = null)
|
|
{
|
|
if (null === $color) {
|
|
$this->background = null;
|
|
|
|
return;
|
|
}
|
|
|
|
if (!isset(static::$availableBackgroundColors[$color])) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'Invalid background color specified: "%s". Expected one of (%s)',
|
|
$color,
|
|
implode(', ', array_keys(static::$availableBackgroundColors))
|
|
));
|
|
}
|
|
|
|
$this->background = static::$availableBackgroundColors[$color];
|
|
}
|
|
|
|
/**
|
|
* Sets some specific style option.
|
|
*
|
|
* @param string $option The option name
|
|
*
|
|
* @throws InvalidArgumentException When the option name isn't defined
|
|
*/
|
|
public function setOption($option)
|
|
{
|
|
if (!isset(static::$availableOptions[$option])) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'Invalid option specified: "%s". Expected one of (%s)',
|
|
$option,
|
|
implode(', ', array_keys(static::$availableOptions))
|
|
));
|
|
}
|
|
|
|
if (!in_array(static::$availableOptions[$option], $this->options)) {
|
|
$this->options[] = static::$availableOptions[$option];
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Unsets some specific style option.
|
|
*
|
|
* @param string $option The option name
|
|
*
|
|
* @throws InvalidArgumentException When the option name isn't defined
|
|
*/
|
|
public function unsetOption($option)
|
|
{
|
|
if (!isset(static::$availableOptions[$option])) {
|
|
throw new InvalidArgumentException(sprintf(
|
|
'Invalid option specified: "%s". Expected one of (%s)',
|
|
$option,
|
|
implode(', ', array_keys(static::$availableOptions))
|
|
));
|
|
}
|
|
|
|
$pos = array_search(static::$availableOptions[$option], $this->options);
|
|
if (false !== $pos) {
|
|
unset($this->options[$pos]);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* {@inheritdoc}
|
|
*/
|
|
public function setOptions(array $options)
|
|
{
|
|
$this->options = array();
|
|
|
|
foreach ($options as $option) {
|
|
$this->setOption($option);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Applies the style to a given text.
|
|
*
|
|
* @param string $text The text to style
|
|
*
|
|
* @return string
|
|
*/
|
|
public function apply($text)
|
|
{
|
|
$setCodes = array();
|
|
$unsetCodes = array();
|
|
|
|
if (null !== $this->foreground) {
|
|
$setCodes[] = $this->foreground['set'];
|
|
$unsetCodes[] = $this->foreground['unset'];
|
|
}
|
|
if (null !== $this->background) {
|
|
$setCodes[] = $this->background['set'];
|
|
$unsetCodes[] = $this->background['unset'];
|
|
}
|
|
if (count($this->options)) {
|
|
foreach ($this->options as $option) {
|
|
$setCodes[] = $option['set'];
|
|
$unsetCodes[] = $option['unset'];
|
|
}
|
|
}
|
|
|
|
if (0 === count($setCodes)) {
|
|
return $text;
|
|
}
|
|
|
|
return sprintf("\033[%sm%s\033[%sm", implode(';', $setCodes), $text, implode(';', $unsetCodes));
|
|
}
|
|
}
|