- Implements scheduler worker for welcome and post notifications

- Updates sending queue worker to save rendered newsletter body
- Updates sending queue router to schedule post notification newsletters

lib/Config/Hooks.php

@@ -149,8 +149,8 @@ class Hooks {
   }
 
   function setupCronWorkers() {
-    add_action('mailpoet_cron_worker', array($this, 'runSchedulerWorker'), 10, 1);
-    add_action('mailpoet_cron_worker', array($this, 'runSendingQueueWorker'), 10, 1);
+    add_action('mailpoet_scheduler_worker', array($this, 'runSchedulerWorker'), 10, 1);
+    add_action('mailpoet_queue_worker', array($this, 'runSendingQueueWorker'), 10, 1);
   }
 
   function runSchedulerWorker($timer) {

lib/Config/Migrator.php

@@ -214,6 +214,9 @@ class Migrator {
     $attributes = array(
       'id mediumint(9) NOT NULL AUTO_INCREMENT,',
       'newsletter_id mediumint(9) NOT NULL,',
+      'newsletter_rendered_body longtext,',
+      'newsletter_rendered_body_hash varchar(250) NULL DEFAULT NULL,',
+      'newsletter_rendered_subject varchar(250) NULL DEFAULT NULL,',
       'subscribers longtext,',
       'status varchar(12) NULL DEFAULT NULL,',
       'priority mediumint(9) NOT NULL DEFAULT 0,',

lib/Config/Populator.php

@@ -185,9 +185,9 @@ class Populator {
         'newsletter_type' => 'notification',
       ),
       array(
-        'name' => 'lastSentData',
+        'name' => 'segments',
         'newsletter_type' => 'notification',
-      ),
+      )
     );
   }
 

lib/Cron/Daemon.php

@@ -37,7 +37,8 @@ class Daemon {
     }
     $this->abortIfStopped($daemon);
     try {
-      do_action('mailpoet_cron_worker', $this->timer);
+      do_action('mailpoet_scheduler_worker', $this->timer);
+      do_action('mailpoet_queue_worker', $this->timer);
     } catch(\Exception $e) {
     }
     $elapsed_time = microtime(true) - $this->timer;

lib/Cron/Workers/Scheduler.php

@@ -4,8 +4,11 @@ namespace MailPoet\Cron\Workers;
 use Carbon\Carbon;
 use MailPoet\Cron\CronHelper;
 use MailPoet\Models\Newsletter;
+use MailPoet\Models\Segment;
 use MailPoet\Models\SendingQueue;
 use MailPoet\Models\SubscriberSegment;
+use MailPoet\Util\Helpers;
+use Cron\CronExpression as Cron;
 
 if(!defined('ABSPATH')) exit;
 
@@ -18,23 +21,29 @@ class Scheduler {
   }
 
   function process() {
-    $this->processScheduledNewsletters();
-  }
-
-  function processScheduledQueues() {
     $scheduled_queues = SendingQueue::where('status', 'scheduled')
-      ->whereLte('scheduled_at', Carbon::now()
-        ->format('Y-m-d H:i:s'))
+      ->whereLte('scheduled_at', Carbon::now()->format('Y-m-d H:i:s'))
       ->findMany();
     if(!count($scheduled_queues)) return;
     foreach($scheduled_queues as $queue) {
       $newsletter = Newsletter::filter('filterWithOptions')
-        ->findOne($queue->newsletter_id)
-        ->asArray();
+        ->findOne($queue->newsletter_id);
+      if(!$newsletter) {
+        $queue->delete();
+      } elseif($newsletter->type === 'welcome') {
+        $this->processWelcomeNewsletter($newsletter, $queue);
+      } elseif($newsletter->type === 'notification') {
+        $this->processPostNotificationNewsletter($newsletter, $queue);
+      }
+      CronHelper::checkExecutionTimer($this->timer);
+    }
+  }
+
+  function processWelcomeNewsletter($newsletter, $queue) {
     $subscriber = unserialize($queue->subscribers);
     $subscriber_in_segment =
       SubscriberSegment::where('subscriber_id', $subscriber['to_process'][0])
-          ->where('segment_id', $newsletter['segment'])
+        ->where('segment_id', $newsletter->segment)
         ->findOne();
     if(!$subscriber_in_segment) {
       $queue->delete();
@@ -42,7 +51,37 @@ class Scheduler {
       $queue->status = null;
       $queue->save();
     }
-      CronHelper::checkExecutionTimer($this->timer);
-    }
+  }
+
+  function processPostNotificationNewsletter($newsletter, $queue) {
+    $subscriber_ids = array();
+    $segments = Segment::whereIn('id', unserialize($newsletter->segments))
+      ->findMany();
+    foreach($segments as $segment) {
+      $subscriber_ids = array_merge(
+        $subscriber_ids,
+        Helpers::arrayColumn(
+          $segment->subscribers()->findArray(),
+          'id'
+        )
+      );
+    }
+    if(empty($subscriber_ids)) return;
+    // TODO: check if newsletter contents changed since last time it was sent
+    $subscriber_ids = array_unique($subscriber_ids);
+    $queue->subscribers = serialize(
+      array(
+        'to_process' => $subscriber_ids
+      )
+    );
+    $queue->count_total = $queue->count_to_process = count($subscriber_ids);
+    $queue->status = null;
+    $queue->save();
+    $new_queue = SendingQueue::create();
+    $new_queue->newsletter_id = $newsletter->id;
+    $schedule = Cron::factory($newsletter->schedule);
+    $new_queue->scheduled_at = $schedule->getNextRunDate()->format('Y-m-d H:i:s');
+    $new_queue->status = 'scheduled';
+    $new_queue->save();
   }
 }

lib/Cron/Workers/SendingQueue.php

@@ -10,7 +10,6 @@ use MailPoet\Models\Subscriber;
 use MailPoet\Newsletter\Renderer\Renderer;
 use MailPoet\Newsletter\Shortcodes\Shortcodes;
 use MailPoet\Util\Helpers;
-use MailPoet\Util\Security;
 
 if(!defined('ABSPATH')) exit;
 
@@ -37,6 +36,8 @@ class SendingQueue {
       if(!$newsletter) {
         continue;
       }
+      $newsletter = $newsletter->asArray();
+      $newsletter['body'] = $this->getNewsletterBodyAndSubject($queue, $newsletter);
       $queue->subscribers = (object) unserialize($queue->subscribers);
       if(!isset($queue->subscribers->processed)) {
         $queue->subscribers->processed = array();
@@ -44,8 +45,6 @@ class SendingQueue {
       if(!isset($queue->subscribers->failed)) {
         $queue->subscribers->failed = array();
       }
-      $newsletter = $newsletter->asArray();
-      $newsletter['body'] = $this->renderNewsletter($newsletter);
       $mailer = $this->configureMailer($newsletter);
       foreach(array_chunk($queue->subscribers->to_process, self::batch_size) as
               $subscribers_ids) {
@@ -67,6 +66,20 @@ class SendingQueue {
     }
   }
 
+  function getNewsletterBodyAndSubject($queue, $newsletter) {
+    // check if newsletter has been rendered, in which case return its contents
+    // or render & and for future use
+    if($queue->newsletter_rendered_body === null) {
+      $newsletter['body'] = $this->renderNewsletter($newsletter);
+      $queue->newsletter_rendered_body = json_encode($newsletter['body']);
+      $queue->newsletter_rendered_body_hash = md5($newsletter['body']['text']);
+      $queue->save();
+    } else {
+      $newsletter['body'] = json_decode($queue->newsletter_rendered_body);
+    }
+    return $newsletter['body'];
+  }
+
   function processBulkSubscribers($mailer, $newsletter, $subscribers, $queue) {
     foreach($subscribers as $subscriber) {
       $processed_newsletters[] =

lib/Newsletter/Scheduler/Scheduler.php

@@ -24,28 +24,33 @@ class Scheduler {
       '#' . $newsletter['nthWeekDay'];
     switch($interval_type) {
       case 'immediately':
-        $cron = '* * * * *';
+        $schedule = '* * * * *';
         break;
-      case 'immediate': //daily
-        $cron = sprintf('0 %s * * *', $hour);
+      case 'immediate':
+      case 'daily':
+        $schedule = sprintf('0 %s * * *', $hour);
         break;
       case 'weekly':
-        $cron = sprintf('0 %s * * %s', $hour, $week_day);
+        $schedule = sprintf('0 %s * * %s', $hour, $week_day);
         break;
       case 'monthly':
-        $cron = sprintf('0 %s %s * *', $hour, $month_day);
+        $schedule = sprintf('0 %s %s * *', $hour, $month_day);
         break;
       case 'nthWeekDay':
-        $cron = sprintf('0 %s ? * %s%s', $hour, $week_day, $nth_week_day);
+        $schedule = sprintf('0 %s ? * %s%s', $hour, $week_day, $nth_week_day);
         break;
     }
     $option_field = NewsletterOptionField::where('name', 'schedule')
       ->findOne()
       ->asArray();
+    $relation = NewsletterOption::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 = $cron;
+    }
+    $relation->value = $schedule;
     $relation->save();
   }
 

lib/Router/SendingQueue.php

@@ -2,14 +2,16 @@
 namespace MailPoet\Router;
 
 use MailPoet\Models\Newsletter;
+use MailPoet\Models\NewsletterOption;
+use MailPoet\Models\NewsletterOptionField;
 use MailPoet\Models\Segment;
 use MailPoet\Util\Helpers;
+use Cron\CronExpression as Cron;
 
 if(!defined('ABSPATH')) exit;
 
 class SendingQueue {
   function add($data) {
-    // check if mailer is properly configured
     try {
       new Mailer(false);
     } catch(\Exception $e) {
@@ -19,18 +21,58 @@ class SendingQueue {
       );
     }
 
-    $queue = \MailPoet\Models\SendingQueue::whereNull('status')
-      ->where('newsletter_id', $data['newsletter_id'])
-      ->findArray();
+    $newsletter = Newsletter::filter('filterWithOptions')
+      ->findOne($data['newsletter_id']);
+    if(!$newsletter) {
+      return array(
+        'result' => false,
+        'errors' => array(__('Newsletter does not exist.'))
+      );
+    }
 
+    $queue = \MailPoet\Models\SendingQueue::whereNull('status')
+      ->where('newsletter_id', $newsletter->id)
+      ->findOne();
     if(!empty($queue)) {
       return array(
         'result' => false,
         'errors' => array(__('Send operation is already in progress.'))
       );
     }
+
+    if($newsletter->type === 'notification') {
+      $option_field = NewsletterOptionField::where('name', 'segments')
+        ->where('newsletter_type', 'notification')
+        ->findOne();
+      $relation = NewsletterOption::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 = serialize($data['segments']);
+      $relation->save();
+
+      $queue = \MailPoet\Models\SendingQueue::where('status', 'scheduled')
+        ->where('newsletter_id', $newsletter->id)
+        ->findOne();
+      if(!$queue) {
         $queue = \MailPoet\Models\SendingQueue::create();
-    $queue->newsletter_id = $data['newsletter_id'];
+        $queue->newsletter_id = $newsletter->id;
+      }
+      $schedule = Cron::factory($newsletter->schedule);
+      $queue->scheduled_at = $schedule->getNextRunDate()->format('Y-m-d H:i:s');
+      $queue->status = 'scheduled';
+      $queue->save();
+      return array(
+        'result' => true,
+        'data' => array(__('Newsletter was scheduled for sending.'))
+      );
+    }
+
+    $queue = \MailPoet\Models\SendingQueue::create();
+    $queue->newsletter_id = $newsletter->id;
     $subscriber_ids = array();
     $segments = Segment::whereIn('id', $data['segments'])
       ->findMany();