first commit
This commit is contained in:
@@ -0,0 +1,154 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_DBLogger
|
||||
*
|
||||
* Action logs data table data store.
|
||||
*
|
||||
* @since 3.0.0
|
||||
*/
|
||||
class ActionScheduler_DBLogger extends ActionScheduler_Logger {
|
||||
|
||||
/**
|
||||
* Add a record to an action log.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
* @param string $message Message to be saved in the log entry.
|
||||
* @param DateTime $date Timestamp of the log entry.
|
||||
*
|
||||
* @return int The log entry ID.
|
||||
*/
|
||||
public function log( $action_id, $message, DateTime $date = null ) {
|
||||
if ( empty( $date ) ) {
|
||||
$date = as_get_datetime_object();
|
||||
} else {
|
||||
$date = clone $date;
|
||||
}
|
||||
|
||||
$date_gmt = $date->format( 'Y-m-d H:i:s' );
|
||||
ActionScheduler_TimezoneHelper::set_local_timezone( $date );
|
||||
$date_local = $date->format( 'Y-m-d H:i:s' );
|
||||
|
||||
/** @var \wpdb $wpdb */ //phpcs:ignore Generic.Commenting.DocComment.MissingShort
|
||||
global $wpdb;
|
||||
$wpdb->insert(
|
||||
$wpdb->actionscheduler_logs,
|
||||
array(
|
||||
'action_id' => $action_id,
|
||||
'message' => $message,
|
||||
'log_date_gmt' => $date_gmt,
|
||||
'log_date_local' => $date_local,
|
||||
),
|
||||
array( '%d', '%s', '%s', '%s' )
|
||||
);
|
||||
|
||||
return $wpdb->insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve an action log entry.
|
||||
*
|
||||
* @param int $entry_id Log entry ID.
|
||||
*
|
||||
* @return ActionScheduler_LogEntry
|
||||
*/
|
||||
public function get_entry( $entry_id ) {
|
||||
/** @var \wpdb $wpdb */ //phpcs:ignore Generic.Commenting.DocComment.MissingShort
|
||||
global $wpdb;
|
||||
$entry = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$wpdb->actionscheduler_logs} WHERE log_id=%d", $entry_id ) );
|
||||
|
||||
return $this->create_entry_from_db_record( $entry );
|
||||
}
|
||||
|
||||
/**
|
||||
* Create an action log entry from a database record.
|
||||
*
|
||||
* @param object $record Log entry database record object.
|
||||
*
|
||||
* @return ActionScheduler_LogEntry
|
||||
*/
|
||||
private function create_entry_from_db_record( $record ) {
|
||||
if ( empty( $record ) ) {
|
||||
return new ActionScheduler_NullLogEntry();
|
||||
}
|
||||
|
||||
if ( is_null( $record->log_date_gmt ) ) {
|
||||
$date = as_get_datetime_object( ActionScheduler_StoreSchema::DEFAULT_DATE );
|
||||
} else {
|
||||
$date = as_get_datetime_object( $record->log_date_gmt );
|
||||
}
|
||||
|
||||
return new ActionScheduler_LogEntry( $record->action_id, $record->message, $date );
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve the an action's log entries from the database.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return ActionScheduler_LogEntry[]
|
||||
*/
|
||||
public function get_logs( $action_id ) {
|
||||
/** @var \wpdb $wpdb */ //phpcs:ignore Generic.Commenting.DocComment.MissingShort
|
||||
global $wpdb;
|
||||
|
||||
$records = $wpdb->get_results( $wpdb->prepare( "SELECT * FROM {$wpdb->actionscheduler_logs} WHERE action_id=%d", $action_id ) );
|
||||
|
||||
return array_map( array( $this, 'create_entry_from_db_record' ), $records );
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the data store.
|
||||
*
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
public function init() {
|
||||
$table_maker = new ActionScheduler_LoggerSchema();
|
||||
$table_maker->init();
|
||||
$table_maker->register_tables();
|
||||
|
||||
parent::init();
|
||||
|
||||
add_action( 'action_scheduler_deleted_action', array( $this, 'clear_deleted_action_logs' ), 10, 1 );
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete the action logs for an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function clear_deleted_action_logs( $action_id ) {
|
||||
/** @var \wpdb $wpdb */ //phpcs:ignore Generic.Commenting.DocComment.MissingShort
|
||||
global $wpdb;
|
||||
$wpdb->delete( $wpdb->actionscheduler_logs, array( 'action_id' => $action_id ), array( '%d' ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Bulk add cancel action log entries.
|
||||
*
|
||||
* @param array $action_ids List of action ID.
|
||||
*/
|
||||
public function bulk_log_cancel_actions( $action_ids ) {
|
||||
if ( empty( $action_ids ) ) {
|
||||
return;
|
||||
}
|
||||
|
||||
/** @var \wpdb $wpdb */ //phpcs:ignore Generic.Commenting.DocComment.MissingShort
|
||||
global $wpdb;
|
||||
$date = as_get_datetime_object();
|
||||
$date_gmt = $date->format( 'Y-m-d H:i:s' );
|
||||
ActionScheduler_TimezoneHelper::set_local_timezone( $date );
|
||||
$date_local = $date->format( 'Y-m-d H:i:s' );
|
||||
$message = __( 'action canceled', 'woocommerce' );
|
||||
$format = '(%d, ' . $wpdb->prepare( '%s, %s, %s', $message, $date_gmt, $date_local ) . ')';
|
||||
$sql_query = "INSERT {$wpdb->actionscheduler_logs} (action_id, message, log_date_gmt, log_date_local) VALUES ";
|
||||
$value_rows = array();
|
||||
|
||||
foreach ( $action_ids as $action_id ) {
|
||||
$value_rows[] = $wpdb->prepare( $format, $action_id ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
$sql_query .= implode( ',', $value_rows );
|
||||
|
||||
$wpdb->query( $sql_query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,877 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_DBStore
|
||||
*
|
||||
* Action data table data store.
|
||||
*
|
||||
* @since 3.0.0
|
||||
*/
|
||||
class ActionScheduler_DBStore extends ActionScheduler_Store {
|
||||
|
||||
/**
|
||||
* Used to share information about the before_date property of claims internally.
|
||||
*
|
||||
* This is used in preference to passing the same information as a method param
|
||||
* for backwards-compatibility reasons.
|
||||
*
|
||||
* @var DateTime|null
|
||||
*/
|
||||
private $claim_before_date = null;
|
||||
|
||||
/** @var int */
|
||||
protected static $max_args_length = 8000;
|
||||
|
||||
/** @var int */
|
||||
protected static $max_index_length = 191;
|
||||
|
||||
/**
|
||||
* Initialize the data store
|
||||
*
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
public function init() {
|
||||
$table_maker = new ActionScheduler_StoreSchema();
|
||||
$table_maker->init();
|
||||
$table_maker->register_tables();
|
||||
}
|
||||
|
||||
/**
|
||||
* Save an action.
|
||||
*
|
||||
* @param ActionScheduler_Action $action Action object.
|
||||
* @param DateTime $date Optional schedule date. Default null.
|
||||
*
|
||||
* @return int Action ID.
|
||||
* @throws RuntimeException Throws exception when saving the action fails.
|
||||
*/
|
||||
public function save_action( ActionScheduler_Action $action, \DateTime $date = null ) {
|
||||
try {
|
||||
|
||||
$this->validate_action( $action );
|
||||
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$data = array(
|
||||
'hook' => $action->get_hook(),
|
||||
'status' => ( $action->is_finished() ? self::STATUS_COMPLETE : self::STATUS_PENDING ),
|
||||
'scheduled_date_gmt' => $this->get_scheduled_date_string( $action, $date ),
|
||||
'scheduled_date_local' => $this->get_scheduled_date_string_local( $action, $date ),
|
||||
'schedule' => serialize( $action->get_schedule() ), // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions.serialize_serialize
|
||||
'group_id' => $this->get_group_id( $action->get_group() ),
|
||||
);
|
||||
$args = wp_json_encode( $action->get_args() );
|
||||
if ( strlen( $args ) <= static::$max_index_length ) {
|
||||
$data['args'] = $args;
|
||||
} else {
|
||||
$data['args'] = $this->hash_args( $args );
|
||||
$data['extended_args'] = $args;
|
||||
}
|
||||
|
||||
$table_name = ! empty( $wpdb->actionscheduler_actions ) ? $wpdb->actionscheduler_actions : $wpdb->prefix . 'actionscheduler_actions';
|
||||
$wpdb->insert( $table_name, $data );
|
||||
$action_id = $wpdb->insert_id;
|
||||
|
||||
if ( is_wp_error( $action_id ) ) {
|
||||
throw new \RuntimeException( $action_id->get_error_message() );
|
||||
} elseif ( empty( $action_id ) ) {
|
||||
throw new \RuntimeException( $wpdb->last_error ? $wpdb->last_error : __( 'Database error.', 'woocommerce' ) );
|
||||
}
|
||||
|
||||
do_action( 'action_scheduler_stored_action', $action_id );
|
||||
|
||||
return $action_id;
|
||||
} catch ( \Exception $e ) {
|
||||
/* translators: %s: error message */
|
||||
throw new \RuntimeException( sprintf( __( 'Error saving action: %s', 'woocommerce' ), $e->getMessage() ), 0 );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate a hash from json_encoded $args using MD5 as this isn't for security.
|
||||
*
|
||||
* @param string $args JSON encoded action args.
|
||||
* @return string
|
||||
*/
|
||||
protected function hash_args( $args ) {
|
||||
return md5( $args );
|
||||
}
|
||||
|
||||
/**
|
||||
* Get action args query param value from action args.
|
||||
*
|
||||
* @param array $args Action args.
|
||||
* @return string
|
||||
*/
|
||||
protected function get_args_for_query( $args ) {
|
||||
$encoded = wp_json_encode( $args );
|
||||
if ( strlen( $encoded ) <= static::$max_index_length ) {
|
||||
return $encoded;
|
||||
}
|
||||
return $this->hash_args( $encoded );
|
||||
}
|
||||
/**
|
||||
* Get a group's ID based on its name/slug.
|
||||
*
|
||||
* @param string $slug The string name of a group.
|
||||
* @param bool $create_if_not_exists Whether to create the group if it does not already exist. Default, true - create the group.
|
||||
*
|
||||
* @return int The group's ID, if it exists or is created, or 0 if it does not exist and is not created.
|
||||
*/
|
||||
protected function get_group_id( $slug, $create_if_not_exists = true ) {
|
||||
if ( empty( $slug ) ) {
|
||||
return 0;
|
||||
}
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$group_id = (int) $wpdb->get_var( $wpdb->prepare( "SELECT group_id FROM {$wpdb->actionscheduler_groups} WHERE slug=%s", $slug ) );
|
||||
if ( empty( $group_id ) && $create_if_not_exists ) {
|
||||
$group_id = $this->create_group( $slug );
|
||||
}
|
||||
|
||||
return $group_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Create an action group.
|
||||
*
|
||||
* @param string $slug Group slug.
|
||||
*
|
||||
* @return int Group ID.
|
||||
*/
|
||||
protected function create_group( $slug ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$wpdb->insert( $wpdb->actionscheduler_groups, array( 'slug' => $slug ) );
|
||||
|
||||
return (int) $wpdb->insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return ActionScheduler_Action
|
||||
*/
|
||||
public function fetch_action( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$data = $wpdb->get_row(
|
||||
$wpdb->prepare(
|
||||
"SELECT a.*, g.slug AS `group` FROM {$wpdb->actionscheduler_actions} a LEFT JOIN {$wpdb->actionscheduler_groups} g ON a.group_id=g.group_id WHERE a.action_id=%d",
|
||||
$action_id
|
||||
)
|
||||
);
|
||||
|
||||
if ( empty( $data ) ) {
|
||||
return $this->get_null_action();
|
||||
}
|
||||
|
||||
if ( ! empty( $data->extended_args ) ) {
|
||||
$data->args = $data->extended_args;
|
||||
unset( $data->extended_args );
|
||||
}
|
||||
|
||||
// Convert NULL dates to zero dates.
|
||||
$date_fields = array(
|
||||
'scheduled_date_gmt',
|
||||
'scheduled_date_local',
|
||||
'last_attempt_gmt',
|
||||
'last_attempt_gmt',
|
||||
);
|
||||
foreach ( $date_fields as $date_field ) {
|
||||
if ( is_null( $data->$date_field ) ) {
|
||||
$data->$date_field = ActionScheduler_StoreSchema::DEFAULT_DATE;
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
$action = $this->make_action_from_db_record( $data );
|
||||
} catch ( ActionScheduler_InvalidActionException $exception ) {
|
||||
do_action( 'action_scheduler_failed_fetch_action', $action_id, $exception );
|
||||
return $this->get_null_action();
|
||||
}
|
||||
|
||||
return $action;
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a null action.
|
||||
*
|
||||
* @return ActionScheduler_NullAction
|
||||
*/
|
||||
protected function get_null_action() {
|
||||
return new ActionScheduler_NullAction();
|
||||
}
|
||||
|
||||
/**
|
||||
* Create an action from a database record.
|
||||
*
|
||||
* @param object $data Action database record.
|
||||
*
|
||||
* @return ActionScheduler_Action|ActionScheduler_CanceledAction|ActionScheduler_FinishedAction
|
||||
*/
|
||||
protected function make_action_from_db_record( $data ) {
|
||||
|
||||
$hook = $data->hook;
|
||||
$args = json_decode( $data->args, true );
|
||||
$schedule = unserialize( $data->schedule ); // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions.serialize_unserialize
|
||||
|
||||
$this->validate_args( $args, $data->action_id );
|
||||
$this->validate_schedule( $schedule, $data->action_id );
|
||||
|
||||
if ( empty( $schedule ) ) {
|
||||
$schedule = new ActionScheduler_NullSchedule();
|
||||
}
|
||||
$group = $data->group ? $data->group : '';
|
||||
|
||||
return ActionScheduler::factory()->get_stored_action( $data->status, $data->hook, $args, $schedule, $group );
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the SQL statement to query (or count) actions.
|
||||
*
|
||||
* @since x.x.x $query['status'] accepts array of statuses instead of a single status.
|
||||
*
|
||||
* @param array $query Filtering options.
|
||||
* @param string $select_or_count Whether the SQL should select and return the IDs or just the row count.
|
||||
*
|
||||
* @return string SQL statement already properly escaped.
|
||||
* @throws InvalidArgumentException If the query is invalid.
|
||||
*/
|
||||
protected function get_query_actions_sql( array $query, $select_or_count = 'select' ) {
|
||||
|
||||
if ( ! in_array( $select_or_count, array( 'select', 'count' ), true ) ) {
|
||||
throw new InvalidArgumentException( __( 'Invalid value for select or count parameter. Cannot query actions.', 'woocommerce' ) );
|
||||
}
|
||||
|
||||
$query = wp_parse_args(
|
||||
$query,
|
||||
array(
|
||||
'hook' => '',
|
||||
'args' => null,
|
||||
'date' => null,
|
||||
'date_compare' => '<=',
|
||||
'modified' => null,
|
||||
'modified_compare' => '<=',
|
||||
'group' => '',
|
||||
'status' => '',
|
||||
'claimed' => null,
|
||||
'per_page' => 5,
|
||||
'offset' => 0,
|
||||
'orderby' => 'date',
|
||||
'order' => 'ASC',
|
||||
)
|
||||
);
|
||||
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$sql = ( 'count' === $select_or_count ) ? 'SELECT count(a.action_id)' : 'SELECT a.action_id';
|
||||
$sql .= " FROM {$wpdb->actionscheduler_actions} a";
|
||||
$sql_params = array();
|
||||
|
||||
if ( ! empty( $query['group'] ) || 'group' === $query['orderby'] ) {
|
||||
$sql .= " LEFT JOIN {$wpdb->actionscheduler_groups} g ON g.group_id=a.group_id";
|
||||
}
|
||||
|
||||
$sql .= ' WHERE 1=1';
|
||||
|
||||
if ( ! empty( $query['group'] ) ) {
|
||||
$sql .= ' AND g.slug=%s';
|
||||
$sql_params[] = $query['group'];
|
||||
}
|
||||
|
||||
if ( $query['hook'] ) {
|
||||
$sql .= ' AND a.hook=%s';
|
||||
$sql_params[] = $query['hook'];
|
||||
}
|
||||
if ( ! is_null( $query['args'] ) ) {
|
||||
$sql .= ' AND a.args=%s';
|
||||
$sql_params[] = $this->get_args_for_query( $query['args'] );
|
||||
}
|
||||
|
||||
if ( $query['status'] ) {
|
||||
$statuses = (array) $query['status'];
|
||||
$placeholders = array_fill( 0, count( $statuses ), '%s' );
|
||||
$sql .= ' AND a.status IN (' . join( ', ', $placeholders ) . ')';
|
||||
$sql_params = array_merge( $sql_params, array_values( $statuses ) );
|
||||
}
|
||||
|
||||
if ( $query['date'] instanceof \DateTime ) {
|
||||
$date = clone $query['date'];
|
||||
$date->setTimezone( new \DateTimeZone( 'UTC' ) );
|
||||
$date_string = $date->format( 'Y-m-d H:i:s' );
|
||||
$comparator = $this->validate_sql_comparator( $query['date_compare'] );
|
||||
$sql .= " AND a.scheduled_date_gmt $comparator %s";
|
||||
$sql_params[] = $date_string;
|
||||
}
|
||||
|
||||
if ( $query['modified'] instanceof \DateTime ) {
|
||||
$modified = clone $query['modified'];
|
||||
$modified->setTimezone( new \DateTimeZone( 'UTC' ) );
|
||||
$date_string = $modified->format( 'Y-m-d H:i:s' );
|
||||
$comparator = $this->validate_sql_comparator( $query['modified_compare'] );
|
||||
$sql .= " AND a.last_attempt_gmt $comparator %s";
|
||||
$sql_params[] = $date_string;
|
||||
}
|
||||
|
||||
if ( true === $query['claimed'] ) {
|
||||
$sql .= ' AND a.claim_id != 0';
|
||||
} elseif ( false === $query['claimed'] ) {
|
||||
$sql .= ' AND a.claim_id = 0';
|
||||
} elseif ( ! is_null( $query['claimed'] ) ) {
|
||||
$sql .= ' AND a.claim_id = %d';
|
||||
$sql_params[] = $query['claimed'];
|
||||
}
|
||||
|
||||
if ( ! empty( $query['search'] ) ) {
|
||||
$sql .= ' AND (a.hook LIKE %s OR (a.extended_args IS NULL AND a.args LIKE %s) OR a.extended_args LIKE %s';
|
||||
for ( $i = 0; $i < 3; $i++ ) {
|
||||
$sql_params[] = sprintf( '%%%s%%', $query['search'] );
|
||||
}
|
||||
|
||||
$search_claim_id = (int) $query['search'];
|
||||
if ( $search_claim_id ) {
|
||||
$sql .= ' OR a.claim_id = %d';
|
||||
$sql_params[] = $search_claim_id;
|
||||
}
|
||||
|
||||
$sql .= ')';
|
||||
}
|
||||
|
||||
if ( 'select' === $select_or_count ) {
|
||||
if ( 'ASC' === strtoupper( $query['order'] ) ) {
|
||||
$order = 'ASC';
|
||||
} else {
|
||||
$order = 'DESC';
|
||||
}
|
||||
switch ( $query['orderby'] ) {
|
||||
case 'hook':
|
||||
$sql .= " ORDER BY a.hook $order";
|
||||
break;
|
||||
case 'group':
|
||||
$sql .= " ORDER BY g.slug $order";
|
||||
break;
|
||||
case 'modified':
|
||||
$sql .= " ORDER BY a.last_attempt_gmt $order";
|
||||
break;
|
||||
case 'none':
|
||||
break;
|
||||
case 'action_id':
|
||||
$sql .= " ORDER BY a.action_id $order";
|
||||
break;
|
||||
case 'date':
|
||||
default:
|
||||
$sql .= " ORDER BY a.scheduled_date_gmt $order";
|
||||
break;
|
||||
}
|
||||
|
||||
if ( $query['per_page'] > 0 ) {
|
||||
$sql .= ' LIMIT %d, %d';
|
||||
$sql_params[] = $query['offset'];
|
||||
$sql_params[] = $query['per_page'];
|
||||
}
|
||||
}
|
||||
|
||||
if ( ! empty( $sql_params ) ) {
|
||||
$sql = $wpdb->prepare( $sql, $sql_params ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
|
||||
return $sql;
|
||||
}
|
||||
|
||||
/**
|
||||
* Query for action count or list of action IDs.
|
||||
*
|
||||
* @since x.x.x $query['status'] accepts array of statuses instead of a single status.
|
||||
*
|
||||
* @see ActionScheduler_Store::query_actions for $query arg usage.
|
||||
*
|
||||
* @param array $query Query filtering options.
|
||||
* @param string $query_type Whether to select or count the results. Defaults to select.
|
||||
*
|
||||
* @return string|array|null The IDs of actions matching the query. Null on failure.
|
||||
*/
|
||||
public function query_actions( $query = array(), $query_type = 'select' ) {
|
||||
/** @var wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$sql = $this->get_query_actions_sql( $query, $query_type );
|
||||
|
||||
return ( 'count' === $query_type ) ? $wpdb->get_var( $sql ) : $wpdb->get_col( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.DirectDatabaseQuery.NoSql, WordPress.DB.DirectDatabaseQuery.NoCaching
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a count of all actions in the store, grouped by status.
|
||||
*
|
||||
* @return array Set of 'status' => int $count pairs for statuses with 1 or more actions of that status.
|
||||
*/
|
||||
public function action_counts() {
|
||||
global $wpdb;
|
||||
|
||||
$sql = "SELECT a.status, count(a.status) as 'count'";
|
||||
$sql .= " FROM {$wpdb->actionscheduler_actions} a";
|
||||
$sql .= ' GROUP BY a.status';
|
||||
|
||||
$actions_count_by_status = array();
|
||||
$action_stati_and_labels = $this->get_status_labels();
|
||||
|
||||
foreach ( $wpdb->get_results( $sql ) as $action_data ) { // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
// Ignore any actions with invalid status.
|
||||
if ( array_key_exists( $action_data->status, $action_stati_and_labels ) ) {
|
||||
$actions_count_by_status[ $action_data->status ] = $action_data->count;
|
||||
}
|
||||
}
|
||||
|
||||
return $actions_count_by_status;
|
||||
}
|
||||
|
||||
/**
|
||||
* Cancel an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return void
|
||||
* @throws \InvalidArgumentException If the action update failed.
|
||||
*/
|
||||
public function cancel_action( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$updated = $wpdb->update(
|
||||
$wpdb->actionscheduler_actions,
|
||||
array( 'status' => self::STATUS_CANCELED ),
|
||||
array( 'action_id' => $action_id ),
|
||||
array( '%s' ),
|
||||
array( '%d' )
|
||||
);
|
||||
if ( false === $updated ) {
|
||||
/* translators: %s: action ID */
|
||||
throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'woocommerce' ), $action_id ) );
|
||||
}
|
||||
do_action( 'action_scheduler_canceled_action', $action_id );
|
||||
}
|
||||
|
||||
/**
|
||||
* Cancel pending actions by hook.
|
||||
*
|
||||
* @since 3.0.0
|
||||
*
|
||||
* @param string $hook Hook name.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function cancel_actions_by_hook( $hook ) {
|
||||
$this->bulk_cancel_actions( array( 'hook' => $hook ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Cancel pending actions by group.
|
||||
*
|
||||
* @param string $group Group slug.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function cancel_actions_by_group( $group ) {
|
||||
$this->bulk_cancel_actions( array( 'group' => $group ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Bulk cancel actions.
|
||||
*
|
||||
* @since 3.0.0
|
||||
*
|
||||
* @param array $query_args Query parameters.
|
||||
*/
|
||||
protected function bulk_cancel_actions( $query_args ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
if ( ! is_array( $query_args ) ) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Don't cancel actions that are already canceled.
|
||||
if ( isset( $query_args['status'] ) && self::STATUS_CANCELED === $query_args['status'] ) {
|
||||
return;
|
||||
}
|
||||
|
||||
$action_ids = true;
|
||||
$query_args = wp_parse_args(
|
||||
$query_args,
|
||||
array(
|
||||
'per_page' => 1000,
|
||||
'status' => self::STATUS_PENDING,
|
||||
'orderby' => 'action_id',
|
||||
)
|
||||
);
|
||||
|
||||
while ( $action_ids ) {
|
||||
$action_ids = $this->query_actions( $query_args );
|
||||
if ( empty( $action_ids ) ) {
|
||||
break;
|
||||
}
|
||||
|
||||
$format = array_fill( 0, count( $action_ids ), '%d' );
|
||||
$query_in = '(' . implode( ',', $format ) . ')';
|
||||
$parameters = $action_ids;
|
||||
array_unshift( $parameters, self::STATUS_CANCELED );
|
||||
|
||||
$wpdb->query(
|
||||
$wpdb->prepare(
|
||||
"UPDATE {$wpdb->actionscheduler_actions} SET status = %s WHERE action_id IN {$query_in}", // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
|
||||
$parameters
|
||||
)
|
||||
);
|
||||
|
||||
do_action( 'action_scheduler_bulk_cancel_actions', $action_ids );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
* @throws \InvalidArgumentException If the action deletion failed.
|
||||
*/
|
||||
public function delete_action( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$deleted = $wpdb->delete( $wpdb->actionscheduler_actions, array( 'action_id' => $action_id ), array( '%d' ) );
|
||||
if ( empty( $deleted ) ) {
|
||||
throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'woocommerce' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
|
||||
}
|
||||
do_action( 'action_scheduler_deleted_action', $action_id );
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the schedule date for an action.
|
||||
*
|
||||
* @param string $action_id Action ID.
|
||||
*
|
||||
* @return \DateTime The local date the action is scheduled to run, or the date that it ran.
|
||||
*/
|
||||
public function get_date( $action_id ) {
|
||||
$date = $this->get_date_gmt( $action_id );
|
||||
ActionScheduler_TimezoneHelper::set_local_timezone( $date );
|
||||
return $date;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the GMT schedule date for an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @throws \InvalidArgumentException If action cannot be identified.
|
||||
* @return \DateTime The GMT date the action is scheduled to run, or the date that it ran.
|
||||
*/
|
||||
protected function get_date_gmt( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$record = $wpdb->get_row( $wpdb->prepare( "SELECT * FROM {$wpdb->actionscheduler_actions} WHERE action_id=%d", $action_id ) );
|
||||
if ( empty( $record ) ) {
|
||||
throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'woocommerce' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
|
||||
}
|
||||
if ( self::STATUS_PENDING === $record->status ) {
|
||||
return as_get_datetime_object( $record->scheduled_date_gmt );
|
||||
} else {
|
||||
return as_get_datetime_object( $record->last_attempt_gmt );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Stake a claim on actions.
|
||||
*
|
||||
* @param int $max_actions Maximum number of action to include in claim.
|
||||
* @param \DateTime $before_date Jobs must be schedule before this date. Defaults to now.
|
||||
* @param array $hooks Hooks to filter for.
|
||||
* @param string $group Group to filter for.
|
||||
*
|
||||
* @return ActionScheduler_ActionClaim
|
||||
*/
|
||||
public function stake_claim( $max_actions = 10, \DateTime $before_date = null, $hooks = array(), $group = '' ) {
|
||||
$claim_id = $this->generate_claim_id();
|
||||
|
||||
$this->claim_before_date = $before_date;
|
||||
$this->claim_actions( $claim_id, $max_actions, $before_date, $hooks, $group );
|
||||
$action_ids = $this->find_actions_by_claim_id( $claim_id );
|
||||
$this->claim_before_date = null;
|
||||
|
||||
return new ActionScheduler_ActionClaim( $claim_id, $action_ids );
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate a new action claim.
|
||||
*
|
||||
* @return int Claim ID.
|
||||
*/
|
||||
protected function generate_claim_id() {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$now = as_get_datetime_object();
|
||||
$wpdb->insert( $wpdb->actionscheduler_claims, array( 'date_created_gmt' => $now->format( 'Y-m-d H:i:s' ) ) );
|
||||
|
||||
return $wpdb->insert_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark actions claimed.
|
||||
*
|
||||
* @param string $claim_id Claim Id.
|
||||
* @param int $limit Number of action to include in claim.
|
||||
* @param \DateTime $before_date Should use UTC timezone.
|
||||
* @param array $hooks Hooks to filter for.
|
||||
* @param string $group Group to filter for.
|
||||
*
|
||||
* @return int The number of actions that were claimed.
|
||||
* @throws \InvalidArgumentException Throws InvalidArgumentException if group doesn't exist.
|
||||
* @throws \RuntimeException Throws RuntimeException if unable to claim action.
|
||||
*/
|
||||
protected function claim_actions( $claim_id, $limit, \DateTime $before_date = null, $hooks = array(), $group = '' ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$now = as_get_datetime_object();
|
||||
$date = is_null( $before_date ) ? $now : clone $before_date;
|
||||
|
||||
// can't use $wpdb->update() because of the <= condition.
|
||||
$update = "UPDATE {$wpdb->actionscheduler_actions} SET claim_id=%d, last_attempt_gmt=%s, last_attempt_local=%s";
|
||||
$params = array(
|
||||
$claim_id,
|
||||
$now->format( 'Y-m-d H:i:s' ),
|
||||
current_time( 'mysql' ),
|
||||
);
|
||||
|
||||
$where = 'WHERE claim_id = 0 AND scheduled_date_gmt <= %s AND status=%s';
|
||||
$params[] = $date->format( 'Y-m-d H:i:s' );
|
||||
$params[] = self::STATUS_PENDING;
|
||||
|
||||
if ( ! empty( $hooks ) ) {
|
||||
$placeholders = array_fill( 0, count( $hooks ), '%s' );
|
||||
$where .= ' AND hook IN (' . join( ', ', $placeholders ) . ')';
|
||||
$params = array_merge( $params, array_values( $hooks ) );
|
||||
}
|
||||
|
||||
if ( ! empty( $group ) ) {
|
||||
|
||||
$group_id = $this->get_group_id( $group, false );
|
||||
|
||||
// throw exception if no matching group found, this matches ActionScheduler_wpPostStore's behaviour.
|
||||
if ( empty( $group_id ) ) {
|
||||
/* translators: %s: group name */
|
||||
throw new InvalidArgumentException( sprintf( __( 'The group "%s" does not exist.', 'woocommerce' ), $group ) );
|
||||
}
|
||||
|
||||
$where .= ' AND group_id = %d';
|
||||
$params[] = $group_id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets the order-by clause used in the action claim query.
|
||||
*
|
||||
* @since x.x.x
|
||||
*
|
||||
* @param string $order_by_sql
|
||||
*/
|
||||
$order = apply_filters( 'action_scheduler_claim_actions_order_by', 'ORDER BY attempts ASC, scheduled_date_gmt ASC, action_id ASC' );
|
||||
$params[] = $limit;
|
||||
|
||||
$sql = $wpdb->prepare( "{$update} {$where} {$order} LIMIT %d", $params ); // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared, WordPress.DB.PreparedSQLPlaceholders
|
||||
$rows_affected = $wpdb->query( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
|
||||
if ( false === $rows_affected ) {
|
||||
throw new \RuntimeException( __( 'Unable to claim actions. Database error.', 'woocommerce' ) );
|
||||
}
|
||||
|
||||
return (int) $rows_affected;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of active claims.
|
||||
*
|
||||
* @return int
|
||||
*/
|
||||
public function get_claim_count() {
|
||||
global $wpdb;
|
||||
|
||||
$sql = "SELECT COUNT(DISTINCT claim_id) FROM {$wpdb->actionscheduler_actions} WHERE claim_id != 0 AND status IN ( %s, %s)";
|
||||
$sql = $wpdb->prepare( $sql, array( self::STATUS_PENDING, self::STATUS_RUNNING ) ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
|
||||
return (int) $wpdb->get_var( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
|
||||
/**
|
||||
* Return an action's claim ID, as stored in the claim_id column.
|
||||
*
|
||||
* @param string $action_id Action ID.
|
||||
* @return mixed
|
||||
*/
|
||||
public function get_claim_id( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$sql = "SELECT claim_id FROM {$wpdb->actionscheduler_actions} WHERE action_id=%d";
|
||||
$sql = $wpdb->prepare( $sql, $action_id ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
|
||||
return (int) $wpdb->get_var( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve the action IDs of action in a claim.
|
||||
*
|
||||
* @param int $claim_id Claim ID.
|
||||
* @return int[]
|
||||
*/
|
||||
public function find_actions_by_claim_id( $claim_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$action_ids = array();
|
||||
$before_date = isset( $this->claim_before_date ) ? $this->claim_before_date : as_get_datetime_object();
|
||||
$cut_off = $before_date->format( 'Y-m-d H:i:s' );
|
||||
|
||||
$sql = $wpdb->prepare(
|
||||
"SELECT action_id, scheduled_date_gmt FROM {$wpdb->actionscheduler_actions} WHERE claim_id = %d",
|
||||
$claim_id
|
||||
);
|
||||
|
||||
// Verify that the scheduled date for each action is within the expected bounds (in some unusual
|
||||
// cases, we cannot depend on MySQL to honor all of the WHERE conditions we specify).
|
||||
foreach ( $wpdb->get_results( $sql ) as $claimed_action ) { // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
if ( $claimed_action->scheduled_date_gmt <= $cut_off ) {
|
||||
$action_ids[] = absint( $claimed_action->action_id );
|
||||
}
|
||||
}
|
||||
|
||||
return $action_ids;
|
||||
}
|
||||
|
||||
/**
|
||||
* Release actions from a claim and delete the claim.
|
||||
*
|
||||
* @param ActionScheduler_ActionClaim $claim Claim object.
|
||||
*/
|
||||
public function release_claim( ActionScheduler_ActionClaim $claim ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$wpdb->update( $wpdb->actionscheduler_actions, array( 'claim_id' => 0 ), array( 'claim_id' => $claim->get_id() ), array( '%d' ), array( '%d' ) );
|
||||
$wpdb->delete( $wpdb->actionscheduler_claims, array( 'claim_id' => $claim->get_id() ), array( '%d' ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Remove the claim from an action.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function unclaim_action( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$wpdb->update(
|
||||
$wpdb->actionscheduler_actions,
|
||||
array( 'claim_id' => 0 ),
|
||||
array( 'action_id' => $action_id ),
|
||||
array( '%s' ),
|
||||
array( '%d' )
|
||||
);
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark an action as failed.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
* @throws \InvalidArgumentException Throw an exception if action was not updated.
|
||||
*/
|
||||
public function mark_failure( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$updated = $wpdb->update(
|
||||
$wpdb->actionscheduler_actions,
|
||||
array( 'status' => self::STATUS_FAILED ),
|
||||
array( 'action_id' => $action_id ),
|
||||
array( '%s' ),
|
||||
array( '%d' )
|
||||
);
|
||||
if ( empty( $updated ) ) {
|
||||
throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'woocommerce' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Add execution message to action log.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return void
|
||||
*/
|
||||
public function log_execution( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
|
||||
$sql = "UPDATE {$wpdb->actionscheduler_actions} SET attempts = attempts+1, status=%s, last_attempt_gmt = %s, last_attempt_local = %s WHERE action_id = %d";
|
||||
$sql = $wpdb->prepare( $sql, self::STATUS_RUNNING, current_time( 'mysql', true ), current_time( 'mysql' ), $action_id ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
$wpdb->query( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark an action as complete.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return void
|
||||
* @throws \InvalidArgumentException Throw an exception if action was not updated.
|
||||
*/
|
||||
public function mark_complete( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$updated = $wpdb->update(
|
||||
$wpdb->actionscheduler_actions,
|
||||
array(
|
||||
'status' => self::STATUS_COMPLETE,
|
||||
'last_attempt_gmt' => current_time( 'mysql', true ),
|
||||
'last_attempt_local' => current_time( 'mysql' ),
|
||||
),
|
||||
array( 'action_id' => $action_id ),
|
||||
array( '%s' ),
|
||||
array( '%d' )
|
||||
);
|
||||
if ( empty( $updated ) ) {
|
||||
throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'woocommerce' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
|
||||
}
|
||||
|
||||
/**
|
||||
* Fires after a scheduled action has been completed.
|
||||
*
|
||||
* @since 3.4.2
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
do_action( 'action_scheduler_completed_action', $action_id );
|
||||
}
|
||||
|
||||
/**
|
||||
* Get an action's status.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*
|
||||
* @return string
|
||||
* @throws \InvalidArgumentException Throw an exception if not status was found for action_id.
|
||||
* @throws \RuntimeException Throw an exception if action status could not be retrieved.
|
||||
*/
|
||||
public function get_status( $action_id ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$sql = "SELECT status FROM {$wpdb->actionscheduler_actions} WHERE action_id=%d";
|
||||
$sql = $wpdb->prepare( $sql, $action_id ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
$status = $wpdb->get_var( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
|
||||
|
||||
if ( null === $status ) {
|
||||
throw new \InvalidArgumentException( __( 'Invalid action ID. No status found.', 'woocommerce' ) );
|
||||
} elseif ( empty( $status ) ) {
|
||||
throw new \RuntimeException( __( 'Unknown status found for action.', 'woocommerce' ) );
|
||||
} else {
|
||||
return $status;
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,426 @@
|
||||
<?php
|
||||
|
||||
use ActionScheduler_Store as Store;
|
||||
use Action_Scheduler\Migration\Runner;
|
||||
use Action_Scheduler\Migration\Config;
|
||||
use Action_Scheduler\Migration\Controller;
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_HybridStore
|
||||
*
|
||||
* A wrapper around multiple stores that fetches data from both.
|
||||
*
|
||||
* @since 3.0.0
|
||||
*/
|
||||
class ActionScheduler_HybridStore extends Store {
|
||||
const DEMARKATION_OPTION = 'action_scheduler_hybrid_store_demarkation';
|
||||
|
||||
private $primary_store;
|
||||
private $secondary_store;
|
||||
private $migration_runner;
|
||||
|
||||
/**
|
||||
* @var int The dividing line between IDs of actions created
|
||||
* by the primary and secondary stores.
|
||||
*
|
||||
* Methods that accept an action ID will compare the ID against
|
||||
* this to determine which store will contain that ID. In almost
|
||||
* all cases, the ID should come from the primary store, but if
|
||||
* client code is bypassing the API functions and fetching IDs
|
||||
* from elsewhere, then there is a chance that an unmigrated ID
|
||||
* might be requested.
|
||||
*/
|
||||
private $demarkation_id = 0;
|
||||
|
||||
/**
|
||||
* ActionScheduler_HybridStore constructor.
|
||||
*
|
||||
* @param Config $config Migration config object.
|
||||
*/
|
||||
public function __construct( Config $config = null ) {
|
||||
$this->demarkation_id = (int) get_option( self::DEMARKATION_OPTION, 0 );
|
||||
if ( empty( $config ) ) {
|
||||
$config = Controller::instance()->get_migration_config_object();
|
||||
}
|
||||
$this->primary_store = $config->get_destination_store();
|
||||
$this->secondary_store = $config->get_source_store();
|
||||
$this->migration_runner = new Runner( $config );
|
||||
}
|
||||
|
||||
/**
|
||||
* Initialize the table data store tables.
|
||||
*
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
public function init() {
|
||||
add_action( 'action_scheduler/created_table', [ $this, 'set_autoincrement' ], 10, 2 );
|
||||
$this->primary_store->init();
|
||||
$this->secondary_store->init();
|
||||
remove_action( 'action_scheduler/created_table', [ $this, 'set_autoincrement' ], 10 );
|
||||
}
|
||||
|
||||
/**
|
||||
* When the actions table is created, set its autoincrement
|
||||
* value to be one higher than the posts table to ensure that
|
||||
* there are no ID collisions.
|
||||
*
|
||||
* @param string $table_name
|
||||
* @param string $table_suffix
|
||||
*
|
||||
* @return void
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
public function set_autoincrement( $table_name, $table_suffix ) {
|
||||
if ( ActionScheduler_StoreSchema::ACTIONS_TABLE === $table_suffix ) {
|
||||
if ( empty( $this->demarkation_id ) ) {
|
||||
$this->demarkation_id = $this->set_demarkation_id();
|
||||
}
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
/**
|
||||
* A default date of '0000-00-00 00:00:00' is invalid in MySQL 5.7 when configured with
|
||||
* sql_mode including both STRICT_TRANS_TABLES and NO_ZERO_DATE.
|
||||
*/
|
||||
$default_date = new DateTime( 'tomorrow' );
|
||||
$null_action = new ActionScheduler_NullAction();
|
||||
$date_gmt = $this->get_scheduled_date_string( $null_action, $default_date );
|
||||
$date_local = $this->get_scheduled_date_string_local( $null_action, $default_date );
|
||||
|
||||
$row_count = $wpdb->insert(
|
||||
$wpdb->{ActionScheduler_StoreSchema::ACTIONS_TABLE},
|
||||
[
|
||||
'action_id' => $this->demarkation_id,
|
||||
'hook' => '',
|
||||
'status' => '',
|
||||
'scheduled_date_gmt' => $date_gmt,
|
||||
'scheduled_date_local' => $date_local,
|
||||
'last_attempt_gmt' => $date_gmt,
|
||||
'last_attempt_local' => $date_local,
|
||||
]
|
||||
);
|
||||
if ( $row_count > 0 ) {
|
||||
$wpdb->delete(
|
||||
$wpdb->{ActionScheduler_StoreSchema::ACTIONS_TABLE},
|
||||
[ 'action_id' => $this->demarkation_id ]
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Store the demarkation id in WP options.
|
||||
*
|
||||
* @param int $id The ID to set as the demarkation point between the two stores
|
||||
* Leave null to use the next ID from the WP posts table.
|
||||
*
|
||||
* @return int The new ID.
|
||||
*
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
private function set_demarkation_id( $id = null ) {
|
||||
if ( empty( $id ) ) {
|
||||
/** @var \wpdb $wpdb */
|
||||
global $wpdb;
|
||||
$id = (int) $wpdb->get_var( "SELECT MAX(ID) FROM $wpdb->posts" );
|
||||
$id ++;
|
||||
}
|
||||
update_option( self::DEMARKATION_OPTION, $id );
|
||||
|
||||
return $id;
|
||||
}
|
||||
|
||||
/**
|
||||
* Find the first matching action from the secondary store.
|
||||
* If it exists, migrate it to the primary store immediately.
|
||||
* After it migrates, the secondary store will logically contain
|
||||
* the next matching action, so return the result thence.
|
||||
*
|
||||
* @param string $hook
|
||||
* @param array $params
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
public function find_action( $hook, $params = [] ) {
|
||||
$found_unmigrated_action = $this->secondary_store->find_action( $hook, $params );
|
||||
if ( ! empty( $found_unmigrated_action ) ) {
|
||||
$this->migrate( [ $found_unmigrated_action ] );
|
||||
}
|
||||
|
||||
return $this->primary_store->find_action( $hook, $params );
|
||||
}
|
||||
|
||||
/**
|
||||
* Find actions matching the query in the secondary source first.
|
||||
* If any are found, migrate them immediately. Then the secondary
|
||||
* store will contain the canonical results.
|
||||
*
|
||||
* @param array $query
|
||||
* @param string $query_type Whether to select or count the results. Default, select.
|
||||
*
|
||||
* @return int[]
|
||||
*/
|
||||
public function query_actions( $query = [], $query_type = 'select' ) {
|
||||
$found_unmigrated_actions = $this->secondary_store->query_actions( $query, 'select' );
|
||||
if ( ! empty( $found_unmigrated_actions ) ) {
|
||||
$this->migrate( $found_unmigrated_actions );
|
||||
}
|
||||
|
||||
return $this->primary_store->query_actions( $query, $query_type );
|
||||
}
|
||||
|
||||
/**
|
||||
* Get a count of all actions in the store, grouped by status
|
||||
*
|
||||
* @return array Set of 'status' => int $count pairs for statuses with 1 or more actions of that status.
|
||||
*/
|
||||
public function action_counts() {
|
||||
$unmigrated_actions_count = $this->secondary_store->action_counts();
|
||||
$migrated_actions_count = $this->primary_store->action_counts();
|
||||
$actions_count_by_status = array();
|
||||
|
||||
foreach ( $this->get_status_labels() as $status_key => $status_label ) {
|
||||
|
||||
$count = 0;
|
||||
|
||||
if ( isset( $unmigrated_actions_count[ $status_key ] ) ) {
|
||||
$count += $unmigrated_actions_count[ $status_key ];
|
||||
}
|
||||
|
||||
if ( isset( $migrated_actions_count[ $status_key ] ) ) {
|
||||
$count += $migrated_actions_count[ $status_key ];
|
||||
}
|
||||
|
||||
$actions_count_by_status[ $status_key ] = $count;
|
||||
}
|
||||
|
||||
$actions_count_by_status = array_filter( $actions_count_by_status );
|
||||
|
||||
return $actions_count_by_status;
|
||||
}
|
||||
|
||||
/**
|
||||
* If any actions would have been claimed by the secondary store,
|
||||
* migrate them immediately, then ask the primary store for the
|
||||
* canonical claim.
|
||||
*
|
||||
* @param int $max_actions
|
||||
* @param DateTime|null $before_date
|
||||
*
|
||||
* @return ActionScheduler_ActionClaim
|
||||
*/
|
||||
public function stake_claim( $max_actions = 10, DateTime $before_date = null, $hooks = array(), $group = '' ) {
|
||||
$claim = $this->secondary_store->stake_claim( $max_actions, $before_date, $hooks, $group );
|
||||
|
||||
$claimed_actions = $claim->get_actions();
|
||||
if ( ! empty( $claimed_actions ) ) {
|
||||
$this->migrate( $claimed_actions );
|
||||
}
|
||||
|
||||
$this->secondary_store->release_claim( $claim );
|
||||
|
||||
return $this->primary_store->stake_claim( $max_actions, $before_date, $hooks, $group );
|
||||
}
|
||||
|
||||
/**
|
||||
* Migrate a list of actions to the table data store.
|
||||
*
|
||||
* @param array $action_ids List of action IDs.
|
||||
*/
|
||||
private function migrate( $action_ids ) {
|
||||
$this->migration_runner->migrate_actions( $action_ids );
|
||||
}
|
||||
|
||||
/**
|
||||
* Save an action to the primary store.
|
||||
*
|
||||
* @param ActionScheduler_Action $action Action object to be saved.
|
||||
* @param DateTime $date Optional. Schedule date. Default null.
|
||||
*
|
||||
* @return int The action ID
|
||||
*/
|
||||
public function save_action( ActionScheduler_Action $action, DateTime $date = null ) {
|
||||
return $this->primary_store->save_action( $action, $date );
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve an existing action whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function fetch_action( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id, true );
|
||||
if ( $store ) {
|
||||
return $store->fetch_action( $action_id );
|
||||
} else {
|
||||
return new ActionScheduler_NullAction();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Cancel an existing action whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function cancel_action( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
$store->cancel_action( $action_id );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete an existing action whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function delete_action( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
$store->delete_action( $action_id );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the schedule date an existing action whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function get_date( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
return $store->get_date( $action_id );
|
||||
} else {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark an existing action as failed whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function mark_failure( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
$store->mark_failure( $action_id );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Log the execution of an existing action whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function log_execution( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
$store->log_execution( $action_id );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark an existing action complete whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function mark_complete( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
$store->mark_complete( $action_id );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get an existing action status whether migrated or not.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function get_status( $action_id ) {
|
||||
$store = $this->get_store_from_action_id( $action_id );
|
||||
if ( $store ) {
|
||||
return $store->get_status( $action_id );
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return which store an action is stored in.
|
||||
*
|
||||
* @param int $action_id ID of the action.
|
||||
* @param bool $primary_first Optional flag indicating search the primary store first.
|
||||
* @return ActionScheduler_Store
|
||||
*/
|
||||
protected function get_store_from_action_id( $action_id, $primary_first = false ) {
|
||||
if ( $primary_first ) {
|
||||
$stores = [
|
||||
$this->primary_store,
|
||||
$this->secondary_store,
|
||||
];
|
||||
} elseif ( $action_id < $this->demarkation_id ) {
|
||||
$stores = [
|
||||
$this->secondary_store,
|
||||
$this->primary_store,
|
||||
];
|
||||
} else {
|
||||
$stores = [
|
||||
$this->primary_store,
|
||||
];
|
||||
}
|
||||
|
||||
foreach ( $stores as $store ) {
|
||||
$action = $store->fetch_action( $action_id );
|
||||
if ( ! is_a( $action, 'ActionScheduler_NullAction' ) ) {
|
||||
return $store;
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
/* * * * * * * * * * * * * * * * * * * * * * * * * * *
|
||||
* All claim-related functions should operate solely
|
||||
* on the primary store.
|
||||
* * * * * * * * * * * * * * * * * * * * * * * * * * */
|
||||
|
||||
/**
|
||||
* Get the claim count from the table data store.
|
||||
*/
|
||||
public function get_claim_count() {
|
||||
return $this->primary_store->get_claim_count();
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve the claim ID for an action from the table data store.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function get_claim_id( $action_id ) {
|
||||
return $this->primary_store->get_claim_id( $action_id );
|
||||
}
|
||||
|
||||
/**
|
||||
* Release a claim in the table data store.
|
||||
*
|
||||
* @param ActionScheduler_ActionClaim $claim Claim object.
|
||||
*/
|
||||
public function release_claim( ActionScheduler_ActionClaim $claim ) {
|
||||
$this->primary_store->release_claim( $claim );
|
||||
}
|
||||
|
||||
/**
|
||||
* Release claims on an action in the table data store.
|
||||
*
|
||||
* @param int $action_id Action ID.
|
||||
*/
|
||||
public function unclaim_action( $action_id ) {
|
||||
$this->primary_store->unclaim_action( $action_id );
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve a list of action IDs by claim.
|
||||
*
|
||||
* @param int $claim_id Claim ID.
|
||||
*/
|
||||
public function find_actions_by_claim_id( $claim_id ) {
|
||||
return $this->primary_store->find_actions_by_claim_id( $claim_id );
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,240 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_wpCommentLogger
|
||||
*/
|
||||
class ActionScheduler_wpCommentLogger extends ActionScheduler_Logger {
|
||||
const AGENT = 'ActionScheduler';
|
||||
const TYPE = 'action_log';
|
||||
|
||||
/**
|
||||
* @param string $action_id
|
||||
* @param string $message
|
||||
* @param DateTime $date
|
||||
*
|
||||
* @return string The log entry ID
|
||||
*/
|
||||
public function log( $action_id, $message, DateTime $date = NULL ) {
|
||||
if ( empty($date) ) {
|
||||
$date = as_get_datetime_object();
|
||||
} else {
|
||||
$date = as_get_datetime_object( clone $date );
|
||||
}
|
||||
$comment_id = $this->create_wp_comment( $action_id, $message, $date );
|
||||
return $comment_id;
|
||||
}
|
||||
|
||||
protected function create_wp_comment( $action_id, $message, DateTime $date ) {
|
||||
|
||||
$comment_date_gmt = $date->format('Y-m-d H:i:s');
|
||||
ActionScheduler_TimezoneHelper::set_local_timezone( $date );
|
||||
$comment_data = array(
|
||||
'comment_post_ID' => $action_id,
|
||||
'comment_date' => $date->format('Y-m-d H:i:s'),
|
||||
'comment_date_gmt' => $comment_date_gmt,
|
||||
'comment_author' => self::AGENT,
|
||||
'comment_content' => $message,
|
||||
'comment_agent' => self::AGENT,
|
||||
'comment_type' => self::TYPE,
|
||||
);
|
||||
return wp_insert_comment($comment_data);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $entry_id
|
||||
*
|
||||
* @return ActionScheduler_LogEntry
|
||||
*/
|
||||
public function get_entry( $entry_id ) {
|
||||
$comment = $this->get_comment( $entry_id );
|
||||
if ( empty($comment) || $comment->comment_type != self::TYPE ) {
|
||||
return new ActionScheduler_NullLogEntry();
|
||||
}
|
||||
|
||||
$date = as_get_datetime_object( $comment->comment_date_gmt );
|
||||
ActionScheduler_TimezoneHelper::set_local_timezone( $date );
|
||||
return new ActionScheduler_LogEntry( $comment->comment_post_ID, $comment->comment_content, $date );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $action_id
|
||||
*
|
||||
* @return ActionScheduler_LogEntry[]
|
||||
*/
|
||||
public function get_logs( $action_id ) {
|
||||
$status = 'all';
|
||||
if ( get_post_status($action_id) == 'trash' ) {
|
||||
$status = 'post-trashed';
|
||||
}
|
||||
$comments = get_comments(array(
|
||||
'post_id' => $action_id,
|
||||
'orderby' => 'comment_date_gmt',
|
||||
'order' => 'ASC',
|
||||
'type' => self::TYPE,
|
||||
'status' => $status,
|
||||
));
|
||||
$logs = array();
|
||||
foreach ( $comments as $c ) {
|
||||
$entry = $this->get_entry( $c );
|
||||
if ( !empty($entry) ) {
|
||||
$logs[] = $entry;
|
||||
}
|
||||
}
|
||||
return $logs;
|
||||
}
|
||||
|
||||
protected function get_comment( $comment_id ) {
|
||||
return get_comment( $comment_id );
|
||||
}
|
||||
|
||||
|
||||
|
||||
/**
|
||||
* @param WP_Comment_Query $query
|
||||
*/
|
||||
public function filter_comment_queries( $query ) {
|
||||
foreach ( array('ID', 'parent', 'post_author', 'post_name', 'post_parent', 'type', 'post_type', 'post_id', 'post_ID') as $key ) {
|
||||
if ( !empty($query->query_vars[$key]) ) {
|
||||
return; // don't slow down queries that wouldn't include action_log comments anyway
|
||||
}
|
||||
}
|
||||
$query->query_vars['action_log_filter'] = TRUE;
|
||||
add_filter( 'comments_clauses', array( $this, 'filter_comment_query_clauses' ), 10, 2 );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param array $clauses
|
||||
* @param WP_Comment_Query $query
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
public function filter_comment_query_clauses( $clauses, $query ) {
|
||||
if ( !empty($query->query_vars['action_log_filter']) ) {
|
||||
$clauses['where'] .= $this->get_where_clause();
|
||||
}
|
||||
return $clauses;
|
||||
}
|
||||
|
||||
/**
|
||||
* Make sure Action Scheduler logs are excluded from comment feeds, which use WP_Query, not
|
||||
* the WP_Comment_Query class handled by @see self::filter_comment_queries().
|
||||
*
|
||||
* @param string $where
|
||||
* @param WP_Query $query
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
public function filter_comment_feed( $where, $query ) {
|
||||
if ( is_comment_feed() ) {
|
||||
$where .= $this->get_where_clause();
|
||||
}
|
||||
return $where;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return a SQL clause to exclude Action Scheduler comments.
|
||||
*
|
||||
* @return string
|
||||
*/
|
||||
protected function get_where_clause() {
|
||||
global $wpdb;
|
||||
return sprintf( " AND {$wpdb->comments}.comment_type != '%s'", self::TYPE );
|
||||
}
|
||||
|
||||
/**
|
||||
* Remove action log entries from wp_count_comments()
|
||||
*
|
||||
* @param array $stats
|
||||
* @param int $post_id
|
||||
*
|
||||
* @return object
|
||||
*/
|
||||
public function filter_comment_count( $stats, $post_id ) {
|
||||
global $wpdb;
|
||||
|
||||
if ( 0 === $post_id ) {
|
||||
$stats = $this->get_comment_count();
|
||||
}
|
||||
|
||||
return $stats;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve the comment counts from our cache, or the database if the cached version isn't set.
|
||||
*
|
||||
* @return object
|
||||
*/
|
||||
protected function get_comment_count() {
|
||||
global $wpdb;
|
||||
|
||||
$stats = get_transient( 'as_comment_count' );
|
||||
|
||||
if ( ! $stats ) {
|
||||
$stats = array();
|
||||
|
||||
$count = $wpdb->get_results( "SELECT comment_approved, COUNT( * ) AS num_comments FROM {$wpdb->comments} WHERE comment_type NOT IN('order_note','action_log') GROUP BY comment_approved", ARRAY_A );
|
||||
|
||||
$total = 0;
|
||||
$stats = array();
|
||||
$approved = array( '0' => 'moderated', '1' => 'approved', 'spam' => 'spam', 'trash' => 'trash', 'post-trashed' => 'post-trashed' );
|
||||
|
||||
foreach ( (array) $count as $row ) {
|
||||
// Don't count post-trashed toward totals
|
||||
if ( 'post-trashed' != $row['comment_approved'] && 'trash' != $row['comment_approved'] ) {
|
||||
$total += $row['num_comments'];
|
||||
}
|
||||
if ( isset( $approved[ $row['comment_approved'] ] ) ) {
|
||||
$stats[ $approved[ $row['comment_approved'] ] ] = $row['num_comments'];
|
||||
}
|
||||
}
|
||||
|
||||
$stats['total_comments'] = $total;
|
||||
$stats['all'] = $total;
|
||||
|
||||
foreach ( $approved as $key ) {
|
||||
if ( empty( $stats[ $key ] ) ) {
|
||||
$stats[ $key ] = 0;
|
||||
}
|
||||
}
|
||||
|
||||
$stats = (object) $stats;
|
||||
set_transient( 'as_comment_count', $stats );
|
||||
}
|
||||
|
||||
return $stats;
|
||||
}
|
||||
|
||||
/**
|
||||
* Delete comment count cache whenever there is new comment or the status of a comment changes. Cache
|
||||
* will be regenerated next time ActionScheduler_wpCommentLogger::filter_comment_count() is called.
|
||||
*/
|
||||
public function delete_comment_count_cache() {
|
||||
delete_transient( 'as_comment_count' );
|
||||
}
|
||||
|
||||
/**
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
public function init() {
|
||||
add_action( 'action_scheduler_before_process_queue', array( $this, 'disable_comment_counting' ), 10, 0 );
|
||||
add_action( 'action_scheduler_after_process_queue', array( $this, 'enable_comment_counting' ), 10, 0 );
|
||||
|
||||
parent::init();
|
||||
|
||||
add_action( 'pre_get_comments', array( $this, 'filter_comment_queries' ), 10, 1 );
|
||||
add_action( 'wp_count_comments', array( $this, 'filter_comment_count' ), 20, 2 ); // run after WC_Comments::wp_count_comments() to make sure we exclude order notes and action logs
|
||||
add_action( 'comment_feed_where', array( $this, 'filter_comment_feed' ), 10, 2 );
|
||||
|
||||
// Delete comments count cache whenever there is a new comment or a comment status changes
|
||||
add_action( 'wp_insert_comment', array( $this, 'delete_comment_count_cache' ) );
|
||||
add_action( 'wp_set_comment_status', array( $this, 'delete_comment_count_cache' ) );
|
||||
}
|
||||
|
||||
public function disable_comment_counting() {
|
||||
wp_defer_comment_counting(true);
|
||||
}
|
||||
public function enable_comment_counting() {
|
||||
wp_defer_comment_counting(false);
|
||||
}
|
||||
|
||||
}
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,58 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_wpPostStore_PostStatusRegistrar
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
class ActionScheduler_wpPostStore_PostStatusRegistrar {
|
||||
public function register() {
|
||||
register_post_status( ActionScheduler_Store::STATUS_RUNNING, array_merge( $this->post_status_args(), $this->post_status_running_labels() ) );
|
||||
register_post_status( ActionScheduler_Store::STATUS_FAILED, array_merge( $this->post_status_args(), $this->post_status_failed_labels() ) );
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the args array for the post type definition
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
protected function post_status_args() {
|
||||
$args = array(
|
||||
'public' => false,
|
||||
'exclude_from_search' => false,
|
||||
'show_in_admin_all_list' => true,
|
||||
'show_in_admin_status_list' => true,
|
||||
);
|
||||
|
||||
return apply_filters( 'action_scheduler_post_status_args', $args );
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the args array for the post type definition
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
protected function post_status_failed_labels() {
|
||||
$labels = array(
|
||||
'label' => _x( 'Failed', 'post', 'woocommerce' ),
|
||||
/* translators: %s: count */
|
||||
'label_count' => _n_noop( 'Failed <span class="count">(%s)</span>', 'Failed <span class="count">(%s)</span>', 'woocommerce' ),
|
||||
);
|
||||
|
||||
return apply_filters( 'action_scheduler_post_status_failed_labels', $labels );
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the args array for the post type definition
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
protected function post_status_running_labels() {
|
||||
$labels = array(
|
||||
'label' => _x( 'In-Progress', 'post', 'woocommerce' ),
|
||||
/* translators: %s: count */
|
||||
'label_count' => _n_noop( 'In-Progress <span class="count">(%s)</span>', 'In-Progress <span class="count">(%s)</span>', 'woocommerce' ),
|
||||
);
|
||||
|
||||
return apply_filters( 'action_scheduler_post_status_running_labels', $labels );
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,50 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_wpPostStore_PostTypeRegistrar
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
class ActionScheduler_wpPostStore_PostTypeRegistrar {
|
||||
public function register() {
|
||||
register_post_type( ActionScheduler_wpPostStore::POST_TYPE, $this->post_type_args() );
|
||||
}
|
||||
|
||||
/**
|
||||
* Build the args array for the post type definition
|
||||
*
|
||||
* @return array
|
||||
*/
|
||||
protected function post_type_args() {
|
||||
$args = array(
|
||||
'label' => __( 'Scheduled Actions', 'woocommerce' ),
|
||||
'description' => __( 'Scheduled actions are hooks triggered on a cetain date and time.', 'woocommerce' ),
|
||||
'public' => false,
|
||||
'map_meta_cap' => true,
|
||||
'hierarchical' => false,
|
||||
'supports' => array('title', 'editor','comments'),
|
||||
'rewrite' => false,
|
||||
'query_var' => false,
|
||||
'can_export' => true,
|
||||
'ep_mask' => EP_NONE,
|
||||
'labels' => array(
|
||||
'name' => __( 'Scheduled Actions', 'woocommerce' ),
|
||||
'singular_name' => __( 'Scheduled Action', 'woocommerce' ),
|
||||
'menu_name' => _x( 'Scheduled Actions', 'Admin menu name', 'woocommerce' ),
|
||||
'add_new' => __( 'Add', 'woocommerce' ),
|
||||
'add_new_item' => __( 'Add New Scheduled Action', 'woocommerce' ),
|
||||
'edit' => __( 'Edit', 'woocommerce' ),
|
||||
'edit_item' => __( 'Edit Scheduled Action', 'woocommerce' ),
|
||||
'new_item' => __( 'New Scheduled Action', 'woocommerce' ),
|
||||
'view' => __( 'View Action', 'woocommerce' ),
|
||||
'view_item' => __( 'View Action', 'woocommerce' ),
|
||||
'search_items' => __( 'Search Scheduled Actions', 'woocommerce' ),
|
||||
'not_found' => __( 'No actions found', 'woocommerce' ),
|
||||
'not_found_in_trash' => __( 'No actions found in trash', 'woocommerce' ),
|
||||
),
|
||||
);
|
||||
|
||||
$args = apply_filters('action_scheduler_post_type_args', $args);
|
||||
return $args;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,26 @@
|
||||
<?php
|
||||
|
||||
/**
|
||||
* Class ActionScheduler_wpPostStore_TaxonomyRegistrar
|
||||
* @codeCoverageIgnore
|
||||
*/
|
||||
class ActionScheduler_wpPostStore_TaxonomyRegistrar {
|
||||
public function register() {
|
||||
register_taxonomy( ActionScheduler_wpPostStore::GROUP_TAXONOMY, ActionScheduler_wpPostStore::POST_TYPE, $this->taxonomy_args() );
|
||||
}
|
||||
|
||||
protected function taxonomy_args() {
|
||||
$args = array(
|
||||
'label' => __( 'Action Group', 'woocommerce' ),
|
||||
'public' => false,
|
||||
'hierarchical' => false,
|
||||
'show_admin_column' => true,
|
||||
'query_var' => false,
|
||||
'rewrite' => false,
|
||||
);
|
||||
|
||||
$args = apply_filters( 'action_scheduler_taxonomy_args', $args );
|
||||
return $args;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user