<?php
namespace Elementor\Core\Experiments;
use Elementor\Core\Experiments\Manager as Experiments_Manager;
use Elementor\Plugin;
if ( ! defined( 'ABSPATH' ) ) {
exit; // Exit if accessed directly
}
class Wp_Cli extends \WP_CLI_Command {
/**
* Activate an Experiment
*
* ## EXAMPLES
*
* 1. wp elementor experiments activate container
* - This will activate the Container experiment.
*
* @param array $args
* @param array|null $assoc_args - Arguments from WP CLI command.
*/
public function activate( $args, $assoc_args ) {
if ( empty( $args[0] ) ) {
\WP_CLI::error( 'Please specify an experiment.' );
}
$is_network = $this->is_network( $assoc_args );
// Activate experiment callback.
$activate = function ( $site = '', $id = null ) use ( $args, $is_network ) {
$success = 'Experiment activated successfully';
$error = 'Cannot activate experiment';
if ( $is_network ) {
$success .= " for site {$site}";
$error .= " for site {$site}";
}
$experiments_manager = Plugin::instance()->experiments;
$option = $experiments_manager->get_feature_option_key( $args[0] );
update_option( $option, Experiments_Manager::STATE_ACTIVE );
try {
\WP_CLI::success( $success );
} catch ( \Exception $e ) {
\WP_CLI::error( $error );
}
};
if ( $is_network ) {
$this->foreach_sites( $activate );
} else {
$activate();
}
}
/**
* Deactivate an Experiment
*
* ## EXAMPLES
*
* 1. wp elementor experiments deactivate container
* - This will deactivate the Container experiment.
*
* @param array $args
* @param array|null $assoc_args - Arguments from WP CLI command.
*/
public function deactivate( $args, $assoc_args ) {
if ( empty( $args[0] ) ) {
\WP_CLI::error( 'Please specify an experiment.' );
}
$is_network = $this->is_network( $assoc_args );
// Activate experiment callback.
$activate = function ( $site = '' ) use ( $args, $is_network ) {
$success = 'Experiment deactivated successfully';
$error = 'Cannot deactivate experiment';
if ( $is_network ) {
$success .= " for site {$site}";
$error .= " for site {$site}";
}
$experiments_manager = Plugin::instance()->experiments;
$option = $experiments_manager->get_feature_option_key( $args[0] );
update_option( $option, Experiments_Manager::STATE_INACTIVE );
try {
\WP_CLI::success( $success );
} catch ( \Exception $e ) {
\WP_CLI::error( $error );
}
};
if ( $is_network ) {
$this->foreach_sites( $activate );
} else {
$activate();
}
}
/**
* Determine if the current website is a multisite.
*
* @param array|null $assoc_args - Arguments from WP CLI command.
*
* @return bool
*/
private function is_network( $assoc_args ) {
return ! empty( $assoc_args['network'] ) && is_multisite();
}
/**
* Iterate over network sites and execute a callback.
*
* @param callable $callback - Callback to execute. Gets the site name & id as parameters.
*
* @return void
*/
private function foreach_sites( callable $callback ) {
$blog_ids = get_sites( [
'fields' => 'ids',
'number' => 0,
] );
foreach ( $blog_ids as $blog_id ) {
switch_to_blog( $blog_id );
$callback( get_option( 'home' ), $blog_id );
restore_current_blog();
}
}
}