insideout10 / wordlift-plugin

src/ext/action-scheduler/classes/schema/ActionScheduler_LoggerSchema.php

Indentation +55 added lines, -55 removed lines

@@ -8,35 +8,35 @@ 
  * Creates a custom table for storing action logs
  */
 class ActionScheduler_LoggerSchema extends ActionScheduler_Abstract_Schema {
-	const LOG_TABLE = 'actionscheduler_logs';
+    const LOG_TABLE = 'actionscheduler_logs';
 
-	/**
-	 * @var int Increment this value to trigger a schema update.
-	 */
-	protected $schema_version = 3;
+    /**
+     * @var int Increment this value to trigger a schema update.
+     */
+    protected $schema_version = 3;
 
-	public function __construct() {
-		$this->tables = array(
-			self::LOG_TABLE,
-		);
-	}
+    public function __construct() {
+        $this->tables = array(
+            self::LOG_TABLE,
+        );
+    }
 
-	/**
-	 * Performs additional setup work required to support this schema.
-	 */
-	public function init() {
-		add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_3_0' ), 10, 2 );
-	}
+    /**
+     * Performs additional setup work required to support this schema.
+     */
+    public function init() {
+        add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_3_0' ), 10, 2 );
+    }
 
-	protected function get_table_definition( $table ) {
-		global $wpdb;
-		$table_name      = $wpdb->$table;
-		$charset_collate = $wpdb->get_charset_collate();
-		switch ( $table ) {
+    protected function get_table_definition( $table ) {
+        global $wpdb;
+        $table_name      = $wpdb->$table;
+        $charset_collate = $wpdb->get_charset_collate();
+        switch ( $table ) {
 
-			case self::LOG_TABLE:
-				$default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
-				return "CREATE TABLE $table_name (
+            case self::LOG_TABLE:
+                $default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
+                return "CREATE TABLE $table_name (
 				        log_id bigint(20) unsigned NOT NULL auto_increment,
 				        action_id bigint(20) unsigned NOT NULL,
 				        message text NOT NULL,
@@ -47,43 +47,43 @@ 
 				        KEY log_date_gmt (log_date_gmt)
 				        ) $charset_collate";
 
-			default:
-				return '';
-		}
-	}
+            default:
+                return '';
+        }
+    }
 
-	/**
-	 * Update the logs table schema, allowing datetime fields to be NULL.
-	 *
-	 * This is needed because the NOT NULL constraint causes a conflict with some versions of MySQL
-	 * configured with sql_mode=NO_ZERO_DATE, which can for instance lead to tables not being created.
-	 *
-	 * Most other schema updates happen via ActionScheduler_Abstract_Schema::update_table(), however
-	 * that method relies on dbDelta() and this change is not possible when using that function.
-	 *
-	 * @param string $table Name of table being updated.
-	 * @param string $db_version The existing schema version of the table.
-	 */
-	public function update_schema_3_0( $table, $db_version ) {
-		global $wpdb;
+    /**
+     * Update the logs table schema, allowing datetime fields to be NULL.
+     *
+     * This is needed because the NOT NULL constraint causes a conflict with some versions of MySQL
+     * configured with sql_mode=NO_ZERO_DATE, which can for instance lead to tables not being created.
+     *
+     * Most other schema updates happen via ActionScheduler_Abstract_Schema::update_table(), however
+     * that method relies on dbDelta() and this change is not possible when using that function.
+     *
+     * @param string $table Name of table being updated.
+     * @param string $db_version The existing schema version of the table.
+     */
+    public function update_schema_3_0( $table, $db_version ) {
+        global $wpdb;
 
-		if ( 'actionscheduler_logs' !== $table || version_compare( $db_version, '3', '>=' ) ) {
-			return;
-		}
+        if ( 'actionscheduler_logs' !== $table || version_compare( $db_version, '3', '>=' ) ) {
+            return;
+        }
 
-		// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-		$table_name   = $wpdb->prefix . 'actionscheduler_logs';
-		$table_list   = $wpdb->get_col( "SHOW TABLES LIKE '{$table_name}'" );
-		$default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
+        // phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
+        $table_name   = $wpdb->prefix . 'actionscheduler_logs';
+        $table_list   = $wpdb->get_col( "SHOW TABLES LIKE '{$table_name}'" );
+        $default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
 
-		if ( ! empty( $table_list ) ) {
-			$query = "
+        if ( ! empty( $table_list ) ) {
+            $query = "
 				ALTER TABLE {$table_name}
 				MODIFY COLUMN log_date_gmt datetime NULL default '{$default_date}',
 				MODIFY COLUMN log_date_local datetime NULL default '{$default_date}'
 			";
-			$wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
-		}
-		// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-	}
+            $wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+        }
+        // phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
+    }
 }

Spacing +9 added lines, -9 removed lines

@@ -25,14 +25,14 @@ 
 	 * Performs additional setup work required to support this schema.
 	 */
 	public function init() {
-		add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_3_0' ), 10, 2 );
+		add_action('action_scheduler_before_schema_update', array($this, 'update_schema_3_0'), 10, 2);
 	}
 
-	protected function get_table_definition( $table ) {
+	protected function get_table_definition($table) {
 		global $wpdb;
 		$table_name      = $wpdb->$table;
 		$charset_collate = $wpdb->get_charset_collate();
-		switch ( $table ) {
+		switch ($table) {
 
 			case self::LOG_TABLE:
 				$default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
@@ -64,25 +64,25 @@ 
 	 * @param string $table Name of table being updated.
 	 * @param string $db_version The existing schema version of the table.
 	 */
-	public function update_schema_3_0( $table, $db_version ) {
+	public function update_schema_3_0($table, $db_version) {
 		global $wpdb;
 
-		if ( 'actionscheduler_logs' !== $table || version_compare( $db_version, '3', '>=' ) ) {
+		if ('actionscheduler_logs' !== $table || version_compare($db_version, '3', '>=')) {
 			return;
 		}
 
 		// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-		$table_name   = $wpdb->prefix . 'actionscheduler_logs';
-		$table_list   = $wpdb->get_col( "SHOW TABLES LIKE '{$table_name}'" );
+		$table_name   = $wpdb->prefix.'actionscheduler_logs';
+		$table_list   = $wpdb->get_col("SHOW TABLES LIKE '{$table_name}'");
 		$default_date = ActionScheduler_StoreSchema::DEFAULT_DATE;
 
-		if ( ! empty( $table_list ) ) {
+		if ( ! empty($table_list)) {
 			$query = "
 				ALTER TABLE {$table_name}
 				MODIFY COLUMN log_date_gmt datetime NULL default '{$default_date}',
 				MODIFY COLUMN log_date_local datetime NULL default '{$default_date}'
 			";
-			$wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+			$wpdb->query($query); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 		}
 		// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
 	}

src/ext/action-scheduler/classes/schema/ActionScheduler_StoreSchema.php

Indentation +65 added lines, -65 removed lines

@@ -8,41 +8,41 @@ 
  * Creates custom tables for storing scheduled actions
  */
 class ActionScheduler_StoreSchema extends ActionScheduler_Abstract_Schema {
-	const ACTIONS_TABLE = 'actionscheduler_actions';
-	const CLAIMS_TABLE  = 'actionscheduler_claims';
-	const GROUPS_TABLE  = 'actionscheduler_groups';
-	const DEFAULT_DATE  = '0000-00-00 00:00:00';
+    const ACTIONS_TABLE = 'actionscheduler_actions';
+    const CLAIMS_TABLE  = 'actionscheduler_claims';
+    const GROUPS_TABLE  = 'actionscheduler_groups';
+    const DEFAULT_DATE  = '0000-00-00 00:00:00';
 
-	/**
-	 * @var int Increment this value to trigger a schema update.
-	 */
-	protected $schema_version = 6;
+    /**
+     * @var int Increment this value to trigger a schema update.
+     */
+    protected $schema_version = 6;
 
-	public function __construct() {
-		$this->tables = array(
-			self::ACTIONS_TABLE,
-			self::CLAIMS_TABLE,
-			self::GROUPS_TABLE,
-		);
-	}
+    public function __construct() {
+        $this->tables = array(
+            self::ACTIONS_TABLE,
+            self::CLAIMS_TABLE,
+            self::GROUPS_TABLE,
+        );
+    }
 
-	/**
-	 * Performs additional setup work required to support this schema.
-	 */
-	public function init() {
-		add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_5_0' ), 10, 2 );
-	}
+    /**
+     * Performs additional setup work required to support this schema.
+     */
+    public function init() {
+        add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_5_0' ), 10, 2 );
+    }
 
-	protected function get_table_definition( $table ) {
-		global $wpdb;
-		$table_name       = $wpdb->$table;
-		$charset_collate  = $wpdb->get_charset_collate();
-		$max_index_length = 191; // @see wp_get_db_schema()
-		$default_date     = self::DEFAULT_DATE;
-		switch ( $table ) {
+    protected function get_table_definition( $table ) {
+        global $wpdb;
+        $table_name       = $wpdb->$table;
+        $charset_collate  = $wpdb->get_charset_collate();
+        $max_index_length = 191; // @see wp_get_db_schema()
+        $default_date     = self::DEFAULT_DATE;
+        switch ( $table ) {
 
-			case self::ACTIONS_TABLE:
-				return "CREATE TABLE {$table_name} (
+            case self::ACTIONS_TABLE:
+                return "CREATE TABLE {$table_name} (
 				        action_id bigint(20) unsigned NOT NULL auto_increment,
 				        hook varchar(191) NOT NULL,
 				        status varchar(20) NOT NULL,
@@ -66,61 +66,61 @@ 
 				        KEY `claim_id_status_scheduled_date_gmt` (`claim_id`, `status`, `scheduled_date_gmt`)
 				        ) $charset_collate";
 
-			case self::CLAIMS_TABLE:
-				return "CREATE TABLE {$table_name} (
+            case self::CLAIMS_TABLE:
+                return "CREATE TABLE {$table_name} (
 				        claim_id bigint(20) unsigned NOT NULL auto_increment,
 				        date_created_gmt datetime NULL default '${default_date}',
 				        PRIMARY KEY  (claim_id),
 				        KEY date_created_gmt (date_created_gmt)
 				        ) $charset_collate";
 
-			case self::GROUPS_TABLE:
-				return "CREATE TABLE {$table_name} (
+            case self::GROUPS_TABLE:
+                return "CREATE TABLE {$table_name} (
 				        group_id bigint(20) unsigned NOT NULL auto_increment,
 				        slug varchar(255) NOT NULL,
 				        PRIMARY KEY  (group_id),
 				        KEY slug (slug($max_index_length))
 				        ) $charset_collate";
 
-			default:
-				return '';
-		}
-	}
+            default:
+                return '';
+        }
+    }
 
-	/**
-	 * Update the actions table schema, allowing datetime fields to be NULL.
-	 *
-	 * This is needed because the NOT NULL constraint causes a conflict with some versions of MySQL
-	 * configured with sql_mode=NO_ZERO_DATE, which can for instance lead to tables not being created.
-	 *
-	 * Most other schema updates happen via ActionScheduler_Abstract_Schema::update_table(), however
-	 * that method relies on dbDelta() and this change is not possible when using that function.
-	 *
-	 * @param string $table Name of table being updated.
-	 * @param string $db_version The existing schema version of the table.
-	 */
-	public function update_schema_5_0( $table, $db_version ) {
-		global $wpdb;
+    /**
+     * Update the actions table schema, allowing datetime fields to be NULL.
+     *
+     * This is needed because the NOT NULL constraint causes a conflict with some versions of MySQL
+     * configured with sql_mode=NO_ZERO_DATE, which can for instance lead to tables not being created.
+     *
+     * Most other schema updates happen via ActionScheduler_Abstract_Schema::update_table(), however
+     * that method relies on dbDelta() and this change is not possible when using that function.
+     *
+     * @param string $table Name of table being updated.
+     * @param string $db_version The existing schema version of the table.
+     */
+    public function update_schema_5_0( $table, $db_version ) {
+        global $wpdb;
 
-		if ( 'actionscheduler_actions' !== $table || version_compare( $db_version, '5', '>=' ) ) {
-			return;
-		}
+        if ( 'actionscheduler_actions' !== $table || version_compare( $db_version, '5', '>=' ) ) {
+            return;
+        }
 
-		// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-		$table_name   = $wpdb->prefix . 'actionscheduler_actions';
-		$table_list   = $wpdb->get_col( "SHOW TABLES LIKE '${table_name}'" );
-		$default_date = self::DEFAULT_DATE;
+        // phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
+        $table_name   = $wpdb->prefix . 'actionscheduler_actions';
+        $table_list   = $wpdb->get_col( "SHOW TABLES LIKE '${table_name}'" );
+        $default_date = self::DEFAULT_DATE;
 
-		if ( ! empty( $table_list ) ) {
-			$query = "
+        if ( ! empty( $table_list ) ) {
+            $query = "
 				ALTER TABLE ${table_name}
 				MODIFY COLUMN scheduled_date_gmt datetime NULL default '${default_date}',
 				MODIFY COLUMN scheduled_date_local datetime NULL default '${default_date}',
 				MODIFY COLUMN last_attempt_gmt datetime NULL default '${default_date}',
 				MODIFY COLUMN last_attempt_local datetime NULL default '${default_date}'
 		";
-			$wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
-		}
-		// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-	}
+            $wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+        }
+        // phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
+    }
 }

Spacing +9 added lines, -9 removed lines

@@ -30,16 +30,16 @@ 
 	 * Performs additional setup work required to support this schema.
 	 */
 	public function init() {
-		add_action( 'action_scheduler_before_schema_update', array( $this, 'update_schema_5_0' ), 10, 2 );
+		add_action('action_scheduler_before_schema_update', array($this, 'update_schema_5_0'), 10, 2);
 	}
 
-	protected function get_table_definition( $table ) {
+	protected function get_table_definition($table) {
 		global $wpdb;
 		$table_name       = $wpdb->$table;
 		$charset_collate  = $wpdb->get_charset_collate();
 		$max_index_length = 191; // @see wp_get_db_schema()
 		$default_date     = self::DEFAULT_DATE;
-		switch ( $table ) {
+		switch ($table) {
 
 			case self::ACTIONS_TABLE:
 				return "CREATE TABLE {$table_name} (
@@ -99,19 +99,19 @@ 
 	 * @param string $table Name of table being updated.
 	 * @param string $db_version The existing schema version of the table.
 	 */
-	public function update_schema_5_0( $table, $db_version ) {
+	public function update_schema_5_0($table, $db_version) {
 		global $wpdb;
 
-		if ( 'actionscheduler_actions' !== $table || version_compare( $db_version, '5', '>=' ) ) {
+		if ('actionscheduler_actions' !== $table || version_compare($db_version, '5', '>=')) {
 			return;
 		}
 
 		// phpcs:disable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-		$table_name   = $wpdb->prefix . 'actionscheduler_actions';
-		$table_list   = $wpdb->get_col( "SHOW TABLES LIKE '${table_name}'" );
+		$table_name   = $wpdb->prefix.'actionscheduler_actions';
+		$table_list   = $wpdb->get_col("SHOW TABLES LIKE '${table_name}'");
 		$default_date = self::DEFAULT_DATE;
 
-		if ( ! empty( $table_list ) ) {
+		if ( ! empty($table_list)) {
 			$query = "
 				ALTER TABLE ${table_name}
 				MODIFY COLUMN scheduled_date_gmt datetime NULL default '${default_date}',
@@ -119,7 +119,7 @@ 
 				MODIFY COLUMN last_attempt_gmt datetime NULL default '${default_date}',
 				MODIFY COLUMN last_attempt_local datetime NULL default '${default_date}'
 		";
-			$wpdb->query( $query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+			$wpdb->query($query); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 		}
 		// phpcs:enable WordPress.DB.PreparedSQL.InterpolatedNotPrepared
 	}

src/ext/action-scheduler/classes/ActionScheduler_Compatibility.php

Indentation +87 added lines, -87 removed lines

@@ -5,105 +5,105 @@
  */
 class ActionScheduler_Compatibility {
 
-	/**
-	 * Converts a shorthand byte value to an integer byte value.
-	 *
-	 * Wrapper for wp_convert_hr_to_bytes(), moved to load.php in WordPress 4.6 from media.php
-	 *
-	 * @link https://secure.php.net/manual/en/function.ini-get.php
-	 * @link https://secure.php.net/manual/en/faq.using.php#faq.using.shorthandbytes
-	 *
-	 * @param string $value A (PHP ini) byte value, either shorthand or ordinary.
-	 * @return int An integer byte value.
-	 */
-	public static function convert_hr_to_bytes( $value ) {
-		if ( function_exists( 'wp_convert_hr_to_bytes' ) ) {
-			return wp_convert_hr_to_bytes( $value );
-		}
+    /**
+     * Converts a shorthand byte value to an integer byte value.
+     *
+     * Wrapper for wp_convert_hr_to_bytes(), moved to load.php in WordPress 4.6 from media.php
+     *
+     * @link https://secure.php.net/manual/en/function.ini-get.php
+     * @link https://secure.php.net/manual/en/faq.using.php#faq.using.shorthandbytes
+     *
+     * @param string $value A (PHP ini) byte value, either shorthand or ordinary.
+     * @return int An integer byte value.
+     */
+    public static function convert_hr_to_bytes( $value ) {
+        if ( function_exists( 'wp_convert_hr_to_bytes' ) ) {
+            return wp_convert_hr_to_bytes( $value );
+        }
 
-		$value = strtolower( trim( $value ) );
-		$bytes = (int) $value;
+        $value = strtolower( trim( $value ) );
+        $bytes = (int) $value;
 
-		if ( false !== strpos( $value, 'g' ) ) {
-			$bytes *= GB_IN_BYTES;
-		} elseif ( false !== strpos( $value, 'm' ) ) {
-			$bytes *= MB_IN_BYTES;
-		} elseif ( false !== strpos( $value, 'k' ) ) {
-			$bytes *= KB_IN_BYTES;
-		}
+        if ( false !== strpos( $value, 'g' ) ) {
+            $bytes *= GB_IN_BYTES;
+        } elseif ( false !== strpos( $value, 'm' ) ) {
+            $bytes *= MB_IN_BYTES;
+        } elseif ( false !== strpos( $value, 'k' ) ) {
+            $bytes *= KB_IN_BYTES;
+        }
 
-		// Deal with large (float) values which run into the maximum integer size.
-		return min( $bytes, PHP_INT_MAX );
-	}
+        // Deal with large (float) values which run into the maximum integer size.
+        return min( $bytes, PHP_INT_MAX );
+    }
 
-	/**
-	 * Attempts to raise the PHP memory limit for memory intensive processes.
-	 *
-	 * Only allows raising the existing limit and prevents lowering it.
-	 *
-	 * Wrapper for wp_raise_memory_limit(), added in WordPress v4.6.0
-	 *
-	 * @return bool|int|string The limit that was set or false on failure.
-	 */
-	public static function raise_memory_limit() {
-		if ( function_exists( 'wp_raise_memory_limit' ) ) {
-			return wp_raise_memory_limit( 'admin' );
-		}
+    /**
+     * Attempts to raise the PHP memory limit for memory intensive processes.
+     *
+     * Only allows raising the existing limit and prevents lowering it.
+     *
+     * Wrapper for wp_raise_memory_limit(), added in WordPress v4.6.0
+     *
+     * @return bool|int|string The limit that was set or false on failure.
+     */
+    public static function raise_memory_limit() {
+        if ( function_exists( 'wp_raise_memory_limit' ) ) {
+            return wp_raise_memory_limit( 'admin' );
+        }
 
-		$current_limit     = @ini_get( 'memory_limit' );
-		$current_limit_int = self::convert_hr_to_bytes( $current_limit );
+        $current_limit     = @ini_get( 'memory_limit' );
+        $current_limit_int = self::convert_hr_to_bytes( $current_limit );
 
-		if ( -1 === $current_limit_int ) {
-			return false;
-		}
+        if ( -1 === $current_limit_int ) {
+            return false;
+        }
 
-		$wp_max_limit       = WP_MAX_MEMORY_LIMIT;
-		$wp_max_limit_int   = self::convert_hr_to_bytes( $wp_max_limit );
-		$filtered_limit     = apply_filters( 'admin_memory_limit', $wp_max_limit );
-		$filtered_limit_int = self::convert_hr_to_bytes( $filtered_limit );
+        $wp_max_limit       = WP_MAX_MEMORY_LIMIT;
+        $wp_max_limit_int   = self::convert_hr_to_bytes( $wp_max_limit );
+        $filtered_limit     = apply_filters( 'admin_memory_limit', $wp_max_limit );
+        $filtered_limit_int = self::convert_hr_to_bytes( $filtered_limit );
 
-		if ( -1 === $filtered_limit_int || ( $filtered_limit_int > $wp_max_limit_int && $filtered_limit_int > $current_limit_int ) ) {
-			if ( false !== @ini_set( 'memory_limit', $filtered_limit ) ) {
-				return $filtered_limit;
-			} else {
-				return false;
-			}
-		} elseif ( -1 === $wp_max_limit_int || $wp_max_limit_int > $current_limit_int ) {
-			if ( false !== @ini_set( 'memory_limit', $wp_max_limit ) ) {
-				return $wp_max_limit;
-			} else {
-				return false;
-			}
-		}
-		return false;
-	}
+        if ( -1 === $filtered_limit_int || ( $filtered_limit_int > $wp_max_limit_int && $filtered_limit_int > $current_limit_int ) ) {
+            if ( false !== @ini_set( 'memory_limit', $filtered_limit ) ) {
+                return $filtered_limit;
+            } else {
+                return false;
+            }
+        } elseif ( -1 === $wp_max_limit_int || $wp_max_limit_int > $current_limit_int ) {
+            if ( false !== @ini_set( 'memory_limit', $wp_max_limit ) ) {
+                return $wp_max_limit;
+            } else {
+                return false;
+            }
+        }
+        return false;
+    }
 
-	/**
-	 * Attempts to raise the PHP timeout for time intensive processes.
-	 *
-	 * Only allows raising the existing limit and prevents lowering it. Wrapper for wc_set_time_limit(), when available.
-	 *
-	 * @param int $limit The time limit in seconds.
-	 */
-	public static function raise_time_limit( $limit = 0 ) {
-		$limit              = (int) $limit;
-		$max_execution_time = (int) ini_get( 'max_execution_time' );
+    /**
+     * Attempts to raise the PHP timeout for time intensive processes.
+     *
+     * Only allows raising the existing limit and prevents lowering it. Wrapper for wc_set_time_limit(), when available.
+     *
+     * @param int $limit The time limit in seconds.
+     */
+    public static function raise_time_limit( $limit = 0 ) {
+        $limit              = (int) $limit;
+        $max_execution_time = (int) ini_get( 'max_execution_time' );
 
-		/*
+        /*
 		 * If the max execution time is already unlimited (zero), or if it exceeds or is equal to the proposed
 		 * limit, there is no reason for us to make further changes (we never want to lower it).
 		 */
-		if (
-			0 === $max_execution_time
-			|| ( $max_execution_time >= $limit && $limit !== 0 )
-		) {
-			return;
-		}
+        if (
+            0 === $max_execution_time
+            || ( $max_execution_time >= $limit && $limit !== 0 )
+        ) {
+            return;
+        }
 
-		if ( function_exists( 'wc_set_time_limit' ) ) {
-			wc_set_time_limit( $limit );
-		} elseif ( function_exists( 'set_time_limit' ) && false === strpos( ini_get( 'disable_functions' ), 'set_time_limit' ) && ! ini_get( 'safe_mode' ) ) { // phpcs:ignore PHPCompatibility.IniDirectives.RemovedIniDirectives.safe_modeDeprecatedRemoved
-			@set_time_limit( $limit ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
-		}
-	}
+        if ( function_exists( 'wc_set_time_limit' ) ) {
+            wc_set_time_limit( $limit );
+        } elseif ( function_exists( 'set_time_limit' ) && false === strpos( ini_get( 'disable_functions' ), 'set_time_limit' ) && ! ini_get( 'safe_mode' ) ) { // phpcs:ignore PHPCompatibility.IniDirectives.RemovedIniDirectives.safe_modeDeprecatedRemoved
+            @set_time_limit( $limit ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
+        }
+    }
 }

Spacing +25 added lines, -25 removed lines

@@ -16,24 +16,24 @@ 
 	 * @param string $value A (PHP ini) byte value, either shorthand or ordinary.
 	 * @return int An integer byte value.
 	 */
-	public static function convert_hr_to_bytes( $value ) {
-		if ( function_exists( 'wp_convert_hr_to_bytes' ) ) {
-			return wp_convert_hr_to_bytes( $value );
+	public static function convert_hr_to_bytes($value) {
+		if (function_exists('wp_convert_hr_to_bytes')) {
+			return wp_convert_hr_to_bytes($value);
 		}
 
-		$value = strtolower( trim( $value ) );
+		$value = strtolower(trim($value));
 		$bytes = (int) $value;
 
-		if ( false !== strpos( $value, 'g' ) ) {
+		if (false !== strpos($value, 'g')) {
 			$bytes *= GB_IN_BYTES;
-		} elseif ( false !== strpos( $value, 'm' ) ) {
+		} elseif (false !== strpos($value, 'm')) {
 			$bytes *= MB_IN_BYTES;
-		} elseif ( false !== strpos( $value, 'k' ) ) {
+		} elseif (false !== strpos($value, 'k')) {
 			$bytes *= KB_IN_BYTES;
 		}
 
 		// Deal with large (float) values which run into the maximum integer size.
-		return min( $bytes, PHP_INT_MAX );
+		return min($bytes, PHP_INT_MAX);
 	}
 
 	/**
@@ -46,30 +46,30 @@ 
 	 * @return bool|int|string The limit that was set or false on failure.
 	 */
 	public static function raise_memory_limit() {
-		if ( function_exists( 'wp_raise_memory_limit' ) ) {
-			return wp_raise_memory_limit( 'admin' );
+		if (function_exists('wp_raise_memory_limit')) {
+			return wp_raise_memory_limit('admin');
 		}
 
-		$current_limit     = @ini_get( 'memory_limit' );
-		$current_limit_int = self::convert_hr_to_bytes( $current_limit );
+		$current_limit     = @ini_get('memory_limit');
+		$current_limit_int = self::convert_hr_to_bytes($current_limit);
 
 		if ( -1 === $current_limit_int ) {
 			return false;
 		}
 
 		$wp_max_limit       = WP_MAX_MEMORY_LIMIT;
-		$wp_max_limit_int   = self::convert_hr_to_bytes( $wp_max_limit );
-		$filtered_limit     = apply_filters( 'admin_memory_limit', $wp_max_limit );
-		$filtered_limit_int = self::convert_hr_to_bytes( $filtered_limit );
+		$wp_max_limit_int   = self::convert_hr_to_bytes($wp_max_limit);
+		$filtered_limit     = apply_filters('admin_memory_limit', $wp_max_limit);
+		$filtered_limit_int = self::convert_hr_to_bytes($filtered_limit);
 
-		if ( -1 === $filtered_limit_int || ( $filtered_limit_int > $wp_max_limit_int && $filtered_limit_int > $current_limit_int ) ) {
-			if ( false !== @ini_set( 'memory_limit', $filtered_limit ) ) {
+		if ( -1 === $filtered_limit_int || ($filtered_limit_int > $wp_max_limit_int && $filtered_limit_int > $current_limit_int) ) {
+			if (false !== @ini_set('memory_limit', $filtered_limit)) {
 				return $filtered_limit;
 			} else {
 				return false;
 			}
 		} elseif ( -1 === $wp_max_limit_int || $wp_max_limit_int > $current_limit_int ) {
-			if ( false !== @ini_set( 'memory_limit', $wp_max_limit ) ) {
+			if (false !== @ini_set('memory_limit', $wp_max_limit)) {
 				return $wp_max_limit;
 			} else {
 				return false;
@@ -85,9 +85,9 @@ 
 	 *
 	 * @param int $limit The time limit in seconds.
 	 */
-	public static function raise_time_limit( $limit = 0 ) {
+	public static function raise_time_limit($limit = 0) {
 		$limit              = (int) $limit;
-		$max_execution_time = (int) ini_get( 'max_execution_time' );
+		$max_execution_time = (int) ini_get('max_execution_time');
 
 		/*
 		 * If the max execution time is already unlimited (zero), or if it exceeds or is equal to the proposed
@@ -95,15 +95,15 @@ 
 		 */
 		if (
 			0 === $max_execution_time
-			|| ( $max_execution_time >= $limit && $limit !== 0 )
+			|| ($max_execution_time >= $limit && $limit !== 0)
 		) {
 			return;
 		}
 
-		if ( function_exists( 'wc_set_time_limit' ) ) {
-			wc_set_time_limit( $limit );
-		} elseif ( function_exists( 'set_time_limit' ) && false === strpos( ini_get( 'disable_functions' ), 'set_time_limit' ) && ! ini_get( 'safe_mode' ) ) { // phpcs:ignore PHPCompatibility.IniDirectives.RemovedIniDirectives.safe_modeDeprecatedRemoved
-			@set_time_limit( $limit ); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
+		if (function_exists('wc_set_time_limit')) {
+			wc_set_time_limit($limit);
+		} elseif (function_exists('set_time_limit') && false === strpos(ini_get('disable_functions'), 'set_time_limit') && ! ini_get('safe_mode')) { // phpcs:ignore PHPCompatibility.IniDirectives.RemovedIniDirectives.safe_modeDeprecatedRemoved
+			@set_time_limit($limit); // phpcs:ignore WordPress.PHP.NoSilencedErrors.Discouraged
 		}
 	}
 }

src/ext/action-scheduler/classes/ActionScheduler_QueueRunner.php

Indentation +207 added lines, -207 removed lines

@@ -4,214 +4,214 @@
  * Class ActionScheduler_QueueRunner
  */
 class ActionScheduler_QueueRunner extends ActionScheduler_Abstract_QueueRunner {
-	const WP_CRON_HOOK = 'action_scheduler_run_queue';
-
-	const WP_CRON_SCHEDULE = 'every_minute';
-
-	/** @var ActionScheduler_AsyncRequest_QueueRunner */
-	protected $async_request;
-
-	/** @var ActionScheduler_QueueRunner  */
-	private static $runner = null;
-
-	/**
-	 * @return ActionScheduler_QueueRunner
-	 * @codeCoverageIgnore
-	 */
-	public static function instance() {
-		if ( empty( self::$runner ) ) {
-			$class        = apply_filters( 'action_scheduler_queue_runner_class', 'ActionScheduler_QueueRunner' );
-			self::$runner = new $class();
-		}
-		return self::$runner;
-	}
-
-	/**
-	 * ActionScheduler_QueueRunner constructor.
-	 *
-	 * @param ActionScheduler_Store             $store
-	 * @param ActionScheduler_FatalErrorMonitor $monitor
-	 * @param ActionScheduler_QueueCleaner      $cleaner
-	 */
-	public function __construct( ActionScheduler_Store $store = null, ActionScheduler_FatalErrorMonitor $monitor = null, ActionScheduler_QueueCleaner $cleaner = null, ActionScheduler_AsyncRequest_QueueRunner $async_request = null ) {
-		parent::__construct( $store, $monitor, $cleaner );
-
-		if ( $async_request === null ) {
-			$async_request = new ActionScheduler_AsyncRequest_QueueRunner( $this->store );
-		}
-
-		$this->async_request = $async_request;
-	}
-
-	/**
-	 * @codeCoverageIgnore
-	 */
-	public function init() {
-
-		add_filter( 'cron_schedules', array( self::instance(), 'add_wp_cron_schedule' ) );
-
-		// Check for and remove any WP Cron hook scheduled by Action Scheduler < 3.0.0, which didn't include the $context param
-		$next_timestamp = wp_next_scheduled( self::WP_CRON_HOOK );
-		if ( $next_timestamp ) {
-			wp_unschedule_event( $next_timestamp, self::WP_CRON_HOOK );
-		}
-
-		$cron_context = array( 'WP Cron' );
-
-		if ( ! wp_next_scheduled( self::WP_CRON_HOOK, $cron_context ) ) {
-			$schedule = apply_filters( 'action_scheduler_run_schedule', self::WP_CRON_SCHEDULE );
-			wp_schedule_event( time(), $schedule, self::WP_CRON_HOOK, $cron_context );
-		}
-
-		add_action( self::WP_CRON_HOOK, array( self::instance(), 'run' ) );
-		$this->hook_dispatch_async_request();
-	}
-
-	/**
-	 * Hook check for dispatching an async request.
-	 */
-	public function hook_dispatch_async_request() {
-		add_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
-	}
-
-	/**
-	 * Unhook check for dispatching an async request.
-	 */
-	public function unhook_dispatch_async_request() {
-		remove_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
-	}
-
-	/**
-	 * Check if we should dispatch an async request to process actions.
-	 *
-	 * This method is attached to 'shutdown', so is called frequently. To avoid slowing down
-	 * the site, it mitigates the work performed in each request by:
-	 * 1. checking if it's in the admin context and then
-	 * 2. haven't run on the 'shutdown' hook within the lock time (60 seconds by default)
-	 * 3. haven't exceeded the number of allowed batches.
-	 *
-	 * The order of these checks is important, because they run from a check on a value:
-	 * 1. in memory - is_admin() maps to $GLOBALS or the WP_ADMIN constant
-	 * 2. in memory - transients use autoloaded options by default
-	 * 3. from a database query - has_maximum_concurrent_batches() run the query
-	 *    $this->store->get_claim_count() to find the current number of claims in the DB.
-	 *
-	 * If all of these conditions are met, then we request an async runner check whether it
-	 * should dispatch a request to process pending actions.
-	 */
-	public function maybe_dispatch_async_request() {
-		if ( is_admin() && ! ActionScheduler::lock()->is_locked( 'async-request-runner' ) ) {
-			// Only start an async queue at most once every 60 seconds
-			ActionScheduler::lock()->set( 'async-request-runner' );
-			$this->async_request->maybe_dispatch();
-		}
-	}
-
-	/**
-	 * Process actions in the queue. Attached to self::WP_CRON_HOOK i.e. 'action_scheduler_run_queue'
-	 *
-	 * The $context param of this method defaults to 'WP Cron', because prior to Action Scheduler 3.0.0
-	 * that was the only context in which this method was run, and the self::WP_CRON_HOOK hook had no context
-	 * passed along with it. New code calling this method directly, or by triggering the self::WP_CRON_HOOK,
-	 * should set a context as the first parameter. For an example of this, refer to the code seen in
-	 *
-	 * @see ActionScheduler_AsyncRequest_QueueRunner::handle()
-	 *
-	 * @param string $context Optional identifer for the context in which this action is being processed, e.g. 'WP CLI' or 'WP Cron'
-	 *        Generally, this should be capitalised and not localised as it's a proper noun.
-	 * @return int The number of actions processed.
-	 */
-	public function run( $context = 'WP Cron' ) {
-		ActionScheduler_Compatibility::raise_memory_limit();
-		ActionScheduler_Compatibility::raise_time_limit( $this->get_time_limit() );
-		do_action( 'action_scheduler_before_process_queue' );
-		$this->run_cleanup();
-		$processed_actions = 0;
-		if ( false === $this->has_maximum_concurrent_batches() ) {
-			$batch_size = apply_filters( 'action_scheduler_queue_runner_batch_size', 25 );
-			do {
-				$processed_actions_in_batch = $this->do_batch( $batch_size, $context );
-				$processed_actions         += $processed_actions_in_batch;
-			} while ( $processed_actions_in_batch > 0 && ! $this->batch_limits_exceeded( $processed_actions ) ); // keep going until we run out of actions, time, or memory
-		}
-
-		do_action( 'action_scheduler_after_process_queue' );
-		return $processed_actions;
-	}
-
-	/**
-	 * Process a batch of actions pending in the queue.
-	 *
-	 * Actions are processed by claiming a set of pending actions then processing each one until either the batch
-	 * size is completed, or memory or time limits are reached, defined by @see $this->batch_limits_exceeded().
-	 *
-	 * @param int    $size The maximum number of actions to process in the batch.
-	 * @param string $context Optional identifer for the context in which this action is being processed, e.g. 'WP CLI' or 'WP Cron'
-	 *        Generally, this should be capitalised and not localised as it's a proper noun.
-	 * @return int The number of actions processed.
-	 */
-	protected function do_batch( $size = 100, $context = '' ) {
-		$claim = $this->store->stake_claim( $size );
-		$this->monitor->attach( $claim );
-		$processed_actions = 0;
-
-		foreach ( $claim->get_actions() as $action_id ) {
-			// bail if we lost the claim
-			if ( ! in_array( $action_id, $this->store->find_actions_by_claim_id( $claim->get_id() ) ) ) {
-				break;
-			}
-			$this->process_action( $action_id, $context );
-			$processed_actions++;
-
-			if ( $this->batch_limits_exceeded( $processed_actions ) ) {
-				break;
-			}
-		}
-		$this->store->release_claim( $claim );
-		$this->monitor->detach();
-		$this->clear_caches();
-		return $processed_actions;
-	}
-
-	/**
-	 * Flush the cache if possible (intended for use after a batch of actions has been processed).
-	 *
-	 * This is useful because running large batches can eat up memory and because invalid data can accrue in the
-	 * runtime cache, which may lead to unexpected results.
-	 */
-	protected function clear_caches() {
-		/*
+    const WP_CRON_HOOK = 'action_scheduler_run_queue';
+
+    const WP_CRON_SCHEDULE = 'every_minute';
+
+    /** @var ActionScheduler_AsyncRequest_QueueRunner */
+    protected $async_request;
+
+    /** @var ActionScheduler_QueueRunner  */
+    private static $runner = null;
+
+    /**
+     * @return ActionScheduler_QueueRunner
+     * @codeCoverageIgnore
+     */
+    public static function instance() {
+        if ( empty( self::$runner ) ) {
+            $class        = apply_filters( 'action_scheduler_queue_runner_class', 'ActionScheduler_QueueRunner' );
+            self::$runner = new $class();
+        }
+        return self::$runner;
+    }
+
+    /**
+     * ActionScheduler_QueueRunner constructor.
+     *
+     * @param ActionScheduler_Store             $store
+     * @param ActionScheduler_FatalErrorMonitor $monitor
+     * @param ActionScheduler_QueueCleaner      $cleaner
+     */
+    public function __construct( ActionScheduler_Store $store = null, ActionScheduler_FatalErrorMonitor $monitor = null, ActionScheduler_QueueCleaner $cleaner = null, ActionScheduler_AsyncRequest_QueueRunner $async_request = null ) {
+        parent::__construct( $store, $monitor, $cleaner );
+
+        if ( $async_request === null ) {
+            $async_request = new ActionScheduler_AsyncRequest_QueueRunner( $this->store );
+        }
+
+        $this->async_request = $async_request;
+    }
+
+    /**
+     * @codeCoverageIgnore
+     */
+    public function init() {
+
+        add_filter( 'cron_schedules', array( self::instance(), 'add_wp_cron_schedule' ) );
+
+        // Check for and remove any WP Cron hook scheduled by Action Scheduler < 3.0.0, which didn't include the $context param
+        $next_timestamp = wp_next_scheduled( self::WP_CRON_HOOK );
+        if ( $next_timestamp ) {
+            wp_unschedule_event( $next_timestamp, self::WP_CRON_HOOK );
+        }
+
+        $cron_context = array( 'WP Cron' );
+
+        if ( ! wp_next_scheduled( self::WP_CRON_HOOK, $cron_context ) ) {
+            $schedule = apply_filters( 'action_scheduler_run_schedule', self::WP_CRON_SCHEDULE );
+            wp_schedule_event( time(), $schedule, self::WP_CRON_HOOK, $cron_context );
+        }
+
+        add_action( self::WP_CRON_HOOK, array( self::instance(), 'run' ) );
+        $this->hook_dispatch_async_request();
+    }
+
+    /**
+     * Hook check for dispatching an async request.
+     */
+    public function hook_dispatch_async_request() {
+        add_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
+    }
+
+    /**
+     * Unhook check for dispatching an async request.
+     */
+    public function unhook_dispatch_async_request() {
+        remove_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
+    }
+
+    /**
+     * Check if we should dispatch an async request to process actions.
+     *
+     * This method is attached to 'shutdown', so is called frequently. To avoid slowing down
+     * the site, it mitigates the work performed in each request by:
+     * 1. checking if it's in the admin context and then
+     * 2. haven't run on the 'shutdown' hook within the lock time (60 seconds by default)
+     * 3. haven't exceeded the number of allowed batches.
+     *
+     * The order of these checks is important, because they run from a check on a value:
+     * 1. in memory - is_admin() maps to $GLOBALS or the WP_ADMIN constant
+     * 2. in memory - transients use autoloaded options by default
+     * 3. from a database query - has_maximum_concurrent_batches() run the query
+     *    $this->store->get_claim_count() to find the current number of claims in the DB.
+     *
+     * If all of these conditions are met, then we request an async runner check whether it
+     * should dispatch a request to process pending actions.
+     */
+    public function maybe_dispatch_async_request() {
+        if ( is_admin() && ! ActionScheduler::lock()->is_locked( 'async-request-runner' ) ) {
+            // Only start an async queue at most once every 60 seconds
+            ActionScheduler::lock()->set( 'async-request-runner' );
+            $this->async_request->maybe_dispatch();
+        }
+    }
+
+    /**
+     * Process actions in the queue. Attached to self::WP_CRON_HOOK i.e. 'action_scheduler_run_queue'
+     *
+     * The $context param of this method defaults to 'WP Cron', because prior to Action Scheduler 3.0.0
+     * that was the only context in which this method was run, and the self::WP_CRON_HOOK hook had no context
+     * passed along with it. New code calling this method directly, or by triggering the self::WP_CRON_HOOK,
+     * should set a context as the first parameter. For an example of this, refer to the code seen in
+     *
+     * @see ActionScheduler_AsyncRequest_QueueRunner::handle()
+     *
+     * @param string $context Optional identifer for the context in which this action is being processed, e.g. 'WP CLI' or 'WP Cron'
+     *        Generally, this should be capitalised and not localised as it's a proper noun.
+     * @return int The number of actions processed.
+     */
+    public function run( $context = 'WP Cron' ) {
+        ActionScheduler_Compatibility::raise_memory_limit();
+        ActionScheduler_Compatibility::raise_time_limit( $this->get_time_limit() );
+        do_action( 'action_scheduler_before_process_queue' );
+        $this->run_cleanup();
+        $processed_actions = 0;
+        if ( false === $this->has_maximum_concurrent_batches() ) {
+            $batch_size = apply_filters( 'action_scheduler_queue_runner_batch_size', 25 );
+            do {
+                $processed_actions_in_batch = $this->do_batch( $batch_size, $context );
+                $processed_actions         += $processed_actions_in_batch;
+            } while ( $processed_actions_in_batch > 0 && ! $this->batch_limits_exceeded( $processed_actions ) ); // keep going until we run out of actions, time, or memory
+        }
+
+        do_action( 'action_scheduler_after_process_queue' );
+        return $processed_actions;
+    }
+
+    /**
+     * Process a batch of actions pending in the queue.
+     *
+     * Actions are processed by claiming a set of pending actions then processing each one until either the batch
+     * size is completed, or memory or time limits are reached, defined by @see $this->batch_limits_exceeded().
+     *
+     * @param int    $size The maximum number of actions to process in the batch.
+     * @param string $context Optional identifer for the context in which this action is being processed, e.g. 'WP CLI' or 'WP Cron'
+     *        Generally, this should be capitalised and not localised as it's a proper noun.
+     * @return int The number of actions processed.
+     */
+    protected function do_batch( $size = 100, $context = '' ) {
+        $claim = $this->store->stake_claim( $size );
+        $this->monitor->attach( $claim );
+        $processed_actions = 0;
+
+        foreach ( $claim->get_actions() as $action_id ) {
+            // bail if we lost the claim
+            if ( ! in_array( $action_id, $this->store->find_actions_by_claim_id( $claim->get_id() ) ) ) {
+                break;
+            }
+            $this->process_action( $action_id, $context );
+            $processed_actions++;
+
+            if ( $this->batch_limits_exceeded( $processed_actions ) ) {
+                break;
+            }
+        }
+        $this->store->release_claim( $claim );
+        $this->monitor->detach();
+        $this->clear_caches();
+        return $processed_actions;
+    }
+
+    /**
+     * Flush the cache if possible (intended for use after a batch of actions has been processed).
+     *
+     * This is useful because running large batches can eat up memory and because invalid data can accrue in the
+     * runtime cache, which may lead to unexpected results.
+     */
+    protected function clear_caches() {
+        /*
 		 * Calling wp_cache_flush_runtime() lets us clear the runtime cache without invalidating the external object
 		 * cache, so we will always prefer this when it is available (but it was only introduced in WordPress 6.0).
 		 */
-		if ( function_exists( 'wp_cache_flush_runtime' ) ) {
-			wp_cache_flush_runtime();
-		} elseif (
-			! wp_using_ext_object_cache()
-			/**
-			 * When an external object cache is in use, and when wp_cache_flush_runtime() is not available, then
-			 * normally the cache will not be flushed after processing a batch of actions (to avoid a performance
-			 * penalty for other processes).
-			 *
-			 * This filter makes it possible to override this behavior and always flush the cache, even if an external
-			 * object cache is in use.
-			 *
-			 * @since 1.0
-			 *
-			 * @param bool $flush_cache If the cache should be flushed.
-			 */
-			|| apply_filters( 'action_scheduler_queue_runner_flush_cache', false )
-		) {
-			wp_cache_flush();
-		}
-	}
-
-	public function add_wp_cron_schedule( $schedules ) {
-		$schedules['every_minute'] = array(
-			'interval' => 60, // in seconds
-			'display'  => __( 'Every minute', 'action-scheduler' ),
-		);
-
-		return $schedules;
-	}
+        if ( function_exists( 'wp_cache_flush_runtime' ) ) {
+            wp_cache_flush_runtime();
+        } elseif (
+            ! wp_using_ext_object_cache()
+            /**
+             * When an external object cache is in use, and when wp_cache_flush_runtime() is not available, then
+             * normally the cache will not be flushed after processing a batch of actions (to avoid a performance
+             * penalty for other processes).
+             *
+             * This filter makes it possible to override this behavior and always flush the cache, even if an external
+             * object cache is in use.
+             *
+             * @since 1.0
+             *
+             * @param bool $flush_cache If the cache should be flushed.
+             */
+            || apply_filters( 'action_scheduler_queue_runner_flush_cache', false )
+        ) {
+            wp_cache_flush();
+        }
+    }
+
+    public function add_wp_cron_schedule( $schedules ) {
+        $schedules['every_minute'] = array(
+            'interval' => 60, // in seconds
+            'display'  => __( 'Every minute', 'action-scheduler' ),
+        );
+
+        return $schedules;
+    }
 }

Spacing +39 added lines, -39 removed lines

@@ -19,8 +19,8 @@ 
 	 * @codeCoverageIgnore
 	 */
 	public static function instance() {
-		if ( empty( self::$runner ) ) {
-			$class        = apply_filters( 'action_scheduler_queue_runner_class', 'ActionScheduler_QueueRunner' );
+		if (empty(self::$runner)) {
+			$class        = apply_filters('action_scheduler_queue_runner_class', 'ActionScheduler_QueueRunner');
 			self::$runner = new $class();
 		}
 		return self::$runner;
@@ -33,11 +33,11 @@ 
 	 * @param ActionScheduler_FatalErrorMonitor $monitor
 	 * @param ActionScheduler_QueueCleaner      $cleaner
 	 */
-	public function __construct( ActionScheduler_Store $store = null, ActionScheduler_FatalErrorMonitor $monitor = null, ActionScheduler_QueueCleaner $cleaner = null, ActionScheduler_AsyncRequest_QueueRunner $async_request = null ) {
-		parent::__construct( $store, $monitor, $cleaner );
+	public function __construct(ActionScheduler_Store $store = null, ActionScheduler_FatalErrorMonitor $monitor = null, ActionScheduler_QueueCleaner $cleaner = null, ActionScheduler_AsyncRequest_QueueRunner $async_request = null) {
+		parent::__construct($store, $monitor, $cleaner);
 
-		if ( $async_request === null ) {
-			$async_request = new ActionScheduler_AsyncRequest_QueueRunner( $this->store );
+		if ($async_request === null) {
+			$async_request = new ActionScheduler_AsyncRequest_QueueRunner($this->store);
 		}
 
 		$this->async_request = $async_request;
@@ -48,22 +48,22 @@ 
 	 */
 	public function init() {
 
-		add_filter( 'cron_schedules', array( self::instance(), 'add_wp_cron_schedule' ) );
+		add_filter('cron_schedules', array(self::instance(), 'add_wp_cron_schedule'));
 
 		// Check for and remove any WP Cron hook scheduled by Action Scheduler < 3.0.0, which didn't include the $context param
-		$next_timestamp = wp_next_scheduled( self::WP_CRON_HOOK );
-		if ( $next_timestamp ) {
-			wp_unschedule_event( $next_timestamp, self::WP_CRON_HOOK );
+		$next_timestamp = wp_next_scheduled(self::WP_CRON_HOOK);
+		if ($next_timestamp) {
+			wp_unschedule_event($next_timestamp, self::WP_CRON_HOOK);
 		}
 
-		$cron_context = array( 'WP Cron' );
+		$cron_context = array('WP Cron');
 
-		if ( ! wp_next_scheduled( self::WP_CRON_HOOK, $cron_context ) ) {
-			$schedule = apply_filters( 'action_scheduler_run_schedule', self::WP_CRON_SCHEDULE );
-			wp_schedule_event( time(), $schedule, self::WP_CRON_HOOK, $cron_context );
+		if ( ! wp_next_scheduled(self::WP_CRON_HOOK, $cron_context)) {
+			$schedule = apply_filters('action_scheduler_run_schedule', self::WP_CRON_SCHEDULE);
+			wp_schedule_event(time(), $schedule, self::WP_CRON_HOOK, $cron_context);
 		}
 
-		add_action( self::WP_CRON_HOOK, array( self::instance(), 'run' ) );
+		add_action(self::WP_CRON_HOOK, array(self::instance(), 'run'));
 		$this->hook_dispatch_async_request();
 	}
 
@@ -71,14 +71,14 @@ 
 	 * Hook check for dispatching an async request.
 	 */
 	public function hook_dispatch_async_request() {
-		add_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
+		add_action('shutdown', array($this, 'maybe_dispatch_async_request'));
 	}
 
 	/**
 	 * Unhook check for dispatching an async request.
 	 */
 	public function unhook_dispatch_async_request() {
-		remove_action( 'shutdown', array( $this, 'maybe_dispatch_async_request' ) );
+		remove_action('shutdown', array($this, 'maybe_dispatch_async_request'));
 	}
 
 	/**
@@ -100,9 +100,9 @@ 
 	 * should dispatch a request to process pending actions.
 	 */
 	public function maybe_dispatch_async_request() {
-		if ( is_admin() && ! ActionScheduler::lock()->is_locked( 'async-request-runner' ) ) {
+		if (is_admin() && ! ActionScheduler::lock()->is_locked('async-request-runner')) {
 			// Only start an async queue at most once every 60 seconds
-			ActionScheduler::lock()->set( 'async-request-runner' );
+			ActionScheduler::lock()->set('async-request-runner');
 			$this->async_request->maybe_dispatch();
 		}
 	}
@@ -121,21 +121,21 @@ 
 	 *        Generally, this should be capitalised and not localised as it's a proper noun.
 	 * @return int The number of actions processed.
 	 */
-	public function run( $context = 'WP Cron' ) {
+	public function run($context = 'WP Cron') {
 		ActionScheduler_Compatibility::raise_memory_limit();
-		ActionScheduler_Compatibility::raise_time_limit( $this->get_time_limit() );
-		do_action( 'action_scheduler_before_process_queue' );
+		ActionScheduler_Compatibility::raise_time_limit($this->get_time_limit());
+		do_action('action_scheduler_before_process_queue');
 		$this->run_cleanup();
 		$processed_actions = 0;
-		if ( false === $this->has_maximum_concurrent_batches() ) {
-			$batch_size = apply_filters( 'action_scheduler_queue_runner_batch_size', 25 );
+		if (false === $this->has_maximum_concurrent_batches()) {
+			$batch_size = apply_filters('action_scheduler_queue_runner_batch_size', 25);
 			do {
-				$processed_actions_in_batch = $this->do_batch( $batch_size, $context );
+				$processed_actions_in_batch = $this->do_batch($batch_size, $context);
 				$processed_actions         += $processed_actions_in_batch;
-			} while ( $processed_actions_in_batch > 0 && ! $this->batch_limits_exceeded( $processed_actions ) ); // keep going until we run out of actions, time, or memory
+			} while ($processed_actions_in_batch > 0 && ! $this->batch_limits_exceeded($processed_actions)); // keep going until we run out of actions, time, or memory
 		}
 
-		do_action( 'action_scheduler_after_process_queue' );
+		do_action('action_scheduler_after_process_queue');
 		return $processed_actions;
 	}
 
@@ -150,24 +150,24 @@ 
 	 *        Generally, this should be capitalised and not localised as it's a proper noun.
 	 * @return int The number of actions processed.
 	 */
-	protected function do_batch( $size = 100, $context = '' ) {
-		$claim = $this->store->stake_claim( $size );
-		$this->monitor->attach( $claim );
+	protected function do_batch($size = 100, $context = '') {
+		$claim = $this->store->stake_claim($size);
+		$this->monitor->attach($claim);
 		$processed_actions = 0;
 
-		foreach ( $claim->get_actions() as $action_id ) {
+		foreach ($claim->get_actions() as $action_id) {
 			// bail if we lost the claim
-			if ( ! in_array( $action_id, $this->store->find_actions_by_claim_id( $claim->get_id() ) ) ) {
+			if ( ! in_array($action_id, $this->store->find_actions_by_claim_id($claim->get_id()))) {
 				break;
 			}
-			$this->process_action( $action_id, $context );
+			$this->process_action($action_id, $context);
 			$processed_actions++;
 
-			if ( $this->batch_limits_exceeded( $processed_actions ) ) {
+			if ($this->batch_limits_exceeded($processed_actions)) {
 				break;
 			}
 		}
-		$this->store->release_claim( $claim );
+		$this->store->release_claim($claim);
 		$this->monitor->detach();
 		$this->clear_caches();
 		return $processed_actions;
@@ -184,7 +184,7 @@ 
 		 * Calling wp_cache_flush_runtime() lets us clear the runtime cache without invalidating the external object
 		 * cache, so we will always prefer this when it is available (but it was only introduced in WordPress 6.0).
 		 */
-		if ( function_exists( 'wp_cache_flush_runtime' ) ) {
+		if (function_exists('wp_cache_flush_runtime')) {
 			wp_cache_flush_runtime();
 		} elseif (
 			! wp_using_ext_object_cache()
@@ -200,16 +200,16 @@ 
 			 *
 			 * @param bool $flush_cache If the cache should be flushed.
 			 */
-			|| apply_filters( 'action_scheduler_queue_runner_flush_cache', false )
+			|| apply_filters('action_scheduler_queue_runner_flush_cache', false)
 		) {
 			wp_cache_flush();
 		}
 	}
 
-	public function add_wp_cron_schedule( $schedules ) {
+	public function add_wp_cron_schedule($schedules) {
 		$schedules['every_minute'] = array(
 			'interval' => 60, // in seconds
-			'display'  => __( 'Every minute', 'action-scheduler' ),
+			'display'  => __('Every minute', 'action-scheduler'),
 		);
 
 		return $schedules;

src/ext/action-scheduler/classes/data-stores/ActionScheduler_DBStore.php

Indentation +1043 added lines, -1043 removed lines

@@ -9,1055 +9,1055 @@
  */
 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, checks if this is a unique action before actually saving.
-	 *
-	 * @param ActionScheduler_Action $action         Action object.
-	 * @param \DateTime              $scheduled_date Optional schedule date. Default null.
-	 *
-	 * @return int                  Action ID.
-	 * @throws RuntimeException     Throws exception when saving the action fails.
-	 */
-	public function save_unique_action( ActionScheduler_Action $action, \DateTime $scheduled_date = null ) {
-		return $this->save_action_to_db( $action, $scheduled_date, true );
-	}
-
-	/**
-	 * Save an action. Can save duplicate action as well, prefer using `save_unique_action` instead.
-	 *
-	 * @param ActionScheduler_Action $action Action object.
-	 * @param \DateTime              $scheduled_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 $scheduled_date = null ) {
-		return $this->save_action_to_db( $action, $scheduled_date, false );
-	}
-
-	/**
-	 * Save an action.
-	 *
-	 * @param ActionScheduler_Action $action Action object.
-	 * @param ?DateTime              $date Optional schedule date. Default null.
-	 * @param bool                   $unique Whether the action should be unique.
-	 *
-	 * @return int Action ID.
-	 * @throws RuntimeException     Throws exception when saving the action fails.
-	 */
-	private function save_action_to_db( ActionScheduler_Action $action, DateTime $date = null, $unique = false ) {
-		global $wpdb;
-
-		try {
-			$this->validate_action( $action );
-
-			$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;
-			}
-
-			$insert_sql = $this->build_insert_sql( $data, $unique );
-
-			// phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared -- $insert_sql should be already prepared.
-			$wpdb->query( $insert_sql );
-			$action_id = $wpdb->insert_id;
-
-			if ( is_wp_error( $action_id ) ) {
-				throw new \RuntimeException( $action_id->get_error_message() );
-			} elseif ( empty( $action_id ) ) {
-				if ( $unique ) {
-					return 0;
-				}
-				throw new \RuntimeException( $wpdb->last_error ? $wpdb->last_error : __( 'Database error.', 'action-scheduler' ) );
-			}
-
-			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', 'action-scheduler' ), $e->getMessage() ), 0 );
-		}
-	}
-
-	/**
-	 * Helper function to build insert query.
-	 *
-	 * @param array $data Row data for action.
-	 * @param bool  $unique Whether the action should be unique.
-	 *
-	 * @return string Insert query.
-	 */
-	private function build_insert_sql( array $data, $unique ) {
-		global $wpdb;
-		$columns      = array_keys( $data );
-		$values       = array_values( $data );
-		$placeholders = array_map( array( $this, 'get_placeholder_for_column' ), $columns );
-
-		$table_name = ! empty( $wpdb->actionscheduler_actions ) ? $wpdb->actionscheduler_actions : $wpdb->prefix . 'actionscheduler_actions';
-
-		$column_sql      = '`' . implode( '`, `', $columns ) . '`';
-		$placeholder_sql = implode( ', ', $placeholders );
-		$where_clause    = $this->build_where_clause_for_insert( $data, $table_name, $unique );
-		// phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $column_sql and $where_clause are already prepared. $placeholder_sql is hardcoded.
-		$insert_query = $wpdb->prepare(
-			"
+    /**
+     * 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, checks if this is a unique action before actually saving.
+     *
+     * @param ActionScheduler_Action $action         Action object.
+     * @param \DateTime              $scheduled_date Optional schedule date. Default null.
+     *
+     * @return int                  Action ID.
+     * @throws RuntimeException     Throws exception when saving the action fails.
+     */
+    public function save_unique_action( ActionScheduler_Action $action, \DateTime $scheduled_date = null ) {
+        return $this->save_action_to_db( $action, $scheduled_date, true );
+    }
+
+    /**
+     * Save an action. Can save duplicate action as well, prefer using `save_unique_action` instead.
+     *
+     * @param ActionScheduler_Action $action Action object.
+     * @param \DateTime              $scheduled_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 $scheduled_date = null ) {
+        return $this->save_action_to_db( $action, $scheduled_date, false );
+    }
+
+    /**
+     * Save an action.
+     *
+     * @param ActionScheduler_Action $action Action object.
+     * @param ?DateTime              $date Optional schedule date. Default null.
+     * @param bool                   $unique Whether the action should be unique.
+     *
+     * @return int Action ID.
+     * @throws RuntimeException     Throws exception when saving the action fails.
+     */
+    private function save_action_to_db( ActionScheduler_Action $action, DateTime $date = null, $unique = false ) {
+        global $wpdb;
+
+        try {
+            $this->validate_action( $action );
+
+            $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;
+            }
+
+            $insert_sql = $this->build_insert_sql( $data, $unique );
+
+            // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared -- $insert_sql should be already prepared.
+            $wpdb->query( $insert_sql );
+            $action_id = $wpdb->insert_id;
+
+            if ( is_wp_error( $action_id ) ) {
+                throw new \RuntimeException( $action_id->get_error_message() );
+            } elseif ( empty( $action_id ) ) {
+                if ( $unique ) {
+                    return 0;
+                }
+                throw new \RuntimeException( $wpdb->last_error ? $wpdb->last_error : __( 'Database error.', 'action-scheduler' ) );
+            }
+
+            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', 'action-scheduler' ), $e->getMessage() ), 0 );
+        }
+    }
+
+    /**
+     * Helper function to build insert query.
+     *
+     * @param array $data Row data for action.
+     * @param bool  $unique Whether the action should be unique.
+     *
+     * @return string Insert query.
+     */
+    private function build_insert_sql( array $data, $unique ) {
+        global $wpdb;
+        $columns      = array_keys( $data );
+        $values       = array_values( $data );
+        $placeholders = array_map( array( $this, 'get_placeholder_for_column' ), $columns );
+
+        $table_name = ! empty( $wpdb->actionscheduler_actions ) ? $wpdb->actionscheduler_actions : $wpdb->prefix . 'actionscheduler_actions';
+
+        $column_sql      = '`' . implode( '`, `', $columns ) . '`';
+        $placeholder_sql = implode( ', ', $placeholders );
+        $where_clause    = $this->build_where_clause_for_insert( $data, $table_name, $unique );
+        // phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $column_sql and $where_clause are already prepared. $placeholder_sql is hardcoded.
+        $insert_query = $wpdb->prepare(
+            "
 INSERT INTO $table_name ( $column_sql )
 SELECT $placeholder_sql FROM DUAL
 WHERE ( $where_clause ) IS NULL",
-			$values
-		);
-		// phpcs:enable
-
-		return $insert_query;
-	}
-
-	/**
-	 * Helper method to build where clause for action insert statement.
-	 *
-	 * @param array  $data Row data for action.
-	 * @param string $table_name Action table name.
-	 * @param bool   $unique Where action should be unique.
-	 *
-	 * @return string Where clause to be used with insert.
-	 */
-	private function build_where_clause_for_insert( $data, $table_name, $unique ) {
-		global $wpdb;
-
-		if ( ! $unique ) {
-			return 'SELECT NULL FROM DUAL';
-		}
-
-		$pending_statuses            = array(
-			ActionScheduler_Store::STATUS_PENDING,
-			ActionScheduler_Store::STATUS_RUNNING,
-		);
-		$pending_status_placeholders = implode( ', ', array_fill( 0, count( $pending_statuses ), '%s' ) );
-		// phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $pending_status_placeholders is hardcoded.
-		$where_clause = $wpdb->prepare(
-			"
+            $values
+        );
+        // phpcs:enable
+
+        return $insert_query;
+    }
+
+    /**
+     * Helper method to build where clause for action insert statement.
+     *
+     * @param array  $data Row data for action.
+     * @param string $table_name Action table name.
+     * @param bool   $unique Where action should be unique.
+     *
+     * @return string Where clause to be used with insert.
+     */
+    private function build_where_clause_for_insert( $data, $table_name, $unique ) {
+        global $wpdb;
+
+        if ( ! $unique ) {
+            return 'SELECT NULL FROM DUAL';
+        }
+
+        $pending_statuses            = array(
+            ActionScheduler_Store::STATUS_PENDING,
+            ActionScheduler_Store::STATUS_RUNNING,
+        );
+        $pending_status_placeholders = implode( ', ', array_fill( 0, count( $pending_statuses ), '%s' ) );
+        // phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $pending_status_placeholders is hardcoded.
+        $where_clause = $wpdb->prepare(
+            "
 SELECT action_id FROM $table_name
 WHERE status IN ( $pending_status_placeholders )
 AND hook = %s
 AND `group_id` = %d
 ",
-			array_merge(
-				$pending_statuses,
-				array(
-					$data['hook'],
-					$data['group_id'],
-				)
-			)
-		);
-		// phpcs:enable
-
-		return "$where_clause" . ' LIMIT 1';
-	}
-
-	/**
-	 * Helper method to get $wpdb->prepare placeholder for a given column name.
-	 *
-	 * @param string $column_name Name of column in actions table.
-	 *
-	 * @return string Placeholder to use for given column.
-	 */
-	private function get_placeholder_for_column( $column_name ) {
-		$string_columns = array(
-			'hook',
-			'status',
-			'scheduled_date_gmt',
-			'scheduled_date_local',
-			'args',
-			'schedule',
-			'last_attempt_gmt',
-			'last_attempt_local',
-			'extended_args',
-		);
-
-		return in_array( $column_name, $string_columns ) ? '%s' : '%d';
-	}
-
-	/**
-	 * 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 ( $data->$date_field === null ) {
-				$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 3.3.0 $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.', 'action-scheduler' ) );
-		}
-
-		$query = wp_parse_args(
-			$query,
-			array(
-				'hook'                  => '',
-				'args'                  => null,
-				'partial_args_matching' => 'off', // can be 'like' or 'json'
-				'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;
-
-		$db_server_info = is_callable( array( $wpdb, 'db_server_info' ) ) ? $wpdb->db_server_info() : $wpdb->db_version();
-		if ( false !== strpos( $db_server_info, 'MariaDB' ) ) {
-			$supports_json = version_compare(
-				PHP_VERSION_ID >= 80016 ? $wpdb->db_version() : preg_replace( '/[^0-9.].*/', '', str_replace( '5.5.5-', '', $db_server_info ) ),
-				'10.2',
-				'>='
-			);
-		} else {
-			$supports_json = version_compare( $wpdb->db_version(), '5.7', '>=' );
-		}
-
-		$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 ( ! empty( $query['hook'] ) ) {
-			$sql         .= ' AND a.hook=%s';
-			$sql_params[] = $query['hook'];
-		}
-
-		if ( $query['args'] !== null ) {
-			switch ( $query['partial_args_matching'] ) {
-				case 'json':
-					if ( ! $supports_json ) {
-						throw new \RuntimeException( __( 'JSON partial matching not supported in your environment. Please check your MySQL/MariaDB version.', 'action-scheduler' ) );
-					}
-					$supported_types = array(
-						'integer' => '%d',
-						'boolean' => '%s',
-						'double'  => '%f',
-						'string'  => '%s',
-					);
-					foreach ( $query['args'] as $key => $value ) {
-						$value_type = gettype( $value );
-						if ( 'boolean' === $value_type ) {
-							$value = $value ? 'true' : 'false';
-						}
-						$placeholder = isset( $supported_types[ $value_type ] ) ? $supported_types[ $value_type ] : false;
-						if ( ! $placeholder ) {
-							throw new \RuntimeException(
-								sprintf(
-								/* translators: %s: provided value type */
-									__( 'The value type for the JSON partial matching is not supported. Must be either integer, boolean, double or string. %s type provided.', 'action-scheduler' ),
-									$value_type
-								)
-							);
-						}
-						$sql         .= ' AND JSON_EXTRACT(a.args, %s)=' . $placeholder;
-						$sql_params[] = '$.' . $key;
-						$sql_params[] = $value;
-					}
-					break;
-				case 'like':
-					foreach ( $query['args'] as $key => $value ) {
-						$sql         .= ' AND a.args LIKE %s';
-						$json_partial = $wpdb->esc_like( trim( json_encode( array( $key => $value ) ), '{}' ) );
-						$sql_params[] = "%{$json_partial}%";
-					}
-					break;
-				case 'off':
-					$sql         .= ' AND a.args=%s';
-					$sql_params[] = $this->get_args_for_query( $query['args'] );
-					break;
-				default:
-					throw new \RuntimeException( __( 'Unknown partial args matching value.', 'action-scheduler' ) );
-			}
-		}
-
-		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 ( $query['claimed'] !== null ) {
-			$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 3.3.0 $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', 'action-scheduler' ), $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', 'action-scheduler' ), $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', 'action-scheduler' ), $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 = $before_date === null ? $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.', 'action-scheduler' ), $group ) );
-			}
-
-			$where   .= ' AND group_id = %d';
-			$params[] = $group_id;
-		}
-
-		/**
-		 * Sets the order-by clause used in the action claim query.
-		 *
-		 * @since 3.4.0
-		 *
-		 * @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.', 'action-scheduler' ) );
-		}
-
-		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', 'action-scheduler' ), $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', 'action-scheduler' ), $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.', 'action-scheduler' ) );
-		} elseif ( empty( $status ) ) {
-			throw new \RuntimeException( __( 'Unknown status found for action.', 'action-scheduler' ) );
-		} else {
-			return $status;
-		}
-	}
+            array_merge(
+                $pending_statuses,
+                array(
+                    $data['hook'],
+                    $data['group_id'],
+                )
+            )
+        );
+        // phpcs:enable
+
+        return "$where_clause" . ' LIMIT 1';
+    }
+
+    /**
+     * Helper method to get $wpdb->prepare placeholder for a given column name.
+     *
+     * @param string $column_name Name of column in actions table.
+     *
+     * @return string Placeholder to use for given column.
+     */
+    private function get_placeholder_for_column( $column_name ) {
+        $string_columns = array(
+            'hook',
+            'status',
+            'scheduled_date_gmt',
+            'scheduled_date_local',
+            'args',
+            'schedule',
+            'last_attempt_gmt',
+            'last_attempt_local',
+            'extended_args',
+        );
+
+        return in_array( $column_name, $string_columns ) ? '%s' : '%d';
+    }
+
+    /**
+     * 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 ( $data->$date_field === null ) {
+                $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 3.3.0 $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.', 'action-scheduler' ) );
+        }
+
+        $query = wp_parse_args(
+            $query,
+            array(
+                'hook'                  => '',
+                'args'                  => null,
+                'partial_args_matching' => 'off', // can be 'like' or 'json'
+                '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;
+
+        $db_server_info = is_callable( array( $wpdb, 'db_server_info' ) ) ? $wpdb->db_server_info() : $wpdb->db_version();
+        if ( false !== strpos( $db_server_info, 'MariaDB' ) ) {
+            $supports_json = version_compare(
+                PHP_VERSION_ID >= 80016 ? $wpdb->db_version() : preg_replace( '/[^0-9.].*/', '', str_replace( '5.5.5-', '', $db_server_info ) ),
+                '10.2',
+                '>='
+            );
+        } else {
+            $supports_json = version_compare( $wpdb->db_version(), '5.7', '>=' );
+        }
+
+        $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 ( ! empty( $query['hook'] ) ) {
+            $sql         .= ' AND a.hook=%s';
+            $sql_params[] = $query['hook'];
+        }
+
+        if ( $query['args'] !== null ) {
+            switch ( $query['partial_args_matching'] ) {
+                case 'json':
+                    if ( ! $supports_json ) {
+                        throw new \RuntimeException( __( 'JSON partial matching not supported in your environment. Please check your MySQL/MariaDB version.', 'action-scheduler' ) );
+                    }
+                    $supported_types = array(
+                        'integer' => '%d',
+                        'boolean' => '%s',
+                        'double'  => '%f',
+                        'string'  => '%s',
+                    );
+                    foreach ( $query['args'] as $key => $value ) {
+                        $value_type = gettype( $value );
+                        if ( 'boolean' === $value_type ) {
+                            $value = $value ? 'true' : 'false';
+                        }
+                        $placeholder = isset( $supported_types[ $value_type ] ) ? $supported_types[ $value_type ] : false;
+                        if ( ! $placeholder ) {
+                            throw new \RuntimeException(
+                                sprintf(
+                                /* translators: %s: provided value type */
+                                    __( 'The value type for the JSON partial matching is not supported. Must be either integer, boolean, double or string. %s type provided.', 'action-scheduler' ),
+                                    $value_type
+                                )
+                            );
+                        }
+                        $sql         .= ' AND JSON_EXTRACT(a.args, %s)=' . $placeholder;
+                        $sql_params[] = '$.' . $key;
+                        $sql_params[] = $value;
+                    }
+                    break;
+                case 'like':
+                    foreach ( $query['args'] as $key => $value ) {
+                        $sql         .= ' AND a.args LIKE %s';
+                        $json_partial = $wpdb->esc_like( trim( json_encode( array( $key => $value ) ), '{}' ) );
+                        $sql_params[] = "%{$json_partial}%";
+                    }
+                    break;
+                case 'off':
+                    $sql         .= ' AND a.args=%s';
+                    $sql_params[] = $this->get_args_for_query( $query['args'] );
+                    break;
+                default:
+                    throw new \RuntimeException( __( 'Unknown partial args matching value.', 'action-scheduler' ) );
+            }
+        }
+
+        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 ( $query['claimed'] !== null ) {
+            $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 3.3.0 $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', 'action-scheduler' ), $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', 'action-scheduler' ), $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', 'action-scheduler' ), $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 = $before_date === null ? $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.', 'action-scheduler' ), $group ) );
+            }
+
+            $where   .= ' AND group_id = %d';
+            $params[] = $group_id;
+        }
+
+        /**
+         * Sets the order-by clause used in the action claim query.
+         *
+         * @since 3.4.0
+         *
+         * @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.', 'action-scheduler' ) );
+        }
+
+        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', 'action-scheduler' ), $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', 'action-scheduler' ), $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.', 'action-scheduler' ) );
+        } elseif ( empty( $status ) ) {
+            throw new \RuntimeException( __( 'Unknown status found for action.', 'action-scheduler' ) );
+        } else {
+            return $status;
+        }
+    }
 }

Spacing +230 added lines, -230 removed lines

@@ -45,8 +45,8 @@ 
 	 * @return int                  Action ID.
 	 * @throws RuntimeException     Throws exception when saving the action fails.
 	 */
-	public function save_unique_action( ActionScheduler_Action $action, \DateTime $scheduled_date = null ) {
-		return $this->save_action_to_db( $action, $scheduled_date, true );
+	public function save_unique_action(ActionScheduler_Action $action, \DateTime $scheduled_date = null) {
+		return $this->save_action_to_db($action, $scheduled_date, true);
 	}
 
 	/**
@@ -58,8 +58,8 @@ 
 	 * @return int Action ID.
 	 * @throws RuntimeException     Throws exception when saving the action fails.
 	 */
-	public function save_action( ActionScheduler_Action $action, \DateTime $scheduled_date = null ) {
-		return $this->save_action_to_db( $action, $scheduled_date, false );
+	public function save_action(ActionScheduler_Action $action, \DateTime $scheduled_date = null) {
+		return $this->save_action_to_db($action, $scheduled_date, false);
 	}
 
 	/**
@@ -72,50 +72,50 @@ 
 	 * @return int Action ID.
 	 * @throws RuntimeException     Throws exception when saving the action fails.
 	 */
-	private function save_action_to_db( ActionScheduler_Action $action, DateTime $date = null, $unique = false ) {
+	private function save_action_to_db(ActionScheduler_Action $action, DateTime $date = null, $unique = false) {
 		global $wpdb;
 
 		try {
-			$this->validate_action( $action );
+			$this->validate_action($action);
 
 			$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() ),
+				'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 ) {
+			$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['args']          = $this->hash_args($args);
 				$data['extended_args'] = $args;
 			}
 
-			$insert_sql = $this->build_insert_sql( $data, $unique );
+			$insert_sql = $this->build_insert_sql($data, $unique);
 
 			// phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared -- $insert_sql should be already prepared.
-			$wpdb->query( $insert_sql );
+			$wpdb->query($insert_sql);
 			$action_id = $wpdb->insert_id;
 
-			if ( is_wp_error( $action_id ) ) {
-				throw new \RuntimeException( $action_id->get_error_message() );
-			} elseif ( empty( $action_id ) ) {
-				if ( $unique ) {
+			if (is_wp_error($action_id)) {
+				throw new \RuntimeException($action_id->get_error_message());
+			} elseif (empty($action_id)) {
+				if ($unique) {
 					return 0;
 				}
-				throw new \RuntimeException( $wpdb->last_error ? $wpdb->last_error : __( 'Database error.', 'action-scheduler' ) );
+				throw new \RuntimeException($wpdb->last_error ? $wpdb->last_error : __('Database error.', 'action-scheduler'));
 			}
 
-			do_action( 'action_scheduler_stored_action', $action_id );
+			do_action('action_scheduler_stored_action', $action_id);
 
 			return $action_id;
-		} catch ( \Exception $e ) {
+		} catch (\Exception $e) {
 			/* translators: %s: error message */
-			throw new \RuntimeException( sprintf( __( 'Error saving action: %s', 'action-scheduler' ), $e->getMessage() ), 0 );
+			throw new \RuntimeException(sprintf(__('Error saving action: %s', 'action-scheduler'), $e->getMessage()), 0);
 		}
 	}
 
@@ -127,17 +127,17 @@ 
 	 *
 	 * @return string Insert query.
 	 */
-	private function build_insert_sql( array $data, $unique ) {
+	private function build_insert_sql(array $data, $unique) {
 		global $wpdb;
-		$columns      = array_keys( $data );
-		$values       = array_values( $data );
-		$placeholders = array_map( array( $this, 'get_placeholder_for_column' ), $columns );
+		$columns      = array_keys($data);
+		$values       = array_values($data);
+		$placeholders = array_map(array($this, 'get_placeholder_for_column'), $columns);
 
-		$table_name = ! empty( $wpdb->actionscheduler_actions ) ? $wpdb->actionscheduler_actions : $wpdb->prefix . 'actionscheduler_actions';
+		$table_name = ! empty($wpdb->actionscheduler_actions) ? $wpdb->actionscheduler_actions : $wpdb->prefix.'actionscheduler_actions';
 
-		$column_sql      = '`' . implode( '`, `', $columns ) . '`';
-		$placeholder_sql = implode( ', ', $placeholders );
-		$where_clause    = $this->build_where_clause_for_insert( $data, $table_name, $unique );
+		$column_sql      = '`'.implode('`, `', $columns).'`';
+		$placeholder_sql = implode(', ', $placeholders);
+		$where_clause    = $this->build_where_clause_for_insert($data, $table_name, $unique);
 		// phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $column_sql and $where_clause are already prepared. $placeholder_sql is hardcoded.
 		$insert_query = $wpdb->prepare(
 			"
@@ -160,18 +160,18 @@ 
 	 *
 	 * @return string Where clause to be used with insert.
 	 */
-	private function build_where_clause_for_insert( $data, $table_name, $unique ) {
+	private function build_where_clause_for_insert($data, $table_name, $unique) {
 		global $wpdb;
 
-		if ( ! $unique ) {
+		if ( ! $unique) {
 			return 'SELECT NULL FROM DUAL';
 		}
 
-		$pending_statuses            = array(
+		$pending_statuses = array(
 			ActionScheduler_Store::STATUS_PENDING,
 			ActionScheduler_Store::STATUS_RUNNING,
 		);
-		$pending_status_placeholders = implode( ', ', array_fill( 0, count( $pending_statuses ), '%s' ) );
+		$pending_status_placeholders = implode(', ', array_fill(0, count($pending_statuses), '%s'));
 		// phpcs:disable WordPress.DB.PreparedSQL.NotPrepared, WordPress.DB.PreparedSQL.InterpolatedNotPrepared -- $pending_status_placeholders is hardcoded.
 		$where_clause = $wpdb->prepare(
 			"
@@ -190,7 +190,7 @@ 
 		);
 		// phpcs:enable
 
-		return "$where_clause" . ' LIMIT 1';
+		return "$where_clause".' LIMIT 1';
 	}
 
 	/**
@@ -200,7 +200,7 @@ 
 	 *
 	 * @return string Placeholder to use for given column.
 	 */
-	private function get_placeholder_for_column( $column_name ) {
+	private function get_placeholder_for_column($column_name) {
 		$string_columns = array(
 			'hook',
 			'status',
@@ -213,7 +213,7 @@ 
 			'extended_args',
 		);
 
-		return in_array( $column_name, $string_columns ) ? '%s' : '%d';
+		return in_array($column_name, $string_columns) ? '%s' : '%d';
 	}
 
 	/**
@@ -222,8 +222,8 @@ 
 	 * @param string $args JSON encoded action args.
 	 * @return string
 	 */
-	protected function hash_args( $args ) {
-		return md5( $args );
+	protected function hash_args($args) {
+		return md5($args);
 	}
 
 	/**
@@ -232,12 +232,12 @@ 
 	 * @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 ) {
+	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 );
+		return $this->hash_args($encoded);
 	}
 	/**
 	 * Get a group's ID based on its name/slug.
@@ -247,15 +247,15 @@ 
 	 *
 	 * @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 ) ) {
+	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 );
+		$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;
@@ -268,10 +268,10 @@ 
 	 *
 	 * @return int Group ID.
 	 */
-	protected function create_group( $slug ) {
+	protected function create_group($slug) {
 		/** @var \wpdb $wpdb */
 		global $wpdb;
-		$wpdb->insert( $wpdb->actionscheduler_groups, array( 'slug' => $slug ) );
+		$wpdb->insert($wpdb->actionscheduler_groups, array('slug' => $slug));
 
 		return (int) $wpdb->insert_id;
 	}
@@ -283,7 +283,7 @@ 
 	 *
 	 * @return ActionScheduler_Action
 	 */
-	public function fetch_action( $action_id ) {
+	public function fetch_action($action_id) {
 		/** @var \wpdb $wpdb */
 		global $wpdb;
 		$data = $wpdb->get_row(
@@ -293,13 +293,13 @@ 
 			)
 		);
 
-		if ( empty( $data ) ) {
+		if (empty($data)) {
 			return $this->get_null_action();
 		}
 
-		if ( ! empty( $data->extended_args ) ) {
+		if ( ! empty($data->extended_args)) {
 			$data->args = $data->extended_args;
-			unset( $data->extended_args );
+			unset($data->extended_args);
 		}
 
 		// Convert NULL dates to zero dates.
@@ -309,16 +309,16 @@ 
 			'last_attempt_gmt',
 			'last_attempt_gmt',
 		);
-		foreach ( $date_fields as $date_field ) {
-			if ( $data->$date_field === null ) {
+		foreach ($date_fields as $date_field) {
+			if ($data->$date_field === null) {
 				$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 );
+			$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();
 		}
 
@@ -341,21 +341,21 @@ 
 	 *
 	 * @return ActionScheduler_Action|ActionScheduler_CanceledAction|ActionScheduler_FinishedAction
 	 */
-	protected function make_action_from_db_record( $data ) {
+	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
+		$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 );
+		$this->validate_args($args, $data->action_id);
+		$this->validate_schedule($schedule, $data->action_id);
 
-		if ( empty( $schedule ) ) {
+		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 );
+		return ActionScheduler::factory()->get_stored_action($data->status, $data->hook, $args, $schedule, $group);
 	}
 
 	/**
@@ -369,10 +369,10 @@ 
 	 * @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' ) {
+	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.', 'action-scheduler' ) );
+		if ( ! in_array($select_or_count, array('select', 'count'), true)) {
+			throw new InvalidArgumentException(__('Invalid value for select or count parameter. Cannot query actions.', 'action-scheduler'));
 		}
 
 		$query = wp_parse_args(
@@ -398,42 +398,42 @@ 
 		/** @var \wpdb $wpdb */
 		global $wpdb;
 
-		$db_server_info = is_callable( array( $wpdb, 'db_server_info' ) ) ? $wpdb->db_server_info() : $wpdb->db_version();
-		if ( false !== strpos( $db_server_info, 'MariaDB' ) ) {
+		$db_server_info = is_callable(array($wpdb, 'db_server_info')) ? $wpdb->db_server_info() : $wpdb->db_version();
+		if (false !== strpos($db_server_info, 'MariaDB')) {
 			$supports_json = version_compare(
-				PHP_VERSION_ID >= 80016 ? $wpdb->db_version() : preg_replace( '/[^0-9.].*/', '', str_replace( '5.5.5-', '', $db_server_info ) ),
+				PHP_VERSION_ID >= 80016 ? $wpdb->db_version() : preg_replace('/[^0-9.].*/', '', str_replace('5.5.5-', '', $db_server_info)),
 				'10.2',
 				'>='
 			);
 		} else {
-			$supports_json = version_compare( $wpdb->db_version(), '5.7', '>=' );
+			$supports_json = version_compare($wpdb->db_version(), '5.7', '>=');
 		}
 
-		$sql        = ( 'count' === $select_or_count ) ? 'SELECT count(a.action_id)' : 'SELECT a.action_id';
+		$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'] ) {
+		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'] ) ) {
+		if ( ! empty($query['group'])) {
 			$sql         .= ' AND g.slug=%s';
 			$sql_params[] = $query['group'];
 		}
 
-		if ( ! empty( $query['hook'] ) ) {
+		if ( ! empty($query['hook'])) {
 			$sql         .= ' AND a.hook=%s';
 			$sql_params[] = $query['hook'];
 		}
 
-		if ( $query['args'] !== null ) {
-			switch ( $query['partial_args_matching'] ) {
+		if ($query['args'] !== null) {
+			switch ($query['partial_args_matching']) {
 				case 'json':
-					if ( ! $supports_json ) {
-						throw new \RuntimeException( __( 'JSON partial matching not supported in your environment. Please check your MySQL/MariaDB version.', 'action-scheduler' ) );
+					if ( ! $supports_json) {
+						throw new \RuntimeException(__('JSON partial matching not supported in your environment. Please check your MySQL/MariaDB version.', 'action-scheduler'));
 					}
 					$supported_types = array(
 						'integer' => '%d',
@@ -441,84 +441,84 @@ 
 						'double'  => '%f',
 						'string'  => '%s',
 					);
-					foreach ( $query['args'] as $key => $value ) {
-						$value_type = gettype( $value );
-						if ( 'boolean' === $value_type ) {
+					foreach ($query['args'] as $key => $value) {
+						$value_type = gettype($value);
+						if ('boolean' === $value_type) {
 							$value = $value ? 'true' : 'false';
 						}
-						$placeholder = isset( $supported_types[ $value_type ] ) ? $supported_types[ $value_type ] : false;
-						if ( ! $placeholder ) {
+						$placeholder = isset($supported_types[$value_type]) ? $supported_types[$value_type] : false;
+						if ( ! $placeholder) {
 							throw new \RuntimeException(
 								sprintf(
 								/* translators: %s: provided value type */
-									__( 'The value type for the JSON partial matching is not supported. Must be either integer, boolean, double or string. %s type provided.', 'action-scheduler' ),
+									__('The value type for the JSON partial matching is not supported. Must be either integer, boolean, double or string. %s type provided.', 'action-scheduler'),
 									$value_type
 								)
 							);
 						}
-						$sql         .= ' AND JSON_EXTRACT(a.args, %s)=' . $placeholder;
-						$sql_params[] = '$.' . $key;
+						$sql         .= ' AND JSON_EXTRACT(a.args, %s)='.$placeholder;
+						$sql_params[] = '$.'.$key;
 						$sql_params[] = $value;
 					}
 					break;
 				case 'like':
-					foreach ( $query['args'] as $key => $value ) {
+					foreach ($query['args'] as $key => $value) {
 						$sql         .= ' AND a.args LIKE %s';
-						$json_partial = $wpdb->esc_like( trim( json_encode( array( $key => $value ) ), '{}' ) );
+						$json_partial = $wpdb->esc_like(trim(json_encode(array($key => $value)), '{}'));
 						$sql_params[] = "%{$json_partial}%";
 					}
 					break;
 				case 'off':
 					$sql         .= ' AND a.args=%s';
-					$sql_params[] = $this->get_args_for_query( $query['args'] );
+					$sql_params[] = $this->get_args_for_query($query['args']);
 					break;
 				default:
-					throw new \RuntimeException( __( 'Unknown partial args matching value.', 'action-scheduler' ) );
+					throw new \RuntimeException(__('Unknown partial args matching value.', 'action-scheduler'));
 			}
 		}
 
-		if ( $query['status'] ) {
+		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 ) );
+			$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 ) {
+		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'] );
+			$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 ) {
+		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'] );
+			$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'] ) {
+		if (true === $query['claimed']) {
 			$sql .= ' AND a.claim_id != 0';
-		} elseif ( false === $query['claimed'] ) {
+		} elseif (false === $query['claimed']) {
 			$sql .= ' AND a.claim_id = 0';
-		} elseif ( $query['claimed'] !== null ) {
+		} elseif ($query['claimed'] !== null) {
 			$sql         .= ' AND a.claim_id = %d';
 			$sql_params[] = $query['claimed'];
 		}
 
-		if ( ! empty( $query['search'] ) ) {
+		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'] );
+			for ($i = 0; $i < 3; $i++) {
+				$sql_params[] = sprintf('%%%s%%', $query['search']);
 			}
 
 			$search_claim_id = (int) $query['search'];
-			if ( $search_claim_id ) {
+			if ($search_claim_id) {
 				$sql         .= ' OR a.claim_id = %d';
 				$sql_params[] = $search_claim_id;
 			}
@@ -526,13 +526,13 @@ 
 			$sql .= ')';
 		}
 
-		if ( 'select' === $select_or_count ) {
-			if ( 'ASC' === strtoupper( $query['order'] ) ) {
+		if ('select' === $select_or_count) {
+			if ('ASC' === strtoupper($query['order'])) {
 				$order = 'ASC';
 			} else {
 				$order = 'DESC';
 			}
-			switch ( $query['orderby'] ) {
+			switch ($query['orderby']) {
 				case 'hook':
 					$sql .= " ORDER BY a.hook $order";
 					break;
@@ -553,15 +553,15 @@ 
 					break;
 			}
 
-			if ( $query['per_page'] > 0 ) {
+			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
+		if ( ! empty($sql_params)) {
+			$sql = $wpdb->prepare($sql, $sql_params); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 		}
 
 		return $sql;
@@ -579,13 +579,13 @@ 
 	 *
 	 * @return string|array|null The IDs of actions matching the query. Null on failure.
 	 */
-	public function query_actions( $query = array(), $query_type = 'select' ) {
+	public function query_actions($query = array(), $query_type = 'select') {
 		/** @var wpdb $wpdb */
 		global $wpdb;
 
-		$sql = $this->get_query_actions_sql( $query, $query_type );
+		$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
+		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
 	}
 
 	/**
@@ -603,10 +603,10 @@ 
 		$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
+		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;
+			if (array_key_exists($action_data->status, $action_stati_and_labels)) {
+				$actions_count_by_status[$action_data->status] = $action_data->count;
 			}
 		}
 
@@ -621,22 +621,22 @@ 
 	 * @return void
 	 * @throws \InvalidArgumentException If the action update failed.
 	 */
-	public function cancel_action( $action_id ) {
+	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' )
+			array('status' => self::STATUS_CANCELED),
+			array('action_id' => $action_id),
+			array('%s'),
+			array('%d')
 		);
-		if ( false === $updated ) {
+		if (false === $updated) {
 			/* translators: %s: action ID */
-			throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+			throw new \InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id));
 		}
-		do_action( 'action_scheduler_canceled_action', $action_id );
+		do_action('action_scheduler_canceled_action', $action_id);
 	}
 
 	/**
@@ -648,8 +648,8 @@ 
 	 *
 	 * @return void
 	 */
-	public function cancel_actions_by_hook( $hook ) {
-		$this->bulk_cancel_actions( array( 'hook' => $hook ) );
+	public function cancel_actions_by_hook($hook) {
+		$this->bulk_cancel_actions(array('hook' => $hook));
 	}
 
 	/**
@@ -659,8 +659,8 @@ 
 	 *
 	 * @return void
 	 */
-	public function cancel_actions_by_group( $group ) {
-		$this->bulk_cancel_actions( array( 'group' => $group ) );
+	public function cancel_actions_by_group($group) {
+		$this->bulk_cancel_actions(array('group' => $group));
 	}
 
 	/**
@@ -670,16 +670,16 @@ 
 	 *
 	 * @param array $query_args Query parameters.
 	 */
-	protected function bulk_cancel_actions( $query_args ) {
+	protected function bulk_cancel_actions($query_args) {
 		/** @var \wpdb $wpdb */
 		global $wpdb;
 
-		if ( ! is_array( $query_args ) ) {
+		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'] ) {
+		if (isset($query_args['status']) && self::STATUS_CANCELED === $query_args['status']) {
 			return;
 		}
 
@@ -693,16 +693,16 @@ 
 			)
 		);
 
-		while ( $action_ids ) {
-			$action_ids = $this->query_actions( $query_args );
-			if ( empty( $action_ids ) ) {
+		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 ) . ')';
+			$format     = array_fill(0, count($action_ids), '%d');
+			$query_in   = '('.implode(',', $format).')';
 			$parameters = $action_ids;
-			array_unshift( $parameters, self::STATUS_CANCELED );
+			array_unshift($parameters, self::STATUS_CANCELED);
 
 			$wpdb->query(
 				$wpdb->prepare(
@@ -711,7 +711,7 @@ 
 				)
 			);
 
-			do_action( 'action_scheduler_bulk_cancel_actions', $action_ids );
+			do_action('action_scheduler_bulk_cancel_actions', $action_ids);
 		}
 	}
 
@@ -721,14 +721,14 @@ 
 	 * @param int $action_id Action ID.
 	 * @throws \InvalidArgumentException If the action deletion failed.
 	 */
-	public function delete_action( $action_id ) {
+	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', 'action-scheduler' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
+		$deleted = $wpdb->delete($wpdb->actionscheduler_actions, array('action_id' => $action_id), array('%d'));
+		if (empty($deleted)) {
+			throw new \InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id)); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
 		}
-		do_action( 'action_scheduler_deleted_action', $action_id );
+		do_action('action_scheduler_deleted_action', $action_id);
 	}
 
 	/**
@@ -738,9 +738,9 @@ 
 	 *
 	 * @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 );
+	public function get_date($action_id) {
+		$date = $this->get_date_gmt($action_id);
+		ActionScheduler_TimezoneHelper::set_local_timezone($date);
 		return $date;
 	}
 
@@ -752,17 +752,17 @@ 
 	 * @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 ) {
+	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', 'action-scheduler' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
+		$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', 'action-scheduler'), $action_id)); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
 		}
-		if ( self::STATUS_PENDING === $record->status ) {
-			return as_get_datetime_object( $record->scheduled_date_gmt );
+		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 );
+			return as_get_datetime_object($record->last_attempt_gmt);
 		}
 	}
 
@@ -776,15 +776,15 @@ 
 	 *
 	 * @return ActionScheduler_ActionClaim
 	 */
-	public function stake_claim( $max_actions = 10, \DateTime $before_date = null, $hooks = array(), $group = '' ) {
+	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_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 );
+		return new ActionScheduler_ActionClaim($claim_id, $action_ids);
 	}
 
 	/**
@@ -796,7 +796,7 @@ 
 		/** @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' ) ) );
+		$wpdb->insert($wpdb->actionscheduler_claims, array('date_created_gmt' => $now->format('Y-m-d H:i:s')));
 
 		return $wpdb->insert_id;
 	}
@@ -814,7 +814,7 @@ 
 	 * @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 = '' ) {
+	protected function claim_actions($claim_id, $limit, \DateTime $before_date = null, $hooks = array(), $group = '') {
 		/** @var \wpdb $wpdb */
 		global $wpdb;
 
@@ -825,28 +825,28 @@ 
 		$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' ),
+			$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[] = $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($hooks)) {
+			$placeholders = array_fill(0, count($hooks), '%s');
+			$where       .= ' AND hook IN ('.join(', ', $placeholders).')';
+			$params       = array_merge($params, array_values($hooks));
 		}
 
-		if ( ! empty( $group ) ) {
+		if ( ! empty($group)) {
 
-			$group_id = $this->get_group_id( $group, false );
+			$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 ) ) {
+			if (empty($group_id)) {
 				/* translators: %s: group name */
-				throw new InvalidArgumentException( sprintf( __( 'The group "%s" does not exist.', 'action-scheduler' ), $group ) );
+				throw new InvalidArgumentException(sprintf(__('The group "%s" does not exist.', 'action-scheduler'), $group));
 			}
 
 			$where   .= ' AND group_id = %d';
@@ -860,13 +860,13 @@ 
 		 *
 		 * @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' );
+		$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.', 'action-scheduler' ) );
+		$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.', 'action-scheduler'));
 		}
 
 		return (int) $rows_affected;
@@ -881,9 +881,9 @@ 
 		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
+		$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 (int) $wpdb->get_var($sql); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 	}
 
 	/**
@@ -892,14 +892,14 @@ 
 	 * @param string $action_id Action ID.
 	 * @return mixed
 	 */
-	public function get_claim_id( $action_id ) {
+	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
+		$sql = $wpdb->prepare($sql, $action_id); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 
-		return (int) $wpdb->get_var( $sql ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+		return (int) $wpdb->get_var($sql); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 	}
 
 	/**
@@ -908,13 +908,13 @@ 
 	 * @param  int $claim_id Claim ID.
 	 * @return int[]
 	 */
-	public function find_actions_by_claim_id( $claim_id ) {
+	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' );
+		$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",
@@ -923,9 +923,9 @@ 
 
 		// 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 );
+		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);
 			}
 		}
 
@@ -937,11 +937,11 @@ 
 	 *
 	 * @param ActionScheduler_ActionClaim $claim Claim object.
 	 */
-	public function release_claim( ActionScheduler_ActionClaim $claim ) {
+	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' ) );
+		$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'));
 	}
 
 	/**
@@ -951,15 +951,15 @@ 
 	 *
 	 * @return void
 	 */
-	public function unclaim_action( $action_id ) {
+	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' )
+			array('claim_id' => 0),
+			array('action_id' => $action_id),
+			array('%s'),
+			array('%d')
 		);
 	}
 
@@ -969,18 +969,18 @@ 
 	 * @param int $action_id Action ID.
 	 * @throws \InvalidArgumentException Throw an exception if action was not updated.
 	 */
-	public function mark_failure( $action_id ) {
+	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' )
+			array('status' => self::STATUS_FAILED),
+			array('action_id' => $action_id),
+			array('%s'),
+			array('%d')
 		);
-		if ( empty( $updated ) ) {
-			throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
+		if (empty($updated)) {
+			throw new \InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id)); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
 		}
 	}
 
@@ -991,13 +991,13 @@ 
 	 *
 	 * @return void
 	 */
-	public function log_execution( $action_id ) {
+	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
+		$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
 	}
 
 	/**
@@ -1008,22 +1008,22 @@ 
 	 * @return void
 	 * @throws \InvalidArgumentException Throw an exception if action was not updated.
 	 */
-	public function mark_complete( $action_id ) {
+	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' ),
+				'last_attempt_gmt'   => current_time('mysql', true),
+				'last_attempt_local' => current_time('mysql'),
 			),
-			array( 'action_id' => $action_id ),
-			array( '%s' ),
-			array( '%d' )
+			array('action_id' => $action_id),
+			array('%s'),
+			array('%d')
 		);
-		if ( empty( $updated ) ) {
-			throw new \InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) ); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
+		if (empty($updated)) {
+			throw new \InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id)); //phpcs:ignore WordPress.WP.I18n.MissingTranslatorsComment
 		}
 
 		/**
@@ -1033,7 +1033,7 @@ 
 		 *
 		 * @param int $action_id Action ID.
 		 */
-		do_action( 'action_scheduler_completed_action', $action_id );
+		do_action('action_scheduler_completed_action', $action_id);
 	}
 
 	/**
@@ -1045,17 +1045,17 @@ 
 	 * @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 ) {
+	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
+		$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.', 'action-scheduler' ) );
-		} elseif ( empty( $status ) ) {
-			throw new \RuntimeException( __( 'Unknown status found for action.', 'action-scheduler' ) );
+		if (null === $status) {
+			throw new \InvalidArgumentException(__('Invalid action ID. No status found.', 'action-scheduler'));
+		} elseif (empty($status)) {
+			throw new \RuntimeException(__('Unknown status found for action.', 'action-scheduler'));
 		} else {
 			return $status;
 		}

src/ext/action-scheduler/classes/data-stores/ActionScheduler_wpCommentLogger.php

Indentation +238 added lines, -238 removed lines

@@ -4,243 +4,243 @@
  * 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 );
-	}
+    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 );
+    }
 
 }

Spacing +55 added lines, -55 removed lines

@@ -14,30 +14,30 @@ 
 	 *
 	 * @return string The log entry ID
 	 */
-	public function log( $action_id, $message, DateTime $date = null ) {
-		if ( empty( $date ) ) {
+	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 );
+			$date = as_get_datetime_object(clone $date);
 		}
-		$comment_id = $this->create_wp_comment( $action_id, $message, $date );
+		$comment_id = $this->create_wp_comment($action_id, $message, $date);
 		return $comment_id;
 	}
 
-	protected function create_wp_comment( $action_id, $message, DateTime $date ) {
+	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_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'     => $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 );
+		return wp_insert_comment($comment_data);
 	}
 
 	/**
@@ -45,15 +45,15 @@ 
 	 *
 	 * @return ActionScheduler_LogEntry
 	 */
-	public function get_entry( $entry_id ) {
-		$comment = $this->get_comment( $entry_id );
-		if ( empty( $comment ) || $comment->comment_type != self::TYPE ) {
+	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 );
+		$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);
 	}
 
 	/**
@@ -61,9 +61,9 @@ 
 	 *
 	 * @return ActionScheduler_LogEntry[]
 	 */
-	public function get_logs( $action_id ) {
+	public function get_logs($action_id) {
 		$status = 'all';
-		if ( get_post_status( $action_id ) == 'trash' ) {
+		if (get_post_status($action_id) == 'trash') {
 			$status = 'post-trashed';
 		}
 		$comments = get_comments(
@@ -75,31 +75,31 @@ 
 				'status'  => $status,
 			)
 		);
-		$logs     = array();
-		foreach ( $comments as $c ) {
-			$entry = $this->get_entry( $c );
-			if ( ! empty( $entry ) ) {
+		$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 );
+	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 ] ) ) {
+	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 );
+		add_filter('comments_clauses', array($this, 'filter_comment_query_clauses'), 10, 2);
 	}
 
 	/**
@@ -108,8 +108,8 @@ 
 	 *
 	 * @return array
 	 */
-	public function filter_comment_query_clauses( $clauses, $query ) {
-		if ( ! empty( $query->query_vars['action_log_filter'] ) ) {
+	public function filter_comment_query_clauses($clauses, $query) {
+		if ( ! empty($query->query_vars['action_log_filter'])) {
 			$clauses['where'] .= $this->get_where_clause();
 		}
 		return $clauses;
@@ -124,8 +124,8 @@ 
 	 *
 	 * @return string
 	 */
-	public function filter_comment_feed( $where, $query ) {
-		if ( is_comment_feed() ) {
+	public function filter_comment_feed($where, $query) {
+		if (is_comment_feed()) {
 			$where .= $this->get_where_clause();
 		}
 		return $where;
@@ -138,7 +138,7 @@ 
 	 */
 	protected function get_where_clause() {
 		global $wpdb;
-		return sprintf( " AND {$wpdb->comments}.comment_type != '%s'", self::TYPE );
+		return sprintf(" AND {$wpdb->comments}.comment_type != '%s'", self::TYPE);
 	}
 
 	/**
@@ -149,10 +149,10 @@ 
 	 *
 	 * @return object
 	 */
-	public function filter_comment_count( $stats, $post_id ) {
+	public function filter_comment_count($stats, $post_id) {
 		global $wpdb;
 
-		if ( 0 === $post_id ) {
+		if (0 === $post_id) {
 			$stats = $this->get_comment_count();
 		}
 
@@ -167,12 +167,12 @@ 
 	protected function get_comment_count() {
 		global $wpdb;
 
-		$stats = get_transient( 'as_comment_count' );
+		$stats = get_transient('as_comment_count');
 
-		if ( ! $stats ) {
+		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 );
+			$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();
@@ -184,27 +184,27 @@ 
 				'post-trashed' => 'post-trashed',
 			);
 
-			foreach ( (array) $count as $row ) {
+			foreach ((array) $count as $row) {
 				// Don't count post-trashed toward totals
-				if ( 'post-trashed' != $row['comment_approved'] && 'trash' != $row['comment_approved'] ) {
+				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'];
+				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;
+			foreach ($approved as $key) {
+				if (empty($stats[$key])) {
+					$stats[$key] = 0;
 				}
 			}
 
 			$stats = (object) $stats;
-			set_transient( 'as_comment_count', $stats );
+			set_transient('as_comment_count', $stats);
 		}
 
 		return $stats;
@@ -215,32 +215,32 @@ 
 	 * will be regenerated next time ActionScheduler_wpCommentLogger::filter_comment_count() is called.
 	 */
 	public function delete_comment_count_cache() {
-		delete_transient( 'as_comment_count' );
+		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 );
+		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 );
+		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' ) );
+		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 );
+		wp_defer_comment_counting(true);
 	}
 	public function enable_comment_counting() {
-		wp_defer_comment_counting( false );
+		wp_defer_comment_counting(false);
 	}
 
 }

src/ext/action-scheduler/classes/data-stores/ActionScheduler_DBLogger.php

Indentation +142 added lines, -142 removed lines

@@ -9,146 +9,146 @@
  */
 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 ( $record->log_date_gmt === null ) {
-			$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', 'action-scheduler' );
-		$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
-	}
+    /**
+     * 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 ( $record->log_date_gmt === null ) {
+            $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', 'action-scheduler' );
+        $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
+    }
 }

Spacing +32 added lines, -32 removed lines

@@ -18,16 +18,16 @@ 
 	 *
 	 * @return int     The log entry ID.
 	 */
-	public function log( $action_id, $message, DateTime $date = null ) {
-		if ( empty( $date ) ) {
+	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' );
+		$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;
@@ -39,7 +39,7 @@ 
 				'log_date_gmt'   => $date_gmt,
 				'log_date_local' => $date_local,
 			),
-			array( '%d', '%s', '%s', '%s' )
+			array('%d', '%s', '%s', '%s')
 		);
 
 		return $wpdb->insert_id;
@@ -52,12 +52,12 @@ 
 	 *
 	 * @return ActionScheduler_LogEntry
 	 */
-	public function get_entry( $entry_id ) {
+	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 ) );
+		$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 );
+		return $this->create_entry_from_db_record($entry);
 	}
 
 	/**
@@ -67,18 +67,18 @@ 
 	 *
 	 * @return ActionScheduler_LogEntry
 	 */
-	private function create_entry_from_db_record( $record ) {
-		if ( empty( $record ) ) {
+	private function create_entry_from_db_record($record) {
+		if (empty($record)) {
 			return new ActionScheduler_NullLogEntry();
 		}
 
-		if ( $record->log_date_gmt === null ) {
-			$date = as_get_datetime_object( ActionScheduler_StoreSchema::DEFAULT_DATE );
+		if ($record->log_date_gmt === null) {
+			$date = as_get_datetime_object(ActionScheduler_StoreSchema::DEFAULT_DATE);
 		} else {
-			$date = as_get_datetime_object( $record->log_date_gmt );
+			$date = as_get_datetime_object($record->log_date_gmt);
 		}
 
-		return new ActionScheduler_LogEntry( $record->action_id, $record->message, $date );
+		return new ActionScheduler_LogEntry($record->action_id, $record->message, $date);
 	}
 
 	/**
@@ -88,13 +88,13 @@ 
 	 *
 	 * @return ActionScheduler_LogEntry[]
 	 */
-	public function get_logs( $action_id ) {
+	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 ) );
+		$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 );
+		return array_map(array($this, 'create_entry_from_db_record'), $records);
 	}
 
 	/**
@@ -109,7 +109,7 @@ 
 
 		parent::init();
 
-		add_action( 'action_scheduler_deleted_action', array( $this, 'clear_deleted_action_logs' ), 10, 1 );
+		add_action('action_scheduler_deleted_action', array($this, 'clear_deleted_action_logs'), 10, 1);
 	}
 
 	/**
@@ -117,10 +117,10 @@ 
 	 *
 	 * @param int $action_id Action ID.
 	 */
-	public function clear_deleted_action_logs( $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' ) );
+		$wpdb->delete($wpdb->actionscheduler_logs, array('action_id' => $action_id), array('%d'));
 	}
 
 	/**
@@ -128,27 +128,27 @@ 
 	 *
 	 * @param array $action_ids List of action ID.
 	 */
-	public function bulk_log_cancel_actions( $action_ids ) {
-		if ( empty( $action_ids ) ) {
+	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', 'action-scheduler' );
-		$format     = '(%d, ' . $wpdb->prepare( '%s, %s, %s', $message, $date_gmt, $date_local ) . ')';
+		$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', 'action-scheduler');
+		$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
+		foreach ($action_ids as $action_id) {
+			$value_rows[] = $wpdb->prepare($format, $action_id); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 		}
-		$sql_query .= implode( ',', $value_rows );
+		$sql_query .= implode(',', $value_rows);
 
-		$wpdb->query( $sql_query ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+		$wpdb->query($sql_query); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 	}
 }

src/ext/action-scheduler/classes/data-stores/ActionScheduler_wpPostStore.php

Indentation +1059 added lines, -1059 removed lines

@@ -4,1072 +4,1072 @@
  * Class ActionScheduler_wpPostStore
  */
 class ActionScheduler_wpPostStore extends ActionScheduler_Store {
-	const POST_TYPE         = 'scheduled-action';
-	const GROUP_TAXONOMY    = 'action-group';
-	const SCHEDULE_META_KEY = '_action_manager_schedule';
-	const DEPENDENCIES_MET  = 'as-post-store-dependencies-met';
-
-	/**
-	 * 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;
-
-	/**
-	 * Local Timezone.
-	 *
-	 * @var DateTimeZone
-	 */
-	protected $local_timezone = null;
-
-	/**
-	 * Save action.
-	 *
-	 * @param ActionScheduler_Action $action Scheduled Action.
-	 * @param DateTime               $scheduled_date Scheduled Date.
-	 *
-	 * @throws RuntimeException Throws an exception if the action could not be saved.
-	 * @return int
-	 */
-	public function save_action( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
-		try {
-			$this->validate_action( $action );
-			$post_array = $this->create_post_array( $action, $scheduled_date );
-			$post_id    = $this->save_post_array( $post_array );
-			$this->save_post_schedule( $post_id, $action->get_schedule() );
-			$this->save_action_group( $post_id, $action->get_group() );
-			do_action( 'action_scheduler_stored_action', $post_id );
-			return $post_id;
-		} catch ( Exception $e ) {
-			/* translators: %s: action error message */
-			throw new RuntimeException( sprintf( __( 'Error saving action: %s', 'action-scheduler' ), $e->getMessage() ), 0 );
-		}
-	}
-
-	/**
-	 * Create post array.
-	 *
-	 * @param ActionScheduler_Action $action Scheduled Action.
-	 * @param DateTime               $scheduled_date Scheduled Date.
-	 *
-	 * @return array Returns an array of post data.
-	 */
-	protected function create_post_array( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
-		$post = array(
-			'post_type'     => self::POST_TYPE,
-			'post_title'    => $action->get_hook(),
-			'post_content'  => wp_json_encode( $action->get_args() ),
-			'post_status'   => ( $action->is_finished() ? 'publish' : 'pending' ),
-			'post_date_gmt' => $this->get_scheduled_date_string( $action, $scheduled_date ),
-			'post_date'     => $this->get_scheduled_date_string_local( $action, $scheduled_date ),
-		);
-		return $post;
-	}
-
-	/**
-	 * Save post array.
-	 *
-	 * @param array $post_array Post array.
-	 * @return int Returns the post ID.
-	 * @throws RuntimeException Throws an exception if the action could not be saved.
-	 */
-	protected function save_post_array( $post_array ) {
-		add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
-
-		$has_kses = false !== has_filter( 'content_save_pre', 'wp_filter_post_kses' );
-
-		if ( $has_kses ) {
-			// Prevent KSES from corrupting JSON in post_content.
-			kses_remove_filters();
-		}
-
-		$post_id = wp_insert_post( $post_array );
-
-		if ( $has_kses ) {
-			kses_init_filters();
-		}
-
-		remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
-
-		if ( is_wp_error( $post_id ) || empty( $post_id ) ) {
-			throw new RuntimeException( __( 'Unable to save action.', 'action-scheduler' ) );
-		}
-		return $post_id;
-	}
-
-	/**
-	 * Filter insert post data.
-	 *
-	 * @param array $postdata Post data to filter.
-	 *
-	 * @return array
-	 */
-	public function filter_insert_post_data( $postdata ) {
-		if ( self::POST_TYPE === $postdata['post_type'] ) {
-			$postdata['post_author'] = 0;
-			if ( 'future' === $postdata['post_status'] ) {
-				$postdata['post_status'] = 'publish';
-			}
-		}
-		return $postdata;
-	}
-
-	/**
-	 * Create a (probably unique) post name for scheduled actions in a more performant manner than wp_unique_post_slug().
-	 *
-	 * When an action's post status is transitioned to something other than 'draft', 'pending' or 'auto-draft, like 'publish'
-	 * or 'failed' or 'trash', WordPress will find a unique slug (stored in post_name column) using the wp_unique_post_slug()
-	 * function. This is done to ensure URL uniqueness. The approach taken by wp_unique_post_slug() is to iterate over existing
-	 * post_name values that match, and append a number 1 greater than the largest. This makes sense when manually creating a
-	 * post from the Edit Post screen. It becomes a bottleneck when automatically processing thousands of actions, with a
-	 * database containing thousands of related post_name values.
-	 *
-	 * WordPress 5.1 introduces the 'pre_wp_unique_post_slug' filter for plugins to address this issue.
-	 *
-	 * We can short-circuit WordPress's wp_unique_post_slug() approach using the 'pre_wp_unique_post_slug' filter. This
-	 * method is available to be used as a callback on that filter. It provides a more scalable approach to generating a
-	 * post_name/slug that is probably unique. Because Action Scheduler never actually uses the post_name field, or an
-	 * action's slug, being probably unique is good enough.
-	 *
-	 * For more backstory on this issue, see:
-	 * - https://github.com/woocommerce/action-scheduler/issues/44 and
-	 * - https://core.trac.wordpress.org/ticket/21112
-	 *
-	 * @param string $override_slug Short-circuit return value.
-	 * @param string $slug          The desired slug (post_name).
-	 * @param int    $post_ID       Post ID.
-	 * @param string $post_status   The post status.
-	 * @param string $post_type     Post type.
-	 * @return string
-	 */
-	public function set_unique_post_slug( $override_slug, $slug, $post_ID, $post_status, $post_type ) {
-		if ( self::POST_TYPE === $post_type ) {
-			$override_slug = uniqid( self::POST_TYPE . '-', true ) . '-' . wp_generate_password( 32, false );
-		}
-		return $override_slug;
-	}
-
-	/**
-	 * Save post schedule.
-	 *
-	 * @param int    $post_id  Post ID of the scheduled action.
-	 * @param string $schedule Schedule to save.
-	 *
-	 * @return void
-	 */
-	protected function save_post_schedule( $post_id, $schedule ) {
-		update_post_meta( $post_id, self::SCHEDULE_META_KEY, $schedule );
-	}
-
-	/**
-	 * Save action group.
-	 *
-	 * @param int    $post_id Post ID.
-	 * @param string $group   Group to save.
-	 * @return void
-	 */
-	protected function save_action_group( $post_id, $group ) {
-		if ( empty( $group ) ) {
-			wp_set_object_terms( $post_id, array(), self::GROUP_TAXONOMY, false );
-		} else {
-			wp_set_object_terms( $post_id, array( $group ), self::GROUP_TAXONOMY, false );
-		}
-	}
-
-	/**
-	 * Fetch actions.
-	 *
-	 * @param int $action_id Action ID.
-	 * @return object
-	 */
-	public function fetch_action( $action_id ) {
-		$post = $this->get_post( $action_id );
-		if ( empty( $post ) || self::POST_TYPE !== $post->post_type ) {
-			return $this->get_null_action();
-		}
-
-		try {
-			$action = $this->make_action_from_post( $post );
-		} catch ( ActionScheduler_InvalidActionException $exception ) {
-			do_action( 'action_scheduler_failed_fetch_action', $post->ID, $exception );
-			return $this->get_null_action();
-		}
-
-		return $action;
-	}
-
-	/**
-	 * Get post.
-	 *
-	 * @param string $action_id - Action ID.
-	 * @return WP_Post|null
-	 */
-	protected function get_post( $action_id ) {
-		if ( empty( $action_id ) ) {
-			return null;
-		}
-		return get_post( $action_id );
-	}
-
-	/**
-	 * Get NULL action.
-	 *
-	 * @return ActionScheduler_NullAction
-	 */
-	protected function get_null_action() {
-		return new ActionScheduler_NullAction();
-	}
-
-	/**
-	 * Make action from post.
-	 *
-	 * @param WP_Post $post Post object.
-	 * @return WP_Post
-	 */
-	protected function make_action_from_post( $post ) {
-		$hook = $post->post_title;
-
-		$args = json_decode( $post->post_content, true );
-		$this->validate_args( $args, $post->ID );
-
-		$schedule = get_post_meta( $post->ID, self::SCHEDULE_META_KEY, true );
-		$this->validate_schedule( $schedule, $post->ID );
-
-		$group = wp_get_object_terms( $post->ID, self::GROUP_TAXONOMY, array( 'fields' => 'names' ) );
-		$group = empty( $group ) ? '' : reset( $group );
-
-		return ActionScheduler::factory()->get_stored_action( $this->get_action_status_by_post_status( $post->post_status ), $hook, $args, $schedule, $group );
-	}
-
-	/**
-	 * Get action status by post status.
-	 *
-	 * @param string $post_status Post status.
-	 *
-	 * @throws InvalidArgumentException Throw InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
-	 * @return string
-	 */
-	protected function get_action_status_by_post_status( $post_status ) {
-
-		switch ( $post_status ) {
-			case 'publish':
-				$action_status = self::STATUS_COMPLETE;
-				break;
-			case 'trash':
-				$action_status = self::STATUS_CANCELED;
-				break;
-			default:
-				if ( ! array_key_exists( $post_status, $this->get_status_labels() ) ) {
-					throw new InvalidArgumentException( sprintf( 'Invalid post status: "%s". No matching action status available.', $post_status ) );
-				}
-				$action_status = $post_status;
-				break;
-		}
-
-		return $action_status;
-	}
-
-	/**
-	 * Get post status by action status.
-	 *
-	 * @param string $action_status Action status.
-	 *
-	 * @throws InvalidArgumentException Throws InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
-	 * @return string
-	 */
-	protected function get_post_status_by_action_status( $action_status ) {
-
-		switch ( $action_status ) {
-			case self::STATUS_COMPLETE:
-				$post_status = 'publish';
-				break;
-			case self::STATUS_CANCELED:
-				$post_status = 'trash';
-				break;
-			default:
-				if ( ! array_key_exists( $action_status, $this->get_status_labels() ) ) {
-					throw new InvalidArgumentException( sprintf( 'Invalid action status: "%s".', $action_status ) );
-				}
-				$post_status = $action_status;
-				break;
-		}
-
-		return $post_status;
-	}
-
-	/**
-	 * Returns the SQL statement to query (or count) actions.
-	 *
-	 * @param array  $query            - Filtering options.
-	 * @param string $select_or_count  - Whether the SQL should select and return the IDs or just the row count.
-	 *
-	 * @throws InvalidArgumentException - Throw InvalidArgumentException if $select_or_count not count or select.
-	 * @return string SQL statement. The returned SQL is already properly escaped.
-	 */
-	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 schedule. Cannot save action.', 'action-scheduler' ) );
-		}
-
-		$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',
-				'search'           => '',
-			)
-		);
-
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-		$sql        = ( 'count' === $select_or_count ) ? 'SELECT count(p.ID)' : 'SELECT p.ID ';
-		$sql       .= "FROM {$wpdb->posts} p";
-		$sql_params = array();
-		if ( empty( $query['group'] ) && 'group' === $query['orderby'] ) {
-			$sql .= " LEFT JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
-			$sql .= " LEFT JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
-			$sql .= " LEFT JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
-		} elseif ( ! empty( $query['group'] ) ) {
-			$sql         .= " INNER JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
-			$sql         .= " INNER JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
-			$sql         .= " INNER JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
-			$sql         .= ' AND t.slug=%s';
-			$sql_params[] = $query['group'];
-		}
-		$sql         .= ' WHERE post_type=%s';
-		$sql_params[] = self::POST_TYPE;
-		if ( $query['hook'] ) {
-			$sql         .= ' AND p.post_title=%s';
-			$sql_params[] = $query['hook'];
-		}
-		if ( $query['args'] !== null ) {
-			$sql         .= ' AND p.post_content=%s';
-			$sql_params[] = wp_json_encode( $query['args'] );
-		}
-
-		if ( $query['status'] ) {
-			$post_statuses = array_map( array( $this, 'get_post_status_by_action_status' ), (array) $query['status'] );
-			$placeholders  = array_fill( 0, count( $post_statuses ), '%s' );
-			$sql          .= ' AND p.post_status IN (' . join( ', ', $placeholders ) . ')';
-			$sql_params    = array_merge( $sql_params, array_values( $post_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 p.post_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 p.post_modified_gmt $comparator %s";
-			$sql_params[] = $date_string;
-		}
-
-		if ( true === $query['claimed'] ) {
-			$sql .= " AND p.post_password != ''";
-		} elseif ( false === $query['claimed'] ) {
-			$sql .= " AND p.post_password = ''";
-		} elseif ( $query['claimed'] !== null ) {
-			$sql         .= ' AND p.post_password = %s';
-			$sql_params[] = $query['claimed'];
-		}
-
-		if ( ! empty( $query['search'] ) ) {
-			$sql .= ' AND (p.post_title LIKE %s OR p.post_content LIKE %s OR p.post_password LIKE %s)';
-			for ( $i = 0; $i < 3; $i++ ) {
-				$sql_params[] = sprintf( '%%%s%%', $query['search'] );
-			}
-		}
-
-		if ( 'select' === $select_or_count ) {
-			switch ( $query['orderby'] ) {
-				case 'hook':
-					$orderby = 'p.post_title';
-					break;
-				case 'group':
-					$orderby = 't.name';
-					break;
-				case 'status':
-					$orderby = 'p.post_status';
-					break;
-				case 'modified':
-					$orderby = 'p.post_modified';
-					break;
-				case 'claim_id':
-					$orderby = 'p.post_password';
-					break;
-				case 'schedule':
-				case 'date':
-				default:
-					$orderby = 'p.post_date_gmt';
-					break;
-			}
-			if ( 'ASC' === strtoupper( $query['order'] ) ) {
-				$order = 'ASC';
-			} else {
-				$order = 'DESC';
-			}
-			$sql .= " ORDER BY $orderby $order";
-			if ( $query['per_page'] > 0 ) {
-				$sql         .= ' LIMIT %d, %d';
-				$sql_params[] = $query['offset'];
-				$sql_params[] = $query['per_page'];
-			}
-		}
-
-		return $wpdb->prepare( $sql, $sql_params ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
-	}
-
-	/**
-	 * Query for action count or list of action IDs.
-	 *
-	 * @since 3.3.0 $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' ) {
-		/**
-		 * Global $wpdb object.
-		 *
-		 * @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.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching,WordPress.DB.PreparedSQL.NotPrepared
-	}
-
-	/**
-	 * Get a count of all actions in the store, grouped by status
-	 *
-	 * @return array
-	 */
-	public function action_counts() {
-
-		$action_counts_by_status = array();
-		$action_stati_and_labels = $this->get_status_labels();
-		$posts_count_by_status   = (array) wp_count_posts( self::POST_TYPE, 'readable' );
-
-		foreach ( $posts_count_by_status as $post_status_name => $count ) {
-
-			try {
-				$action_status_name = $this->get_action_status_by_post_status( $post_status_name );
-			} catch ( Exception $e ) {
-				// Ignore any post statuses that aren't for actions.
-				continue;
-			}
-			if ( array_key_exists( $action_status_name, $action_stati_and_labels ) ) {
-				$action_counts_by_status[ $action_status_name ] = $count;
-			}
-		}
-
-		return $action_counts_by_status;
-	}
-
-	/**
-	 * Cancel action.
-	 *
-	 * @param int $action_id Action ID.
-	 *
-	 * @throws InvalidArgumentException If $action_id is not identified.
-	 */
-	public function cancel_action( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
-			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
-		}
-		do_action( 'action_scheduler_canceled_action', $action_id );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
-		wp_trash_post( $action_id );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
-	}
-
-	/**
-	 * Delete action.
-	 *
-	 * @param int $action_id Action ID.
-	 * @return void
-	 * @throws InvalidArgumentException If action is not identified.
-	 */
-	public function delete_action( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
-			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
-		}
-		do_action( 'action_scheduler_deleted_action', $action_id );
-
-		wp_delete_post( $action_id, true );
-	}
-
-	/**
-	 * Get date for claim id.
-	 *
-	 * @param int $action_id Action ID.
-	 * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
-	 */
-	public function get_date( $action_id ) {
-		$next = $this->get_date_gmt( $action_id );
-		return ActionScheduler_TimezoneHelper::set_local_timezone( $next );
-	}
-
-	/**
-	 * Get Date GMT.
-	 *
-	 * @param int $action_id Action ID.
-	 *
-	 * @throws InvalidArgumentException If $action_id is not identified.
-	 * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
-	 */
-	public function get_date_gmt( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
-			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
-		}
-		if ( 'publish' === $post->post_status ) {
-			return as_get_datetime_object( $post->post_modified_gmt );
-		} else {
-			return as_get_datetime_object( $post->post_date_gmt );
-		}
-	}
-
-	/**
-	 * Stake claim.
-	 *
-	 * @param int      $max_actions Maximum number of actions.
-	 * @param DateTime $before_date Jobs must be schedule before this date. Defaults to now.
-	 * @param array    $hooks       Claim only actions with a hook or hooks.
-	 * @param string   $group       Claim only actions in the given group.
-	 *
-	 * @return ActionScheduler_ActionClaim
-	 * @throws RuntimeException When there is an error staking a claim.
-	 * @throws InvalidArgumentException When the given group is not valid.
-	 */
-	public function stake_claim( $max_actions = 10, DateTime $before_date = null, $hooks = array(), $group = '' ) {
-		$this->claim_before_date = $before_date;
-		$claim_id                = $this->generate_claim_id();
-		$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 );
-	}
-
-	/**
-	 * Get claim count.
-	 *
-	 * @return int
-	 */
-	public function get_claim_count() {
-		global $wpdb;
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching
-		return $wpdb->get_var(
-			$wpdb->prepare(
-				"SELECT COUNT(DISTINCT post_password) FROM {$wpdb->posts} WHERE post_password != '' AND post_type = %s AND post_status IN ('in-progress','pending')",
-				array( self::POST_TYPE )
-			)
-		);
-	}
-
-	/**
-	 * Generate claim id.
-	 *
-	 * @return string
-	 */
-	protected function generate_claim_id() {
-		$claim_id = md5( microtime( true ) . wp_rand( 0, 1000 ) );
-		return substr( $claim_id, 0, 20 ); // to fit in db field with 20 char limit.
-	}
-
-	/**
-	 * Claim actions.
-	 *
-	 * @param string   $claim_id    Claim ID.
-	 * @param int      $limit       Limit.
-	 * @param DateTime $before_date Should use UTC timezone.
-	 * @param array    $hooks       Claim only actions with a hook or hooks.
-	 * @param string   $group       Claim only actions in the given group.
-	 *
-	 * @return int The number of actions that were claimed.
-	 * @throws RuntimeException  When there is a database error.
-	 */
-	protected function claim_actions( $claim_id, $limit, DateTime $before_date = null, $hooks = array(), $group = '' ) {
-		// Set up initial variables.
-		$date      = null === $before_date ? as_get_datetime_object() : clone $before_date;
-		$limit_ids = ! empty( $group );
-		$ids       = $limit_ids ? $this->get_actions_by_group( $group, $limit, $date ) : array();
-
-		// If limiting by IDs and no posts found, then return early since we have nothing to update.
-		if ( $limit_ids && 0 === count( $ids ) ) {
-			return 0;
-		}
-
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		/*
+    const POST_TYPE         = 'scheduled-action';
+    const GROUP_TAXONOMY    = 'action-group';
+    const SCHEDULE_META_KEY = '_action_manager_schedule';
+    const DEPENDENCIES_MET  = 'as-post-store-dependencies-met';
+
+    /**
+     * 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;
+
+    /**
+     * Local Timezone.
+     *
+     * @var DateTimeZone
+     */
+    protected $local_timezone = null;
+
+    /**
+     * Save action.
+     *
+     * @param ActionScheduler_Action $action Scheduled Action.
+     * @param DateTime               $scheduled_date Scheduled Date.
+     *
+     * @throws RuntimeException Throws an exception if the action could not be saved.
+     * @return int
+     */
+    public function save_action( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
+        try {
+            $this->validate_action( $action );
+            $post_array = $this->create_post_array( $action, $scheduled_date );
+            $post_id    = $this->save_post_array( $post_array );
+            $this->save_post_schedule( $post_id, $action->get_schedule() );
+            $this->save_action_group( $post_id, $action->get_group() );
+            do_action( 'action_scheduler_stored_action', $post_id );
+            return $post_id;
+        } catch ( Exception $e ) {
+            /* translators: %s: action error message */
+            throw new RuntimeException( sprintf( __( 'Error saving action: %s', 'action-scheduler' ), $e->getMessage() ), 0 );
+        }
+    }
+
+    /**
+     * Create post array.
+     *
+     * @param ActionScheduler_Action $action Scheduled Action.
+     * @param DateTime               $scheduled_date Scheduled Date.
+     *
+     * @return array Returns an array of post data.
+     */
+    protected function create_post_array( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
+        $post = array(
+            'post_type'     => self::POST_TYPE,
+            'post_title'    => $action->get_hook(),
+            'post_content'  => wp_json_encode( $action->get_args() ),
+            'post_status'   => ( $action->is_finished() ? 'publish' : 'pending' ),
+            'post_date_gmt' => $this->get_scheduled_date_string( $action, $scheduled_date ),
+            'post_date'     => $this->get_scheduled_date_string_local( $action, $scheduled_date ),
+        );
+        return $post;
+    }
+
+    /**
+     * Save post array.
+     *
+     * @param array $post_array Post array.
+     * @return int Returns the post ID.
+     * @throws RuntimeException Throws an exception if the action could not be saved.
+     */
+    protected function save_post_array( $post_array ) {
+        add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
+        add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
+
+        $has_kses = false !== has_filter( 'content_save_pre', 'wp_filter_post_kses' );
+
+        if ( $has_kses ) {
+            // Prevent KSES from corrupting JSON in post_content.
+            kses_remove_filters();
+        }
+
+        $post_id = wp_insert_post( $post_array );
+
+        if ( $has_kses ) {
+            kses_init_filters();
+        }
+
+        remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
+        remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
+
+        if ( is_wp_error( $post_id ) || empty( $post_id ) ) {
+            throw new RuntimeException( __( 'Unable to save action.', 'action-scheduler' ) );
+        }
+        return $post_id;
+    }
+
+    /**
+     * Filter insert post data.
+     *
+     * @param array $postdata Post data to filter.
+     *
+     * @return array
+     */
+    public function filter_insert_post_data( $postdata ) {
+        if ( self::POST_TYPE === $postdata['post_type'] ) {
+            $postdata['post_author'] = 0;
+            if ( 'future' === $postdata['post_status'] ) {
+                $postdata['post_status'] = 'publish';
+            }
+        }
+        return $postdata;
+    }
+
+    /**
+     * Create a (probably unique) post name for scheduled actions in a more performant manner than wp_unique_post_slug().
+     *
+     * When an action's post status is transitioned to something other than 'draft', 'pending' or 'auto-draft, like 'publish'
+     * or 'failed' or 'trash', WordPress will find a unique slug (stored in post_name column) using the wp_unique_post_slug()
+     * function. This is done to ensure URL uniqueness. The approach taken by wp_unique_post_slug() is to iterate over existing
+     * post_name values that match, and append a number 1 greater than the largest. This makes sense when manually creating a
+     * post from the Edit Post screen. It becomes a bottleneck when automatically processing thousands of actions, with a
+     * database containing thousands of related post_name values.
+     *
+     * WordPress 5.1 introduces the 'pre_wp_unique_post_slug' filter for plugins to address this issue.
+     *
+     * We can short-circuit WordPress's wp_unique_post_slug() approach using the 'pre_wp_unique_post_slug' filter. This
+     * method is available to be used as a callback on that filter. It provides a more scalable approach to generating a
+     * post_name/slug that is probably unique. Because Action Scheduler never actually uses the post_name field, or an
+     * action's slug, being probably unique is good enough.
+     *
+     * For more backstory on this issue, see:
+     * - https://github.com/woocommerce/action-scheduler/issues/44 and
+     * - https://core.trac.wordpress.org/ticket/21112
+     *
+     * @param string $override_slug Short-circuit return value.
+     * @param string $slug          The desired slug (post_name).
+     * @param int    $post_ID       Post ID.
+     * @param string $post_status   The post status.
+     * @param string $post_type     Post type.
+     * @return string
+     */
+    public function set_unique_post_slug( $override_slug, $slug, $post_ID, $post_status, $post_type ) {
+        if ( self::POST_TYPE === $post_type ) {
+            $override_slug = uniqid( self::POST_TYPE . '-', true ) . '-' . wp_generate_password( 32, false );
+        }
+        return $override_slug;
+    }
+
+    /**
+     * Save post schedule.
+     *
+     * @param int    $post_id  Post ID of the scheduled action.
+     * @param string $schedule Schedule to save.
+     *
+     * @return void
+     */
+    protected function save_post_schedule( $post_id, $schedule ) {
+        update_post_meta( $post_id, self::SCHEDULE_META_KEY, $schedule );
+    }
+
+    /**
+     * Save action group.
+     *
+     * @param int    $post_id Post ID.
+     * @param string $group   Group to save.
+     * @return void
+     */
+    protected function save_action_group( $post_id, $group ) {
+        if ( empty( $group ) ) {
+            wp_set_object_terms( $post_id, array(), self::GROUP_TAXONOMY, false );
+        } else {
+            wp_set_object_terms( $post_id, array( $group ), self::GROUP_TAXONOMY, false );
+        }
+    }
+
+    /**
+     * Fetch actions.
+     *
+     * @param int $action_id Action ID.
+     * @return object
+     */
+    public function fetch_action( $action_id ) {
+        $post = $this->get_post( $action_id );
+        if ( empty( $post ) || self::POST_TYPE !== $post->post_type ) {
+            return $this->get_null_action();
+        }
+
+        try {
+            $action = $this->make_action_from_post( $post );
+        } catch ( ActionScheduler_InvalidActionException $exception ) {
+            do_action( 'action_scheduler_failed_fetch_action', $post->ID, $exception );
+            return $this->get_null_action();
+        }
+
+        return $action;
+    }
+
+    /**
+     * Get post.
+     *
+     * @param string $action_id - Action ID.
+     * @return WP_Post|null
+     */
+    protected function get_post( $action_id ) {
+        if ( empty( $action_id ) ) {
+            return null;
+        }
+        return get_post( $action_id );
+    }
+
+    /**
+     * Get NULL action.
+     *
+     * @return ActionScheduler_NullAction
+     */
+    protected function get_null_action() {
+        return new ActionScheduler_NullAction();
+    }
+
+    /**
+     * Make action from post.
+     *
+     * @param WP_Post $post Post object.
+     * @return WP_Post
+     */
+    protected function make_action_from_post( $post ) {
+        $hook = $post->post_title;
+
+        $args = json_decode( $post->post_content, true );
+        $this->validate_args( $args, $post->ID );
+
+        $schedule = get_post_meta( $post->ID, self::SCHEDULE_META_KEY, true );
+        $this->validate_schedule( $schedule, $post->ID );
+
+        $group = wp_get_object_terms( $post->ID, self::GROUP_TAXONOMY, array( 'fields' => 'names' ) );
+        $group = empty( $group ) ? '' : reset( $group );
+
+        return ActionScheduler::factory()->get_stored_action( $this->get_action_status_by_post_status( $post->post_status ), $hook, $args, $schedule, $group );
+    }
+
+    /**
+     * Get action status by post status.
+     *
+     * @param string $post_status Post status.
+     *
+     * @throws InvalidArgumentException Throw InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
+     * @return string
+     */
+    protected function get_action_status_by_post_status( $post_status ) {
+
+        switch ( $post_status ) {
+            case 'publish':
+                $action_status = self::STATUS_COMPLETE;
+                break;
+            case 'trash':
+                $action_status = self::STATUS_CANCELED;
+                break;
+            default:
+                if ( ! array_key_exists( $post_status, $this->get_status_labels() ) ) {
+                    throw new InvalidArgumentException( sprintf( 'Invalid post status: "%s". No matching action status available.', $post_status ) );
+                }
+                $action_status = $post_status;
+                break;
+        }
+
+        return $action_status;
+    }
+
+    /**
+     * Get post status by action status.
+     *
+     * @param string $action_status Action status.
+     *
+     * @throws InvalidArgumentException Throws InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
+     * @return string
+     */
+    protected function get_post_status_by_action_status( $action_status ) {
+
+        switch ( $action_status ) {
+            case self::STATUS_COMPLETE:
+                $post_status = 'publish';
+                break;
+            case self::STATUS_CANCELED:
+                $post_status = 'trash';
+                break;
+            default:
+                if ( ! array_key_exists( $action_status, $this->get_status_labels() ) ) {
+                    throw new InvalidArgumentException( sprintf( 'Invalid action status: "%s".', $action_status ) );
+                }
+                $post_status = $action_status;
+                break;
+        }
+
+        return $post_status;
+    }
+
+    /**
+     * Returns the SQL statement to query (or count) actions.
+     *
+     * @param array  $query            - Filtering options.
+     * @param string $select_or_count  - Whether the SQL should select and return the IDs or just the row count.
+     *
+     * @throws InvalidArgumentException - Throw InvalidArgumentException if $select_or_count not count or select.
+     * @return string SQL statement. The returned SQL is already properly escaped.
+     */
+    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 schedule. Cannot save action.', 'action-scheduler' ) );
+        }
+
+        $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',
+                'search'           => '',
+            )
+        );
+
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+        $sql        = ( 'count' === $select_or_count ) ? 'SELECT count(p.ID)' : 'SELECT p.ID ';
+        $sql       .= "FROM {$wpdb->posts} p";
+        $sql_params = array();
+        if ( empty( $query['group'] ) && 'group' === $query['orderby'] ) {
+            $sql .= " LEFT JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
+            $sql .= " LEFT JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
+            $sql .= " LEFT JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
+        } elseif ( ! empty( $query['group'] ) ) {
+            $sql         .= " INNER JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
+            $sql         .= " INNER JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
+            $sql         .= " INNER JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
+            $sql         .= ' AND t.slug=%s';
+            $sql_params[] = $query['group'];
+        }
+        $sql         .= ' WHERE post_type=%s';
+        $sql_params[] = self::POST_TYPE;
+        if ( $query['hook'] ) {
+            $sql         .= ' AND p.post_title=%s';
+            $sql_params[] = $query['hook'];
+        }
+        if ( $query['args'] !== null ) {
+            $sql         .= ' AND p.post_content=%s';
+            $sql_params[] = wp_json_encode( $query['args'] );
+        }
+
+        if ( $query['status'] ) {
+            $post_statuses = array_map( array( $this, 'get_post_status_by_action_status' ), (array) $query['status'] );
+            $placeholders  = array_fill( 0, count( $post_statuses ), '%s' );
+            $sql          .= ' AND p.post_status IN (' . join( ', ', $placeholders ) . ')';
+            $sql_params    = array_merge( $sql_params, array_values( $post_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 p.post_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 p.post_modified_gmt $comparator %s";
+            $sql_params[] = $date_string;
+        }
+
+        if ( true === $query['claimed'] ) {
+            $sql .= " AND p.post_password != ''";
+        } elseif ( false === $query['claimed'] ) {
+            $sql .= " AND p.post_password = ''";
+        } elseif ( $query['claimed'] !== null ) {
+            $sql         .= ' AND p.post_password = %s';
+            $sql_params[] = $query['claimed'];
+        }
+
+        if ( ! empty( $query['search'] ) ) {
+            $sql .= ' AND (p.post_title LIKE %s OR p.post_content LIKE %s OR p.post_password LIKE %s)';
+            for ( $i = 0; $i < 3; $i++ ) {
+                $sql_params[] = sprintf( '%%%s%%', $query['search'] );
+            }
+        }
+
+        if ( 'select' === $select_or_count ) {
+            switch ( $query['orderby'] ) {
+                case 'hook':
+                    $orderby = 'p.post_title';
+                    break;
+                case 'group':
+                    $orderby = 't.name';
+                    break;
+                case 'status':
+                    $orderby = 'p.post_status';
+                    break;
+                case 'modified':
+                    $orderby = 'p.post_modified';
+                    break;
+                case 'claim_id':
+                    $orderby = 'p.post_password';
+                    break;
+                case 'schedule':
+                case 'date':
+                default:
+                    $orderby = 'p.post_date_gmt';
+                    break;
+            }
+            if ( 'ASC' === strtoupper( $query['order'] ) ) {
+                $order = 'ASC';
+            } else {
+                $order = 'DESC';
+            }
+            $sql .= " ORDER BY $orderby $order";
+            if ( $query['per_page'] > 0 ) {
+                $sql         .= ' LIMIT %d, %d';
+                $sql_params[] = $query['offset'];
+                $sql_params[] = $query['per_page'];
+            }
+        }
+
+        return $wpdb->prepare( $sql, $sql_params ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+    }
+
+    /**
+     * Query for action count or list of action IDs.
+     *
+     * @since 3.3.0 $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' ) {
+        /**
+         * Global $wpdb object.
+         *
+         * @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.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching,WordPress.DB.PreparedSQL.NotPrepared
+    }
+
+    /**
+     * Get a count of all actions in the store, grouped by status
+     *
+     * @return array
+     */
+    public function action_counts() {
+
+        $action_counts_by_status = array();
+        $action_stati_and_labels = $this->get_status_labels();
+        $posts_count_by_status   = (array) wp_count_posts( self::POST_TYPE, 'readable' );
+
+        foreach ( $posts_count_by_status as $post_status_name => $count ) {
+
+            try {
+                $action_status_name = $this->get_action_status_by_post_status( $post_status_name );
+            } catch ( Exception $e ) {
+                // Ignore any post statuses that aren't for actions.
+                continue;
+            }
+            if ( array_key_exists( $action_status_name, $action_stati_and_labels ) ) {
+                $action_counts_by_status[ $action_status_name ] = $count;
+            }
+        }
+
+        return $action_counts_by_status;
+    }
+
+    /**
+     * Cancel action.
+     *
+     * @param int $action_id Action ID.
+     *
+     * @throws InvalidArgumentException If $action_id is not identified.
+     */
+    public function cancel_action( $action_id ) {
+        $post = get_post( $action_id );
+        if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+            /* translators: %s is the action ID */
+            throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+        }
+        do_action( 'action_scheduler_canceled_action', $action_id );
+        add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
+        wp_trash_post( $action_id );
+        remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
+    }
+
+    /**
+     * Delete action.
+     *
+     * @param int $action_id Action ID.
+     * @return void
+     * @throws InvalidArgumentException If action is not identified.
+     */
+    public function delete_action( $action_id ) {
+        $post = get_post( $action_id );
+        if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+            /* translators: %s is the action ID */
+            throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+        }
+        do_action( 'action_scheduler_deleted_action', $action_id );
+
+        wp_delete_post( $action_id, true );
+    }
+
+    /**
+     * Get date for claim id.
+     *
+     * @param int $action_id Action ID.
+     * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
+     */
+    public function get_date( $action_id ) {
+        $next = $this->get_date_gmt( $action_id );
+        return ActionScheduler_TimezoneHelper::set_local_timezone( $next );
+    }
+
+    /**
+     * Get Date GMT.
+     *
+     * @param int $action_id Action ID.
+     *
+     * @throws InvalidArgumentException If $action_id is not identified.
+     * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
+     */
+    public function get_date_gmt( $action_id ) {
+        $post = get_post( $action_id );
+        if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+            /* translators: %s is the action ID */
+            throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+        }
+        if ( 'publish' === $post->post_status ) {
+            return as_get_datetime_object( $post->post_modified_gmt );
+        } else {
+            return as_get_datetime_object( $post->post_date_gmt );
+        }
+    }
+
+    /**
+     * Stake claim.
+     *
+     * @param int      $max_actions Maximum number of actions.
+     * @param DateTime $before_date Jobs must be schedule before this date. Defaults to now.
+     * @param array    $hooks       Claim only actions with a hook or hooks.
+     * @param string   $group       Claim only actions in the given group.
+     *
+     * @return ActionScheduler_ActionClaim
+     * @throws RuntimeException When there is an error staking a claim.
+     * @throws InvalidArgumentException When the given group is not valid.
+     */
+    public function stake_claim( $max_actions = 10, DateTime $before_date = null, $hooks = array(), $group = '' ) {
+        $this->claim_before_date = $before_date;
+        $claim_id                = $this->generate_claim_id();
+        $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 );
+    }
+
+    /**
+     * Get claim count.
+     *
+     * @return int
+     */
+    public function get_claim_count() {
+        global $wpdb;
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching
+        return $wpdb->get_var(
+            $wpdb->prepare(
+                "SELECT COUNT(DISTINCT post_password) FROM {$wpdb->posts} WHERE post_password != '' AND post_type = %s AND post_status IN ('in-progress','pending')",
+                array( self::POST_TYPE )
+            )
+        );
+    }
+
+    /**
+     * Generate claim id.
+     *
+     * @return string
+     */
+    protected function generate_claim_id() {
+        $claim_id = md5( microtime( true ) . wp_rand( 0, 1000 ) );
+        return substr( $claim_id, 0, 20 ); // to fit in db field with 20 char limit.
+    }
+
+    /**
+     * Claim actions.
+     *
+     * @param string   $claim_id    Claim ID.
+     * @param int      $limit       Limit.
+     * @param DateTime $before_date Should use UTC timezone.
+     * @param array    $hooks       Claim only actions with a hook or hooks.
+     * @param string   $group       Claim only actions in the given group.
+     *
+     * @return int The number of actions that were claimed.
+     * @throws RuntimeException  When there is a database error.
+     */
+    protected function claim_actions( $claim_id, $limit, DateTime $before_date = null, $hooks = array(), $group = '' ) {
+        // Set up initial variables.
+        $date      = null === $before_date ? as_get_datetime_object() : clone $before_date;
+        $limit_ids = ! empty( $group );
+        $ids       = $limit_ids ? $this->get_actions_by_group( $group, $limit, $date ) : array();
+
+        // If limiting by IDs and no posts found, then return early since we have nothing to update.
+        if ( $limit_ids && 0 === count( $ids ) ) {
+            return 0;
+        }
+
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        /*
 		 * Build up custom query to update the affected posts. Parameters are built as a separate array
 		 * to make it easier to identify where they are in the query.
 		 *
 		 * We can't use $wpdb->update() here because of the "ID IN ..." clause.
 		 */
-		$update = "UPDATE {$wpdb->posts} SET post_password = %s, post_modified_gmt = %s, post_modified = %s";
-		$params = array(
-			$claim_id,
-			current_time( 'mysql', true ),
-			current_time( 'mysql' ),
-		);
-
-		// Build initial WHERE clause.
-		$where    = "WHERE post_type = %s AND post_status = %s AND post_password = ''";
-		$params[] = self::POST_TYPE;
-		$params[] = ActionScheduler_Store::STATUS_PENDING;
-
-		if ( ! empty( $hooks ) ) {
-			$placeholders = array_fill( 0, count( $hooks ), '%s' );
-			$where       .= ' AND post_title IN (' . join( ', ', $placeholders ) . ')';
-			$params       = array_merge( $params, array_values( $hooks ) );
-		}
-
-		/*
+        $update = "UPDATE {$wpdb->posts} SET post_password = %s, post_modified_gmt = %s, post_modified = %s";
+        $params = array(
+            $claim_id,
+            current_time( 'mysql', true ),
+            current_time( 'mysql' ),
+        );
+
+        // Build initial WHERE clause.
+        $where    = "WHERE post_type = %s AND post_status = %s AND post_password = ''";
+        $params[] = self::POST_TYPE;
+        $params[] = ActionScheduler_Store::STATUS_PENDING;
+
+        if ( ! empty( $hooks ) ) {
+            $placeholders = array_fill( 0, count( $hooks ), '%s' );
+            $where       .= ' AND post_title IN (' . join( ', ', $placeholders ) . ')';
+            $params       = array_merge( $params, array_values( $hooks ) );
+        }
+
+        /*
 		 * Add the IDs to the WHERE clause. IDs not escaped because they came directly from a prior DB query.
 		 *
 		 * If we're not limiting by IDs, then include the post_date_gmt clause.
 		 */
-		if ( $limit_ids ) {
-			$where .= ' AND ID IN (' . join( ',', $ids ) . ')';
-		} else {
-			$where   .= ' AND post_date_gmt <= %s';
-			$params[] = $date->format( 'Y-m-d H:i:s' );
-		}
-
-		// Add the ORDER BY clause and,ms limit.
-		$order    = 'ORDER BY menu_order ASC, post_date_gmt ASC, ID ASC LIMIT %d';
-		$params[] = $limit;
-
-		// Run the query and gather results.
-		$rows_affected = $wpdb->query( $wpdb->prepare( "{$update} {$where} {$order}", $params ) ); // phpcs:ignore // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared, WordPress.DB.PreparedSQLPlaceholders.UnfinishedPrepare
-
-		if ( false === $rows_affected ) {
-			throw new RuntimeException( __( 'Unable to claim actions. Database error.', 'action-scheduler' ) );
-		}
-
-		return (int) $rows_affected;
-	}
-
-	/**
-	 * Get IDs of actions within a certain group and up to a certain date/time.
-	 *
-	 * @param string   $group The group to use in finding actions.
-	 * @param int      $limit The number of actions to retrieve.
-	 * @param DateTime $date  DateTime object representing cutoff time for actions. Actions retrieved will be
-	 *                        up to and including this DateTime.
-	 *
-	 * @return array IDs of actions in the appropriate group and before the appropriate time.
-	 * @throws InvalidArgumentException When the group does not exist.
-	 */
-	protected function get_actions_by_group( $group, $limit, DateTime $date ) {
-		// Ensure the group exists before continuing.
-		if ( ! term_exists( $group, self::GROUP_TAXONOMY ) ) {
-			/* translators: %s is the group name */
-			throw new InvalidArgumentException( sprintf( __( 'The group "%s" does not exist.', 'action-scheduler' ), $group ) );
-		}
-
-		// Set up a query for post IDs to use later.
-		$query      = new WP_Query();
-		$query_args = array(
-			'fields'           => 'ids',
-			'post_type'        => self::POST_TYPE,
-			'post_status'      => ActionScheduler_Store::STATUS_PENDING,
-			'has_password'     => false,
-			'posts_per_page'   => $limit * 3,
-			'suppress_filters' => true,
-			'no_found_rows'    => true,
-			'orderby'          => array(
-				'menu_order' => 'ASC',
-				'date'       => 'ASC',
-				'ID'         => 'ASC',
-			),
-			'date_query'       => array(
-				'column'    => 'post_date_gmt',
-				'before'    => $date->format( 'Y-m-d H:i' ),
-				'inclusive' => true,
-			),
-			'tax_query'        => array( // phpcs:ignore WordPress.DB.SlowDBQuery
-				array(
-					'taxonomy'         => self::GROUP_TAXONOMY,
-					'field'            => 'slug',
-					'terms'            => $group,
-					'include_children' => false,
-				),
-			),
-		);
-
-		return $query->query( $query_args );
-	}
-
-	/**
-	 * Find actions by claim ID.
-	 *
-	 * @param string $claim_id Claim ID.
-	 * @return array
-	 */
-	public function find_actions_by_claim_id( $claim_id ) {
-		/**
-		 * Global wpdb object.
-		 *
-		 * @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' );
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		$results = $wpdb->get_results(
-			$wpdb->prepare(
-				"SELECT ID, post_date_gmt FROM {$wpdb->posts} WHERE post_type = %s AND post_password = %s",
-				array(
-					self::POST_TYPE,
-					$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 ( $results as $claimed_action ) {
-			if ( $claimed_action->post_date_gmt <= $cut_off ) {
-				$action_ids[] = absint( $claimed_action->ID );
-			}
-		}
-
-		return $action_ids;
-	}
-
-	/**
-	 * Release claim.
-	 *
-	 * @param ActionScheduler_ActionClaim $claim Claim object to release.
-	 * @return void
-	 * @throws RuntimeException When the claim is not unlocked.
-	 */
-	public function release_claim( ActionScheduler_ActionClaim $claim ) {
-		$action_ids = $this->find_actions_by_claim_id( $claim->get_id() );
-		if ( empty( $action_ids ) ) {
-			return; // nothing to do.
-		}
-		$action_id_string = implode( ',', array_map( 'intval', $action_ids ) );
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		$result = $wpdb->query(
-			$wpdb->prepare(
-				"UPDATE {$wpdb->posts} SET post_password = '' WHERE ID IN ($action_id_string) AND post_password = %s", //phpcs:ignore
-				array(
-					$claim->get_id(),
-				)
-			)
-		);
-		if ( false === $result ) {
-			/* translators: %s: claim ID */
-			throw new RuntimeException( sprintf( __( 'Unable to unlock claim %s. Database error.', 'action-scheduler' ), $claim->get_id() ) );
-		}
-	}
-
-	/**
-	 * Unclaim action.
-	 *
-	 * @param string $action_id Action ID.
-	 * @throws RuntimeException When unable to unlock claim on action ID.
-	 */
-	public function unclaim_action( $action_id ) {
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		//phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		$result = $wpdb->query(
-			$wpdb->prepare(
-				"UPDATE {$wpdb->posts} SET post_password = '' WHERE ID = %d AND post_type = %s",
-				$action_id,
-				self::POST_TYPE
-			)
-		);
-		if ( false === $result ) {
-			/* translators: %s: action ID */
-			throw new RuntimeException( sprintf( __( 'Unable to unlock claim on action %s. Database error.', 'action-scheduler' ), $action_id ) );
-		}
-	}
-
-	/**
-	 * Mark failure on action.
-	 *
-	 * @param int $action_id Action ID.
-	 *
-	 * @return void
-	 * @throws RuntimeException When unable to mark failure on action ID.
-	 */
-	public function mark_failure( $action_id ) {
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		$result = $wpdb->query(
-			$wpdb->prepare( "UPDATE {$wpdb->posts} SET post_status = %s WHERE ID = %d AND post_type = %s", self::STATUS_FAILED, $action_id, self::POST_TYPE )
-		);
-		if ( false === $result ) {
-			/* translators: %s: action ID */
-			throw new RuntimeException( sprintf( __( 'Unable to mark failure on action %s. Database error.', 'action-scheduler' ), $action_id ) );
-		}
-	}
-
-	/**
-	 * Return an action's claim ID, as stored in the post password column
-	 *
-	 * @param int $action_id Action ID.
-	 * @return mixed
-	 */
-	public function get_claim_id( $action_id ) {
-		return $this->get_post_column( $action_id, 'post_password' );
-	}
-
-	/**
-	 * Return an action's status, as stored in the post status column
-	 *
-	 * @param int $action_id Action ID.
-	 *
-	 * @return mixed
-	 * @throws InvalidArgumentException When the action ID is invalid.
-	 */
-	public function get_status( $action_id ) {
-		$status = $this->get_post_column( $action_id, 'post_status' );
-
-		if ( null === $status ) {
-			throw new InvalidArgumentException( __( 'Invalid action ID. No status found.', 'action-scheduler' ) );
-		}
-
-		return $this->get_action_status_by_post_status( $status );
-	}
-
-	/**
-	 * Get post column
-	 *
-	 * @param string $action_id Action ID.
-	 * @param string $column_name Column Name.
-	 *
-	 * @return string|null
-	 */
-	private function get_post_column( $action_id, $column_name ) {
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		return $wpdb->get_var(
-			$wpdb->prepare(
-				"SELECT {$column_name} FROM {$wpdb->posts} WHERE ID=%d AND post_type=%s", // phpcs:ignore
-				$action_id,
-				self::POST_TYPE
-			)
-		);
-	}
-
-	/**
-	 * Log Execution.
-	 *
-	 * @param string $action_id Action ID.
-	 */
-	public function log_execution( $action_id ) {
-		/**
-		 * Global wpdb object.
-		 *
-		 * @var wpdb $wpdb
-		 */
-		global $wpdb;
-
-		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-		$wpdb->query(
-			$wpdb->prepare(
-				"UPDATE {$wpdb->posts} SET menu_order = menu_order+1, post_status=%s, post_modified_gmt = %s, post_modified = %s WHERE ID = %d AND post_type = %s",
-				self::STATUS_RUNNING,
-				current_time( 'mysql', true ),
-				current_time( 'mysql' ),
-				$action_id,
-				self::POST_TYPE
-			)
-		);
-	}
-
-	/**
-	 * Record that an action was completed.
-	 *
-	 * @param string $action_id ID of the completed action.
-	 *
-	 * @throws InvalidArgumentException When the action ID is invalid.
-	 * @throws RuntimeException         When there was an error executing the action.
-	 */
-	public function mark_complete( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
-			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
-		}
-		add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
-		$result = wp_update_post(
-			array(
-				'ID'          => $action_id,
-				'post_status' => 'publish',
-			),
-			true
-		);
-		remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
-		if ( is_wp_error( $result ) ) {
-			throw new RuntimeException( $result->get_error_message() );
-		}
-
-		/**
-		 * 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 );
-	}
-
-	/**
-	 * Mark action as migrated when there is an error deleting the action.
-	 *
-	 * @param int $action_id Action ID.
-	 */
-	public function mark_migrated( $action_id ) {
-		wp_update_post(
-			array(
-				'ID'          => $action_id,
-				'post_status' => 'migrated',
-			)
-		);
-	}
-
-	/**
-	 * Determine whether the post store can be migrated.
-	 *
-	 * @param [type] $setting - Setting value.
-	 * @return bool
-	 */
-	public function migration_dependencies_met( $setting ) {
-		global $wpdb;
-
-		$dependencies_met = get_transient( self::DEPENDENCIES_MET );
-		if ( empty( $dependencies_met ) ) {
-			$maximum_args_length = apply_filters( 'action_scheduler_maximum_args_length', 191 );
-			$found_action        = $wpdb->get_var( // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
-				$wpdb->prepare(
-					"SELECT ID FROM {$wpdb->posts} WHERE post_type = %s AND CHAR_LENGTH(post_content) > %d LIMIT 1",
-					$maximum_args_length,
-					self::POST_TYPE
-				)
-			);
-			$dependencies_met    = $found_action ? 'no' : 'yes';
-			set_transient( self::DEPENDENCIES_MET, $dependencies_met, DAY_IN_SECONDS );
-		}
-
-		return 'yes' === $dependencies_met ? $setting : false;
-	}
-
-	/**
-	 * InnoDB indexes have a maximum size of 767 bytes by default, which is only 191 characters with utf8mb4.
-	 *
-	 * Previously, AS wasn't concerned about args length, as we used the (unindex) post_content column. However,
-	 * as we prepare to move to custom tables, and can use an indexed VARCHAR column instead, we want to warn
-	 * developers of this impending requirement.
-	 *
-	 * @param ActionScheduler_Action $action Action object.
-	 */
-	protected function validate_action( ActionScheduler_Action $action ) {
-		try {
-			parent::validate_action( $action );
-		} catch ( Exception $e ) {
-			/* translators: %s is the error message */
-			$message = sprintf( __( '%s Support for strings longer than this will be removed in a future version.', 'action-scheduler' ), $e->getMessage() );
-			_doing_it_wrong( 'ActionScheduler_Action::$args', esc_html( $message ), '2.1.0' );
-		}
-	}
-
-	/**
-	 * (@codeCoverageIgnore)
-	 */
-	public function init() {
-		add_filter( 'action_scheduler_migration_dependencies_met', array( $this, 'migration_dependencies_met' ) );
-
-		$post_type_registrar = new ActionScheduler_wpPostStore_PostTypeRegistrar();
-		$post_type_registrar->register();
-
-		$post_status_registrar = new ActionScheduler_wpPostStore_PostStatusRegistrar();
-		$post_status_registrar->register();
-
-		$taxonomy_registrar = new ActionScheduler_wpPostStore_TaxonomyRegistrar();
-		$taxonomy_registrar->register();
-	}
+        if ( $limit_ids ) {
+            $where .= ' AND ID IN (' . join( ',', $ids ) . ')';
+        } else {
+            $where   .= ' AND post_date_gmt <= %s';
+            $params[] = $date->format( 'Y-m-d H:i:s' );
+        }
+
+        // Add the ORDER BY clause and,ms limit.
+        $order    = 'ORDER BY menu_order ASC, post_date_gmt ASC, ID ASC LIMIT %d';
+        $params[] = $limit;
+
+        // Run the query and gather results.
+        $rows_affected = $wpdb->query( $wpdb->prepare( "{$update} {$where} {$order}", $params ) ); // phpcs:ignore // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared, WordPress.DB.PreparedSQLPlaceholders.UnfinishedPrepare
+
+        if ( false === $rows_affected ) {
+            throw new RuntimeException( __( 'Unable to claim actions. Database error.', 'action-scheduler' ) );
+        }
+
+        return (int) $rows_affected;
+    }
+
+    /**
+     * Get IDs of actions within a certain group and up to a certain date/time.
+     *
+     * @param string   $group The group to use in finding actions.
+     * @param int      $limit The number of actions to retrieve.
+     * @param DateTime $date  DateTime object representing cutoff time for actions. Actions retrieved will be
+     *                        up to and including this DateTime.
+     *
+     * @return array IDs of actions in the appropriate group and before the appropriate time.
+     * @throws InvalidArgumentException When the group does not exist.
+     */
+    protected function get_actions_by_group( $group, $limit, DateTime $date ) {
+        // Ensure the group exists before continuing.
+        if ( ! term_exists( $group, self::GROUP_TAXONOMY ) ) {
+            /* translators: %s is the group name */
+            throw new InvalidArgumentException( sprintf( __( 'The group "%s" does not exist.', 'action-scheduler' ), $group ) );
+        }
+
+        // Set up a query for post IDs to use later.
+        $query      = new WP_Query();
+        $query_args = array(
+            'fields'           => 'ids',
+            'post_type'        => self::POST_TYPE,
+            'post_status'      => ActionScheduler_Store::STATUS_PENDING,
+            'has_password'     => false,
+            'posts_per_page'   => $limit * 3,
+            'suppress_filters' => true,
+            'no_found_rows'    => true,
+            'orderby'          => array(
+                'menu_order' => 'ASC',
+                'date'       => 'ASC',
+                'ID'         => 'ASC',
+            ),
+            'date_query'       => array(
+                'column'    => 'post_date_gmt',
+                'before'    => $date->format( 'Y-m-d H:i' ),
+                'inclusive' => true,
+            ),
+            'tax_query'        => array( // phpcs:ignore WordPress.DB.SlowDBQuery
+                array(
+                    'taxonomy'         => self::GROUP_TAXONOMY,
+                    'field'            => 'slug',
+                    'terms'            => $group,
+                    'include_children' => false,
+                ),
+            ),
+        );
+
+        return $query->query( $query_args );
+    }
+
+    /**
+     * Find actions by claim ID.
+     *
+     * @param string $claim_id Claim ID.
+     * @return array
+     */
+    public function find_actions_by_claim_id( $claim_id ) {
+        /**
+         * Global wpdb object.
+         *
+         * @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' );
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        $results = $wpdb->get_results(
+            $wpdb->prepare(
+                "SELECT ID, post_date_gmt FROM {$wpdb->posts} WHERE post_type = %s AND post_password = %s",
+                array(
+                    self::POST_TYPE,
+                    $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 ( $results as $claimed_action ) {
+            if ( $claimed_action->post_date_gmt <= $cut_off ) {
+                $action_ids[] = absint( $claimed_action->ID );
+            }
+        }
+
+        return $action_ids;
+    }
+
+    /**
+     * Release claim.
+     *
+     * @param ActionScheduler_ActionClaim $claim Claim object to release.
+     * @return void
+     * @throws RuntimeException When the claim is not unlocked.
+     */
+    public function release_claim( ActionScheduler_ActionClaim $claim ) {
+        $action_ids = $this->find_actions_by_claim_id( $claim->get_id() );
+        if ( empty( $action_ids ) ) {
+            return; // nothing to do.
+        }
+        $action_id_string = implode( ',', array_map( 'intval', $action_ids ) );
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        $result = $wpdb->query(
+            $wpdb->prepare(
+                "UPDATE {$wpdb->posts} SET post_password = '' WHERE ID IN ($action_id_string) AND post_password = %s", //phpcs:ignore
+                array(
+                    $claim->get_id(),
+                )
+            )
+        );
+        if ( false === $result ) {
+            /* translators: %s: claim ID */
+            throw new RuntimeException( sprintf( __( 'Unable to unlock claim %s. Database error.', 'action-scheduler' ), $claim->get_id() ) );
+        }
+    }
+
+    /**
+     * Unclaim action.
+     *
+     * @param string $action_id Action ID.
+     * @throws RuntimeException When unable to unlock claim on action ID.
+     */
+    public function unclaim_action( $action_id ) {
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        //phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        $result = $wpdb->query(
+            $wpdb->prepare(
+                "UPDATE {$wpdb->posts} SET post_password = '' WHERE ID = %d AND post_type = %s",
+                $action_id,
+                self::POST_TYPE
+            )
+        );
+        if ( false === $result ) {
+            /* translators: %s: action ID */
+            throw new RuntimeException( sprintf( __( 'Unable to unlock claim on action %s. Database error.', 'action-scheduler' ), $action_id ) );
+        }
+    }
+
+    /**
+     * Mark failure on action.
+     *
+     * @param int $action_id Action ID.
+     *
+     * @return void
+     * @throws RuntimeException When unable to mark failure on action ID.
+     */
+    public function mark_failure( $action_id ) {
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        $result = $wpdb->query(
+            $wpdb->prepare( "UPDATE {$wpdb->posts} SET post_status = %s WHERE ID = %d AND post_type = %s", self::STATUS_FAILED, $action_id, self::POST_TYPE )
+        );
+        if ( false === $result ) {
+            /* translators: %s: action ID */
+            throw new RuntimeException( sprintf( __( 'Unable to mark failure on action %s. Database error.', 'action-scheduler' ), $action_id ) );
+        }
+    }
+
+    /**
+     * Return an action's claim ID, as stored in the post password column
+     *
+     * @param int $action_id Action ID.
+     * @return mixed
+     */
+    public function get_claim_id( $action_id ) {
+        return $this->get_post_column( $action_id, 'post_password' );
+    }
+
+    /**
+     * Return an action's status, as stored in the post status column
+     *
+     * @param int $action_id Action ID.
+     *
+     * @return mixed
+     * @throws InvalidArgumentException When the action ID is invalid.
+     */
+    public function get_status( $action_id ) {
+        $status = $this->get_post_column( $action_id, 'post_status' );
+
+        if ( null === $status ) {
+            throw new InvalidArgumentException( __( 'Invalid action ID. No status found.', 'action-scheduler' ) );
+        }
+
+        return $this->get_action_status_by_post_status( $status );
+    }
+
+    /**
+     * Get post column
+     *
+     * @param string $action_id Action ID.
+     * @param string $column_name Column Name.
+     *
+     * @return string|null
+     */
+    private function get_post_column( $action_id, $column_name ) {
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        return $wpdb->get_var(
+            $wpdb->prepare(
+                "SELECT {$column_name} FROM {$wpdb->posts} WHERE ID=%d AND post_type=%s", // phpcs:ignore
+                $action_id,
+                self::POST_TYPE
+            )
+        );
+    }
+
+    /**
+     * Log Execution.
+     *
+     * @param string $action_id Action ID.
+     */
+    public function log_execution( $action_id ) {
+        /**
+         * Global wpdb object.
+         *
+         * @var wpdb $wpdb
+         */
+        global $wpdb;
+
+        // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+        $wpdb->query(
+            $wpdb->prepare(
+                "UPDATE {$wpdb->posts} SET menu_order = menu_order+1, post_status=%s, post_modified_gmt = %s, post_modified = %s WHERE ID = %d AND post_type = %s",
+                self::STATUS_RUNNING,
+                current_time( 'mysql', true ),
+                current_time( 'mysql' ),
+                $action_id,
+                self::POST_TYPE
+            )
+        );
+    }
+
+    /**
+     * Record that an action was completed.
+     *
+     * @param string $action_id ID of the completed action.
+     *
+     * @throws InvalidArgumentException When the action ID is invalid.
+     * @throws RuntimeException         When there was an error executing the action.
+     */
+    public function mark_complete( $action_id ) {
+        $post = get_post( $action_id );
+        if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+            /* translators: %s is the action ID */
+            throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+        }
+        add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
+        add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
+        $result = wp_update_post(
+            array(
+                'ID'          => $action_id,
+                'post_status' => 'publish',
+            ),
+            true
+        );
+        remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
+        remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
+        if ( is_wp_error( $result ) ) {
+            throw new RuntimeException( $result->get_error_message() );
+        }
+
+        /**
+         * 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 );
+    }
+
+    /**
+     * Mark action as migrated when there is an error deleting the action.
+     *
+     * @param int $action_id Action ID.
+     */
+    public function mark_migrated( $action_id ) {
+        wp_update_post(
+            array(
+                'ID'          => $action_id,
+                'post_status' => 'migrated',
+            )
+        );
+    }
+
+    /**
+     * Determine whether the post store can be migrated.
+     *
+     * @param [type] $setting - Setting value.
+     * @return bool
+     */
+    public function migration_dependencies_met( $setting ) {
+        global $wpdb;
+
+        $dependencies_met = get_transient( self::DEPENDENCIES_MET );
+        if ( empty( $dependencies_met ) ) {
+            $maximum_args_length = apply_filters( 'action_scheduler_maximum_args_length', 191 );
+            $found_action        = $wpdb->get_var( // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
+                $wpdb->prepare(
+                    "SELECT ID FROM {$wpdb->posts} WHERE post_type = %s AND CHAR_LENGTH(post_content) > %d LIMIT 1",
+                    $maximum_args_length,
+                    self::POST_TYPE
+                )
+            );
+            $dependencies_met    = $found_action ? 'no' : 'yes';
+            set_transient( self::DEPENDENCIES_MET, $dependencies_met, DAY_IN_SECONDS );
+        }
+
+        return 'yes' === $dependencies_met ? $setting : false;
+    }
+
+    /**
+     * InnoDB indexes have a maximum size of 767 bytes by default, which is only 191 characters with utf8mb4.
+     *
+     * Previously, AS wasn't concerned about args length, as we used the (unindex) post_content column. However,
+     * as we prepare to move to custom tables, and can use an indexed VARCHAR column instead, we want to warn
+     * developers of this impending requirement.
+     *
+     * @param ActionScheduler_Action $action Action object.
+     */
+    protected function validate_action( ActionScheduler_Action $action ) {
+        try {
+            parent::validate_action( $action );
+        } catch ( Exception $e ) {
+            /* translators: %s is the error message */
+            $message = sprintf( __( '%s Support for strings longer than this will be removed in a future version.', 'action-scheduler' ), $e->getMessage() );
+            _doing_it_wrong( 'ActionScheduler_Action::$args', esc_html( $message ), '2.1.0' );
+        }
+    }
+
+    /**
+     * (@codeCoverageIgnore)
+     */
+    public function init() {
+        add_filter( 'action_scheduler_migration_dependencies_met', array( $this, 'migration_dependencies_met' ) );
+
+        $post_type_registrar = new ActionScheduler_wpPostStore_PostTypeRegistrar();
+        $post_type_registrar->register();
+
+        $post_status_registrar = new ActionScheduler_wpPostStore_PostStatusRegistrar();
+        $post_status_registrar->register();
+
+        $taxonomy_registrar = new ActionScheduler_wpPostStore_TaxonomyRegistrar();
+        $taxonomy_registrar->register();
+    }
 }

Spacing +210 added lines, -210 removed lines

@@ -35,18 +35,18 @@ 
 	 * @throws RuntimeException Throws an exception if the action could not be saved.
 	 * @return int
 	 */
-	public function save_action( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
+	public function save_action(ActionScheduler_Action $action, DateTime $scheduled_date = null) {
 		try {
-			$this->validate_action( $action );
-			$post_array = $this->create_post_array( $action, $scheduled_date );
-			$post_id    = $this->save_post_array( $post_array );
-			$this->save_post_schedule( $post_id, $action->get_schedule() );
-			$this->save_action_group( $post_id, $action->get_group() );
-			do_action( 'action_scheduler_stored_action', $post_id );
+			$this->validate_action($action);
+			$post_array = $this->create_post_array($action, $scheduled_date);
+			$post_id    = $this->save_post_array($post_array);
+			$this->save_post_schedule($post_id, $action->get_schedule());
+			$this->save_action_group($post_id, $action->get_group());
+			do_action('action_scheduler_stored_action', $post_id);
 			return $post_id;
-		} catch ( Exception $e ) {
+		} catch (Exception $e) {
 			/* translators: %s: action error message */
-			throw new RuntimeException( sprintf( __( 'Error saving action: %s', 'action-scheduler' ), $e->getMessage() ), 0 );
+			throw new RuntimeException(sprintf(__('Error saving action: %s', 'action-scheduler'), $e->getMessage()), 0);
 		}
 	}
 
@@ -58,14 +58,14 @@ 
 	 *
 	 * @return array Returns an array of post data.
 	 */
-	protected function create_post_array( ActionScheduler_Action $action, DateTime $scheduled_date = null ) {
+	protected function create_post_array(ActionScheduler_Action $action, DateTime $scheduled_date = null) {
 		$post = array(
 			'post_type'     => self::POST_TYPE,
 			'post_title'    => $action->get_hook(),
-			'post_content'  => wp_json_encode( $action->get_args() ),
-			'post_status'   => ( $action->is_finished() ? 'publish' : 'pending' ),
-			'post_date_gmt' => $this->get_scheduled_date_string( $action, $scheduled_date ),
-			'post_date'     => $this->get_scheduled_date_string_local( $action, $scheduled_date ),
+			'post_content'  => wp_json_encode($action->get_args()),
+			'post_status'   => ($action->is_finished() ? 'publish' : 'pending'),
+			'post_date_gmt' => $this->get_scheduled_date_string($action, $scheduled_date),
+			'post_date'     => $this->get_scheduled_date_string_local($action, $scheduled_date),
 		);
 		return $post;
 	}
@@ -77,28 +77,28 @@ 
 	 * @return int Returns the post ID.
 	 * @throws RuntimeException Throws an exception if the action could not be saved.
 	 */
-	protected function save_post_array( $post_array ) {
-		add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
+	protected function save_post_array($post_array) {
+		add_filter('wp_insert_post_data', array($this, 'filter_insert_post_data'), 10, 1);
+		add_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10, 5);
 
-		$has_kses = false !== has_filter( 'content_save_pre', 'wp_filter_post_kses' );
+		$has_kses = false !== has_filter('content_save_pre', 'wp_filter_post_kses');
 
-		if ( $has_kses ) {
+		if ($has_kses) {
 			// Prevent KSES from corrupting JSON in post_content.
 			kses_remove_filters();
 		}
 
-		$post_id = wp_insert_post( $post_array );
+		$post_id = wp_insert_post($post_array);
 
-		if ( $has_kses ) {
+		if ($has_kses) {
 			kses_init_filters();
 		}
 
-		remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
+		remove_filter('wp_insert_post_data', array($this, 'filter_insert_post_data'), 10);
+		remove_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10);
 
-		if ( is_wp_error( $post_id ) || empty( $post_id ) ) {
-			throw new RuntimeException( __( 'Unable to save action.', 'action-scheduler' ) );
+		if (is_wp_error($post_id) || empty($post_id)) {
+			throw new RuntimeException(__('Unable to save action.', 'action-scheduler'));
 		}
 		return $post_id;
 	}
@@ -110,10 +110,10 @@ 
 	 *
 	 * @return array
 	 */
-	public function filter_insert_post_data( $postdata ) {
-		if ( self::POST_TYPE === $postdata['post_type'] ) {
+	public function filter_insert_post_data($postdata) {
+		if (self::POST_TYPE === $postdata['post_type']) {
 			$postdata['post_author'] = 0;
-			if ( 'future' === $postdata['post_status'] ) {
+			if ('future' === $postdata['post_status']) {
 				$postdata['post_status'] = 'publish';
 			}
 		}
@@ -148,9 +148,9 @@ 
 	 * @param string $post_type     Post type.
 	 * @return string
 	 */
-	public function set_unique_post_slug( $override_slug, $slug, $post_ID, $post_status, $post_type ) {
-		if ( self::POST_TYPE === $post_type ) {
-			$override_slug = uniqid( self::POST_TYPE . '-', true ) . '-' . wp_generate_password( 32, false );
+	public function set_unique_post_slug($override_slug, $slug, $post_ID, $post_status, $post_type) {
+		if (self::POST_TYPE === $post_type) {
+			$override_slug = uniqid(self::POST_TYPE.'-', true).'-'.wp_generate_password(32, false);
 		}
 		return $override_slug;
 	}
@@ -163,8 +163,8 @@ 
 	 *
 	 * @return void
 	 */
-	protected function save_post_schedule( $post_id, $schedule ) {
-		update_post_meta( $post_id, self::SCHEDULE_META_KEY, $schedule );
+	protected function save_post_schedule($post_id, $schedule) {
+		update_post_meta($post_id, self::SCHEDULE_META_KEY, $schedule);
 	}
 
 	/**
@@ -174,11 +174,11 @@ 
 	 * @param string $group   Group to save.
 	 * @return void
 	 */
-	protected function save_action_group( $post_id, $group ) {
-		if ( empty( $group ) ) {
-			wp_set_object_terms( $post_id, array(), self::GROUP_TAXONOMY, false );
+	protected function save_action_group($post_id, $group) {
+		if (empty($group)) {
+			wp_set_object_terms($post_id, array(), self::GROUP_TAXONOMY, false);
 		} else {
-			wp_set_object_terms( $post_id, array( $group ), self::GROUP_TAXONOMY, false );
+			wp_set_object_terms($post_id, array($group), self::GROUP_TAXONOMY, false);
 		}
 	}
 
@@ -188,16 +188,16 @@ 
 	 * @param int $action_id Action ID.
 	 * @return object
 	 */
-	public function fetch_action( $action_id ) {
-		$post = $this->get_post( $action_id );
-		if ( empty( $post ) || self::POST_TYPE !== $post->post_type ) {
+	public function fetch_action($action_id) {
+		$post = $this->get_post($action_id);
+		if (empty($post) || self::POST_TYPE !== $post->post_type) {
 			return $this->get_null_action();
 		}
 
 		try {
-			$action = $this->make_action_from_post( $post );
-		} catch ( ActionScheduler_InvalidActionException $exception ) {
-			do_action( 'action_scheduler_failed_fetch_action', $post->ID, $exception );
+			$action = $this->make_action_from_post($post);
+		} catch (ActionScheduler_InvalidActionException $exception) {
+			do_action('action_scheduler_failed_fetch_action', $post->ID, $exception);
 			return $this->get_null_action();
 		}
 
@@ -210,11 +210,11 @@ 
 	 * @param string $action_id - Action ID.
 	 * @return WP_Post|null
 	 */
-	protected function get_post( $action_id ) {
-		if ( empty( $action_id ) ) {
+	protected function get_post($action_id) {
+		if (empty($action_id)) {
 			return null;
 		}
-		return get_post( $action_id );
+		return get_post($action_id);
 	}
 
 	/**
@@ -232,19 +232,19 @@ 
 	 * @param WP_Post $post Post object.
 	 * @return WP_Post
 	 */
-	protected function make_action_from_post( $post ) {
+	protected function make_action_from_post($post) {
 		$hook = $post->post_title;
 
-		$args = json_decode( $post->post_content, true );
-		$this->validate_args( $args, $post->ID );
+		$args = json_decode($post->post_content, true);
+		$this->validate_args($args, $post->ID);
 
-		$schedule = get_post_meta( $post->ID, self::SCHEDULE_META_KEY, true );
-		$this->validate_schedule( $schedule, $post->ID );
+		$schedule = get_post_meta($post->ID, self::SCHEDULE_META_KEY, true);
+		$this->validate_schedule($schedule, $post->ID);
 
-		$group = wp_get_object_terms( $post->ID, self::GROUP_TAXONOMY, array( 'fields' => 'names' ) );
-		$group = empty( $group ) ? '' : reset( $group );
+		$group = wp_get_object_terms($post->ID, self::GROUP_TAXONOMY, array('fields' => 'names'));
+		$group = empty($group) ? '' : reset($group);
 
-		return ActionScheduler::factory()->get_stored_action( $this->get_action_status_by_post_status( $post->post_status ), $hook, $args, $schedule, $group );
+		return ActionScheduler::factory()->get_stored_action($this->get_action_status_by_post_status($post->post_status), $hook, $args, $schedule, $group);
 	}
 
 	/**
@@ -255,9 +255,9 @@ 
 	 * @throws InvalidArgumentException Throw InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
 	 * @return string
 	 */
-	protected function get_action_status_by_post_status( $post_status ) {
+	protected function get_action_status_by_post_status($post_status) {
 
-		switch ( $post_status ) {
+		switch ($post_status) {
 			case 'publish':
 				$action_status = self::STATUS_COMPLETE;
 				break;
@@ -265,8 +265,8 @@ 
 				$action_status = self::STATUS_CANCELED;
 				break;
 			default:
-				if ( ! array_key_exists( $post_status, $this->get_status_labels() ) ) {
-					throw new InvalidArgumentException( sprintf( 'Invalid post status: "%s". No matching action status available.', $post_status ) );
+				if ( ! array_key_exists($post_status, $this->get_status_labels())) {
+					throw new InvalidArgumentException(sprintf('Invalid post status: "%s". No matching action status available.', $post_status));
 				}
 				$action_status = $post_status;
 				break;
@@ -283,9 +283,9 @@ 
 	 * @throws InvalidArgumentException Throws InvalidArgumentException if $post_status not in known status fields returned by $this->get_status_labels().
 	 * @return string
 	 */
-	protected function get_post_status_by_action_status( $action_status ) {
+	protected function get_post_status_by_action_status($action_status) {
 
-		switch ( $action_status ) {
+		switch ($action_status) {
 			case self::STATUS_COMPLETE:
 				$post_status = 'publish';
 				break;
@@ -293,8 +293,8 @@ 
 				$post_status = 'trash';
 				break;
 			default:
-				if ( ! array_key_exists( $action_status, $this->get_status_labels() ) ) {
-					throw new InvalidArgumentException( sprintf( 'Invalid action status: "%s".', $action_status ) );
+				if ( ! array_key_exists($action_status, $this->get_status_labels())) {
+					throw new InvalidArgumentException(sprintf('Invalid action status: "%s".', $action_status));
 				}
 				$post_status = $action_status;
 				break;
@@ -312,10 +312,10 @@ 
 	 * @throws InvalidArgumentException - Throw InvalidArgumentException if $select_or_count not count or select.
 	 * @return string SQL statement. The returned SQL is already properly escaped.
 	 */
-	protected function get_query_actions_sql( array $query, $select_or_count = 'select' ) {
+	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 schedule. Cannot save action.', 'action-scheduler' ) );
+		if ( ! in_array($select_or_count, array('select', 'count'), true)) {
+			throw new InvalidArgumentException(__('Invalid schedule. Cannot save action.', 'action-scheduler'));
 		}
 
 		$query = wp_parse_args(
@@ -344,14 +344,14 @@ 
 		 * @var wpdb $wpdb
 		 */
 		global $wpdb;
-		$sql        = ( 'count' === $select_or_count ) ? 'SELECT count(p.ID)' : 'SELECT p.ID ';
+		$sql        = ('count' === $select_or_count) ? 'SELECT count(p.ID)' : 'SELECT p.ID ';
 		$sql       .= "FROM {$wpdb->posts} p";
 		$sql_params = array();
-		if ( empty( $query['group'] ) && 'group' === $query['orderby'] ) {
+		if (empty($query['group']) && 'group' === $query['orderby']) {
 			$sql .= " LEFT JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
 			$sql .= " LEFT JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
 			$sql .= " LEFT JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
-		} elseif ( ! empty( $query['group'] ) ) {
+		} elseif ( ! empty($query['group'])) {
 			$sql         .= " INNER JOIN {$wpdb->term_relationships} tr ON tr.object_id=p.ID";
 			$sql         .= " INNER JOIN {$wpdb->term_taxonomy} tt ON tr.term_taxonomy_id=tt.term_taxonomy_id";
 			$sql         .= " INNER JOIN {$wpdb->terms} t ON tt.term_id=t.term_id";
@@ -360,58 +360,58 @@ 
 		}
 		$sql         .= ' WHERE post_type=%s';
 		$sql_params[] = self::POST_TYPE;
-		if ( $query['hook'] ) {
+		if ($query['hook']) {
 			$sql         .= ' AND p.post_title=%s';
 			$sql_params[] = $query['hook'];
 		}
-		if ( $query['args'] !== null ) {
+		if ($query['args'] !== null) {
 			$sql         .= ' AND p.post_content=%s';
-			$sql_params[] = wp_json_encode( $query['args'] );
+			$sql_params[] = wp_json_encode($query['args']);
 		}
 
-		if ( $query['status'] ) {
-			$post_statuses = array_map( array( $this, 'get_post_status_by_action_status' ), (array) $query['status'] );
-			$placeholders  = array_fill( 0, count( $post_statuses ), '%s' );
-			$sql          .= ' AND p.post_status IN (' . join( ', ', $placeholders ) . ')';
-			$sql_params    = array_merge( $sql_params, array_values( $post_statuses ) );
+		if ($query['status']) {
+			$post_statuses = array_map(array($this, 'get_post_status_by_action_status'), (array) $query['status']);
+			$placeholders  = array_fill(0, count($post_statuses), '%s');
+			$sql          .= ' AND p.post_status IN ('.join(', ', $placeholders).')';
+			$sql_params    = array_merge($sql_params, array_values($post_statuses));
 		}
 
-		if ( $query['date'] instanceof DateTime ) {
+		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'] );
+			$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 p.post_date_gmt $comparator %s";
 			$sql_params[] = $date_string;
 		}
 
-		if ( $query['modified'] instanceof DateTime ) {
+		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'] );
+			$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 p.post_modified_gmt $comparator %s";
 			$sql_params[] = $date_string;
 		}
 
-		if ( true === $query['claimed'] ) {
+		if (true === $query['claimed']) {
 			$sql .= " AND p.post_password != ''";
-		} elseif ( false === $query['claimed'] ) {
+		} elseif (false === $query['claimed']) {
 			$sql .= " AND p.post_password = ''";
-		} elseif ( $query['claimed'] !== null ) {
+		} elseif ($query['claimed'] !== null) {
 			$sql         .= ' AND p.post_password = %s';
 			$sql_params[] = $query['claimed'];
 		}
 
-		if ( ! empty( $query['search'] ) ) {
+		if ( ! empty($query['search'])) {
 			$sql .= ' AND (p.post_title LIKE %s OR p.post_content LIKE %s OR p.post_password LIKE %s)';
-			for ( $i = 0; $i < 3; $i++ ) {
-				$sql_params[] = sprintf( '%%%s%%', $query['search'] );
+			for ($i = 0; $i < 3; $i++) {
+				$sql_params[] = sprintf('%%%s%%', $query['search']);
 			}
 		}
 
-		if ( 'select' === $select_or_count ) {
-			switch ( $query['orderby'] ) {
+		if ('select' === $select_or_count) {
+			switch ($query['orderby']) {
 				case 'hook':
 					$orderby = 'p.post_title';
 					break;
@@ -433,20 +433,20 @@ 
 					$orderby = 'p.post_date_gmt';
 					break;
 			}
-			if ( 'ASC' === strtoupper( $query['order'] ) ) {
+			if ('ASC' === strtoupper($query['order'])) {
 				$order = 'ASC';
 			} else {
 				$order = 'DESC';
 			}
 			$sql .= " ORDER BY $orderby $order";
-			if ( $query['per_page'] > 0 ) {
+			if ($query['per_page'] > 0) {
 				$sql         .= ' LIMIT %d, %d';
 				$sql_params[] = $query['offset'];
 				$sql_params[] = $query['per_page'];
 			}
 		}
 
-		return $wpdb->prepare( $sql, $sql_params ); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
+		return $wpdb->prepare($sql, $sql_params); // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared
 	}
 
 	/**
@@ -461,7 +461,7 @@ 
 	 *
 	 * @return string|array|null The IDs of actions matching the query. Null on failure.
 	 */
-	public function query_actions( $query = array(), $query_type = 'select' ) {
+	public function query_actions($query = array(), $query_type = 'select') {
 		/**
 		 * Global $wpdb object.
 		 *
@@ -469,9 +469,9 @@ 
 		 */
 		global $wpdb;
 
-		$sql = $this->get_query_actions_sql( $query, $query_type );
+		$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.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching,WordPress.DB.PreparedSQL.NotPrepared
+		return ('count' === $query_type) ? $wpdb->get_var($sql) : $wpdb->get_col($sql); // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery,WordPress.DB.DirectDatabaseQuery.NoCaching,WordPress.DB.PreparedSQL.NotPrepared
 	}
 
 	/**
@@ -483,18 +483,18 @@ 
 
 		$action_counts_by_status = array();
 		$action_stati_and_labels = $this->get_status_labels();
-		$posts_count_by_status   = (array) wp_count_posts( self::POST_TYPE, 'readable' );
+		$posts_count_by_status   = (array) wp_count_posts(self::POST_TYPE, 'readable');
 
-		foreach ( $posts_count_by_status as $post_status_name => $count ) {
+		foreach ($posts_count_by_status as $post_status_name => $count) {
 
 			try {
-				$action_status_name = $this->get_action_status_by_post_status( $post_status_name );
-			} catch ( Exception $e ) {
+				$action_status_name = $this->get_action_status_by_post_status($post_status_name);
+			} catch (Exception $e) {
 				// Ignore any post statuses that aren't for actions.
 				continue;
 			}
-			if ( array_key_exists( $action_status_name, $action_stati_and_labels ) ) {
-				$action_counts_by_status[ $action_status_name ] = $count;
+			if (array_key_exists($action_status_name, $action_stati_and_labels)) {
+				$action_counts_by_status[$action_status_name] = $count;
 			}
 		}
 
@@ -508,16 +508,16 @@ 
 	 *
 	 * @throws InvalidArgumentException If $action_id is not identified.
 	 */
-	public function cancel_action( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+	public function cancel_action($action_id) {
+		$post = get_post($action_id);
+		if (empty($post) || (self::POST_TYPE !== $post->post_type)) {
 			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+			throw new InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id));
 		}
-		do_action( 'action_scheduler_canceled_action', $action_id );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
-		wp_trash_post( $action_id );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
+		do_action('action_scheduler_canceled_action', $action_id);
+		add_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10, 5);
+		wp_trash_post($action_id);
+		remove_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10);
 	}
 
 	/**
@@ -527,15 +527,15 @@ 
 	 * @return void
 	 * @throws InvalidArgumentException If action is not identified.
 	 */
-	public function delete_action( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+	public function delete_action($action_id) {
+		$post = get_post($action_id);
+		if (empty($post) || (self::POST_TYPE !== $post->post_type)) {
 			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+			throw new InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id));
 		}
-		do_action( 'action_scheduler_deleted_action', $action_id );
+		do_action('action_scheduler_deleted_action', $action_id);
 
-		wp_delete_post( $action_id, true );
+		wp_delete_post($action_id, true);
 	}
 
 	/**
@@ -544,9 +544,9 @@ 
 	 * @param int $action_id Action ID.
 	 * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
 	 */
-	public function get_date( $action_id ) {
-		$next = $this->get_date_gmt( $action_id );
-		return ActionScheduler_TimezoneHelper::set_local_timezone( $next );
+	public function get_date($action_id) {
+		$next = $this->get_date_gmt($action_id);
+		return ActionScheduler_TimezoneHelper::set_local_timezone($next);
 	}
 
 	/**
@@ -557,16 +557,16 @@ 
 	 * @throws InvalidArgumentException If $action_id is not identified.
 	 * @return ActionScheduler_DateTime The date the action is schedule to run, or the date that it ran.
 	 */
-	public function get_date_gmt( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+	public function get_date_gmt($action_id) {
+		$post = get_post($action_id);
+		if (empty($post) || (self::POST_TYPE !== $post->post_type)) {
 			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+			throw new InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id));
 		}
-		if ( 'publish' === $post->post_status ) {
-			return as_get_datetime_object( $post->post_modified_gmt );
+		if ('publish' === $post->post_status) {
+			return as_get_datetime_object($post->post_modified_gmt);
 		} else {
-			return as_get_datetime_object( $post->post_date_gmt );
+			return as_get_datetime_object($post->post_date_gmt);
 		}
 	}
 
@@ -582,14 +582,14 @@ 
 	 * @throws RuntimeException When there is an error staking a claim.
 	 * @throws InvalidArgumentException When the given group is not valid.
 	 */
-	public function stake_claim( $max_actions = 10, DateTime $before_date = null, $hooks = array(), $group = '' ) {
+	public function stake_claim($max_actions = 10, DateTime $before_date = null, $hooks = array(), $group = '') {
 		$this->claim_before_date = $before_date;
 		$claim_id                = $this->generate_claim_id();
-		$this->claim_actions( $claim_id, $max_actions, $before_date, $hooks, $group );
-		$action_ids              = $this->find_actions_by_claim_id( $claim_id );
+		$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 );
+		return new ActionScheduler_ActionClaim($claim_id, $action_ids);
 	}
 
 	/**
@@ -604,7 +604,7 @@ 
 		return $wpdb->get_var(
 			$wpdb->prepare(
 				"SELECT COUNT(DISTINCT post_password) FROM {$wpdb->posts} WHERE post_password != '' AND post_type = %s AND post_status IN ('in-progress','pending')",
-				array( self::POST_TYPE )
+				array(self::POST_TYPE)
 			)
 		);
 	}
@@ -615,8 +615,8 @@ 
 	 * @return string
 	 */
 	protected function generate_claim_id() {
-		$claim_id = md5( microtime( true ) . wp_rand( 0, 1000 ) );
-		return substr( $claim_id, 0, 20 ); // to fit in db field with 20 char limit.
+		$claim_id = md5(microtime(true).wp_rand(0, 1000));
+		return substr($claim_id, 0, 20); // to fit in db field with 20 char limit.
 	}
 
 	/**
@@ -631,14 +631,14 @@ 
 	 * @return int The number of actions that were claimed.
 	 * @throws RuntimeException  When there is a database error.
 	 */
-	protected function claim_actions( $claim_id, $limit, DateTime $before_date = null, $hooks = array(), $group = '' ) {
+	protected function claim_actions($claim_id, $limit, DateTime $before_date = null, $hooks = array(), $group = '') {
 		// Set up initial variables.
 		$date      = null === $before_date ? as_get_datetime_object() : clone $before_date;
-		$limit_ids = ! empty( $group );
-		$ids       = $limit_ids ? $this->get_actions_by_group( $group, $limit, $date ) : array();
+		$limit_ids = ! empty($group);
+		$ids       = $limit_ids ? $this->get_actions_by_group($group, $limit, $date) : array();
 
 		// If limiting by IDs and no posts found, then return early since we have nothing to update.
-		if ( $limit_ids && 0 === count( $ids ) ) {
+		if ($limit_ids && 0 === count($ids)) {
 			return 0;
 		}
 
@@ -658,8 +658,8 @@ 
 		$update = "UPDATE {$wpdb->posts} SET post_password = %s, post_modified_gmt = %s, post_modified = %s";
 		$params = array(
 			$claim_id,
-			current_time( 'mysql', true ),
-			current_time( 'mysql' ),
+			current_time('mysql', true),
+			current_time('mysql'),
 		);
 
 		// Build initial WHERE clause.
@@ -667,10 +667,10 @@ 
 		$params[] = self::POST_TYPE;
 		$params[] = ActionScheduler_Store::STATUS_PENDING;
 
-		if ( ! empty( $hooks ) ) {
-			$placeholders = array_fill( 0, count( $hooks ), '%s' );
-			$where       .= ' AND post_title IN (' . join( ', ', $placeholders ) . ')';
-			$params       = array_merge( $params, array_values( $hooks ) );
+		if ( ! empty($hooks)) {
+			$placeholders = array_fill(0, count($hooks), '%s');
+			$where       .= ' AND post_title IN ('.join(', ', $placeholders).')';
+			$params       = array_merge($params, array_values($hooks));
 		}
 
 		/*
@@ -678,11 +678,11 @@ 
 		 *
 		 * If we're not limiting by IDs, then include the post_date_gmt clause.
 		 */
-		if ( $limit_ids ) {
-			$where .= ' AND ID IN (' . join( ',', $ids ) . ')';
+		if ($limit_ids) {
+			$where .= ' AND ID IN ('.join(',', $ids).')';
 		} else {
 			$where   .= ' AND post_date_gmt <= %s';
-			$params[] = $date->format( 'Y-m-d H:i:s' );
+			$params[] = $date->format('Y-m-d H:i:s');
 		}
 
 		// Add the ORDER BY clause and,ms limit.
@@ -690,10 +690,10 @@ 
 		$params[] = $limit;
 
 		// Run the query and gather results.
-		$rows_affected = $wpdb->query( $wpdb->prepare( "{$update} {$where} {$order}", $params ) ); // phpcs:ignore // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared, WordPress.DB.PreparedSQLPlaceholders.UnfinishedPrepare
+		$rows_affected = $wpdb->query($wpdb->prepare("{$update} {$where} {$order}", $params)); // phpcs:ignore // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared, WordPress.DB.PreparedSQLPlaceholders.UnfinishedPrepare
 
-		if ( false === $rows_affected ) {
-			throw new RuntimeException( __( 'Unable to claim actions. Database error.', 'action-scheduler' ) );
+		if (false === $rows_affected) {
+			throw new RuntimeException(__('Unable to claim actions. Database error.', 'action-scheduler'));
 		}
 
 		return (int) $rows_affected;
@@ -710,11 +710,11 @@ 
 	 * @return array IDs of actions in the appropriate group and before the appropriate time.
 	 * @throws InvalidArgumentException When the group does not exist.
 	 */
-	protected function get_actions_by_group( $group, $limit, DateTime $date ) {
+	protected function get_actions_by_group($group, $limit, DateTime $date) {
 		// Ensure the group exists before continuing.
-		if ( ! term_exists( $group, self::GROUP_TAXONOMY ) ) {
+		if ( ! term_exists($group, self::GROUP_TAXONOMY)) {
 			/* translators: %s is the group name */
-			throw new InvalidArgumentException( sprintf( __( 'The group "%s" does not exist.', 'action-scheduler' ), $group ) );
+			throw new InvalidArgumentException(sprintf(__('The group "%s" does not exist.', 'action-scheduler'), $group));
 		}
 
 		// Set up a query for post IDs to use later.
@@ -734,7 +734,7 @@ 
 			),
 			'date_query'       => array(
 				'column'    => 'post_date_gmt',
-				'before'    => $date->format( 'Y-m-d H:i' ),
+				'before'    => $date->format('Y-m-d H:i'),
 				'inclusive' => true,
 			),
 			'tax_query'        => array( // phpcs:ignore WordPress.DB.SlowDBQuery
@@ -747,7 +747,7 @@ 
 			),
 		);
 
-		return $query->query( $query_args );
+		return $query->query($query_args);
 	}
 
 	/**
@@ -756,7 +756,7 @@ 
 	 * @param string $claim_id Claim ID.
 	 * @return array
 	 */
-	public function find_actions_by_claim_id( $claim_id ) {
+	public function find_actions_by_claim_id($claim_id) {
 		/**
 		 * Global wpdb object.
 		 *
@@ -765,8 +765,8 @@ 
 		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' );
+		$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');
 
 		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
 		$results = $wpdb->get_results(
@@ -781,9 +781,9 @@ 
 
 		// 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 ( $results as $claimed_action ) {
-			if ( $claimed_action->post_date_gmt <= $cut_off ) {
-				$action_ids[] = absint( $claimed_action->ID );
+		foreach ($results as $claimed_action) {
+			if ($claimed_action->post_date_gmt <= $cut_off) {
+				$action_ids[] = absint($claimed_action->ID);
 			}
 		}
 
@@ -797,12 +797,12 @@ 
 	 * @return void
 	 * @throws RuntimeException When the claim is not unlocked.
 	 */
-	public function release_claim( ActionScheduler_ActionClaim $claim ) {
-		$action_ids = $this->find_actions_by_claim_id( $claim->get_id() );
-		if ( empty( $action_ids ) ) {
+	public function release_claim(ActionScheduler_ActionClaim $claim) {
+		$action_ids = $this->find_actions_by_claim_id($claim->get_id());
+		if (empty($action_ids)) {
 			return; // nothing to do.
 		}
-		$action_id_string = implode( ',', array_map( 'intval', $action_ids ) );
+		$action_id_string = implode(',', array_map('intval', $action_ids));
 		/**
 		 * Global wpdb object.
 		 *
@@ -819,9 +819,9 @@ 
 				)
 			)
 		);
-		if ( false === $result ) {
+		if (false === $result) {
 			/* translators: %s: claim ID */
-			throw new RuntimeException( sprintf( __( 'Unable to unlock claim %s. Database error.', 'action-scheduler' ), $claim->get_id() ) );
+			throw new RuntimeException(sprintf(__('Unable to unlock claim %s. Database error.', 'action-scheduler'), $claim->get_id()));
 		}
 	}
 
@@ -831,7 +831,7 @@ 
 	 * @param string $action_id Action ID.
 	 * @throws RuntimeException When unable to unlock claim on action ID.
 	 */
-	public function unclaim_action( $action_id ) {
+	public function unclaim_action($action_id) {
 		/**
 		 * Global wpdb object.
 		 *
@@ -847,9 +847,9 @@ 
 				self::POST_TYPE
 			)
 		);
-		if ( false === $result ) {
+		if (false === $result) {
 			/* translators: %s: action ID */
-			throw new RuntimeException( sprintf( __( 'Unable to unlock claim on action %s. Database error.', 'action-scheduler' ), $action_id ) );
+			throw new RuntimeException(sprintf(__('Unable to unlock claim on action %s. Database error.', 'action-scheduler'), $action_id));
 		}
 	}
 
@@ -861,7 +861,7 @@ 
 	 * @return void
 	 * @throws RuntimeException When unable to mark failure on action ID.
 	 */
-	public function mark_failure( $action_id ) {
+	public function mark_failure($action_id) {
 		/**
 		 * Global wpdb object.
 		 *
@@ -871,11 +871,11 @@ 
 
 		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
 		$result = $wpdb->query(
-			$wpdb->prepare( "UPDATE {$wpdb->posts} SET post_status = %s WHERE ID = %d AND post_type = %s", self::STATUS_FAILED, $action_id, self::POST_TYPE )
+			$wpdb->prepare("UPDATE {$wpdb->posts} SET post_status = %s WHERE ID = %d AND post_type = %s", self::STATUS_FAILED, $action_id, self::POST_TYPE)
 		);
-		if ( false === $result ) {
+		if (false === $result) {
 			/* translators: %s: action ID */
-			throw new RuntimeException( sprintf( __( 'Unable to mark failure on action %s. Database error.', 'action-scheduler' ), $action_id ) );
+			throw new RuntimeException(sprintf(__('Unable to mark failure on action %s. Database error.', 'action-scheduler'), $action_id));
 		}
 	}
 
@@ -885,8 +885,8 @@ 
 	 * @param int $action_id Action ID.
 	 * @return mixed
 	 */
-	public function get_claim_id( $action_id ) {
-		return $this->get_post_column( $action_id, 'post_password' );
+	public function get_claim_id($action_id) {
+		return $this->get_post_column($action_id, 'post_password');
 	}
 
 	/**
@@ -897,14 +897,14 @@ 
 	 * @return mixed
 	 * @throws InvalidArgumentException When the action ID is invalid.
 	 */
-	public function get_status( $action_id ) {
-		$status = $this->get_post_column( $action_id, 'post_status' );
+	public function get_status($action_id) {
+		$status = $this->get_post_column($action_id, 'post_status');
 
-		if ( null === $status ) {
-			throw new InvalidArgumentException( __( 'Invalid action ID. No status found.', 'action-scheduler' ) );
+		if (null === $status) {
+			throw new InvalidArgumentException(__('Invalid action ID. No status found.', 'action-scheduler'));
 		}
 
-		return $this->get_action_status_by_post_status( $status );
+		return $this->get_action_status_by_post_status($status);
 	}
 
 	/**
@@ -915,7 +915,7 @@ 
 	 *
 	 * @return string|null
 	 */
-	private function get_post_column( $action_id, $column_name ) {
+	private function get_post_column($action_id, $column_name) {
 		/**
 		 * Global wpdb object.
 		 *
@@ -938,7 +938,7 @@ 
 	 *
 	 * @param string $action_id Action ID.
 	 */
-	public function log_execution( $action_id ) {
+	public function log_execution($action_id) {
 		/**
 		 * Global wpdb object.
 		 *
@@ -951,8 +951,8 @@ 
 			$wpdb->prepare(
 				"UPDATE {$wpdb->posts} SET menu_order = menu_order+1, post_status=%s, post_modified_gmt = %s, post_modified = %s WHERE ID = %d AND post_type = %s",
 				self::STATUS_RUNNING,
-				current_time( 'mysql', true ),
-				current_time( 'mysql' ),
+				current_time('mysql', true),
+				current_time('mysql'),
 				$action_id,
 				self::POST_TYPE
 			)
@@ -967,14 +967,14 @@ 
 	 * @throws InvalidArgumentException When the action ID is invalid.
 	 * @throws RuntimeException         When there was an error executing the action.
 	 */
-	public function mark_complete( $action_id ) {
-		$post = get_post( $action_id );
-		if ( empty( $post ) || ( self::POST_TYPE !== $post->post_type ) ) {
+	public function mark_complete($action_id) {
+		$post = get_post($action_id);
+		if (empty($post) || (self::POST_TYPE !== $post->post_type)) {
 			/* translators: %s is the action ID */
-			throw new InvalidArgumentException( sprintf( __( 'Unidentified action %s', 'action-scheduler' ), $action_id ) );
+			throw new InvalidArgumentException(sprintf(__('Unidentified action %s', 'action-scheduler'), $action_id));
 		}
-		add_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10, 1 );
-		add_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10, 5 );
+		add_filter('wp_insert_post_data', array($this, 'filter_insert_post_data'), 10, 1);
+		add_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10, 5);
 		$result = wp_update_post(
 			array(
 				'ID'          => $action_id,
@@ -982,10 +982,10 @@ 
 			),
 			true
 		);
-		remove_filter( 'wp_insert_post_data', array( $this, 'filter_insert_post_data' ), 10 );
-		remove_filter( 'pre_wp_unique_post_slug', array( $this, 'set_unique_post_slug' ), 10 );
-		if ( is_wp_error( $result ) ) {
-			throw new RuntimeException( $result->get_error_message() );
+		remove_filter('wp_insert_post_data', array($this, 'filter_insert_post_data'), 10);
+		remove_filter('pre_wp_unique_post_slug', array($this, 'set_unique_post_slug'), 10);
+		if (is_wp_error($result)) {
+			throw new RuntimeException($result->get_error_message());
 		}
 
 		/**
@@ -995,7 +995,7 @@ 
 		 *
 		 * @param int $action_id Action ID.
 		 */
-		do_action( 'action_scheduler_completed_action', $action_id );
+		do_action('action_scheduler_completed_action', $action_id);
 	}
 
 	/**
@@ -1003,7 +1003,7 @@ 
 	 *
 	 * @param int $action_id Action ID.
 	 */
-	public function mark_migrated( $action_id ) {
+	public function mark_migrated($action_id) {
 		wp_update_post(
 			array(
 				'ID'          => $action_id,
@@ -1018,12 +1018,12 @@ 
 	 * @param [type] $setting - Setting value.
 	 * @return bool
 	 */
-	public function migration_dependencies_met( $setting ) {
+	public function migration_dependencies_met($setting) {
 		global $wpdb;
 
-		$dependencies_met = get_transient( self::DEPENDENCIES_MET );
-		if ( empty( $dependencies_met ) ) {
-			$maximum_args_length = apply_filters( 'action_scheduler_maximum_args_length', 191 );
+		$dependencies_met = get_transient(self::DEPENDENCIES_MET);
+		if (empty($dependencies_met)) {
+			$maximum_args_length = apply_filters('action_scheduler_maximum_args_length', 191);
 			$found_action        = $wpdb->get_var( // phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery, WordPress.DB.DirectDatabaseQuery.NoCaching
 				$wpdb->prepare(
 					"SELECT ID FROM {$wpdb->posts} WHERE post_type = %s AND CHAR_LENGTH(post_content) > %d LIMIT 1",
@@ -1031,8 +1031,8 @@ 
 					self::POST_TYPE
 				)
 			);
-			$dependencies_met    = $found_action ? 'no' : 'yes';
-			set_transient( self::DEPENDENCIES_MET, $dependencies_met, DAY_IN_SECONDS );
+			$dependencies_met = $found_action ? 'no' : 'yes';
+			set_transient(self::DEPENDENCIES_MET, $dependencies_met, DAY_IN_SECONDS);
 		}
 
 		return 'yes' === $dependencies_met ? $setting : false;
@@ -1047,13 +1047,13 @@ 
 	 *
 	 * @param ActionScheduler_Action $action Action object.
 	 */
-	protected function validate_action( ActionScheduler_Action $action ) {
+	protected function validate_action(ActionScheduler_Action $action) {
 		try {
-			parent::validate_action( $action );
-		} catch ( Exception $e ) {
+			parent::validate_action($action);
+		} catch (Exception $e) {
 			/* translators: %s is the error message */
-			$message = sprintf( __( '%s Support for strings longer than this will be removed in a future version.', 'action-scheduler' ), $e->getMessage() );
-			_doing_it_wrong( 'ActionScheduler_Action::$args', esc_html( $message ), '2.1.0' );
+			$message = sprintf(__('%s Support for strings longer than this will be removed in a future version.', 'action-scheduler'), $e->getMessage());
+			_doing_it_wrong('ActionScheduler_Action::$args', esc_html($message), '2.1.0');
 		}
 	}
 
@@ -1061,7 +1061,7 @@ 
 	 * (@codeCoverageIgnore)
 	 */
 	public function init() {
-		add_filter( 'action_scheduler_migration_dependencies_met', array( $this, 'migration_dependencies_met' ) );
+		add_filter('action_scheduler_migration_dependencies_met', array($this, 'migration_dependencies_met'));
 
 		$post_type_registrar = new ActionScheduler_wpPostStore_PostTypeRegistrar();
 		$post_type_registrar->register();

src/ext/action-scheduler/classes/data-stores/ActionScheduler_HybridStore.php

Indentation +406 added lines, -406 removed lines

@@ -13,415 +13,415 @@
  * @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', array( $this, 'set_autoincrement' ), 10, 2 );
-		$this->primary_store->init();
-		$this->secondary_store->init();
-		remove_action( 'action_scheduler/created_table', array( $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},
-				array(
-					'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},
-					array( '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 = array() ) {
-		$found_unmigrated_action = $this->secondary_store->find_action( $hook, $params );
-		if ( ! empty( $found_unmigrated_action ) ) {
-			$this->migrate( array( $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 = array(), $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 = array(
-				$this->primary_store,
-				$this->secondary_store,
-			);
-		} elseif ( $action_id < $this->demarkation_id ) {
-			$stores = array(
-				$this->secondary_store,
-				$this->primary_store,
-			);
-		} else {
-			$stores = array(
-				$this->primary_store,
-			);
-		}
-
-		foreach ( $stores as $store ) {
-			$action = $store->fetch_action( $action_id );
-			if ( ! is_a( $action, 'ActionScheduler_NullAction' ) ) {
-				return $store;
-			}
-		}
-		return null;
-	}
-
-	/*
+    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', array( $this, 'set_autoincrement' ), 10, 2 );
+        $this->primary_store->init();
+        $this->secondary_store->init();
+        remove_action( 'action_scheduler/created_table', array( $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},
+                array(
+                    '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},
+                    array( '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 = array() ) {
+        $found_unmigrated_action = $this->secondary_store->find_action( $hook, $params );
+        if ( ! empty( $found_unmigrated_action ) ) {
+            $this->migrate( array( $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 = array(), $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 = array(
+                $this->primary_store,
+                $this->secondary_store,
+            );
+        } elseif ( $action_id < $this->demarkation_id ) {
+            $stores = array(
+                $this->secondary_store,
+                $this->primary_store,
+            );
+        } else {
+            $stores = array(
+                $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 );
-	}
+    /**
+     * 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 );
+    }
 }

Spacing +92 added lines, -92 removed lines

@@ -37,14 +37,14 @@ 
 	 *
 	 * @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 ) ) {
+	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 );
+		$this->migration_runner = new Runner($config);
 	}
 
 	/**
@@ -53,10 +53,10 @@ 
 	 * @codeCoverageIgnore
 	 */
 	public function init() {
-		add_action( 'action_scheduler/created_table', array( $this, 'set_autoincrement' ), 10, 2 );
+		add_action('action_scheduler/created_table', array($this, 'set_autoincrement'), 10, 2);
 		$this->primary_store->init();
 		$this->secondary_store->init();
-		remove_action( 'action_scheduler/created_table', array( $this, 'set_autoincrement' ), 10 );
+		remove_action('action_scheduler/created_table', array($this, 'set_autoincrement'), 10);
 	}
 
 	/**
@@ -70,9 +70,9 @@ 
 	 * @return void
 	 * @codeCoverageIgnore
 	 */
-	public function set_autoincrement( $table_name, $table_suffix ) {
-		if ( ActionScheduler_StoreSchema::ACTIONS_TABLE === $table_suffix ) {
-			if ( empty( $this->demarkation_id ) ) {
+	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 */
@@ -81,10 +81,10 @@ 
 			 * 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' );
+			$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 );
+			$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},
@@ -98,10 +98,10 @@ 
 					'last_attempt_local'   => $date_local,
 				)
 			);
-			if ( $row_count > 0 ) {
+			if ($row_count > 0) {
 				$wpdb->delete(
 					$wpdb->{ActionScheduler_StoreSchema::ACTIONS_TABLE},
-					array( 'action_id' => $this->demarkation_id )
+					array('action_id' => $this->demarkation_id)
 				);
 			}
 		}
@@ -117,14 +117,14 @@ 
 	 *
 	 * @codeCoverageIgnore
 	 */
-	private function set_demarkation_id( $id = null ) {
-		if ( empty( $id ) ) {
+	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 ++;
+			$id = (int) $wpdb->get_var("SELECT MAX(ID) FROM $wpdb->posts");
+			$id++;
 		}
-		update_option( self::DEMARKATION_OPTION, $id );
+		update_option(self::DEMARKATION_OPTION, $id);
 
 		return $id;
 	}
@@ -140,13 +140,13 @@ 
 	 *
 	 * @return string
 	 */
-	public function find_action( $hook, $params = array() ) {
-		$found_unmigrated_action = $this->secondary_store->find_action( $hook, $params );
-		if ( ! empty( $found_unmigrated_action ) ) {
-			$this->migrate( array( $found_unmigrated_action ) );
+	public function find_action($hook, $params = array()) {
+		$found_unmigrated_action = $this->secondary_store->find_action($hook, $params);
+		if ( ! empty($found_unmigrated_action)) {
+			$this->migrate(array($found_unmigrated_action));
 		}
 
-		return $this->primary_store->find_action( $hook, $params );
+		return $this->primary_store->find_action($hook, $params);
 	}
 
 	/**
@@ -159,13 +159,13 @@ 
 	 *
 	 * @return int[]
 	 */
-	public function query_actions( $query = array(), $query_type = 'select' ) {
-		$found_unmigrated_actions = $this->secondary_store->query_actions( $query, 'select' );
-		if ( ! empty( $found_unmigrated_actions ) ) {
-			$this->migrate( $found_unmigrated_actions );
+	public function query_actions($query = array(), $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 );
+		return $this->primary_store->query_actions($query, $query_type);
 	}
 
 	/**
@@ -178,22 +178,22 @@ 
 		$migrated_actions_count   = $this->primary_store->action_counts();
 		$actions_count_by_status  = array();
 
-		foreach ( $this->get_status_labels() as $status_key => $status_label ) {
+		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($unmigrated_actions_count[$status_key])) {
+				$count += $unmigrated_actions_count[$status_key];
 			}
 
-			if ( isset( $migrated_actions_count[ $status_key ] ) ) {
-				$count += $migrated_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[$status_key] = $count;
 		}
 
-		$actions_count_by_status = array_filter( $actions_count_by_status );
+		$actions_count_by_status = array_filter($actions_count_by_status);
 
 		return $actions_count_by_status;
 	}
@@ -208,17 +208,17 @@ 
 	 *
 	 * @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 );
+	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 );
+		if ( ! empty($claimed_actions)) {
+			$this->migrate($claimed_actions);
 		}
 
-		$this->secondary_store->release_claim( $claim );
+		$this->secondary_store->release_claim($claim);
 
-		return $this->primary_store->stake_claim( $max_actions, $before_date, $hooks, $group );
+		return $this->primary_store->stake_claim($max_actions, $before_date, $hooks, $group);
 	}
 
 	/**
@@ -226,8 +226,8 @@ 
 	 *
 	 * @param array $action_ids List of action IDs.
 	 */
-	private function migrate( $action_ids ) {
-		$this->migration_runner->migrate_actions( $action_ids );
+	private function migrate($action_ids) {
+		$this->migration_runner->migrate_actions($action_ids);
 	}
 
 	/**
@@ -238,8 +238,8 @@ 
 	 *
 	 * @return int The action ID
 	 */
-	public function save_action( ActionScheduler_Action $action, DateTime $date = null ) {
-		return $this->primary_store->save_action( $action, $date );
+	public function save_action(ActionScheduler_Action $action, DateTime $date = null) {
+		return $this->primary_store->save_action($action, $date);
 	}
 
 	/**
@@ -247,10 +247,10 @@ 
 	 *
 	 * @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 );
+	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();
 		}
@@ -261,10 +261,10 @@ 
 	 *
 	 * @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 );
+	public function cancel_action($action_id) {
+		$store = $this->get_store_from_action_id($action_id);
+		if ($store) {
+			$store->cancel_action($action_id);
 		}
 	}
 
@@ -273,10 +273,10 @@ 
 	 *
 	 * @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 );
+	public function delete_action($action_id) {
+		$store = $this->get_store_from_action_id($action_id);
+		if ($store) {
+			$store->delete_action($action_id);
 		}
 	}
 
@@ -285,10 +285,10 @@ 
 	 *
 	 * @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 );
+	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;
 		}
@@ -299,10 +299,10 @@ 
 	 *
 	 * @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 );
+	public function mark_failure($action_id) {
+		$store = $this->get_store_from_action_id($action_id);
+		if ($store) {
+			$store->mark_failure($action_id);
 		}
 	}
 
@@ -311,10 +311,10 @@ 
 	 *
 	 * @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 );
+	public function log_execution($action_id) {
+		$store = $this->get_store_from_action_id($action_id);
+		if ($store) {
+			$store->log_execution($action_id);
 		}
 	}
 
@@ -323,10 +323,10 @@ 
 	 *
 	 * @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 );
+	public function mark_complete($action_id) {
+		$store = $this->get_store_from_action_id($action_id);
+		if ($store) {
+			$store->mark_complete($action_id);
 		}
 	}
 
@@ -335,10 +335,10 @@ 
 	 *
 	 * @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 );
+	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;
 	}
@@ -350,13 +350,13 @@ 
 	 * @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 ) {
+	protected function get_store_from_action_id($action_id, $primary_first = false) {
+		if ($primary_first) {
 			$stores = array(
 				$this->primary_store,
 				$this->secondary_store,
 			);
-		} elseif ( $action_id < $this->demarkation_id ) {
+		} elseif ($action_id < $this->demarkation_id) {
 			$stores = array(
 				$this->secondary_store,
 				$this->primary_store,
@@ -367,9 +367,9 @@ 
 			);
 		}
 
-		foreach ( $stores as $store ) {
-			$action = $store->fetch_action( $action_id );
-			if ( ! is_a( $action, 'ActionScheduler_NullAction' ) ) {
+		foreach ($stores as $store) {
+			$action = $store->fetch_action($action_id);
+			if ( ! is_a($action, 'ActionScheduler_NullAction')) {
 				return $store;
 			}
 		}
@@ -394,8 +394,8 @@ 
 	 *
 	 * @param int $action_id Action ID.
 	 */
-	public function get_claim_id( $action_id ) {
-		return $this->primary_store->get_claim_id( $action_id );
+	public function get_claim_id($action_id) {
+		return $this->primary_store->get_claim_id($action_id);
 	}
 
 	/**
@@ -403,8 +403,8 @@ 
 	 *
 	 * @param ActionScheduler_ActionClaim $claim Claim object.
 	 */
-	public function release_claim( ActionScheduler_ActionClaim $claim ) {
-		$this->primary_store->release_claim( $claim );
+	public function release_claim(ActionScheduler_ActionClaim $claim) {
+		$this->primary_store->release_claim($claim);
 	}
 
 	/**
@@ -412,8 +412,8 @@ 
 	 *
 	 * @param int $action_id Action ID.
 	 */
-	public function unclaim_action( $action_id ) {
-		$this->primary_store->unclaim_action( $action_id );
+	public function unclaim_action($action_id) {
+		$this->primary_store->unclaim_action($action_id);
 	}
 
 	/**
@@ -421,7 +421,7 @@ 
 	 *
 	 * @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 );
+	public function find_actions_by_claim_id($claim_id) {
+		return $this->primary_store->find_actions_by_claim_id($claim_id);
 	}
 }