piratepoet/lib/Newsletter/Scheduler/Scheduler.php

<?php
namespace MailPoet\Newsletter\Scheduler;

use Carbon\Carbon;
use MailPoet\Models\Newsletter;
use MailPoet\Models\NewsletterOption;
use MailPoet\Models\NewsletterOptionField;
use MailPoet\Models\SendingQueue;

class Scheduler {
  const seconds_in_hour = 3600;
  const last_weekday_format = 'L';
  const wp_all_roles = 'mailpoet_all';

  static function postNotification($newsletter_id) {
    $newsletter = Newsletter::filter('filterWithOptions')
      ->findOne($newsletter_id)
      ->asArray();
    $interval_type = $newsletter['intervalType'];
    $hour = (int) $newsletter['timeOfDay'] / self::seconds_in_hour;
    $week_day = $newsletter['weekDay'];
    $month_day = $newsletter['monthDay'];
    $nth_week_day = ($newsletter['nthWeekDay'] === self::last_weekday_format) ?
      $newsletter['nthWeekDay'] :
      '#' . $newsletter['nthWeekDay'];
    switch($interval_type) {
      case 'immediately':
        $schedule = '* * * * *';
        break;
      case 'immediate':
      case 'daily':
        $schedule = sprintf('0 %s * * *', $hour);
        break;
      case 'weekly':
        $schedule = sprintf('0 %s * * %s', $hour, $week_day);
        break;
      case 'monthly':
        $schedule = sprintf('0 %s %s * *', $hour, $month_day);
        break;
      case 'nthWeekDay':
        $schedule = sprintf('0 %s ? * %s%s', $hour, $week_day, $nth_week_day);
        break;
    }
    $option_field = NewsletterOptionField::where('name', 'schedule')
      ->findOne()
      ->asArray();
    $relation = NewsletterOption::where('newsletter_id', $newsletter_id)
      ->where('option_field_id', $option_field['id'])
      ->findOne();
    if(!$relation) {
      $relation = NewsletterOption::create();
      $relation->newsletter_id = $newsletter['id'];
      $relation->option_field_id = $option_field['id'];
    }
    $relation->value = $schedule;
    $relation->save();
  }

  static function welcomeForSegmentSubscription($subscriber_id, array $segments) {
    $newsletters = self::getWelcomeNewsletters();
    if(!count($newsletters)) return;
    foreach($newsletters as $newsletter) {
      if($newsletter['event'] === 'segment' &&
        in_array($newsletter['segment'], $segments)
      ) {
        self::createSendingQueueEntry($newsletter, $subscriber_id);
      }
    }
  }

  static function welcomeForNewWPUser($subscriber_id, array $wp_user, $old_user_data) {
    $newsletters = self::getWelcomeNewsletters();
    if(!count($newsletters)) return;
    foreach($newsletters as $newsletter) {
      if($newsletter['event'] === 'user') {
        if($old_user_data) {
          // do not schedule welcome newsletter if roles have not changed
          $old_role = (array) $old_user_data->roles;
          $new_role = (array) $wp_user->roles;
          if($newsletter['role'] === self::wp_all_roles ||
            !array_diff($old_role, $new_role)
          ) {
            continue;
          }
        }
        if($newsletter['role'] === self::wp_all_roles ||
          in_array($newsletter['role'], $wp_user['roles'])
        ) {
          self::createSendingQueueEntry($newsletter, $subscriber_id);
        }
      }
    }
  }

  private static function getWelcomeNewsletters() {
    return Newsletter::where('type', 'welcome')
      ->whereNull('deleted_at')
      ->filter('filterWithOptions')
      ->findArray();
  }

  private static function createSendingQueueEntry($newsletter, $subscriber_id) {
    $queue = SendingQueue::create();
    $queue->newsletter_id = $newsletter['id'];
    $queue->subscribers = serialize(
      array(
        'to_process' => array($subscriber_id)
      )
    );
    $queue->count_total = $queue->count_to_process = 1;
    $after_time_type = $newsletter['afterTimeType'];
    $after_time_number = $newsletter['afterTimeNumber'];
    $scheduled_at = null;
    $current_time = Carbon::createFromTimestamp(current_time('timestamp'));
    switch($after_time_type) {
      case 'hours':
        $scheduled_at = $current_time->addHours($after_time_number);
        break;
      case 'days':
        $scheduled_at = $current_time->addDays($after_time_number);
        break;
      case 'weeks':
        $scheduled_at = $current_time->addWeeks($after_time_number);
        break;
      default:
        $scheduled_at = $current_time;
    }
    $queue->status = 'scheduled';
    $queue->scheduled_at = $scheduled_at;
    $queue->save();
  }
}