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: 

<?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' => 32,
            'error' => 31,
            'warning' => 33,
            'info' => 34,
        );



        /**
         * @param   string|NULL
         * @return  self
         */
        public function success($str = NULL)
        {
            return $this->color('success')
                ->printString($str);
        }



        /**
         * @param   string|NULL
         * @return  self
         */
        public function error($str = NULL)
        {
            return $this->color('error')
                ->printString($str);
        }



        /**
         * @param   string|NULL
         * @return  self
         */
        public function warning($str = NULL)
        {
            return $this->color('warning')
                ->printString($str);
        }



        /**
         * @param   string|NULL
         * @return  self
         */
        public function info($str = NULL)
        {
            return $this->color('info')
                ->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[0;" . self::$colors[$colorId] . 'm')
                    ->setAutoNewLine($newLineEnabled);
            }

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



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