You are here

function _date_repeat_calc in Date 7.2

Same name and namespace in other branches
  1. 5.2 date_repeat/date_repeat_calc.inc \_date_repeat_calc()
  2. 6.2 date_repeat/date_repeat_calc.inc \_date_repeat_calc()
  3. 6 date_repeat/date_repeat_calc.inc \_date_repeat_calc()
  4. 7.3 date_repeat/date_repeat_calc.inc \_date_repeat_calc()
  5. 7 date_repeat/date_repeat_calc.inc \_date_repeat_calc()

Private implementation of date_repeat_calc().

Compute dates that match the requested rule, within a specified date range.

1 call to _date_repeat_calc()
date_repeat_calc in date_repeat/date_repeat.module
Analyze a RRULE and return dates that match it.

File

date_repeat/date_repeat_calc.inc, line 34
Code to compute the dates that match an iCal RRULE.

Code

function _date_repeat_calc($rrule, $start, $end, $exceptions, $timezone, $additions) {
  module_load_include('inc', 'date_api', 'date_api_ical');
  if (empty($timezone)) {
    $timezone = date_default_timezone();
  }

  // Make sure the 'EXCEPTIONS' string isn't appended to the rule.
  $parts = explode("\n", $rrule);
  if (count($parts)) {
    $rrule = $parts[0];
  }

  // Get the parsed array of rule values.
  $rrule = date_ical_parse_rrule('RRULE:', $rrule);

  // These default values indicate there is no RRULE here.
  if ($rrule['FREQ'] == 'NONE' || isset($rrule['INTERVAL']) && $rrule['INTERVAL'] == 0) {
    return array();
  }

  // Create a date object for the start and end dates.
  $start_date = new DateObject($start, $timezone);

  // Versions of PHP greater than PHP 5.3.5 require that we set an explicit
  // time when using date_modify() or the time may not match the original value.
  // Adding this modifier gives us the same results in both older and newer
  // versions of PHP.
  $modify_time = ' ' . $start_date
    ->format('g:ia');

  // If the rule has an UNTIL, see if that is earlier than the end date.
  if (!empty($rrule['UNTIL'])) {
    $end_date = new DateObject($end, $timezone);
    $until_date = date_ical_date($rrule['UNTIL'], $timezone);
    if (date_format($until_date, 'U') < date_format($end_date, 'U')) {
      $end_date = $until_date;
    }
  }
  elseif (empty($end)) {
    if (!empty($rrule['COUNT'])) {
      $end_date = NULL;
    }
    else {
      return array();
    }
  }
  else {
    $end_date = new DateObject($end, $timezone);
  }

  // Get an integer value for the interval, if none given, '1' is implied.
  if (empty($rrule['INTERVAL'])) {
    $rrule['INTERVAL'] = 1;
  }
  $interval = max(1, $rrule['INTERVAL']);
  $count = isset($rrule['COUNT']) ? $rrule['COUNT'] : NULL;
  if (empty($rrule['FREQ'])) {
    $rrule['FREQ'] = 'DAILY';
  }

  // Make sure DAILY frequency isn't used in places it won't work.
  if (!empty($rrule['BYMONTHDAY']) && !in_array($rrule['FREQ'], array(
    'MONTHLY',
    'YEARLY',
  ))) {
    $rrule['FREQ'] = 'MONTHLY';
  }
  elseif (!empty($rrule['BYDAY']) && !in_array($rrule['FREQ'], array(
    'MONTHLY',
    'WEEKLY',
    'YEARLY',
  ))) {
    $rrule['FREQ'] = 'WEEKLY';
  }

  // Find the time period to jump forward between dates.
  switch ($rrule['FREQ']) {
    case 'DAILY':
      $jump = $interval . ' days';
      break;
    case 'WEEKLY':
      $jump = $interval . ' weeks';
      break;
    case 'MONTHLY':
      $jump = $interval . ' months';
      break;
    case 'YEARLY':
      $jump = $interval . ' years';
      break;
  }
  $rrule = date_repeat_adjust_rrule($rrule, $start_date);

  // The start date always goes into the results, whether or not it meets the
  // rules. RFC 2445 includes examples where the start date DOES NOT meet the
  // rules, but the expected results always include the start date.
  $days = array(
    date_format($start_date, DATE_FORMAT_DATETIME),
  );

  // BYMONTHDAY will look for specific days of the month in one or more months.
  // This process is only valid when frequency is monthly or yearly.
  if (!empty($rrule['BYMONTHDAY'])) {
    $finished = FALSE;
    $current_day = clone $start_date;
    $direction_days = array();

    // Deconstruct the day in case it has a negative modifier.
    foreach ($rrule['BYMONTHDAY'] as $day) {
      preg_match("@(-)?([0-9]{1,2})@", $day, $regs);
      if (!empty($regs[2])) {

        // Convert parameters into full day name, count, and direction.
        $direction_days[$day] = array(
          'direction' => !empty($regs[1]) ? $regs[1] : '+',
          'direction_count' => $regs[2],
        );
      }
    }
    while (!$finished) {
      $period_finished = FALSE;
      while (!$period_finished) {
        foreach ($rrule['BYMONTHDAY'] as $monthday) {
          $day = $direction_days[$monthday];
          $current_day = date_repeat_set_month_day($current_day, NULL, $day['direction_count'], $day['direction'], $timezone, $modify_time);
          date_repeat_add_dates($days, $current_day, $start_date, $end_date, $exceptions, $rrule);
          if ($finished = date_repeat_is_finished($current_day, $days, $count, $end_date)) {
            $period_finished = TRUE;
          }
        }

        // If it's monthly, keep looping through months, one INTERVAL at a time.
        if ($rrule['FREQ'] == 'MONTHLY') {
          if ($finished = date_repeat_is_finished($current_day, $days, $count, $end_date)) {
            $period_finished = TRUE;
          }

          // Back up to first of month and jump.
          $current_day = date_repeat_set_month_day($current_day, NULL, 1, '+', $timezone, $modify_time);
          date_modify($current_day, '+' . $jump . $modify_time);
        }
        else {
          if (date_format($current_day, 'n') == 12) {
            $period_finished = TRUE;
          }
          else {

            // Back up to first of month and jump.
            $current_day = date_repeat_set_month_day($current_day, NULL, 1, '+', $timezone, $modify_time);
            date_modify($current_day, '+1 month' . $modify_time);
          }
        }
      }
      if ($rrule['FREQ'] == 'YEARLY') {

        // Back up to first of year and jump.
        $current_day = date_repeat_set_year_day($current_day, NULL, NULL, 1, '+', $timezone, $modify_time);
        date_modify($current_day, '+' . $jump . $modify_time);
      }
      $finished = date_repeat_is_finished($current_day, $days, $count, $end_date);
    }
  }
  elseif (empty($rrule['BYDAY'])) {

    // $current_day will keep track of where we are in the calculation.
    $current_day = clone $start_date;
    $finished = FALSE;
    $months = !empty($rrule['BYMONTH']) ? $rrule['BYMONTH'] : array();
    while (!$finished) {
      date_repeat_add_dates($days, $current_day, $start_date, $end_date, $exceptions, $rrule);
      $finished = date_repeat_is_finished($current_day, $days, $count, $end_date);
      date_modify($current_day, '+' . $jump . $modify_time);
    }
  }
  else {

    // More complex searches for day names and criteria like '-1SU' or
    // '2TU,2TH', require that we interate through the whole time period
    // checking each BYDAY. Create helper array to pull day names out of iCal
    // day strings.
    $day_names = date_repeat_dow_day_options(FALSE);
    $days_of_week = array_keys($day_names);

    // Parse out information about the BYDAYs and separate them depending on
    // whether they have directional parameters like -1SU or 2TH.
    $month_days = array();
    $week_days = array();

    // Find the right first day of the week to use, iCal rules say Monday
    // should be used if none is specified.
    $week_start_rule = !empty($rrule['WKST']) ? trim($rrule['WKST']) : 'MO';
    $week_start_day = $day_names[$week_start_rule];

    // Make sure the week days array is sorted into week order, we use the
    // $ordered_keys to get the right values into the key and force the array
    // to that order. Needed later when we iterate through each week looking
    // for days so we don't jump to the next week when we hit a day out of
    // order.
    $ordered = date_repeat_days_ordered($week_start_rule);
    $ordered_keys = array_flip($ordered);
    if ($rrule['FREQ'] == 'YEARLY' && !empty($rrule['BYMONTH'])) {

      // Additional cycle to apply month preferences.
      foreach ($rrule['BYMONTH'] as $month) {
        foreach ($rrule['BYDAY'] as $day) {
          preg_match("@(-)?([0-9]+)?([SU|MO|TU|WE|TH|FR|SA]{2})@", trim($day), $regs);

          // Convert parameters into full day name, count, and direction. Add
          // leading zero to first 9 months.
          if (!empty($regs[2])) {
            $direction_days[] = array(
              'day' => $day_names[$regs[3]],
              'direction' => !empty($regs[1]) ? $regs[1] : '+',
              'direction_count' => $regs[2],
              'month' => strlen($month) > 1 ? $month : '0' . $month,
            );
          }
          else {
            $week_days[$ordered_keys[$regs[3]]] = $day_names[$regs[3]];
          }
        }
      }
    }
    else {
      foreach ($rrule['BYDAY'] as $day) {
        preg_match("@(-)?([0-9]+)?([SU|MO|TU|WE|TH|FR|SA]{2})@", trim($day), $regs);
        if (!empty($regs[2])) {

          // Convert parameters into full day name, count, and direction.
          $direction_days[] = array(
            'day' => $day_names[$regs[3]],
            'direction' => !empty($regs[1]) ? $regs[1] : '+',
            'direction_count' => $regs[2],
            'month' => NULL,
          );
        }
        else {
          $week_days[$ordered_keys[$regs[3]]] = $day_names[$regs[3]];
        }
      }
    }
    ksort($week_days);

    // BYDAYs with parameters like -1SU (last Sun) or 2TH (second Thur) need to
    // be processed one month or year at a time.
    if (!empty($direction_days) && in_array($rrule['FREQ'], array(
      'MONTHLY',
      'YEARLY',
    ))) {
      $finished = FALSE;
      $current_day = clone $start_date;
      while (!$finished) {
        foreach ($direction_days as $day) {

          // Find the BYDAY date in the current month.
          if ($rrule['FREQ'] == 'MONTHLY') {
            $current_day = date_repeat_set_month_day($current_day, $day['day'], $day['direction_count'], $day['direction'], $timezone, $modify_time);
          }
          else {
            $current_day = date_repeat_set_year_day($current_day, $day['month'], $day['day'], $day['direction_count'], $day['direction'], $timezone, $modify_time);
          }
          date_repeat_add_dates($days, $current_day, $start_date, $end_date, $exceptions, $rrule);
        }
        $finished = date_repeat_is_finished($current_day, $days, $count, $end_date);

        // Reset to beginning of period before jumping to next period. Needed
        // especially when working with values like 'last Saturday' to be sure
        // we don't skip months like February.
        $year = date_format($current_day, 'Y');
        $month = date_format($current_day, 'n');
        if ($rrule['FREQ'] == 'MONTHLY') {
          date_date_set($current_day, $year, $month, 1);
        }
        else {
          date_date_set($current_day, $year, 1, 1);
        }

        // Jump to the next period.
        date_modify($current_day, '+' . $jump . $modify_time);
      }
    }

    // For BYDAYs without parameters,like TU,TH (every Tues and Thur), we look
    // for every one of those days during the frequency period. Iterate through
    // periods of a WEEK, MONTH, or YEAR, checking for the days of the week
    // that match our criteria for each week in the period, then jumping ahead
    // to the next week, month, or year, an INTERVAL at a time.
    if (!empty($week_days) && in_array($rrule['FREQ'], array(
      'MONTHLY',
      'WEEKLY',
      'YEARLY',
    ))) {
      $finished = FALSE;
      $current_day = clone $start_date;
      $format = $rrule['FREQ'] == 'YEARLY' ? 'Y' : 'n';
      $current_period = date_format($current_day, $format);

      // Back up to the beginning of the week in case we are somewhere in the
      // middle of the possible week days, needed so we don't prematurely jump
      // to the next week. The date_repeat_add_dates() function will keep dates
      // outside the range from getting added.
      if (date_format($current_day, 'l') != $day_names[$day]) {
        date_modify($current_day, '-1 ' . $week_start_day . $modify_time);
      }
      while (!$finished) {
        $period_finished = FALSE;
        while (!$period_finished) {
          $moved = FALSE;
          foreach ($week_days as $delta => $day) {

            // Find the next occurence of each day in this week, only add it if
            // we are still in the current month or year. The
            // date_repeat_add_dates() function is insufficient to test whether
            // to include this date if we are using a rule like 'every other
            // month', so we must explicitly test it here. If we're already on
            // the right day, don't jump or we will prematurely move into the
            // next week.
            if (date_format($current_day, 'l') != $day) {
              date_modify($current_day, '+1 ' . $day . $modify_time);
              $moved = TRUE;
            }
            if ($rrule['FREQ'] == 'WEEKLY' || date_format($current_day, $format) == $current_period) {
              date_repeat_add_dates($days, $current_day, $start_date, $end_date, $exceptions, $rrule);
            }
          }
          $finished = date_repeat_is_finished($current_day, $days, $count, $end_date);

          // Make sure we don't get stuck in endless loop if the current day
          // never got changed above.
          if (!$moved) {
            date_modify($current_day, '+1 day' . $modify_time);
          }

          // If this is a WEEKLY frequency, stop after each week, otherwise,
          // stop when we've moved outside the current period. Jump to the end
          // of the week, then test the period.
          if ($finished || $rrule['FREQ'] == 'WEEKLY') {
            $period_finished = TRUE;
          }
          elseif ($rrule['FREQ'] != 'WEEKLY' && date_format($current_day, $format) != $current_period) {
            $period_finished = TRUE;
          }
        }
        if ($finished) {
          continue;
        }

        // We'll be at the end of a week, month, or year when we get to this
        // point in the code. Go back to the beginning of this period before we
        // jump, to ensure we jump to the first day of the next period.
        switch ($rrule['FREQ']) {
          case 'WEEKLY':
            date_modify($current_day, '+1 ' . $week_start_day . $modify_time);
            date_modify($current_day, '-1 week' . $modify_time);
            break;
          case 'MONTHLY':
            date_modify($current_day, '-' . (date_format($current_day, 'j') - 1) . ' days' . $modify_time);
            date_modify($current_day, '-1 month' . $modify_time);
            break;
          case 'YEARLY':
            date_modify($current_day, '-' . date_format($current_day, 'z') . ' days' . $modify_time);
            date_modify($current_day, '-1 year' . $modify_time);
            break;
        }

        // Jump ahead to the next period to be evaluated.
        date_modify($current_day, '+' . $jump . $modify_time);
        $current_period = date_format($current_day, $format);
        $finished = date_repeat_is_finished($current_day, $days, $count, $end_date);
      }
    }
  }

  // Add additional dates.
  foreach ($additions as $addition) {
    $date = new dateObject($addition . ' ' . $start_date
      ->format('H:i:s'), $timezone);
    $days[] = date_format($date, DATE_FORMAT_DATETIME);
  }
  sort($days);
  return $days;
}