Fix_WPCLI

class Fix_WPCLI {

	/**
	 * Constructor.
	 *
	 * @since CP-1.5.0
	 */
	public function __construct() {
		WP_CLI::add_hook( 'after_wp_load', array( __CLASS__, 'add_cp_version_to_scope' ) );
		WP_CLI::add_hook( 'before_invoke:core check-update', array( __CLASS__, 'correct_core_check_update' ) );
	}

	/**
	 * Add $cp_version to scope.
	 *
	 * @since CP-1.7.3
	 */
	public static function add_cp_version_to_scope() {
		// Add $cp_version to scope.
		if ( ! isset( $GLOBALS['cp_version'] ) ) {
			global $cp_version;
			require ABSPATH . WPINC . '/version.php';
		}
	}

	/**
	 * Fix wp core check-update command.
	 *
	 * @since CP-1.5.0
	 */
	public static function correct_core_check_update() {
		// Add $cp_version to scope.
		global $cp_version;

		// Check for updates. Bail on error.
		// When playing with versions, an empty array is returned if it's not on api.
		wp_version_check( array(), true );
		$core_updates = get_core_updates();
		if ( false === $core_updates || array() === $core_updates ) {
			WP_CLI::error( 'Something went wrong checking for updates.' );
			exit;
		}

		// We are on latest.
		if ( 'latest' === $core_updates[0]->{'response'} ) {
			WP_CLI::success( 'ClassicPress is at the latest version.' );
			exit;
		}

		// Standard options.
		$arg_fields    = 'version,update_type,package_url';
		$format_fields = 'table';

		// Retrieve command line options and parse them.
		global $argv;
		$current_command = implode( ' ', $argv );

		$fields_match = array();
		if ( preg_match_all( '/ --fields=([a-z,_]+)/', $current_command, $fields_match ) > 0 ) {
			$arg_fields = $fields_match[1][0];
		}

		$field_match = array();
		if ( preg_match_all( '/ --field=([a-z_]+)/', $current_command, $field_match ) > 0 ) {
			$arg_fields = $field_match[1][0];
		}

		$cp_format_match = array();
		if ( preg_match_all( '/ --format=([a-z]+)/', $current_command, $format_match ) > 0 ) {
			$format_fields = $format_match[1][0];
		}

		$minor = preg_match( '/ --minor */', $current_command );

		$major = preg_match( '/ --major */', $current_command );

		// Prepare output array.
		$table_output = array();

		// Loop in the update list.
		foreach ( $core_updates as $index => $update ) {

			// Get update type and skip if options tells to.
			$type = WP_CLI\Utils\get_named_sem_ver( $update->{'version'}, $cp_version );
			if ( ( 1 === $major ) && ( 'major' !== $type ) ) {
				continue;
			}
			if ( ( 1 === $minor ) && ( 'patch' === $type ) ) {
				continue;
			}

			$table_output[] = array(
				'version'     => $update->{'version'},
				'package_url' => $update->{'download'},
				'update_type' => $type,
			);

		}

		// Check if the filters left no updates.
		if ( empty( $table_output ) ) {
			WP_CLI::success( 'ClassicPress is at the latest version.' );
			exit;
		}

		// Render output.
		WP_CLI\Utils\format_items( $format_fields, $table_output, $arg_fields );

		// Exit to prevent the core check-update command to continue his work.
		exit;
	}
}