theme-check-cli.php

<?php
/**
 * Run Envato Theme Check from the command line
 */
include 'checkbase.php';
include 'main.php';

class ThemeCheckCLI extends WP_CLI_Command {
    function __construct()
    {
        parent::__construct();
        $this->fetcher = new \WP_CLI\Fetchers\Theme;
    }
    /**
    * Show a list of the current themes
    *
    * ## OPTIONS
    *
    * [--errors=<errors>]
    * : set true to return only themes with errors, false for only without errors. Default: false
    *
    * [--allowed=<allowed>]
    * : (Multisite) set true to return only themes allowed on this site, false for only those not allowed,
    * 'site' for only site-allowed, 'network' for only network-allowed.
    *
    * [--blog_id=<id>]
    * : (Multisite) Blog ID, if different than current
    *
    * @subcommand list
    */
    public function list_themes( $args = array(), $assoc_args = array() )
    {
        $defaults       = array( 'errors' => false, 'allowed' => null, 'blog_id' => 0 );
        $args           = wp_parse_args( $assoc_args, $defaults );
        $args['errors'] = 'true' === $args['errors'];

        if ( ( 'true' == $args['allowed'] ) || ( 'false' == $args['allowed'] ) )
            $args['allowed'] = 'true' === $args['allowed'];

        $themes = wp_get_themes( $args );

        foreach ( $themes as $slug => $theme )
        {
            WP_CLI::line( $slug . ': ' . $theme->get('Name') );
        }
    }
    /**
    * Check a theme
    *
    * <theme>
    * : The theme slug to check
    *
    * [--format=<format>]
    * : set to true to format as json. Default: false
    *
    */
    public function check( $args = array(), $assoc_args = array() )
    {
        global $themechecks;

        checkcount();

        // prevent undefined index errors
        if (!array_key_exists('format', $assoc_args)) {
            $assoc_args['format'] = '';
        }

        // empty array for the json format
        $required_json    = array();
        $warnings_json    = array();
        $recommended_json = array();
        $errors_json      = array();
        $result_json      = array();

        $theme = $this->fetcher->get_check( $args[0] );
        $files = $theme->get_files( null, -1 );
        $css   = $php = $other = array();

        foreach( $files as $key => $filename )
        {
            if ( strpos( $filename, 'tgm-plugin-activation' ) === false && strpos( $filename, 'merlin' ) === false ) {
                if ( substr( $filename, -4 ) == '.php' )
                {
                    $php[ $filename ] = php_strip_whitespace( $filename );
                }
                else if ( substr( $filename, -4 ) == '.css' )
                {
                    $css[ $filename ] = file_get_contents( $filename );
                }
                else
                {
                    $other[ $filename ] = ( ! is_dir( $filename ) ) ? file_get_contents( $filename ) : '';
                }
            }
        }

        $success = run_themechecks($php, $css, $other);
        $errors  = array();

        foreach ( $themechecks as $check )
        {
            if ( $check instanceof themecheck )
            {
                $error = $check->getError();

                if ( ! empty( $error ) )
                {
                    $errors = array_merge( $error, $errors );
                }
            }
        }

        $errors = array_unique( $errors );
        $errors = array_map( 'strip_tags', $errors );
        rsort( $errors );

        // assume to pass unless we see a required or warning message.
        $pass = true;

        foreach ( $errors as $error )
        {
            list( $type, $message ) = explode( ':', $error, 2 );

            if ( 'REQUIRED' == trim( $type ) )
            {
                if ( 'true' == $assoc_args['format'] )
                {
                    array_push( $required_json, "REQUIRED: " . trim( $message ) );
                }
                else
                {
                    WP_CLI::warning( '%rREQUIRED:%n ' . trim( $message ) );
                }
                $pass = false;
            }
            elseif ( 'WARNING' == trim( $type ) )
            {
                if ( 'true' == $assoc_args['format'] )
                {
                    array_push( $warnings_json, "WARNING: " . trim( $message ) );
                }
                else
                {
                    WP_CLI::warning( '%yWARNING:%n ' . trim( $message ) );
                }
                $pass = false;

            }
            elseif ( 'RECOMMENDED' == trim( $type ) )
            {
                if ( 'true' == $assoc_args['format'] )
                {
                    array_push( $recommended_json, "RECOMMENDED: " . trim( $message ) );
                }
                else
                {
                    WP_CLI::warning( '%cRECOMMENDED:%n ' . trim( $message ) );
                }
            }
            else
            {
                if ( 'true' == $assoc_args['format'] )
                {
                    array_push( $errors_json, "ERROR: " . trim( $error ) );
                }
                else
                {
                    WP_CLI::warning( $error );
                }
            }
        }

        WP_CLI::line();

        if ( empty( $errors ) )
        {
            if ( 'true' == $assoc_args['format'] )
            {
                array_push( $result_json, "SUCCESS" );
                array_push( $result_json, "THEME PASSED REVIEW" );
            }
            else
            {
                WP_CLI::success( "THEME PASSED REVIEW" );
            }
        }
        elseif ( true === $pass )
        {
            if ( 'true' == $assoc_args['format'] )
            {
                array_push( $result_json, "SUCCESS" );
                array_push( $result_json, "THEME PASSED REVIEW WITH RECOMMENDED CHANGES" );
            }
            else
            {
                WP_CLI::success( "THEME PASSED REVIEW WITH RECOMMENDED CHANGES" );
            }
        }
        else
        {
            if ( 'true' == $assoc_args['format'] )
            {
                array_push( $result_json, "FAIL" );
                array_push( $result_json, "THEME DID NOT PASS REVIEW" );
            }
            else
            {
                WP_CLI::line( WP_CLI::colorize( "%RFAIL:%n THEME DID NOT PASS REVIEW" ) );
            }
        }

        if ( 'true' == $assoc_args['format'] )
        {
            $output = array (
                'result'      => $result_json,
                'required'    => $required_json,
                'recommended' => $recommended_json,
                'warnings'    => $warnings_json,
                'errors'      => $errors_json
            );
            echo htmlspecialchars_decode( json_encode( $output, JSON_UNESCAPED_SLASHES ) );
        }
    }
    /**
    * Check for the active theme
    *
    * [--format=<format>]
    * : set to true to format as json. Default: false
    *
    */
    public function active( $args = array(), $assoc_args = array() )
    {
        $active_theme = wp_get_theme();
        $theme_folder_name = $active_theme->template;
        // Next four lines set up $themename and $data for wp cli version, as check_main is never run
        global $themename, $data;
        $themename = $theme_folder_name;
        $theme = get_theme_root( $theme_folder_name ) . "/$theme_folder_name";
        $data = tc_get_theme_data( $theme . '/style.css' );
        $this->check( array($theme_folder_name), $assoc_args);
    }
}

class ThemeCheckCLILogger extends WP_CLI\Loggers\Regular {
    public function _line( $message, $label, $color, $handle = STDOUT )
    {
        if ( ! empty( $label ) )
        {
            $label = \cli\Colors::colorize( "$color$label:%n ", $this->in_color );
        }
        $this->write( $handle, "{$label}{$message}\n" );
    }

    function warning( $message )
    {
        $this->_line( WP_CLI::colorize( $message ), '', '', STDERR );
    }
}
WP_CLI::set_logger( new ThemeCheckCLILogger( true ) );

WP_CLI::add_command( 'theme review', 'ThemeCheckCLI' );