initial commit

includes/cli/class-wc-cli-update-command.php

@@ -0,0 +1,81 @@
+<?php
+/**
+ * WC_CLI_Update_Command class file.
+ *
+ * @package WooCommerce\CLI
+ */
+
+if ( ! defined( 'ABSPATH' ) ) {
+	exit;
+}
+
+/**
+ * Allows updates via CLI.
+ *
+ * @version 3.0.0
+ * @package WooCommerce
+ */
+class WC_CLI_Update_Command {
+
+	/**
+	 * Registers the update command.
+	 */
+	public static function register_commands() {
+		WP_CLI::add_command( 'wc update', array( 'WC_CLI_Update_Command', 'update' ) );
+	}
+
+	/**
+	 * Runs all pending WooCommerce database updates.
+	 */
+	public static function update() {
+		global $wpdb;
+
+		$wpdb->hide_errors();
+
+		include_once WC_ABSPATH . 'includes/class-wc-install.php';
+		include_once WC_ABSPATH . 'includes/wc-update-functions.php';
+
+		$current_db_version = get_option( 'woocommerce_db_version' );
+		$update_count       = 0;
+		$callbacks          = WC_Install::get_db_update_callbacks();
+		$callbacks_to_run   = array();
+
+		foreach ( $callbacks as $version => $update_callbacks ) {
+			if ( version_compare( $current_db_version, $version, '<' ) ) {
+				foreach ( $update_callbacks as $update_callback ) {
+					$callbacks_to_run[] = $update_callback;
+				}
+			}
+		}
+
+		if ( empty( $callbacks_to_run ) ) {
+			// Ensure DB version is set to the current WC version to match WP-Admin update routine.
+			WC_Install::update_db_version();
+			/* translators: %s Database version number */
+			WP_CLI::success( sprintf( __( 'No updates required. Database version is %s', 'woocommerce' ), get_option( 'woocommerce_db_version' ) ) );
+			return;
+		}
+
+		/* translators: 1: Number of database updates 2: List of update callbacks */
+		WP_CLI::log( sprintf( __( 'Found %1$d updates (%2$s)', 'woocommerce' ), count( $callbacks_to_run ), implode( ', ', $callbacks_to_run ) ) );
+
+		$progress = \WP_CLI\Utils\make_progress_bar( __( 'Updating database', 'woocommerce' ), count( $callbacks_to_run ) ); // phpcs:ignore PHPCompatibility.LanguageConstructs.NewLanguageConstructs.t_ns_separatorFound
+
+		foreach ( $callbacks_to_run as $update_callback ) {
+			call_user_func( $update_callback );
+			$result = false;
+			while ( $result ) {
+				$result = (bool) call_user_func( $update_callback );
+			}
+			$update_count ++;
+			$progress->tick();
+		}
+
+		$progress->finish();
+
+		WC_Admin_Notices::remove_notice( 'update', true );
+
+		/* translators: 1: Number of database updates performed 2: Database version number */
+		WP_CLI::success( sprintf( __( '%1$d update functions completed. Database version is %2$s', 'woocommerce' ), absint( $update_count ), get_option( 'woocommerce_db_version' ) ) );
+	}
+}