1:   2:   3:   4:   5:   6:   7:   8:   9:  10:  11:  12:  13:  14:  15:  16:  17:  18:  19:  20:  21:  22:  23:  24:  25:  26:  27:  28:  29:  30:  31:  32:  33:  34:  35:  36:  37:  38:  39:  40:  41:  42:  43:  44:  45:  46:  47:  48:  49:  50:  51:  52:  53:  54:  55:  56:  57:  58:  59:  60:  61:  62:  63:  64:  65:  66:  67:  68:  69:  70:  71:  72:  73:  74:  75:  76:  77:  78:  79:  80:  81:  82:  83:  84:  85:  86:  87:  88:  89:  90:  91:  92:  93:  94:  95:  96:  97:  98:  99: 100: 101: 102: 103: 104: 105: 106: 107: 108: 109: 110: 111: 112: 113: 114: 115: 116: 117: 118: 119: 120: 121: 122: 123: 124: 125: 126: 127: 128: 129: 130: 131: 132: 133: 134: 135: 

<?php
    /**
     * Cz CLI Console
     * @author Jan Pecha, <janpecha@email.cz>
     */

    namespace Cz\Cli\Outputs;
    use Cz\Cli\OutputException;

    class ColoredOutput extends BaseOutput
    {
        protected static $colors = array(
            'success' => '0;32',
            'error' => '0;31',
            'warning' => '0;33',
            'info' => '0;34',
            'muted' => '1;30',
        );



        /**
         * @param   string|string[]
         * @return  self
         */
        public function success($str)
        {
            if (!is_array($str))
            {
                $str = func_get_args();
            }

            return $this->color('success')
                ->printString($str);
        }



        /**
         * @param   string|string[]
         * @return  self
         */
        public function error($str)
        {
            if (!is_array($str))
            {
                $str = func_get_args();
            }

            return $this->color('error')
                ->printString($str);
        }



        /**
         * @param   string|string[]
         * @return  self
         */
        public function warning($str)
        {
            if (!is_array($str))
            {
                $str = func_get_args();
            }

            return $this->color('warning')
                ->printString($str);
        }



        /**
         * @param   string|string[]
         * @return  self
         */
        public function info($str)
        {
            if (!is_array($str))
            {
                $str = func_get_args();
            }

            return $this->color('info')
                ->printString($str);
        }



        /**
         * @param   string|string[]
         * @return  self
         */
        public function muted($str)
        {
            if (!is_array($str))
            {
                $str = func_get_args();
            }

            return $this->color('muted')
                ->printString($str);
        }



        protected function color($colorId)
        {
            $newLineEnabled = $this->newLineEnabled;
            $this->setAutoNewLine(FALSE); // disable new lines

            if($colorId === NULL) // reset color
            {
                return $this->output("\033[0m")
                    ->setAutoNewLine($newLineEnabled);
            }
            elseif(isset(self::$colors[$colorId = (string) $colorId]))
            {
                return $this->output("\033[" . self::$colors[$colorId] . 'm')
                    ->setAutoNewLine($newLineEnabled);
            }

            throw new OutputException("Unknow color: $colorId");
        }



        protected function printString($str = NULL)
        {
            return $this->output($str)
                ->color(NULL);
        }
    }