kangana/application/classes/Model/Task.php

<?php defined('SYSPATH') OR die('No direct access allowed.');
/**
 * Task model.
 * Task is a job to send the next letter in a course.
 * @package Models
 * @author Oreolek
 **/
class Model_Task extends ORM {
  const STATUS_UNKNOWN = 0;
  const STATUS_PENDING = 1;
  const STATUS_SENDING = 2;
  const STATUS_SENT = 3;
  const STATUS_CANCELED = 4;

  protected $_has_many = array(
    'letter' => array(
      'model' => 'Letter'
    ),
    'client' => array(
      'model' => 'Client'
    )
  );

  /**
   * @return array validation rules
   **/
  public function rules()
  {
    return array(
      'date' => array(
        array('not_empty'),
        array('date'),
      ),
      'status' => array(
        array('numeric')
      )
    );
  }

  /**
   * Array of field labels.
   * Used in forms.
   **/
  protected $_labels = array(
    'date' => 'Mailing date',
    'status' => 'Status'
  );

  public function execute()
  {
    $letter = ORM::factory('Letter', $this->letter_id);
    $client = ORM::factory('Client', $this->client_id);
    $letter->send($client->email);
    $this->status = STATUS_SENT;
    $this->save();
  }

  /**
   * Get next unsent letter by client. Returns FALSE if there're no letters left to send.
   * @param array $letters array of letter IDs (integers) to send
   * @param int $client_id client ID
   * @return int/bool first unsent letter ID or FALSE
   **/
  public static function next_unsent($client_id, $letters)
  {
    $query = DB::select(array(DB::expr('COUNT(*)'), 'cnt'))
      ->from('tasks');
    if (is_array($letters))
    {
      $query = $query->where('letter_id', 'IN', $letters);
    }
    else
    {
      $query = $query->where('letter_id', '=', $letters);
    }
    $cnt = $query
      ->and_where('status', '=', self::STATUS_SENT)
      ->and_where('client_id', '=', $client_id)
      ->execute()
      ->get('cnt');
    if ($cnt < count($letters))
    {
      $query = DB::select('letter_id')
        ->from('tasks');
      if (is_array($letters))
      {
        $query = $query->where('letter_id', 'IN', $letters);
      }
      else
      {
        $query = $query->where('letter_id', '=', $letters);
      }
      $sent_letters = $query
        ->and_where('status', '=', self::STATUS_SENT)
        ->and_where('client_id', '=', $client_id)
        ->execute()
        ->as_array(NULL, 'letter_id');
      if (is_array($letters))
      {
        $retval = $letters[0];
        $i = 0;
        while ($i < $cnt)
        {
          $retval = $letters[$i];
          if ( ! in_array($retval, $sent_letters))
          {
            break;
          }
          $i++;
        }
        if ($i < $cnt)
        {
          return $retval;
        }
        elseif($i < count( $letters ))
        {
          return $letters[$i];
        }
        return FALSE;
      }
      else
      {
        // count($letters) is > count($sent_letters), $letters is not an array
        // so there are no sent letters
        return $letters;
      }
    }
    return FALSE;
  }

  public static function prepare($client_id, $letter_id)
  {
    return DB::insert('tasks', array('client_id', 'letter_id', 'date', 'status'))
      ->values(array($client_id, $letter_id, date('Y-m-d'), self::STATUS_PENDING))
      ->execute();
  }

  /**
   * Get last sent or prepared letter and check if it's time to send another one.
   **/
  public static function check_period($client_id, $letters, $period)
  {
    $query = DB::select('date')
      ->from('tasks');
    if (is_array($letters))
    {
      $query = $query->where('letter_id', 'IN', $letters);
    }
    else
    {
      $query = $query->where('letter_id', '=', $letters);
    }
    $check = NULL;
    $check = $query
      ->and_where('status', '=', self::STATUS_SENT)
      ->or_where('status', '=', self::STATUS_PENDING)
      ->and_where('client_id', '=', $client_id)
      ->and_where(DB::expr('DATEDIFF(CURDATE(), `date`)'), '=', $period)
      ->execute()
      ->get('date');
    if ( ! empty($check))
      return TRUE;
    return FALSE;
  }

  /**
   * Cancel tasks for specified client. If a course ID parameter is present, cancel only tasks for this course.
   **/
  public static function cancel($client_id, $course_id = 0)
  {
    $query = DB::update('tasks')->set(array('status' => self::STATUS_CANCELED))->where('client_id', '=', $client_id);
    if ($course_id > 0)
    {
      $query->and_where('letter_id', 'IN', Model_Course::get_letter_ids($course_id));
    }
    return $query->execute();
  }
}