Viktor Miller
48f270f6a5
- Downloaded Action Scheduler 3.9.3 from GitHub - Placed in vendor/action-scheduler/ directory - Included in main plugin file before other code for proper initialization - Library will auto-initialize itself when required
122 lines
3.5 KiB
PHP
122 lines
3.5 KiB
PHP
<?php
|
|
|
|
namespace Action_Scheduler\WP_CLI\Action;
|
|
|
|
use function \WP_CLI\Utils\get_flag_value;
|
|
|
|
/**
|
|
* WP-CLI command: action-scheduler action generate
|
|
*/
|
|
class Generate_Command extends \ActionScheduler_WPCLI_Command {
|
|
|
|
/**
|
|
* Execute command.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function execute() {
|
|
$hook = $this->args[0];
|
|
$schedule_start = $this->args[1];
|
|
$callback_args = get_flag_value( $this->assoc_args, 'args', array() );
|
|
$group = get_flag_value( $this->assoc_args, 'group', '' );
|
|
$interval = (int) get_flag_value( $this->assoc_args, 'interval', 0 ); // avoid absint() to support negative intervals
|
|
$count = absint( get_flag_value( $this->assoc_args, 'count', 1 ) );
|
|
|
|
if ( ! empty( $callback_args ) ) {
|
|
$callback_args = json_decode( $callback_args, true );
|
|
}
|
|
|
|
$schedule_start = as_get_datetime_object( $schedule_start );
|
|
|
|
$function_args = array(
|
|
'start' => absint( $schedule_start->format( 'U' ) ),
|
|
'interval' => $interval,
|
|
'count' => $count,
|
|
'hook' => $hook,
|
|
'callback_args' => $callback_args,
|
|
'group' => $group,
|
|
);
|
|
|
|
$function_args = array_values( $function_args );
|
|
|
|
try {
|
|
$actions_added = $this->generate( ...$function_args );
|
|
} catch ( \Exception $e ) {
|
|
$this->print_error( $e );
|
|
}
|
|
|
|
$num_actions_added = count( (array) $actions_added );
|
|
|
|
$this->print_success( $num_actions_added, 'single' );
|
|
}
|
|
|
|
/**
|
|
* Schedule multiple single actions.
|
|
*
|
|
* @param int $schedule_start Starting timestamp of first action.
|
|
* @param int $interval How long to wait between runs.
|
|
* @param int $count Limit number of actions to schedule.
|
|
* @param string $hook The hook to trigger.
|
|
* @param array $args Arguments to pass when the hook triggers.
|
|
* @param string $group The group to assign this job to.
|
|
* @return int[] IDs of actions added.
|
|
*/
|
|
protected function generate( $schedule_start, $interval, $count, $hook, array $args = array(), $group = '' ) {
|
|
$actions_added = array();
|
|
|
|
$progress_bar = \WP_CLI\Utils\make_progress_bar(
|
|
sprintf(
|
|
/* translators: %d is number of actions to create */
|
|
_n( 'Creating %d action', 'Creating %d actions', $count, 'action-scheduler' ),
|
|
number_format_i18n( $count )
|
|
),
|
|
$count
|
|
);
|
|
|
|
for ( $i = 0; $i < $count; $i++ ) {
|
|
$actions_added[] = as_schedule_single_action( $schedule_start + ( $i * $interval ), $hook, $args, $group );
|
|
$progress_bar->tick();
|
|
}
|
|
|
|
$progress_bar->finish();
|
|
|
|
return $actions_added;
|
|
}
|
|
|
|
/**
|
|
* Print a success message with the action ID.
|
|
*
|
|
* @param int $actions_added Number of actions generated.
|
|
* @param string $action_type Type of actions scheduled.
|
|
* @return void
|
|
*/
|
|
protected function print_success( $actions_added, $action_type ) {
|
|
\WP_CLI::success(
|
|
sprintf(
|
|
/* translators: %1$d refers to the total number of tasks added, %2$s is the action type */
|
|
_n( '%1$d %2$s action scheduled.', '%1$d %2$s actions scheduled.', $actions_added, 'action-scheduler' ),
|
|
number_format_i18n( $actions_added ),
|
|
$action_type
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Convert an exception into a WP CLI error.
|
|
*
|
|
* @param \Exception $e The error object.
|
|
* @throws \WP_CLI\ExitException When an error occurs.
|
|
* @return void
|
|
*/
|
|
protected function print_error( \Exception $e ) {
|
|
\WP_CLI::error(
|
|
sprintf(
|
|
/* translators: %s refers to the exception error message. */
|
|
__( 'There was an error creating the scheduled action: %s', 'action-scheduler' ),
|
|
$e->getMessage()
|
|
)
|
|
);
|
|
}
|
|
|
|
}
|