<?php

[0] Fix | Delete

namespace Elementor\Modules\WpCli;

[1] Fix | Delete

[2] Fix | Delete

use Elementor\Plugin;

[3] Fix | Delete

[4] Fix | Delete

if ( ! defined( 'ABSPATH' ) ) {

[5] Fix | Delete

exit; // Exit if accessed directly.

[6] Fix | Delete

}

[7] Fix | Delete

[8] Fix | Delete

/**

[9] Fix | Delete

* Elementor Page Builder cli tools.

[10] Fix | Delete

*/

[11] Fix | Delete

class Update extends \WP_CLI_Command {

[12] Fix | Delete

[13] Fix | Delete

/**

[14] Fix | Delete

* Update the DB after plugin upgrade.

[15] Fix | Delete

*

[16] Fix | Delete

* [--network]

[17] Fix | Delete

* Update DB in all the sites in the network.

[18] Fix | Delete

*

[19] Fix | Delete

* [--force]

[20] Fix | Delete

* Force update even if it's looks like that update is in progress.

[21] Fix | Delete

*

[22] Fix | Delete

*

[23] Fix | Delete

* ## EXAMPLES

[24] Fix | Delete

*

[25] Fix | Delete

* 1. wp elementor update db

[26] Fix | Delete

* - This will Upgrade the DB if needed.

[27] Fix | Delete

*

[28] Fix | Delete

* 2. wp elementor update db --force

[29] Fix | Delete

* - This will Upgrade the DB even if another process is running.

[30] Fix | Delete

*

[31] Fix | Delete

* 3. wp elementor update db --network

[32] Fix | Delete

* - This will Upgrade the DB for each site in the network if needed.

[33] Fix | Delete

*

[34] Fix | Delete

* @since 2.4.0

[35] Fix | Delete

* @access public

[36] Fix | Delete

*

[37] Fix | Delete

* @param $args

[38] Fix | Delete

* @param $assoc_args

[39] Fix | Delete

*/

[40] Fix | Delete

public function db( $args, $assoc_args ) {

[41] Fix | Delete

$network = ! empty( $assoc_args['network'] ) && is_multisite();

[42] Fix | Delete

[43] Fix | Delete

if ( $network ) {

[44] Fix | Delete

$blog_ids = get_sites( [

[45] Fix | Delete

'fields' => 'ids',

[46] Fix | Delete

'number' => 0,

[47] Fix | Delete

] );

[48] Fix | Delete

[49] Fix | Delete

foreach ( $blog_ids as $blog_id ) {

[50] Fix | Delete

switch_to_blog( $blog_id );

[51] Fix | Delete

[52] Fix | Delete

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

[53] Fix | Delete

[54] Fix | Delete

$this->do_db_upgrade( $assoc_args );

[55] Fix | Delete

[56] Fix | Delete

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

[57] Fix | Delete

[58] Fix | Delete

restore_current_blog();

[59] Fix | Delete

}

[60] Fix | Delete

} else {

[61] Fix | Delete

$this->do_db_upgrade( $assoc_args );

[62] Fix | Delete

}

[63] Fix | Delete

}

[64] Fix | Delete

[65] Fix | Delete

protected function get_update_db_manager_class() {

[66] Fix | Delete

return '\Elementor\Core\Upgrade\Manager';

[67] Fix | Delete

}

[68] Fix | Delete

[69] Fix | Delete

protected function do_db_upgrade( $assoc_args ) {

[70] Fix | Delete

$manager_class = $this->get_update_db_manager_class();

[71] Fix | Delete

[72] Fix | Delete

/** @var \Elementor\Core\Upgrade\Manager $manager */

[73] Fix | Delete

$manager = new $manager_class();

[74] Fix | Delete

[75] Fix | Delete

$updater = $manager->get_task_runner();

[76] Fix | Delete

[77] Fix | Delete

if ( $updater->is_process_locked() && empty( $assoc_args['force'] ) ) {

[78] Fix | Delete

\WP_CLI::warning( 'Oops! Process is already running. Use --force to force run.' );

[79] Fix | Delete

return;

[80] Fix | Delete

}

[81] Fix | Delete

[82] Fix | Delete

if ( ! $manager->should_upgrade() ) {

[83] Fix | Delete

\WP_CLI::success( 'The DB is already updated!' );

[84] Fix | Delete

return;

[85] Fix | Delete

}

[86] Fix | Delete

[87] Fix | Delete

$callbacks = $manager->get_upgrade_callbacks();

[88] Fix | Delete

$did_tasks = false;

[89] Fix | Delete

[90] Fix | Delete

if ( ! empty( $callbacks ) ) {

[91] Fix | Delete

Plugin::$instance->logger->get_logger()->info( 'Update DB has been started', [

[92] Fix | Delete

'meta' => [

[93] Fix | Delete

'plugin' => $manager->get_plugin_label(),

[94] Fix | Delete

'from' => $manager->get_current_version(),

[95] Fix | Delete

'to' => $manager->get_new_version(),

[96] Fix | Delete

],

[97] Fix | Delete

] );

[98] Fix | Delete

[99] Fix | Delete

$updater->handle_immediately( $callbacks );

[100] Fix | Delete

$did_tasks = true;

[101] Fix | Delete

}

[102] Fix | Delete

[103] Fix | Delete

$manager->on_runner_complete( $did_tasks );

[104] Fix | Delete

[105] Fix | Delete

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

[106] Fix | Delete

}

[107] Fix | Delete

}

[108] Fix | Delete

[109] Fix | Delete