pictcode / lib / Cake / Console / ConsoleOutput.php @ 995bd018
履歴 | 表示 | アノテート | ダウンロード (7.879 KB)
| 1 | 
      <?php
     | 
  
|---|---|
| 2 | 
      /**
     | 
  
| 3 | 
       * ConsoleOutput file.
     | 
  
| 4 | 
       *
     | 
  
| 5 | 
       * CakePHP(tm) : Rapid Development Framework (http://cakephp.org)
     | 
  
| 6 | 
       * Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
     | 
  
| 7 | 
       *
     | 
  
| 8 | 
       * Licensed under The MIT License
     | 
  
| 9 | 
       * For full copyright and license information, please see the LICENSE.txt
     | 
  
| 10 | 
       * Redistributions of files must retain the above copyright notice.
     | 
  
| 11 | 
       *
     | 
  
| 12 | 
       * @copyright     Copyright (c) Cake Software Foundation, Inc. (http://cakefoundation.org)
     | 
  
| 13 | 
       * @link          http://cakephp.org CakePHP(tm) Project
     | 
  
| 14 | 
       * @since         CakePHP(tm) v 2.0
     | 
  
| 15 | 
       * @license       http://www.opensource.org/licenses/mit-license.php MIT License
     | 
  
| 16 | 
       */
     | 
  
| 17 | 
       | 
  
| 18 | 
      /**
     | 
  
| 19 | 
       * Object wrapper for outputting information from a shell application.
     | 
  
| 20 | 
       * Can be connected to any stream resource that can be used with fopen()
     | 
  
| 21 | 
       *
     | 
  
| 22 | 
       * Can generate colorized output on consoles that support it. There are a few
     | 
  
| 23 | 
       * built in styles
     | 
  
| 24 | 
       *
     | 
  
| 25 | 
       * - `error` Error messages.
     | 
  
| 26 | 
       * - `warning` Warning messages.
     | 
  
| 27 | 
       * - `info` Informational messages.
     | 
  
| 28 | 
       * - `comment` Additional text.
     | 
  
| 29 | 
       * - `question` Magenta text used for user prompts
     | 
  
| 30 | 
       *
     | 
  
| 31 | 
       * By defining styles with addStyle() you can create custom console styles.
     | 
  
| 32 | 
       *
     | 
  
| 33 | 
       * ### Using styles in output
     | 
  
| 34 | 
       *
     | 
  
| 35 | 
       * You can format console output using tags with the name of the style to apply. From inside a shell object
     | 
  
| 36 | 
       *
     | 
  
| 37 | 
       * `$this->out('<warning>Overwrite:</warning> foo.php was overwritten.');`
     | 
  
| 38 | 
       *
     | 
  
| 39 | 
       * This would create orange 'Overwrite:' text, while the rest of the text would remain the normal color.
     | 
  
| 40 | 
       * See ConsoleOutput::styles() to learn more about defining your own styles. Nested styles are not supported
     | 
  
| 41 | 
       * at this time.
     | 
  
| 42 | 
       *
     | 
  
| 43 | 
       * @package       Cake.Console
     | 
  
| 44 | 
       */
     | 
  
| 45 | 
      class ConsoleOutput {  | 
  
| 46 | 
       | 
  
| 47 | 
      /**
     | 
  
| 48 | 
       * Raw output constant - no modification of output text.
     | 
  
| 49 | 
       *
     | 
  
| 50 | 
       * @var int
     | 
  
| 51 | 
       */
     | 
  
| 52 | 
      const RAW = 0;  | 
  
| 53 | 
       | 
  
| 54 | 
      /**
     | 
  
| 55 | 
       * Plain output - tags will be stripped.
     | 
  
| 56 | 
       *
     | 
  
| 57 | 
       * @var int
     | 
  
| 58 | 
       */
     | 
  
| 59 | 
      const PLAIN = 1;  | 
  
| 60 | 
       | 
  
| 61 | 
      /**
     | 
  
| 62 | 
       * Color output - Convert known tags in to ANSI color escape codes.
     | 
  
| 63 | 
       *
     | 
  
| 64 | 
       * @var int
     | 
  
| 65 | 
       */
     | 
  
| 66 | 
      const COLOR = 2;  | 
  
| 67 | 
       | 
  
| 68 | 
      /**
     | 
  
| 69 | 
       * Constant for a newline.
     | 
  
| 70 | 
       *
     | 
  
| 71 | 
       * @var string
     | 
  
| 72 | 
       */
     | 
  
| 73 | 
      const LF = PHP_EOL;  | 
  
| 74 | 
       | 
  
| 75 | 
      /**
     | 
  
| 76 | 
       * File handle for output.
     | 
  
| 77 | 
       *
     | 
  
| 78 | 
       * @var resource
     | 
  
| 79 | 
       */
     | 
  
| 80 | 
      protected $_output;  | 
  
| 81 | 
       | 
  
| 82 | 
      /**
     | 
  
| 83 | 
       * The current output type. Manipulated with ConsoleOutput::outputAs();
     | 
  
| 84 | 
       *
     | 
  
| 85 | 
       * @var int
     | 
  
| 86 | 
       */
     | 
  
| 87 | 
      protected $_outputAs = self::COLOR;  | 
  
| 88 | 
       | 
  
| 89 | 
      /**
     | 
  
| 90 | 
       * text colors used in colored output.
     | 
  
| 91 | 
       *
     | 
  
| 92 | 
       * @var array
     | 
  
| 93 | 
       */
     | 
  
| 94 | 
      protected static $_foregroundColors = array(  | 
  
| 95 | 
      'black' => 30,  | 
  
| 96 | 
      'red' => 31,  | 
  
| 97 | 
      'green' => 32,  | 
  
| 98 | 
      'yellow' => 33,  | 
  
| 99 | 
      'blue' => 34,  | 
  
| 100 | 
      'magenta' => 35,  | 
  
| 101 | 
      'cyan' => 36,  | 
  
| 102 | 
      'white' => 37  | 
  
| 103 | 
      );  | 
  
| 104 | 
       | 
  
| 105 | 
      /**
     | 
  
| 106 | 
       * background colors used in colored output.
     | 
  
| 107 | 
       *
     | 
  
| 108 | 
       * @var array
     | 
  
| 109 | 
       */
     | 
  
| 110 | 
      protected static $_backgroundColors = array(  | 
  
| 111 | 
      'black' => 40,  | 
  
| 112 | 
      'red' => 41,  | 
  
| 113 | 
      'green' => 42,  | 
  
| 114 | 
      'yellow' => 43,  | 
  
| 115 | 
      'blue' => 44,  | 
  
| 116 | 
      'magenta' => 45,  | 
  
| 117 | 
      'cyan' => 46,  | 
  
| 118 | 
      'white' => 47  | 
  
| 119 | 
      );  | 
  
| 120 | 
       | 
  
| 121 | 
      /**
     | 
  
| 122 | 
       * formatting options for colored output
     | 
  
| 123 | 
       *
     | 
  
| 124 | 
       * @var string
     | 
  
| 125 | 
       */
     | 
  
| 126 | 
      protected static $_options = array(  | 
  
| 127 | 
      'bold' => 1,  | 
  
| 128 | 
      'underline' => 4,  | 
  
| 129 | 
      'blink' => 5,  | 
  
| 130 | 
      'reverse' => 7,  | 
  
| 131 | 
      );  | 
  
| 132 | 
       | 
  
| 133 | 
      /**
     | 
  
| 134 | 
       * Styles that are available as tags in console output.
     | 
  
| 135 | 
       * You can modify these styles with ConsoleOutput::styles()
     | 
  
| 136 | 
       *
     | 
  
| 137 | 
       * @var array
     | 
  
| 138 | 
       */
     | 
  
| 139 | 
      protected static $_styles = array(  | 
  
| 140 | 
      'emergency' => array('text' => 'red', 'underline' => true),  | 
  
| 141 | 
      'alert' => array('text' => 'red', 'underline' => true),  | 
  
| 142 | 
      'critical' => array('text' => 'red', 'underline' => true),  | 
  
| 143 | 
      'error' => array('text' => 'red', 'underline' => true),  | 
  
| 144 | 
      'warning' => array('text' => 'yellow'),  | 
  
| 145 | 
      'info' => array('text' => 'cyan'),  | 
  
| 146 | 
      'debug' => array('text' => 'yellow'),  | 
  
| 147 | 
      'success' => array('text' => 'green'),  | 
  
| 148 | 
      'comment' => array('text' => 'blue'),  | 
  
| 149 | 
      'question' => array('text' => 'magenta'),  | 
  
| 150 | 
      'notice' => array('text' => 'cyan')  | 
  
| 151 | 
      );  | 
  
| 152 | 
       | 
  
| 153 | 
      /**
     | 
  
| 154 | 
       * Construct the output object.
     | 
  
| 155 | 
       *
     | 
  
| 156 | 
       * Checks for a pretty console environment. Ansicon and ConEmu allows
     | 
  
| 157 | 
       * pretty consoles on Windows, and is supported.
     | 
  
| 158 | 
       *
     | 
  
| 159 | 
       * @param string $stream The identifier of the stream to write output to.
     | 
  
| 160 | 
       */
     | 
  
| 161 | 
      public function __construct($stream = 'php://stdout') {  | 
  
| 162 | 
      $this->_output = fopen($stream, 'w');  | 
  
| 163 | 
       | 
  
| 164 | 
      if ((DS === '\\' && !(bool)env('ANSICON') && env('ConEmuANSI') !== 'ON') ||  | 
  
| 165 | 
      $stream === 'php://output' ||  | 
  
| 166 | 
      (function_exists('posix_isatty') && !posix_isatty($this->_output))  | 
  
| 167 | 
                      ) {
     | 
  
| 168 | 
      $this->_outputAs = static::PLAIN;  | 
  
| 169 | 
      }  | 
  
| 170 | 
      }  | 
  
| 171 | 
       | 
  
| 172 | 
      /**
     | 
  
| 173 | 
       * Outputs a single or multiple messages to stdout. If no parameters
     | 
  
| 174 | 
       * are passed, outputs just a newline.
     | 
  
| 175 | 
       *
     | 
  
| 176 | 
       * @param string|array $message A string or an array of strings to output
     | 
  
| 177 | 
       * @param int $newlines Number of newlines to append
     | 
  
| 178 | 
       * @return int Returns the number of bytes returned from writing to stdout.
     | 
  
| 179 | 
       */
     | 
  
| 180 | 
      public function write($message, $newlines = 1) {  | 
  
| 181 | 
      if (is_array($message)) {  | 
  
| 182 | 
      $message = implode(static::LF, $message);  | 
  
| 183 | 
      }  | 
  
| 184 | 
      return $this->_write($this->styleText($message . str_repeat(static::LF, $newlines)));  | 
  
| 185 | 
      }  | 
  
| 186 | 
       | 
  
| 187 | 
      /**
     | 
  
| 188 | 
       * Apply styling to text.
     | 
  
| 189 | 
       *
     | 
  
| 190 | 
       * @param string $text Text with styling tags.
     | 
  
| 191 | 
       * @return string String with color codes added.
     | 
  
| 192 | 
       */
     | 
  
| 193 | 
      public function styleText($text) {  | 
  
| 194 | 
      if ($this->_outputAs == static::RAW) {  | 
  
| 195 | 
      return $text;  | 
  
| 196 | 
      }  | 
  
| 197 | 
      if ($this->_outputAs == static::PLAIN) {  | 
  
| 198 | 
      $tags = implode('|', array_keys(static::$_styles));  | 
  
| 199 | 
      return preg_replace('#</?(?:' . $tags . ')>#', '', $text);  | 
  
| 200 | 
      }  | 
  
| 201 | 
      return preg_replace_callback(  | 
  
| 202 | 
      '/<(?P<tag>[a-z0-9-_]+)>(?P<text>.*?)<\/(\1)>/ims', array($this, '_replaceTags'), $text  | 
  
| 203 | 
      );  | 
  
| 204 | 
      }  | 
  
| 205 | 
       | 
  
| 206 | 
      /**
     | 
  
| 207 | 
       * Replace tags with color codes.
     | 
  
| 208 | 
       *
     | 
  
| 209 | 
       * @param array $matches An array of matches to replace.
     | 
  
| 210 | 
       * @return string
     | 
  
| 211 | 
       */
     | 
  
| 212 | 
      protected function _replaceTags($matches) {  | 
  
| 213 | 
      $style = $this->styles($matches['tag']);  | 
  
| 214 | 
      if (empty($style)) {  | 
  
| 215 | 
      return '<' . $matches['tag'] . '>' . $matches['text'] . '</' . $matches['tag'] . '>';  | 
  
| 216 | 
      }  | 
  
| 217 | 
       | 
  
| 218 | 
      $styleInfo = array();  | 
  
| 219 | 
      if (!empty($style['text']) && isset(static::$_foregroundColors[$style['text']])) {  | 
  
| 220 | 
      $styleInfo[] = static::$_foregroundColors[$style['text']];  | 
  
| 221 | 
      }  | 
  
| 222 | 
      if (!empty($style['background']) && isset(static::$_backgroundColors[$style['background']])) {  | 
  
| 223 | 
      $styleInfo[] = static::$_backgroundColors[$style['background']];  | 
  
| 224 | 
      }  | 
  
| 225 | 
      unset($style['text'], $style['background']);  | 
  
| 226 | 
      foreach ($style as $option => $value) {  | 
  
| 227 | 
      if ($value) {  | 
  
| 228 | 
      $styleInfo[] = static::$_options[$option];  | 
  
| 229 | 
      }  | 
  
| 230 | 
      }  | 
  
| 231 | 
      return "\033[" . implode($styleInfo, ';') . 'm' . $matches['text'] . "\033[0m";  | 
  
| 232 | 
      }  | 
  
| 233 | 
       | 
  
| 234 | 
      /**
     | 
  
| 235 | 
       * Writes a message to the output stream.
     | 
  
| 236 | 
       *
     | 
  
| 237 | 
       * @param string $message Message to write.
     | 
  
| 238 | 
       * @return bool success
     | 
  
| 239 | 
       */
     | 
  
| 240 | 
      protected function _write($message) {  | 
  
| 241 | 
      return fwrite($this->_output, $message);  | 
  
| 242 | 
      }  | 
  
| 243 | 
       | 
  
| 244 | 
      /**
     | 
  
| 245 | 
       * Get the current styles offered, or append new ones in.
     | 
  
| 246 | 
       *
     | 
  
| 247 | 
       * ### Get a style definition
     | 
  
| 248 | 
       *
     | 
  
| 249 | 
       * `$this->output->styles('error');`
     | 
  
| 250 | 
       *
     | 
  
| 251 | 
       * ### Get all the style definitions
     | 
  
| 252 | 
       *
     | 
  
| 253 | 
       * `$this->output->styles();`
     | 
  
| 254 | 
       *
     | 
  
| 255 | 
       * ### Create or modify an existing style
     | 
  
| 256 | 
       *
     | 
  
| 257 | 
       * `$this->output->styles('annoy', array('text' => 'purple', 'background' => 'yellow', 'blink' => true));`
     | 
  
| 258 | 
       *
     | 
  
| 259 | 
       * ### Remove a style
     | 
  
| 260 | 
       *
     | 
  
| 261 | 
       * `$this->output->styles('annoy', false);`
     | 
  
| 262 | 
       *
     | 
  
| 263 | 
       * @param string $style The style to get or create.
     | 
  
| 264 | 
       * @param array $definition The array definition of the style to change or create a style
     | 
  
| 265 | 
       *   or false to remove a style.
     | 
  
| 266 | 
       * @return mixed If you are getting styles, the style or null will be returned. If you are creating/modifying
     | 
  
| 267 | 
       *   styles true will be returned.
     | 
  
| 268 | 
       */
     | 
  
| 269 | 
      public function styles($style = null, $definition = null) {  | 
  
| 270 | 
      if ($style === null && $definition === null) {  | 
  
| 271 | 
      return static::$_styles;  | 
  
| 272 | 
      }  | 
  
| 273 | 
      if (is_string($style) && $definition === null) {  | 
  
| 274 | 
      return isset(static::$_styles[$style]) ? static::$_styles[$style] : null;  | 
  
| 275 | 
      }  | 
  
| 276 | 
      if ($definition === false) {  | 
  
| 277 | 
      unset(static::$_styles[$style]);  | 
  
| 278 | 
      return true;  | 
  
| 279 | 
      }  | 
  
| 280 | 
      static::$_styles[$style] = $definition;  | 
  
| 281 | 
      return true;  | 
  
| 282 | 
      }  | 
  
| 283 | 
       | 
  
| 284 | 
      /**
     | 
  
| 285 | 
       * Get/Set the output type to use. The output type how formatting tags are treated.
     | 
  
| 286 | 
       *
     | 
  
| 287 | 
       * @param int $type The output type to use. Should be one of the class constants.
     | 
  
| 288 | 
       * @return mixed Either null or the value if getting.
     | 
  
| 289 | 
       */
     | 
  
| 290 | 
      public function outputAs($type = null) {  | 
  
| 291 | 
      if ($type === null) {  | 
  
| 292 | 
      return $this->_outputAs;  | 
  
| 293 | 
      }  | 
  
| 294 | 
      $this->_outputAs = $type;  | 
  
| 295 | 
      }  | 
  
| 296 | 
       | 
  
| 297 | 
      /**
     | 
  
| 298 | 
       * Clean up and close handles
     | 
  
| 299 | 
       */
     | 
  
| 300 | 
      public function __destruct() {  | 
  
| 301 | 
      if (is_resource($this->_output)) {  | 
  
| 302 | 
      fclose($this->_output);  | 
  
| 303 | 
      }  | 
  
| 304 | 
      }  | 
  
| 305 | 
       | 
  
| 306 | 
      }  |