luyadev/luya

<?php

namespace luya\console;

use luya\helpers\ObjectHelper;
use Yii;
use yii\base\InvalidCallException;
use yii\console\Controller as BaseController;
use yii\helpers\Console;

/**
 * Console Controller base class.
 *
 * Extends the base controller by adding helper methods to output responses based on its
 * muted behavior to run unit tests without respones.
 *
 * @author Basil Suter <basil@nadar.io>
 * @since 1.0.0
 */
abstract class Controller extends BaseController
{
    public function init()
    {
        parent::init();

        // Ensure the console command is running under web application object.
        if (!ObjectHelper::isInstanceOf(Yii::$app, 'yii\console\Application', false)) {
            throw new InvalidCallException("The console controller can only run within a console Application context.");
        }
    }

    /**
     * Helper method to see if the current Application is muted or not. If the Application is muted, no output
     * will displayed.
     *
     * @return bool
     */
    public function isMuted()
    {
        return Yii::$app->mute;
    }

    /**
     * Generates a printable string from a message.
     *
     * If a message is not a string, it will return var export to generate
     * a returnable string from a message.
     *
     * @param mixed $message
     * @return string
     * @since 1.0.22
     */
    public function printableMessage($message)
    {
        return is_scalar($message) ? $message : var_export($message, true);
    }

    /**
     * Helper method for writting console application output, include before and after wrappers.
     *
     * @param string $message The message which is displayed
     * @param string $color A color from {{\yii\helpers\Console}} color constants.
     */
    protected function output($message, $color = null)
    {
        $format = [];
        if (!$this->isMuted()) {
            if ($color !== null) {
                $format[] = $color;
            }
            echo Console::ansiFormat("\r".$this->printableMessage($message)."\n", $format);
        }
    }

    /**
     * Helper method to stop the console command with an error message, outputError returns exit code 1.
     *
     * @param string $message The message which should be displayed.
     * @return int Exit code 1
     */
    public function outputError($message)
    {
        $this->output($message, Console::FG_RED);

        return 1;
    }

    /**
     * Helper method to stop the console command with a success message, outputSuccess returns exit code 0.
     *
     * @param string $message The message which sould be displayed
     * @return int Exit code 0
     */
    public function outputSuccess($message)
    {
        $this->output($message, Console::FG_GREEN);

        return 0;
    }

    /**
     * Helper method to stop the console command with a info message which is threated in case of returns as success
     * but does have a different output color (blue). outputInfo returns exit code 0.
     *
     * @param string $message The message which sould be displayed.
     * @return int Exit code 0
     */
    public function outputInfo($message)
    {
        $this->output($message, Console::FG_CYAN);

        return 0;
    }
}