firepot

Back Back Up Advanced Editor

<?php
namespace Elementor\Modules\WpCli;

use Elementor\Plugin;

if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly
}

/**
 * Elementor Page Builder cli tools.
 */
class Update extends \WP_CLI_Command {

/**
	 * Update the DB after plugin upgrade.
	 *
	 * [--network]
	 *      Update DB in all the sites in the network.
	 *
	 * [--force]
	 *      Force update even if it's looks like that update is in progress.
	 *
	 *
	 * ## EXAMPLES
	 *
	 *  1. wp elementor update db
	 *      - This will Upgrade the DB if needed.
	 *
	 *  2. wp elementor update db --force
	 *      - This will Upgrade the DB even if another process is running.
	 *
	 *  3. wp elementor update db --network
	 *      - This will Upgrade the DB for each site in the network if needed.
	 *
	 * @since  2.4.0
	 * @access public
	 *
	 * @param $args
	 * @param $assoc_args
	 */
	public function db( $args, $assoc_args ) {
		$network = ! empty( $assoc_args['network'] ) && is_multisite();

if ( $network ) {
			$blog_ids = get_sites( [
				'fields' => 'ids',
				'number' => 0,
			] );

foreach ( $blog_ids as $blog_id ) {
				switch_to_blog( $blog_id );

\WP_CLI::line( 'Site #' . $blog_id . ' - ' . get_option( 'blogname' ) );

$this->do_db_upgrade( $assoc_args );

\WP_CLI::success( 'Done! - ' . get_option( 'home' ) );

restore_current_blog();
			}
		} else {
			$this->do_db_upgrade( $assoc_args );
		}
	}

protected function get_update_db_manager_class() {
		return '\Elementor\Core\Upgrade\Manager';
	}

protected function do_db_upgrade( $assoc_args ) {
		$manager_class = $this->get_update_db_manager_class();

/** @var \Elementor\Core\Upgrade\Manager $manager */
		$manager = new $manager_class();

$updater = $manager->get_task_runner();

if ( $updater->is_process_locked() && empty( $assoc_args['force'] ) ) {
			\WP_CLI::warning( 'Oops! Process is already running. Use --force to force run.' );
			return;
		}

if ( ! $manager->should_upgrade() ) {
			\WP_CLI::success( 'The DB is already updated!' );
			return;
		}

$callbacks = $manager->get_upgrade_callbacks();
		$did_tasks = false;

if ( ! empty( $callbacks ) ) {
			Plugin::$instance->logger->get_logger()->info( 'Update DB has been started', [
				'meta' => [
					'plugin' => $manager->get_plugin_label(),
					'from' => $manager->get_current_version(),
					'to' => $manager->get_new_version(),
				],
			] );

$updater->handle_immediately( $callbacks );
			$did_tasks = true;
		}

$manager->on_runner_complete( $did_tasks );

\WP_CLI::success( count( $callbacks ) . ' updates(s) has been applied.' );
	}
}