iqltechnologies / brighty

plugins/woocommerce/packages/action-scheduler/classes/ActionScheduler_NullLogEntry.php

Indentation +4 added lines, -4 removed lines

@@ -4,8 +4,8 @@
  * Class ActionScheduler_NullLogEntry
  */
 class ActionScheduler_NullLogEntry extends ActionScheduler_LogEntry {
-	public function __construct( $action_id = '', $message = '' ) {
-		// nothing to see here
-	}
+    public function __construct( $action_id = '', $message = '' ) {
+        // nothing to see here
+    }
 }
- 
\ No newline at end of file
+    
\ No newline at end of file

plugins/woocommerce/packages/action-scheduler/classes/ActionScheduler_wcSystemStatus.php

Indentation +130 added lines, -130 removed lines

@@ -5,105 +5,105 @@ 
  */
 class ActionScheduler_wcSystemStatus {
 
-	/**
-	 * The active data stores
-	 *
-	 * @var ActionScheduler_Store
-	 */
-	protected $store;
-
-	/**
-	 * Constructor method for ActionScheduler_wcSystemStatus.
-	 *
-	 * @param ActionScheduler_Store $store Active store object.
-	 *
-	 * @return void
-	 */
-	public function __construct( $store ) {
-		$this->store = $store;
-	}
-
-	/**
-	 * Display action data, including number of actions grouped by status and the oldest & newest action in each status.
-	 *
-	 * Helpful to identify issues, like a clogged queue.
-	 */
-	public function render() {
-		$action_counts     = $this->store->action_counts();
-		$status_labels     = $this->store->get_status_labels();
-		$oldest_and_newest = $this->get_oldest_and_newest( array_keys( $status_labels ) );
-
-		$this->get_template( $status_labels, $action_counts, $oldest_and_newest );
-	}
-
-	/**
-	 * Get oldest and newest scheduled dates for a given set of statuses.
-	 *
-	 * @param array $status_keys Set of statuses to find oldest & newest action for.
-	 * @return array
-	 */
-	protected function get_oldest_and_newest( $status_keys ) {
-
-		$oldest_and_newest = array();
-
-		foreach ( $status_keys as $status ) {
-			$oldest_and_newest[ $status ] = array(
-				'oldest' => '–',
-				'newest' => '–',
-			);
-
-			if ( 'in-progress' === $status ) {
-				continue;
-			}
-
-			$oldest_and_newest[ $status ]['oldest'] = $this->get_action_status_date( $status, 'oldest' );
-			$oldest_and_newest[ $status ]['newest'] = $this->get_action_status_date( $status, 'newest' );
-		}
-
-		return $oldest_and_newest;
-	}
-
-	/**
-	 * Get oldest or newest scheduled date for a given status.
-	 *
-	 * @param string $status Action status label/name string.
-	 * @param string $date_type Oldest or Newest.
-	 * @return DateTime
-	 */
-	protected function get_action_status_date( $status, $date_type = 'oldest' ) {
-
-		$order = 'oldest' === $date_type ? 'ASC' : 'DESC';
-
-		$action = $this->store->query_actions(
-			array(
-				'claimed'  => false,
-				'status'   => $status,
-				'per_page' => 1,
-				'order'    => $order,
-			)
-		);
-
-		if ( ! empty( $action ) ) {
-			$date_object = $this->store->get_date( $action[0] );
-			$action_date = $date_object->format( 'Y-m-d H:i:s O' );
-		} else {
-			$action_date = '–';
-		}
-
-		return $action_date;
-	}
-
-	/**
-	 * Get oldest or newest scheduled date for a given status.
-	 *
-	 * @param array $status_labels Set of statuses to find oldest & newest action for.
-	 * @param array $action_counts Number of actions grouped by status.
-	 * @param array $oldest_and_newest Date of the oldest and newest action with each status.
-	 */
-	protected function get_template( $status_labels, $action_counts, $oldest_and_newest ) {
-		$as_version   = ActionScheduler_Versions::instance()->latest_version();
-		$as_datastore = get_class( ActionScheduler_Store::instance() );
-		?>
+    /**
+     * The active data stores
+     *
+     * @var ActionScheduler_Store
+     */
+    protected $store;
+
+    /**
+     * Constructor method for ActionScheduler_wcSystemStatus.
+     *
+     * @param ActionScheduler_Store $store Active store object.
+     *
+     * @return void
+     */
+    public function __construct( $store ) {
+        $this->store = $store;
+    }
+
+    /**
+     * Display action data, including number of actions grouped by status and the oldest & newest action in each status.
+     *
+     * Helpful to identify issues, like a clogged queue.
+     */
+    public function render() {
+        $action_counts     = $this->store->action_counts();
+        $status_labels     = $this->store->get_status_labels();
+        $oldest_and_newest = $this->get_oldest_and_newest( array_keys( $status_labels ) );
+
+        $this->get_template( $status_labels, $action_counts, $oldest_and_newest );
+    }
+
+    /**
+     * Get oldest and newest scheduled dates for a given set of statuses.
+     *
+     * @param array $status_keys Set of statuses to find oldest & newest action for.
+     * @return array
+     */
+    protected function get_oldest_and_newest( $status_keys ) {
+
+        $oldest_and_newest = array();
+
+        foreach ( $status_keys as $status ) {
+            $oldest_and_newest[ $status ] = array(
+                'oldest' => '–',
+                'newest' => '–',
+            );
+
+            if ( 'in-progress' === $status ) {
+                continue;
+            }
+
+            $oldest_and_newest[ $status ]['oldest'] = $this->get_action_status_date( $status, 'oldest' );
+            $oldest_and_newest[ $status ]['newest'] = $this->get_action_status_date( $status, 'newest' );
+        }
+
+        return $oldest_and_newest;
+    }
+
+    /**
+     * Get oldest or newest scheduled date for a given status.
+     *
+     * @param string $status Action status label/name string.
+     * @param string $date_type Oldest or Newest.
+     * @return DateTime
+     */
+    protected function get_action_status_date( $status, $date_type = 'oldest' ) {
+
+        $order = 'oldest' === $date_type ? 'ASC' : 'DESC';
+
+        $action = $this->store->query_actions(
+            array(
+                'claimed'  => false,
+                'status'   => $status,
+                'per_page' => 1,
+                'order'    => $order,
+            )
+        );
+
+        if ( ! empty( $action ) ) {
+            $date_object = $this->store->get_date( $action[0] );
+            $action_date = $date_object->format( 'Y-m-d H:i:s O' );
+        } else {
+            $action_date = '–';
+        }
+
+        return $action_date;
+    }
+
+    /**
+     * Get oldest or newest scheduled date for a given status.
+     *
+     * @param array $status_labels Set of statuses to find oldest & newest action for.
+     * @param array $action_counts Number of actions grouped by status.
+     * @param array $oldest_and_newest Date of the oldest and newest action with each status.
+     */
+    protected function get_template( $status_labels, $action_counts, $oldest_and_newest ) {
+        $as_version   = ActionScheduler_Versions::instance()->latest_version();
+        $as_datastore = get_class( ActionScheduler_Store::instance() );
+        ?>
 
 		<table class="wc_status_table widefat" cellspacing="0">
 			<thead>
@@ -128,39 +128,39 @@ 
 			</thead>
 			<tbody>
 				<?php
-				foreach ( $action_counts as $status => $count ) {
-					// WC uses the 3rd column for export, so we need to display more data in that (hidden when viewed as part of the table) and add an empty 2nd column.
-					printf(
-						'<tr><td>%1$s</td><td>&nbsp;</td><td>%2$s<span style="display: none;">, Oldest: %3$s, Newest: %4$s</span></td><td>%3$s</td><td>%4$s</td></tr>',
-						esc_html( $status_labels[ $status ] ),
-						esc_html( number_format_i18n( $count ) ),
-						esc_html( $oldest_and_newest[ $status ]['oldest'] ),
-						esc_html( $oldest_and_newest[ $status ]['newest'] )
-					);
-				}
-				?>
+                foreach ( $action_counts as $status => $count ) {
+                    // WC uses the 3rd column for export, so we need to display more data in that (hidden when viewed as part of the table) and add an empty 2nd column.
+                    printf(
+                        '<tr><td>%1$s</td><td>&nbsp;</td><td>%2$s<span style="display: none;">, Oldest: %3$s, Newest: %4$s</span></td><td>%3$s</td><td>%4$s</td></tr>',
+                        esc_html( $status_labels[ $status ] ),
+                        esc_html( number_format_i18n( $count ) ),
+                        esc_html( $oldest_and_newest[ $status ]['oldest'] ),
+                        esc_html( $oldest_and_newest[ $status ]['newest'] )
+                    );
+                }
+                ?>
 			</tbody>
 		</table>
 
 		<?php
-	}
-
-	/**
-	 * Is triggered when invoking inaccessible methods in an object context.
-	 *
-	 * @param string $name Name of method called.
-	 * @param array  $arguments Parameters to invoke the method with.
-	 *
-	 * @return mixed
-	 * @link https://php.net/manual/en/language.oop5.overloading.php#language.oop5.overloading.methods
-	 */
-	public function __call( $name, $arguments ) {
-		switch ( $name ) {
-			case 'print':
-				_deprecated_function( __CLASS__ . '::print()', '2.2.4', __CLASS__ . '::render()' );
-				return call_user_func_array( array( $this, 'render' ), $arguments );
-		}
-
-		return null;
-	}
+    }
+
+    /**
+     * Is triggered when invoking inaccessible methods in an object context.
+     *
+     * @param string $name Name of method called.
+     * @param array  $arguments Parameters to invoke the method with.
+     *
+     * @return mixed
+     * @link https://php.net/manual/en/language.oop5.overloading.php#language.oop5.overloading.methods
+     */
+    public function __call( $name, $arguments ) {
+        switch ( $name ) {
+            case 'print':
+                _deprecated_function( __CLASS__ . '::print()', '2.2.4', __CLASS__ . '::render()' );
+                return call_user_func_array( array( $this, 'render' ), $arguments );
+        }
+
+        return null;
+    }
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/Runner.php

Indentation +120 added lines, -120 removed lines

@@ -13,124 +13,124 @@
  * @codeCoverageIgnore
  */
 class Runner {
-	/** @var ActionScheduler_Store */
-	private $source_store;
-
-	/** @var ActionScheduler_Store */
-	private $destination_store;
-
-	/** @var ActionScheduler_Logger */
-	private $source_logger;
-
-	/** @var ActionScheduler_Logger */
-	private $destination_logger;
-
-	/** @var BatchFetcher */
-	private $batch_fetcher;
-
-	/** @var ActionMigrator */
-	private $action_migrator;
-
-	/** @var LogMigrator */
-	private $log_migrator;
-
-	/** @var ProgressBar */
-	private $progress_bar;
-
-	/**
-	 * Runner constructor.
-	 *
-	 * @param Config $config Migration configuration object.
-	 */
-	public function __construct( Config $config ) {
-		$this->source_store       = $config->get_source_store();
-		$this->destination_store  = $config->get_destination_store();
-		$this->source_logger      = $config->get_source_logger();
-		$this->destination_logger = $config->get_destination_logger();
-
-		$this->batch_fetcher = new BatchFetcher( $this->source_store );
-		if ( $config->get_dry_run() ) {
-			$this->log_migrator    = new DryRun_LogMigrator( $this->source_logger, $this->destination_logger );
-			$this->action_migrator = new DryRun_ActionMigrator( $this->source_store, $this->destination_store, $this->log_migrator );
-		} else {
-			$this->log_migrator    = new LogMigrator( $this->source_logger, $this->destination_logger );
-			$this->action_migrator = new ActionMigrator( $this->source_store, $this->destination_store, $this->log_migrator );
-		}
-
-		if ( defined( 'WP_CLI' ) && WP_CLI ) {
-			$this->progress_bar = $config->get_progress_bar();
-		}
-	}
-
-	/**
-	 * Run migration batch.
-	 *
-	 * @param int $batch_size Optional batch size. Default 10.
-	 *
-	 * @return int Size of batch processed.
-	 */
-	public function run( $batch_size = 10 ) {
-		$batch = $this->batch_fetcher->fetch( $batch_size );
-		$batch_size = count( $batch );
-
-		if ( ! $batch_size ) {
-			return 0;
-		}
-
-		if ( $this->progress_bar ) {
-			/* translators: %d: amount of actions */
-			$this->progress_bar->set_message( sprintf( _n( 'Migrating %d action', 'Migrating %d actions', $batch_size, 'woocommerce' ), number_format_i18n( $batch_size ) ) );
-			$this->progress_bar->set_count( $batch_size );
-		}
-
-		$this->migrate_actions( $batch );
-
-		return $batch_size;
-	}
-
-	/**
-	 * Migration a batch of actions.
-	 *
-	 * @param array $action_ids List of action IDs to migrate.
-	 */
-	public function migrate_actions( array $action_ids ) {
-		do_action( 'action_scheduler/migration_batch_starting', $action_ids );
-
-		\ActionScheduler::logger()->unhook_stored_action();
-		$this->destination_logger->unhook_stored_action();
-
-		foreach ( $action_ids as $source_action_id ) {
-			$destination_action_id = $this->action_migrator->migrate( $source_action_id );
-			if ( $destination_action_id ) {
-				$this->destination_logger->log( $destination_action_id, sprintf(
-					/* translators: 1: source action ID 2: source store class 3: destination action ID 4: destination store class */
-					__( 'Migrated action with ID %1$d in %2$s to ID %3$d in %4$s', 'woocommerce' ),
-					$source_action_id,
-					get_class( $this->source_store ),
-					$destination_action_id,
-					get_class( $this->destination_store )
-				) );
-			}
-
-			if ( $this->progress_bar ) {
-				$this->progress_bar->tick();
-			}
-		}
-
-		if ( $this->progress_bar ) {
-			$this->progress_bar->finish();
-		}
-
-		\ActionScheduler::logger()->hook_stored_action();
-
-		do_action( 'action_scheduler/migration_batch_complete', $action_ids );
-	}
-
-	/**
-	 * Initialize destination store and logger.
-	 */
-	public function init_destination() {
-		$this->destination_store->init();
-		$this->destination_logger->init();
-	}
+    /** @var ActionScheduler_Store */
+    private $source_store;
+
+    /** @var ActionScheduler_Store */
+    private $destination_store;
+
+    /** @var ActionScheduler_Logger */
+    private $source_logger;
+
+    /** @var ActionScheduler_Logger */
+    private $destination_logger;
+
+    /** @var BatchFetcher */
+    private $batch_fetcher;
+
+    /** @var ActionMigrator */
+    private $action_migrator;
+
+    /** @var LogMigrator */
+    private $log_migrator;
+
+    /** @var ProgressBar */
+    private $progress_bar;
+
+    /**
+     * Runner constructor.
+     *
+     * @param Config $config Migration configuration object.
+     */
+    public function __construct( Config $config ) {
+        $this->source_store       = $config->get_source_store();
+        $this->destination_store  = $config->get_destination_store();
+        $this->source_logger      = $config->get_source_logger();
+        $this->destination_logger = $config->get_destination_logger();
+
+        $this->batch_fetcher = new BatchFetcher( $this->source_store );
+        if ( $config->get_dry_run() ) {
+            $this->log_migrator    = new DryRun_LogMigrator( $this->source_logger, $this->destination_logger );
+            $this->action_migrator = new DryRun_ActionMigrator( $this->source_store, $this->destination_store, $this->log_migrator );
+        } else {
+            $this->log_migrator    = new LogMigrator( $this->source_logger, $this->destination_logger );
+            $this->action_migrator = new ActionMigrator( $this->source_store, $this->destination_store, $this->log_migrator );
+        }
+
+        if ( defined( 'WP_CLI' ) && WP_CLI ) {
+            $this->progress_bar = $config->get_progress_bar();
+        }
+    }
+
+    /**
+     * Run migration batch.
+     *
+     * @param int $batch_size Optional batch size. Default 10.
+     *
+     * @return int Size of batch processed.
+     */
+    public function run( $batch_size = 10 ) {
+        $batch = $this->batch_fetcher->fetch( $batch_size );
+        $batch_size = count( $batch );
+
+        if ( ! $batch_size ) {
+            return 0;
+        }
+
+        if ( $this->progress_bar ) {
+            /* translators: %d: amount of actions */
+            $this->progress_bar->set_message( sprintf( _n( 'Migrating %d action', 'Migrating %d actions', $batch_size, 'woocommerce' ), number_format_i18n( $batch_size ) ) );
+            $this->progress_bar->set_count( $batch_size );
+        }
+
+        $this->migrate_actions( $batch );
+
+        return $batch_size;
+    }
+
+    /**
+     * Migration a batch of actions.
+     *
+     * @param array $action_ids List of action IDs to migrate.
+     */
+    public function migrate_actions( array $action_ids ) {
+        do_action( 'action_scheduler/migration_batch_starting', $action_ids );
+
+        \ActionScheduler::logger()->unhook_stored_action();
+        $this->destination_logger->unhook_stored_action();
+
+        foreach ( $action_ids as $source_action_id ) {
+            $destination_action_id = $this->action_migrator->migrate( $source_action_id );
+            if ( $destination_action_id ) {
+                $this->destination_logger->log( $destination_action_id, sprintf(
+                    /* translators: 1: source action ID 2: source store class 3: destination action ID 4: destination store class */
+                    __( 'Migrated action with ID %1$d in %2$s to ID %3$d in %4$s', 'woocommerce' ),
+                    $source_action_id,
+                    get_class( $this->source_store ),
+                    $destination_action_id,
+                    get_class( $this->destination_store )
+                ) );
+            }
+
+            if ( $this->progress_bar ) {
+                $this->progress_bar->tick();
+            }
+        }
+
+        if ( $this->progress_bar ) {
+            $this->progress_bar->finish();
+        }
+
+        \ActionScheduler::logger()->hook_stored_action();
+
+        do_action( 'action_scheduler/migration_batch_complete', $action_ids );
+    }
+
+    /**
+     * Initialize destination store and logger.
+     */
+    public function init_destination() {
+        $this->destination_store->init();
+        $this->destination_logger->init();
+    }
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/DryRun_ActionMigrator.php

Indentation +11 added lines, -11 removed lines

@@ -13,16 +13,16 @@
  * @codeCoverageIgnore
  */
 class DryRun_ActionMigrator extends ActionMigrator {
-	/**
-	 * Simulate migrating an action.
-	 *
-	 * @param int $source_action_id Action ID.
-	 *
-	 * @return int
-	 */
-	public function migrate( $source_action_id ) {
-		do_action( 'action_scheduler/migrate_action_dry_run', $source_action_id );
+    /**
+     * Simulate migrating an action.
+     *
+     * @param int $source_action_id Action ID.
+     *
+     * @return int
+     */
+    public function migrate( $source_action_id ) {
+        do_action( 'action_scheduler/migrate_action_dry_run', $source_action_id );
 
-		return 0;
-	}
+        return 0;
+    }
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/ActionMigrator.php

Indentation +93 added lines, -93 removed lines

@@ -13,97 +13,97 @@
  * @codeCoverageIgnore
  */
 class ActionMigrator {
-	/** var ActionScheduler_Store */
-	private $source;
-
-	/** var ActionScheduler_Store */
-	private $destination;
-
-	/** var LogMigrator */
-	private $log_migrator;
-
-	/**
-	 * ActionMigrator constructor.
-	 *
-	 * @param ActionScheduler_Store $source_store Source store object.
-	 * @param ActionScheduler_Store $destination_store Destination store object.
-	 * @param LogMigrator           $log_migrator Log migrator object.
-	 */
-	public function __construct( \ActionScheduler_Store $source_store, \ActionScheduler_Store $destination_store, LogMigrator $log_migrator ) {
-		$this->source       = $source_store;
-		$this->destination  = $destination_store;
-		$this->log_migrator = $log_migrator;
-	}
-
-	/**
-	 * Migrate an action.
-	 *
-	 * @param int $source_action_id Action ID.
-	 *
-	 * @return int 0|new action ID
-	 */
-	public function migrate( $source_action_id ) {
-		try {
-			$action = $this->source->fetch_action( $source_action_id );
-			$status = $this->source->get_status( $source_action_id );
-		} catch ( \Exception $e ) {
-			$action = null;
-			$status = '';
-		}
-
-		if ( is_null( $action ) || empty( $status ) || ! $action->get_schedule()->get_date() ) {
-			// null action or empty status means the fetch operation failed or the action didn't exist
-			// null schedule means it's missing vital data
-			// delete it and move on
-			try {
-				$this->source->delete_action( $source_action_id );
-			} catch ( \Exception $e ) {
-				// nothing to do, it didn't exist in the first place
-			}
-			do_action( 'action_scheduler/no_action_to_migrate', $source_action_id, $this->source, $this->destination );
-
-			return 0;
-		}
-
-		try {
-
-			// Make sure the last attempt date is set correctly for completed and failed actions
-			$last_attempt_date = ( $status !== \ActionScheduler_Store::STATUS_PENDING ) ? $this->source->get_date( $source_action_id ) : null;
-
-			$destination_action_id = $this->destination->save_action( $action, null, $last_attempt_date );
-		} catch ( \Exception $e ) {
-			do_action( 'action_scheduler/migrate_action_failed', $source_action_id, $this->source, $this->destination );
-
-			return 0; // could not save the action in the new store
-		}
-
-		try {
-			switch ( $status ) {
-				case \ActionScheduler_Store::STATUS_FAILED :
-					$this->destination->mark_failure( $destination_action_id );
-					break;
-				case \ActionScheduler_Store::STATUS_CANCELED :
-					$this->destination->cancel_action( $destination_action_id );
-					break;
-			}
-
-			$this->log_migrator->migrate( $source_action_id, $destination_action_id );
-			$this->source->delete_action( $source_action_id );
-
-			$test_action = $this->source->fetch_action( $source_action_id );
-			if ( ! is_a( $test_action, 'ActionScheduler_NullAction' ) ) {
-				throw new \RuntimeException( sprintf( __( 'Unable to remove source migrated action %s', 'woocommerce' ), $source_action_id ) );
-			}
-			do_action( 'action_scheduler/migrated_action', $source_action_id, $destination_action_id, $this->source, $this->destination );
-
-			return $destination_action_id;
-		} catch ( \Exception $e ) {
-			// could not delete from the old store
-			$this->source->mark_migrated( $source_action_id );
-			do_action( 'action_scheduler/migrate_action_incomplete', $source_action_id, $destination_action_id, $this->source, $this->destination );
-			do_action( 'action_scheduler/migrated_action', $source_action_id, $destination_action_id, $this->source, $this->destination );
-
-			return $destination_action_id;
-		}
-	}
+    /** var ActionScheduler_Store */
+    private $source;
+
+    /** var ActionScheduler_Store */
+    private $destination;
+
+    /** var LogMigrator */
+    private $log_migrator;
+
+    /**
+     * ActionMigrator constructor.
+     *
+     * @param ActionScheduler_Store $source_store Source store object.
+     * @param ActionScheduler_Store $destination_store Destination store object.
+     * @param LogMigrator           $log_migrator Log migrator object.
+     */
+    public function __construct( \ActionScheduler_Store $source_store, \ActionScheduler_Store $destination_store, LogMigrator $log_migrator ) {
+        $this->source       = $source_store;
+        $this->destination  = $destination_store;
+        $this->log_migrator = $log_migrator;
+    }
+
+    /**
+     * Migrate an action.
+     *
+     * @param int $source_action_id Action ID.
+     *
+     * @return int 0|new action ID
+     */
+    public function migrate( $source_action_id ) {
+        try {
+            $action = $this->source->fetch_action( $source_action_id );
+            $status = $this->source->get_status( $source_action_id );
+        } catch ( \Exception $e ) {
+            $action = null;
+            $status = '';
+        }
+
+        if ( is_null( $action ) || empty( $status ) || ! $action->get_schedule()->get_date() ) {
+            // null action or empty status means the fetch operation failed or the action didn't exist
+            // null schedule means it's missing vital data
+            // delete it and move on
+            try {
+                $this->source->delete_action( $source_action_id );
+            } catch ( \Exception $e ) {
+                // nothing to do, it didn't exist in the first place
+            }
+            do_action( 'action_scheduler/no_action_to_migrate', $source_action_id, $this->source, $this->destination );
+
+            return 0;
+        }
+
+        try {
+
+            // Make sure the last attempt date is set correctly for completed and failed actions
+            $last_attempt_date = ( $status !== \ActionScheduler_Store::STATUS_PENDING ) ? $this->source->get_date( $source_action_id ) : null;
+
+            $destination_action_id = $this->destination->save_action( $action, null, $last_attempt_date );
+        } catch ( \Exception $e ) {
+            do_action( 'action_scheduler/migrate_action_failed', $source_action_id, $this->source, $this->destination );
+
+            return 0; // could not save the action in the new store
+        }
+
+        try {
+            switch ( $status ) {
+                case \ActionScheduler_Store::STATUS_FAILED :
+                    $this->destination->mark_failure( $destination_action_id );
+                    break;
+                case \ActionScheduler_Store::STATUS_CANCELED :
+                    $this->destination->cancel_action( $destination_action_id );
+                    break;
+            }
+
+            $this->log_migrator->migrate( $source_action_id, $destination_action_id );
+            $this->source->delete_action( $source_action_id );
+
+            $test_action = $this->source->fetch_action( $source_action_id );
+            if ( ! is_a( $test_action, 'ActionScheduler_NullAction' ) ) {
+                throw new \RuntimeException( sprintf( __( 'Unable to remove source migrated action %s', 'woocommerce' ), $source_action_id ) );
+            }
+            do_action( 'action_scheduler/migrated_action', $source_action_id, $destination_action_id, $this->source, $this->destination );
+
+            return $destination_action_id;
+        } catch ( \Exception $e ) {
+            // could not delete from the old store
+            $this->source->mark_migrated( $source_action_id );
+            do_action( 'action_scheduler/migrate_action_incomplete', $source_action_id, $destination_action_id, $this->source, $this->destination );
+            do_action( 'action_scheduler/migrated_action', $source_action_id, $destination_action_id, $this->source, $this->destination );
+
+            return $destination_action_id;
+        }
+    }
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/Scheduler.php

Indentation +111 added lines, -111 removed lines

@@ -13,116 +13,116 @@
  * @codeCoverageIgnore
  */
 class Scheduler {
-	/** Migration action hook. */
-	const HOOK            = 'action_scheduler/migration_hook';
-
-	/** Migration action group. */
-	const GROUP           = 'action-scheduler-migration';
-
-	/**
-	 * Set up the callback for the scheduled job.
-	 */
-	public function hook() {
-		add_action( self::HOOK, array( $this, 'run_migration' ), 10, 0 );
-	}
-
-	/**
-	 * Remove the callback for the scheduled job.
-	 */
-	public function unhook() {
-		remove_action( self::HOOK, array( $this, 'run_migration' ), 10 );
-	}
-
-	/**
-	 * The migration callback.
-	 */
-	public function run_migration() {
-		$migration_runner = $this->get_migration_runner();
-		$count            = $migration_runner->run( $this->get_batch_size() );
-
-		if ( $count === 0 ) {
-			$this->mark_complete();
-		} else {
-			$this->schedule_migration( time() + $this->get_schedule_interval() );
-		}
-	}
-
-	/**
-	 * Mark the migration complete.
-	 */
-	public function mark_complete() {
-		$this->unschedule_migration();
-
-		\ActionScheduler_DataController::mark_migration_complete();
-		do_action( 'action_scheduler/migration_complete' );
-	}
-
-	/**
-	 * Get a flag indicating whether the migration is scheduled.
-	 *
-	 * @return bool Whether there is a pending action in the store to handle the migration
-	 */
-	public function is_migration_scheduled() {
-		$next = as_next_scheduled_action( self::HOOK );
-
-		return ! empty( $next );
-	}
-
-	/**
-	 * Schedule the migration.
-	 *
-	 * @param int $when Optional timestamp to run the next migration batch. Defaults to now.
-	 *
-	 * @return string The action ID
-	 */
-	public function schedule_migration( $when = 0 ) {
-		$next = as_next_scheduled_action( self::HOOK );
-
-		if ( ! empty( $next ) ) {
-			return $next;
-		}
-
-		if ( empty( $when ) ) {
-			$when = time() + MINUTE_IN_SECONDS;
-		}
-
-		return as_schedule_single_action( $when, self::HOOK, array(), self::GROUP );
-	}
-
-	/**
-	 * Remove the scheduled migration action.
-	 */
-	public function unschedule_migration() {
-		as_unschedule_action( self::HOOK, null, self::GROUP );
-	}
-
-	/**
-	 * Get migration batch schedule interval.
-	 *
-	 * @return int Seconds between migration runs. Defaults to 0 seconds to allow chaining migration via Async Runners.
-	 */
-	private function get_schedule_interval() {
-		return (int) apply_filters( 'action_scheduler/migration_interval', 0 );
-	}
-
-	/**
-	 * Get migration batch size.
-	 *
-	 * @return int Number of actions to migrate in each batch. Defaults to 250.
-	 */
-	private function get_batch_size() {
-		return (int) apply_filters( 'action_scheduler/migration_batch_size', 250 );
-	}
-
-	/**
-	 * Get migration runner object.
-	 *
-	 * @return Runner
-	 */
-	private function get_migration_runner() {
-		$config = Controller::instance()->get_migration_config_object();
-
-		return new Runner( $config );
-	}
+    /** Migration action hook. */
+    const HOOK            = 'action_scheduler/migration_hook';
+
+    /** Migration action group. */
+    const GROUP           = 'action-scheduler-migration';
+
+    /**
+     * Set up the callback for the scheduled job.
+     */
+    public function hook() {
+        add_action( self::HOOK, array( $this, 'run_migration' ), 10, 0 );
+    }
+
+    /**
+     * Remove the callback for the scheduled job.
+     */
+    public function unhook() {
+        remove_action( self::HOOK, array( $this, 'run_migration' ), 10 );
+    }
+
+    /**
+     * The migration callback.
+     */
+    public function run_migration() {
+        $migration_runner = $this->get_migration_runner();
+        $count            = $migration_runner->run( $this->get_batch_size() );
+
+        if ( $count === 0 ) {
+            $this->mark_complete();
+        } else {
+            $this->schedule_migration( time() + $this->get_schedule_interval() );
+        }
+    }
+
+    /**
+     * Mark the migration complete.
+     */
+    public function mark_complete() {
+        $this->unschedule_migration();
+
+        \ActionScheduler_DataController::mark_migration_complete();
+        do_action( 'action_scheduler/migration_complete' );
+    }
+
+    /**
+     * Get a flag indicating whether the migration is scheduled.
+     *
+     * @return bool Whether there is a pending action in the store to handle the migration
+     */
+    public function is_migration_scheduled() {
+        $next = as_next_scheduled_action( self::HOOK );
+
+        return ! empty( $next );
+    }
+
+    /**
+     * Schedule the migration.
+     *
+     * @param int $when Optional timestamp to run the next migration batch. Defaults to now.
+     *
+     * @return string The action ID
+     */
+    public function schedule_migration( $when = 0 ) {
+        $next = as_next_scheduled_action( self::HOOK );
+
+        if ( ! empty( $next ) ) {
+            return $next;
+        }
+
+        if ( empty( $when ) ) {
+            $when = time() + MINUTE_IN_SECONDS;
+        }
+
+        return as_schedule_single_action( $when, self::HOOK, array(), self::GROUP );
+    }
+
+    /**
+     * Remove the scheduled migration action.
+     */
+    public function unschedule_migration() {
+        as_unschedule_action( self::HOOK, null, self::GROUP );
+    }
+
+    /**
+     * Get migration batch schedule interval.
+     *
+     * @return int Seconds between migration runs. Defaults to 0 seconds to allow chaining migration via Async Runners.
+     */
+    private function get_schedule_interval() {
+        return (int) apply_filters( 'action_scheduler/migration_interval', 0 );
+    }
+
+    /**
+     * Get migration batch size.
+     *
+     * @return int Number of actions to migrate in each batch. Defaults to 250.
+     */
+    private function get_batch_size() {
+        return (int) apply_filters( 'action_scheduler/migration_batch_size', 250 );
+    }
+
+    /**
+     * Get migration runner object.
+     *
+     * @return Runner
+     */
+    private function get_migration_runner() {
+        $config = Controller::instance()->get_migration_config_object();
+
+        return new Runner( $config );
+    }
 
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/Config.php

Indentation +148 added lines, -148 removed lines

@@ -17,152 +17,152 @@
  * A config builder for the ActionScheduler\Migration\Runner class
  */
 class Config {
-	/** @var ActionScheduler_Store */
-	private $source_store;
-
-	/** @var ActionScheduler_Logger */
-	private $source_logger;
-
-	/** @var ActionScheduler_Store */
-	private $destination_store;
-
-	/** @var ActionScheduler_Logger */
-	private $destination_logger;
-
-	/** @var Progress bar */
-	private $progress_bar;
-
-	/** @var bool */
-	private $dry_run = false;
-
-	/**
-	 * Config constructor.
-	 */
-	public function __construct() {
-
-	}
-
-	/**
-	 * Get the configured source store.
-	 *
-	 * @return ActionScheduler_Store
-	 */
-	public function get_source_store() {
-		if ( empty( $this->source_store ) ) {
-			throw new \RuntimeException( __( 'Source store must be configured before running a migration', 'woocommerce' ) );
-		}
-
-		return $this->source_store;
-	}
-
-	/**
-	 * Set the configured source store.
-	 *
-	 * @param ActionScheduler_Store $store Source store object.
-	 */
-	public function set_source_store( Store $store ) {
-		$this->source_store = $store;
-	}
-
-	/**
-	 * Get the configured source loger.
-	 *
-	 * @return ActionScheduler_Logger
-	 */
-	public function get_source_logger() {
-		if ( empty( $this->source_logger ) ) {
-			throw new \RuntimeException( __( 'Source logger must be configured before running a migration', 'woocommerce' ) );
-		}
-
-		return $this->source_logger;
-	}
-
-	/**
-	 * Set the configured source logger.
-	 *
-	 * @param ActionScheduler_Logger $logger
-	 */
-	public function set_source_logger( Logger $logger ) {
-		$this->source_logger = $logger;
-	}
-
-	/**
-	 * Get the configured destination store.
-	 *
-	 * @return ActionScheduler_Store
-	 */
-	public function get_destination_store() {
-		if ( empty( $this->destination_store ) ) {
-			throw new \RuntimeException( __( 'Destination store must be configured before running a migration', 'woocommerce' ) );
-		}
-
-		return $this->destination_store;
-	}
-
-	/**
-	 * Set the configured destination store.
-	 *
-	 * @param ActionScheduler_Store $store
-	 */
-	public function set_destination_store( Store $store ) {
-		$this->destination_store = $store;
-	}
-
-	/**
-	 * Get the configured destination logger.
-	 *
-	 * @return ActionScheduler_Logger
-	 */
-	public function get_destination_logger() {
-		if ( empty( $this->destination_logger ) ) {
-			throw new \RuntimeException( __( 'Destination logger must be configured before running a migration', 'woocommerce' ) );
-		}
-
-		return $this->destination_logger;
-	}
-
-	/**
-	 * Set the configured destination logger.
-	 *
-	 * @param ActionScheduler_Logger $logger
-	 */
-	public function set_destination_logger( Logger $logger ) {
-		$this->destination_logger = $logger;
-	}
-
-	/**
-	 * Get flag indicating whether it's a dry run.
-	 *
-	 * @return bool
-	 */
-	public function get_dry_run() {
-		return $this->dry_run;
-	}
-
-	/**
-	 * Set flag indicating whether it's a dry run.
-	 *
-	 * @param bool $dry_run
-	 */
-	public function set_dry_run( $dry_run ) {
-		$this->dry_run = (bool) $dry_run;
-	}
-
-	/**
-	 * Get progress bar object.
-	 *
-	 * @return ActionScheduler\WPCLI\ProgressBar
-	 */
-	public function get_progress_bar() {
-		return $this->progress_bar;
-	}
-
-	/**
-	 * Set progress bar object.
-	 *
-	 * @param ActionScheduler\WPCLI\ProgressBar $progress_bar
-	 */
-	public function set_progress_bar( ProgressBar $progress_bar ) {
-		$this->progress_bar = $progress_bar;
-	}
+    /** @var ActionScheduler_Store */
+    private $source_store;
+
+    /** @var ActionScheduler_Logger */
+    private $source_logger;
+
+    /** @var ActionScheduler_Store */
+    private $destination_store;
+
+    /** @var ActionScheduler_Logger */
+    private $destination_logger;
+
+    /** @var Progress bar */
+    private $progress_bar;
+
+    /** @var bool */
+    private $dry_run = false;
+
+    /**
+     * Config constructor.
+     */
+    public function __construct() {
+
+    }
+
+    /**
+     * Get the configured source store.
+     *
+     * @return ActionScheduler_Store
+     */
+    public function get_source_store() {
+        if ( empty( $this->source_store ) ) {
+            throw new \RuntimeException( __( 'Source store must be configured before running a migration', 'woocommerce' ) );
+        }
+
+        return $this->source_store;
+    }
+
+    /**
+     * Set the configured source store.
+     *
+     * @param ActionScheduler_Store $store Source store object.
+     */
+    public function set_source_store( Store $store ) {
+        $this->source_store = $store;
+    }
+
+    /**
+     * Get the configured source loger.
+     *
+     * @return ActionScheduler_Logger
+     */
+    public function get_source_logger() {
+        if ( empty( $this->source_logger ) ) {
+            throw new \RuntimeException( __( 'Source logger must be configured before running a migration', 'woocommerce' ) );
+        }
+
+        return $this->source_logger;
+    }
+
+    /**
+     * Set the configured source logger.
+     *
+     * @param ActionScheduler_Logger $logger
+     */
+    public function set_source_logger( Logger $logger ) {
+        $this->source_logger = $logger;
+    }
+
+    /**
+     * Get the configured destination store.
+     *
+     * @return ActionScheduler_Store
+     */
+    public function get_destination_store() {
+        if ( empty( $this->destination_store ) ) {
+            throw new \RuntimeException( __( 'Destination store must be configured before running a migration', 'woocommerce' ) );
+        }
+
+        return $this->destination_store;
+    }
+
+    /**
+     * Set the configured destination store.
+     *
+     * @param ActionScheduler_Store $store
+     */
+    public function set_destination_store( Store $store ) {
+        $this->destination_store = $store;
+    }
+
+    /**
+     * Get the configured destination logger.
+     *
+     * @return ActionScheduler_Logger
+     */
+    public function get_destination_logger() {
+        if ( empty( $this->destination_logger ) ) {
+            throw new \RuntimeException( __( 'Destination logger must be configured before running a migration', 'woocommerce' ) );
+        }
+
+        return $this->destination_logger;
+    }
+
+    /**
+     * Set the configured destination logger.
+     *
+     * @param ActionScheduler_Logger $logger
+     */
+    public function set_destination_logger( Logger $logger ) {
+        $this->destination_logger = $logger;
+    }
+
+    /**
+     * Get flag indicating whether it's a dry run.
+     *
+     * @return bool
+     */
+    public function get_dry_run() {
+        return $this->dry_run;
+    }
+
+    /**
+     * Set flag indicating whether it's a dry run.
+     *
+     * @param bool $dry_run
+     */
+    public function set_dry_run( $dry_run ) {
+        $this->dry_run = (bool) $dry_run;
+    }
+
+    /**
+     * Get progress bar object.
+     *
+     * @return ActionScheduler\WPCLI\ProgressBar
+     */
+    public function get_progress_bar() {
+        return $this->progress_bar;
+    }
+
+    /**
+     * Set progress bar object.
+     *
+     * @param ActionScheduler\WPCLI\ProgressBar $progress_bar
+     */
+    public function set_progress_bar( ProgressBar $progress_bar ) {
+        $this->progress_bar = $progress_bar;
+    }
 }

plugins/woocommerce/packages/action-scheduler/classes/migration/BatchFetcher.php

Indentation +61 added lines, -61 removed lines

@@ -16,71 +16,71 @@
  * @codeCoverageIgnore
  */
 class BatchFetcher {
-	/** var ActionScheduler_Store */
-	private $store;
+    /** var ActionScheduler_Store */
+    private $store;
 
-	/**
-	 * BatchFetcher constructor.
-	 *
-	 * @param ActionScheduler_Store $source_store Source store object.
-	 */
-	public function __construct( Store $source_store ) {
-		$this->store = $source_store;
-	}
+    /**
+     * BatchFetcher constructor.
+     *
+     * @param ActionScheduler_Store $source_store Source store object.
+     */
+    public function __construct( Store $source_store ) {
+        $this->store = $source_store;
+    }
 
-	/**
-	 * Retrieve a list of actions.
-	 *
-	 * @param int $count The number of actions to retrieve
-	 *
-	 * @return int[] A list of action IDs
-	 */
-	public function fetch( $count = 10 ) {
-		foreach ( $this->get_query_strategies( $count ) as $query ) {
-			$action_ids = $this->store->query_actions( $query );
-			if ( ! empty( $action_ids ) ) {
-				return $action_ids;
-			}
-		}
+    /**
+     * Retrieve a list of actions.
+     *
+     * @param int $count The number of actions to retrieve
+     *
+     * @return int[] A list of action IDs
+     */
+    public function fetch( $count = 10 ) {
+        foreach ( $this->get_query_strategies( $count ) as $query ) {
+            $action_ids = $this->store->query_actions( $query );
+            if ( ! empty( $action_ids ) ) {
+                return $action_ids;
+            }
+        }
 
-		return [];
-	}
+        return [];
+    }
 
-	/**
-	 * Generate a list of prioritized of action search parameters.
-	 *
-	 * @param int $count Number of actions to find.
-	 *
-	 * @return array
-	 */
-	private function get_query_strategies( $count ) {
-		$now  = as_get_datetime_object();
-		$args = [
-			'date'     => $now,
-			'per_page' => $count,
-			'offset'   => 0,
-			'orderby'  => 'date',
-			'order'    => 'ASC',
-		];
+    /**
+     * Generate a list of prioritized of action search parameters.
+     *
+     * @param int $count Number of actions to find.
+     *
+     * @return array
+     */
+    private function get_query_strategies( $count ) {
+        $now  = as_get_datetime_object();
+        $args = [
+            'date'     => $now,
+            'per_page' => $count,
+            'offset'   => 0,
+            'orderby'  => 'date',
+            'order'    => 'ASC',
+        ];
 
-		$priorities = [
-			Store::STATUS_PENDING,
-			Store::STATUS_FAILED,
-			Store::STATUS_CANCELED,
-			Store::STATUS_COMPLETE,
-			Store::STATUS_RUNNING,
-			'', // any other unanticipated status
-		];
+        $priorities = [
+            Store::STATUS_PENDING,
+            Store::STATUS_FAILED,
+            Store::STATUS_CANCELED,
+            Store::STATUS_COMPLETE,
+            Store::STATUS_RUNNING,
+            '', // any other unanticipated status
+        ];
 
-		foreach ( $priorities as $status ) {
-			yield wp_parse_args( [
-				'status'       => $status,
-				'date_compare' => '<=',
-			], $args );
-			yield wp_parse_args( [
-				'status'       => $status,
-				'date_compare' => '>=',
-			], $args );
-		}
-	}
+        foreach ( $priorities as $status ) {
+            yield wp_parse_args( [
+                'status'       => $status,
+                'date_compare' => '<=',
+            ], $args );
+            yield wp_parse_args( [
+                'status'       => $status,
+                'date_compare' => '>=',
+            ], $args );
+        }
+    }
 }
\ No newline at end of file

plugins/woocommerce/packages/action-scheduler/classes/migration/Controller.php

Indentation +195 added lines, -195 removed lines

@@ -19,85 +19,85 @@ 
  * @codeCoverageIgnore
  */
 class Controller {
-	private static $instance;
-
-	/** @var Action_Scheduler\Migration\Scheduler */
-	private $migration_scheduler;
-
-	/** @var string */
-	private $store_classname;
-
-	/** @var string */
-	private $logger_classname;
-
-	/** @var bool */
-	private $migrate_custom_store;
-
-	/**
-	 * Controller constructor.
-	 *
-	 * @param Scheduler $migration_scheduler Migration scheduler object.
-	 */
-	protected function __construct( Scheduler $migration_scheduler ) {
-		$this->migration_scheduler = $migration_scheduler;
-		$this->store_classname     = '';
-	}
-
-	/**
-	 * Set the action store class name.
-	 *
-	 * @param string $class Classname of the store class.
-	 *
-	 * @return string
-	 */
-	public function get_store_class( $class ) {
-		if ( \ActionScheduler_DataController::is_migration_complete() ) {
-			return \ActionScheduler_DataController::DATASTORE_CLASS;
-		} elseif ( \ActionScheduler_Store::DEFAULT_CLASS !== $class ) {
-			$this->store_classname = $class;
-			return $class;
-		} else {
-			return 'ActionScheduler_HybridStore';
-		}
-	}
-
-	/**
-	 * Set the action logger class name.
-	 *
-	 * @param string $class Classname of the logger class.
-	 *
-	 * @return string
-	 */
-	public function get_logger_class( $class ) {
-		\ActionScheduler_Store::instance();
-
-		if ( $this->has_custom_datastore() ) {
-			$this->logger_classname = $class;
-			return $class;
-		} else {
-			return \ActionScheduler_DataController::LOGGER_CLASS;
-		}
-	}
-
-	/**
-	 * Get flag indicating whether a custom datastore is in use.
-	 *
-	 * @return bool
-	 */
-	public function has_custom_datastore() {
-		return (bool) $this->store_classname;
-	}
-
-	/**
-	 * Set up the background migration process.
-	 *
-	 * @return void
-	 */
-	public function schedule_migration() {
-		$logging_tables = new ActionScheduler_LoggerSchema();
-		$store_tables   = new ActionScheduler_StoreSchema();
-
-		/*
+    private static $instance;
+
+    /** @var Action_Scheduler\Migration\Scheduler */
+    private $migration_scheduler;
+
+    /** @var string */
+    private $store_classname;
+
+    /** @var string */
+    private $logger_classname;
+
+    /** @var bool */
+    private $migrate_custom_store;
+
+    /**
+     * Controller constructor.
+     *
+     * @param Scheduler $migration_scheduler Migration scheduler object.
+     */
+    protected function __construct( Scheduler $migration_scheduler ) {
+        $this->migration_scheduler = $migration_scheduler;
+        $this->store_classname     = '';
+    }
+
+    /**
+     * Set the action store class name.
+     *
+     * @param string $class Classname of the store class.
+     *
+     * @return string
+     */
+    public function get_store_class( $class ) {
+        if ( \ActionScheduler_DataController::is_migration_complete() ) {
+            return \ActionScheduler_DataController::DATASTORE_CLASS;
+        } elseif ( \ActionScheduler_Store::DEFAULT_CLASS !== $class ) {
+            $this->store_classname = $class;
+            return $class;
+        } else {
+            return 'ActionScheduler_HybridStore';
+        }
+    }
+
+    /**
+     * Set the action logger class name.
+     *
+     * @param string $class Classname of the logger class.
+     *
+     * @return string
+     */
+    public function get_logger_class( $class ) {
+        \ActionScheduler_Store::instance();
+
+        if ( $this->has_custom_datastore() ) {
+            $this->logger_classname = $class;
+            return $class;
+        } else {
+            return \ActionScheduler_DataController::LOGGER_CLASS;
+        }
+    }
+
+    /**
+     * Get flag indicating whether a custom datastore is in use.
+     *
+     * @return bool
+     */
+    public function has_custom_datastore() {
+        return (bool) $this->store_classname;
+    }
+
+    /**
+     * Set up the background migration process.
+     *
+     * @return void
+     */
+    public function schedule_migration() {
+        $logging_tables = new ActionScheduler_LoggerSchema();
+        $store_tables   = new ActionScheduler_StoreSchema();
+
+        /*
 		 * In some unusual cases, the expected tables may not have been created. In such cases
 		 * we do not schedule a migration as doing so will lead to fatal error conditions.
 		 *
@@ -107,120 +107,120 @@ 
 		 *
 		 * @see https://github.com/woocommerce/action-scheduler/issues/653
 		 */
-		if (
-			ActionScheduler_DataController::is_migration_complete()
-			|| $this->migration_scheduler->is_migration_scheduled()
-			|| ! $store_tables->tables_exist()
-			|| ! $logging_tables->tables_exist()
-		) {
-			return;
-		}
-
-		$this->migration_scheduler->schedule_migration();
-	}
-
-	/**
-	 * Get the default migration config object
-	 *
-	 * @return ActionScheduler\Migration\Config
-	 */
-	public function get_migration_config_object() {
-		static $config = null;
-
-		if ( ! $config ) {
-			$source_store  = $this->store_classname ? new $this->store_classname() : new \ActionScheduler_wpPostStore();
-			$source_logger = $this->logger_classname ? new $this->logger_classname() : new \ActionScheduler_wpCommentLogger();
-
-			$config = new Config();
-			$config->set_source_store( $source_store );
-			$config->set_source_logger( $source_logger );
-			$config->set_destination_store( new \ActionScheduler_DBStoreMigrator() );
-			$config->set_destination_logger( new \ActionScheduler_DBLogger() );
-
-			if ( defined( 'WP_CLI' ) && WP_CLI ) {
-				$config->set_progress_bar( new ProgressBar( '', 0 ) );
-			}
-		}
-
-		return apply_filters( 'action_scheduler/migration_config', $config );
-	}
-
-	/**
-	 * Hook dashboard migration notice.
-	 */
-	public function hook_admin_notices() {
-		if ( ! $this->allow_migration() || \ActionScheduler_DataController::is_migration_complete() ) {
-			return;
-		}
-		add_action( 'admin_notices', array( $this, 'display_migration_notice' ), 10, 0 );
-	}
-
-	/**
-	 * Show a dashboard notice that migration is in progress.
-	 */
-	public function display_migration_notice() {
-		printf( '<div class="notice notice-warning"><p>%s</p></div>', esc_html__( 'Action Scheduler migration in progress. The list of scheduled actions may be incomplete.', 'woocommerce' ) );
-	}
-
-	/**
-	 * Add store classes. Hook migration.
-	 */
-	private function hook() {
-		add_filter( 'action_scheduler_store_class', array( $this, 'get_store_class' ), 100, 1 );
-		add_filter( 'action_scheduler_logger_class', array( $this, 'get_logger_class' ), 100, 1 );
-		add_action( 'init', array( $this, 'maybe_hook_migration' ) );
-		add_action( 'wp_loaded', array( $this, 'schedule_migration' ) );
-
-		// Action Scheduler may be displayed as a Tools screen or WooCommerce > Status administration screen
-		add_action( 'load-tools_page_action-scheduler', array( $this, 'hook_admin_notices' ), 10, 0 );
-		add_action( 'load-woocommerce_page_wc-status', array( $this, 'hook_admin_notices' ), 10, 0 );
-	}
-
-	/**
-	 * Possibly hook the migration scheduler action.
-	 *
-	 * @author Jeremy Pry
-	 */
-	public function maybe_hook_migration() {
-		if ( ! $this->allow_migration() || \ActionScheduler_DataController::is_migration_complete() ) {
-			return;
-		}
-
-		$this->migration_scheduler->hook();
-	}
-
-	/**
-	 * Allow datastores to enable migration to AS tables.
-	 */
-	public function allow_migration() {
-		if ( ! \ActionScheduler_DataController::dependencies_met() ) {
-			return false;
-		}
-
-		if ( null === $this->migrate_custom_store ) {
-			$this->migrate_custom_store = apply_filters( 'action_scheduler_migrate_data_store', false );
-		}
-
-		return ( ! $this->has_custom_datastore() ) || $this->migrate_custom_store;
-	}
-
-	/**
-	 * Proceed with the migration if the dependencies have been met.
-	 */
-	public static function init() {
-		if ( \ActionScheduler_DataController::dependencies_met() ) {
-			self::instance()->hook();
-		}
-	}
-
-	/**
-	 * Singleton factory.
-	 */
-	public static function instance() {
-		if ( ! isset( self::$instance ) ) {
-			self::$instance = new static( new Scheduler() );
-		}
-
-		return self::$instance;
-	}
+        if (
+            ActionScheduler_DataController::is_migration_complete()
+            || $this->migration_scheduler->is_migration_scheduled()
+            || ! $store_tables->tables_exist()
+            || ! $logging_tables->tables_exist()
+        ) {
+            return;
+        }
+
+        $this->migration_scheduler->schedule_migration();
+    }
+
+    /**
+     * Get the default migration config object
+     *
+     * @return ActionScheduler\Migration\Config
+     */
+    public function get_migration_config_object() {
+        static $config = null;
+
+        if ( ! $config ) {
+            $source_store  = $this->store_classname ? new $this->store_classname() : new \ActionScheduler_wpPostStore();
+            $source_logger = $this->logger_classname ? new $this->logger_classname() : new \ActionScheduler_wpCommentLogger();
+
+            $config = new Config();
+            $config->set_source_store( $source_store );
+            $config->set_source_logger( $source_logger );
+            $config->set_destination_store( new \ActionScheduler_DBStoreMigrator() );
+            $config->set_destination_logger( new \ActionScheduler_DBLogger() );
+
+            if ( defined( 'WP_CLI' ) && WP_CLI ) {
+                $config->set_progress_bar( new ProgressBar( '', 0 ) );
+            }
+        }
+
+        return apply_filters( 'action_scheduler/migration_config', $config );
+    }
+
+    /**
+     * Hook dashboard migration notice.
+     */
+    public function hook_admin_notices() {
+        if ( ! $this->allow_migration() || \ActionScheduler_DataController::is_migration_complete() ) {
+            return;
+        }
+        add_action( 'admin_notices', array( $this, 'display_migration_notice' ), 10, 0 );
+    }
+
+    /**
+     * Show a dashboard notice that migration is in progress.
+     */
+    public function display_migration_notice() {
+        printf( '<div class="notice notice-warning"><p>%s</p></div>', esc_html__( 'Action Scheduler migration in progress. The list of scheduled actions may be incomplete.', 'woocommerce' ) );
+    }
+
+    /**
+     * Add store classes. Hook migration.
+     */
+    private function hook() {
+        add_filter( 'action_scheduler_store_class', array( $this, 'get_store_class' ), 100, 1 );
+        add_filter( 'action_scheduler_logger_class', array( $this, 'get_logger_class' ), 100, 1 );
+        add_action( 'init', array( $this, 'maybe_hook_migration' ) );
+        add_action( 'wp_loaded', array( $this, 'schedule_migration' ) );
+
+        // Action Scheduler may be displayed as a Tools screen or WooCommerce > Status administration screen
+        add_action( 'load-tools_page_action-scheduler', array( $this, 'hook_admin_notices' ), 10, 0 );
+        add_action( 'load-woocommerce_page_wc-status', array( $this, 'hook_admin_notices' ), 10, 0 );
+    }
+
+    /**
+     * Possibly hook the migration scheduler action.
+     *
+     * @author Jeremy Pry
+     */
+    public function maybe_hook_migration() {
+        if ( ! $this->allow_migration() || \ActionScheduler_DataController::is_migration_complete() ) {
+            return;
+        }
+
+        $this->migration_scheduler->hook();
+    }
+
+    /**
+     * Allow datastores to enable migration to AS tables.
+     */
+    public function allow_migration() {
+        if ( ! \ActionScheduler_DataController::dependencies_met() ) {
+            return false;
+        }
+
+        if ( null === $this->migrate_custom_store ) {
+            $this->migrate_custom_store = apply_filters( 'action_scheduler_migrate_data_store', false );
+        }
+
+        return ( ! $this->has_custom_datastore() ) || $this->migrate_custom_store;
+    }
+
+    /**
+     * Proceed with the migration if the dependencies have been met.
+     */
+    public static function init() {
+        if ( \ActionScheduler_DataController::dependencies_met() ) {
+            self::instance()->hook();
+        }
+    }
+
+    /**
+     * Singleton factory.
+     */
+    public static function instance() {
+        if ( ! isset( self::$instance ) ) {
+            self::$instance = new static( new Scheduler() );
+        }
+
+        return self::$instance;
+    }
 }