mirror of
https://github.com/pronamic/woocommerce-subscriptions.git
synced 2025-10-13 12:52:56 +00:00
198 lines
7.6 KiB
PHP
198 lines
7.6 KiB
PHP
<?php
|
|
/**
|
|
* Scheduler for subscription events that uses the Action Scheduler
|
|
*
|
|
* @class WCS_Action_Scheduler
|
|
* @version 2.0.0
|
|
* @package WooCommerce Subscriptions/Classes
|
|
* @category Class
|
|
* @author Prospress
|
|
*/
|
|
class WCS_Action_Scheduler extends WCS_Scheduler {
|
|
|
|
/*@protected Array of $action_hook => $date_type values */
|
|
protected $action_hooks = array(
|
|
'woocommerce_scheduled_subscription_trial_end' => 'trial_end',
|
|
'woocommerce_scheduled_subscription_payment' => 'next_payment',
|
|
'woocommerce_scheduled_subscription_payment_retry' => 'payment_retry',
|
|
'woocommerce_scheduled_subscription_expiration' => 'end',
|
|
);
|
|
|
|
/**
|
|
* Maybe set a schedule action if the new date is in the future
|
|
*
|
|
* @param object $subscription An instance of a WC_Subscription object
|
|
* @param string $date_type Can be 'trial_end', 'next_payment', 'payment_retry', 'end', 'end_of_prepaid_term' or a custom date type
|
|
* @param string $datetime A MySQL formated date/time string in the GMT/UTC timezone.
|
|
*/
|
|
public function update_date( $subscription, $date_type, $datetime ) {
|
|
|
|
if ( in_array( $date_type, $this->get_date_types_to_schedule() ) ) {
|
|
|
|
$action_hook = $this->get_scheduled_action_hook( $subscription, $date_type );
|
|
|
|
if ( ! empty( $action_hook ) ) {
|
|
|
|
$action_args = $this->get_action_args( $date_type, $subscription );
|
|
$timestamp = wcs_date_to_time( $datetime );
|
|
$next_scheduled = wc_next_scheduled_action( $action_hook, $action_args );
|
|
|
|
if ( $next_scheduled !== $timestamp ) {
|
|
|
|
// Maybe clear the existing schedule for this hook
|
|
$this->unschedule_actions( $action_hook, $action_args );
|
|
|
|
// Only reschedule if it's in the future
|
|
if ( $timestamp > current_time( 'timestamp', true ) && ( 'payment_retry' == $date_type || 'active' == $subscription->get_status() ) ) {
|
|
wc_schedule_single_action( $timestamp, $action_hook, $action_args );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Delete a date from the action scheduler queue
|
|
*
|
|
* @param object $subscription An instance of a WC_Subscription object
|
|
* @param string $date_type Can be 'trial_end', 'next_payment', 'end', 'end_of_prepaid_term' or a custom date type
|
|
*/
|
|
public function delete_date( $subscription, $date_type ) {
|
|
$this->update_date( $subscription, $date_type, 0 );
|
|
}
|
|
|
|
/**
|
|
* When a subscription's status is updated, maybe schedule an event
|
|
*
|
|
* @param object $subscription An instance of a WC_Subscription object
|
|
* @param string $date_type Can be 'trial_end', 'next_payment', 'end', 'end_of_prepaid_term' or a custom date type
|
|
* @param string $datetime A MySQL formated date/time string in the GMT/UTC timezone.
|
|
*/
|
|
public function update_status( $subscription, $new_status, $old_status ) {
|
|
|
|
switch ( $new_status ) {
|
|
case 'active' :
|
|
|
|
foreach ( $this->action_hooks as $action_hook => $date_type ) {
|
|
|
|
$event_time = $subscription->get_time( $date_type );
|
|
|
|
// If there's no payment retry date, avoid calling get_action_args() because it calls the resource intensive WC_Subscription::get_last_order() / get_related_orders()
|
|
if ( 'payment_retry' === $date_type && 0 === $event_time ) {
|
|
continue;
|
|
}
|
|
|
|
$action_args = $this->get_action_args( $date_type, $subscription );
|
|
$next_scheduled = wc_next_scheduled_action( $action_hook, $action_args );
|
|
|
|
// Maybe clear the existing schedule for this hook
|
|
if ( false !== $next_scheduled && $next_scheduled != $event_time ) {
|
|
$this->unschedule_actions( $action_hook, $action_args );
|
|
}
|
|
|
|
if ( 0 != $event_time && $event_time > current_time( 'timestamp', true ) && $next_scheduled != $event_time ) {
|
|
wc_schedule_single_action( $event_time, $action_hook, $action_args );
|
|
}
|
|
}
|
|
break;
|
|
case 'pending-cancel' :
|
|
|
|
// Now that we have the current times, clear the scheduled hooks
|
|
foreach ( $this->action_hooks as $action_hook => $date_type ) {
|
|
$this->unschedule_actions( $action_hook, $this->get_action_args( $date_type, $subscription ) );
|
|
}
|
|
|
|
$end_time = $subscription->get_time( 'end' ); // This will have been set to the correct date already
|
|
$action_args = $this->get_action_args( 'end', $subscription );
|
|
$next_scheduled = wc_next_scheduled_action( 'woocommerce_scheduled_subscription_end_of_prepaid_term', $action_args );
|
|
|
|
if ( false !== $next_scheduled && $next_scheduled != $end_time ) {
|
|
$this->unschedule_actions( 'woocommerce_scheduled_subscription_end_of_prepaid_term', $action_args );
|
|
}
|
|
|
|
// The end date was set in WC_Subscriptions::update_dates() to the appropriate value, so we can schedule our action for that time
|
|
if ( $end_time > current_time( 'timestamp', true ) && $next_scheduled != $end_time ) {
|
|
wc_schedule_single_action( $end_time, 'woocommerce_scheduled_subscription_end_of_prepaid_term', $action_args );
|
|
}
|
|
break;
|
|
case 'on-hold' :
|
|
case 'cancelled' :
|
|
case 'switched' :
|
|
case 'expired' :
|
|
case 'trash' :
|
|
foreach ( $this->action_hooks as $action_hook => $date_type ) {
|
|
$this->unschedule_actions( $action_hook, $this->get_action_args( $date_type, $subscription ) );
|
|
}
|
|
$this->unschedule_actions( 'woocommerce_scheduled_subscription_end_of_prepaid_term', $this->get_action_args( 'end', $subscription ) );
|
|
break;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the hook to use in the action scheduler for the date type
|
|
*
|
|
* @param object $subscription An instance of WC_Subscription to get the hook for
|
|
* @param string $date_type Can be 'trial_end', 'next_payment', 'expiration', 'end_of_prepaid_term' or a custom date type
|
|
*/
|
|
protected function get_scheduled_action_hook( $subscription, $date_type ) {
|
|
|
|
$hook = '';
|
|
|
|
switch ( $date_type ) {
|
|
case 'next_payment' :
|
|
$hook = 'woocommerce_scheduled_subscription_payment';
|
|
break;
|
|
case 'payment_retry' :
|
|
$hook = 'woocommerce_scheduled_subscription_payment_retry';
|
|
break;
|
|
case 'trial_end' :
|
|
$hook = 'woocommerce_scheduled_subscription_trial_end';
|
|
break;
|
|
case 'end' :
|
|
// End dates may need either an expiration or end of prepaid term hook, depending on the status
|
|
if ( $subscription->has_status( 'cancelled' ) ) {
|
|
$hook = 'woocommerce_scheduled_subscription_end_of_prepaid_term';
|
|
} elseif ( $subscription->has_status( 'active' ) ) {
|
|
$hook = 'woocommerce_scheduled_subscription_expiration';
|
|
}
|
|
break;
|
|
}
|
|
|
|
return apply_filters( 'woocommerce_subscriptions_scheduled_action_hook', $hook, $date_type );
|
|
}
|
|
|
|
/**
|
|
* Get the args to set on the scheduled action.
|
|
*
|
|
* @param string $date_type Can be 'trial_end', 'next_payment', 'expiration', 'end_of_prepaid_term' or a custom date type
|
|
* @param object $subscription An instance of WC_Subscription to get the hook for
|
|
* @return array Array of name => value pairs stored against the scheduled action.
|
|
*/
|
|
protected function get_action_args( $date_type, $subscription ) {
|
|
|
|
if ( 'payment_retry' == $date_type ) {
|
|
|
|
$last_order_id = $subscription->get_last_order( 'ids', 'renewal' );
|
|
$action_args = array( 'order_id' => $last_order_id );
|
|
|
|
} else {
|
|
$action_args = array( 'subscription_id' => $subscription->get_id() );
|
|
}
|
|
|
|
return apply_filters( 'woocommerce_subscriptions_scheduled_action_args', $action_args, $date_type, $subscription );
|
|
}
|
|
|
|
/**
|
|
* Get the args to set on the scheduled action.
|
|
*
|
|
* @param string $$action_hook Name of event used as the hook for the scheduled action.
|
|
* @param array $action_args Array of name => value pairs stored against the scheduled action.
|
|
*/
|
|
protected function unschedule_actions( $action_hook, $action_args ) {
|
|
do {
|
|
wc_unschedule_action( $action_hook, $action_args );
|
|
$next_scheduled = wc_next_scheduled_action( $action_hook, $action_args );
|
|
} while ( false !== $next_scheduled );
|
|
}
|
|
}
|