<?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.' );
}
}