improved byday support

[sope] / sope-ical / NGiCal / iCalRecurrenceRule.m

diff --git a/sope-ical/NGiCal/iCalRecurrenceRule.m b/sope-ical/NGiCal/iCalRecurrenceRule.m
index 55a1a1db2af1700838885290e8dce59cff9d1f6f..7a03c4673aff7f467c015006847992af46e7965e 100644 (file)
--- a/sope-ical/NGiCal/iCalRecurrenceRule.m
+++ b/sope-ical/NGiCal/iCalRecurrenceRule.m
@@ -144,7 +144,7 @@
     if (c0 == 'f' || c0 == 'F') return iCalWeekDayFriday;
 
     c1 = [_day characterAtIndex:1];
-    if (c0 == 't' || c0 == 't') {
+    if (c0 == 't' || c0 == 'T') {
       if (c1 == 'u' || c1 == 'U') return iCalWeekDayTuesday;
       if (c1 == 'h' || c1 == 'H') return iCalWeekDayThursday;
     }
@@ -206,7 +206,7 @@
   NSMutableString *s;
   unsigned        i, mask, day;
   BOOL            needsComma;
-
+  
   s          = [NSMutableString stringWithCapacity:20];
   needsComma = NO;
   mask       = self->byDay.mask;
@@ -216,6 +216,10 @@
     if (mask & day) {
       if (needsComma)
         [s appendString:@","];
+      else if (self->byDay.useOccurence)
+       // Note: we only support one occurrence currently
+       [s appendFormat:@"%i", self->byDayOccurence1];
+      
       [s appendString:[self iCalRepresentationForWeekDay:day]];
       needsComma = YES;
     }
@@ -260,6 +264,7 @@
 /* properties */
 
 - (void)setFreq:(NSString *)_freq {
+  // TODO: shouldn't we preserve what the user gives us?
   _freq = [_freq uppercaseString];
   if ([_freq isEqualToString:@"WEEKLY"])
     self->frequency = iCalRecurrenceFrequenceWeekly;
@@ -299,17 +304,60 @@
 }
 
 - (void)setByday:(NSString *)_byDayList {
+  // TODO: each day can have an associated occurence, eg:
+  //        +1MO,+2TU,-9WE
   NSArray  *days;
   unsigned i, count;
-
+  
+  /* reset mask */
   self->byDay.mask = 0;
+  self->byDay.useOccurence = 0;
+  self->byDayOccurence1 = 0;
+  
   days  = [_byDayList componentsSeparatedByString:@","];
   for (i = 0, count = [days count]; i < count; i++) {
     NSString    *iCalDay;
     iCalWeekDay day;
+    unsigned    len;
+    unichar     c0;
+    int         occurence;
+    
+    iCalDay = [days objectAtIndex:i]; // eg: MO or TU
+    if ((len = [iCalDay length]) == 0) {
+      [self errorWithFormat:@"found an empty day in byday list: '%@'", 
+             _byDayList];
+      continue;
+    }
+    
+    c0 = [iCalDay characterAtIndex:0];
+    if (((c0 == '+' || c0 == '-') && len > 2) || (isdigit(c0) && len > 1)) {
+      int offset;
+      
+      occurence = [iCalDay intValue];
+
+      offset = 1;
+      while (offset < len && isdigit([iCalDay characterAtIndex:offset]))
+       offset++;
+      
+      iCalDay = [iCalDay substringFromIndex:offset];
+      
+      if (self->byDay.useOccurence) {
+       [self errorWithFormat:
+               @"we only supported one occurence (occ=%i,day=%@): '%@'", 
+               occurence, iCalDay, _byDayList];
+       continue;
+      }
+      
+      self->byDay.useOccurence = 1;
+      self->byDayOccurence1 = occurence;
+    }
+    else if (self->byDay.useOccurence) {
+      [self errorWithFormat:
+             @"a byday occurence was specified on one day, but not on others"
+             @" (unsupported): '%@'", _byDayList];
+    }
     
-    iCalDay = [days objectAtIndex:i];
-    day     = [self weekDayFromICalRepresentation:iCalDay];
+    day = [self weekDayFromICalRepresentation:iCalDay];
     self->byDay.mask |= day;
   }
 }
@@ -354,4 +402,8 @@
   return s;
 }
 
+- (NSString *)description {
+  return [self iCalRepresentation];
+}
+
 @end /* iCalRecurrenceRule */