]> Pileus Git - lackey/blobdiff - cals/ical.c
Fix date and ical test cases
[lackey] / cals / ical.c
index aafbc99d425521982f63ef6b20e6eea828e16e3a..bd41b721c899715ff29853eff22e8de320cc8d37 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2012 Andy Spencer <andy753421@gmail.com>
+ * Copyright (C) 2012-2013 Andy Spencer <andy753421@gmail.com>
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -29,6 +29,7 @@
 
 /* Local types */
 typedef struct {
+       cal_t *cal;
        icalcomponent *comp;
        struct icaltimetype start;
        struct icaltimetype end;
@@ -79,7 +80,8 @@ static icaltimetype to_itime(date_t time)
        };
 }
 
-static void add_recur(icalarray *array, icalcomponent *comp,
+static void add_recur(cal_t *cal,
+               icalarray *array, icalcomponent *comp,
                icaltimetype start, icaltimetype end,
                icalcomponent_kind which)
 {
@@ -99,45 +101,54 @@ static void add_recur(icalarray *array, icalcomponent *comp,
                cend   = icalcomponent_get_dtend(comp);
                length = icaltime_subtract(cend, cstart);
 
-               rrule  = icalcomponent_get_first_property(comp, ICAL_RRULE_PROPERTY);
-               recur  = icalproperty_get_rrule(rrule);
-               iter   = icalrecur_iterator_new(recur, cstart);
-
-               /* Add recurrences */
+               /* Full day event */
                if (icaltime_is_null_time(cstart) ||
                    which == ICAL_VTODO_COMPONENT) {
                        icalarray_append(array, &(ical_inst){
+                               .cal   = cal,
                                .comp  = comp,
                                .start = cstart,
                                .end   = cend,
                        });
-               } else while (1) {
-                       istart = iend = icalrecur_iterator_next(iter);
-                       if (icaltime_is_null_time(istart))
-                               break;    // no more instances
-                       if (!icaltime_is_null_time(cend))
-                               iend = icaltime_add(iend, length);
-
-                       if (icaltime_compare(iend, start) <= 0)
-                               continue; // instance ends before start time
-                       if (icaltime_compare(istart, end) >= 0)
-                               break;    // instance begins after stop time
-
-                       icalarray_append(array, &(ical_inst){
-                               .comp  = comp,
-                               .start = istart,
-                               .end   = iend,
-                       });
                }
 
-               icalrecur_iterator_free(iter);
+               /* Add all recurrences */
+               rrule = icalcomponent_get_first_property(comp, ICAL_RRULE_PROPERTY);
+               while (rrule) {
+                       recur = icalproperty_get_rrule(rrule);
+                       iter  = icalrecur_iterator_new(recur, cstart);
+
+                       /* Add recurrence for this rrule */
+                       while (1) {
+                               istart = iend = icalrecur_iterator_next(iter);
+                               if (icaltime_is_null_time(istart))
+                                       break;    // no more instances
+                               if (!icaltime_is_null_time(cend))
+                                       iend = icaltime_add(iend, length);
+
+                               if (icaltime_compare(iend, start) <= 0)
+                                       continue; // instance ends before start time
+                               if (icaltime_compare(istart, end) >= 0)
+                                       break;    // instance begins after stop time
+
+                               icalarray_append(array, &(ical_inst){
+                                       .cal   = cal,
+                                       .comp  = comp,
+                                       .start = istart,
+                                       .end   = iend,
+                               });
+                       }
+
+                       icalrecur_iterator_free(iter);
+                       rrule = icalcomponent_get_next_property(comp, ICAL_RRULE_PROPERTY);
+               }
        }
 
        /* Add children */
        icalcomponent_kind find = ICAL_ANY_COMPONENT;
        icalcomponent *child = icalcomponent_get_first_component(comp, find);
        while (child) {
-               add_recur(array, child, start, end, which);
+               add_recur(cal, array, child, start, end, which);
                child = icalcomponent_get_next_component(comp, find);
        }
 }
@@ -148,17 +159,16 @@ static void read_icals(void)
                if (cal->comp == NULL && cal->location) {
                        wordexp_t wexp;
                        wordexp(cal->location, &wexp, WRDE_NOCMD);
-                       if (wexp.we_wordc == 0)
-                               continue;
-                       FILE *file = fopen(wexp.we_wordv[0], "r");
-                       wordfree(&wexp);
-                       if (!file)
-                               continue;
-
                        icalparser *parser = icalparser_new();
-                       icalparser_set_gen_data(parser, file);
+                       for (int i = 0; i < wexp.we_wordc; i++) {
+                               FILE *file = fopen(wexp.we_wordv[i], "r");
+                               if (!file)
+                                       continue;
+                               icalparser_set_gen_data(parser, file);
+                       }
                        cal->comp = icalparser_parse(parser, (void*)fgets);
                        icalparser_free(parser);
+                       wordfree(&wexp);
                }
        }
 }
@@ -175,6 +185,7 @@ static event_t *to_event(ical_inst *inst)
        event->cat   = icalproperty_get_value_as_string_r(prop);
        event->start = to_date(inst->start);
        event->end   = to_date(inst->end);
+       event->cal   = inst->cal;
        return event;
 }
 
@@ -213,6 +224,7 @@ static todo_t *to_todo(ical_inst *inst)
                       perc ? icalproperty_get_percentcomplete(perc) : 0;
        todo->start  = to_date(inst->start);
        todo->due    = to_date(icalcomponent_get_due(inst->comp));
+       todo->cal    = inst->cal;
        return todo;
 }
 
@@ -292,7 +304,7 @@ event_t *ical_events(date_t _start, date_t _end)
        icaltimetype end   = to_itime(_end);
        icalarray *array = icalarray_new(sizeof(ical_inst), 1);
        for (ical_t *cal = calendars; cal; cal = cal->next)
-               add_recur(array, cal->comp, start, end, ICAL_VEVENT_COMPONENT);
+               add_recur(&cal->cal, array, cal->comp, start, end, ICAL_VEVENT_COMPONENT);
        icalarray_sort(array, ical_compare);
        event_t *events = to_events(array);
        icalarray_free(array);
@@ -309,7 +321,7 @@ todo_t *ical_todos(date_t _start, date_t _end)
        icaltimetype end   = to_itime(_end);
        icalarray *array = icalarray_new(sizeof(ical_inst), 1);
        for (ical_t *cal = calendars; cal; cal = cal->next)
-               add_recur(array, cal->comp, start, end, ICAL_VTODO_COMPONENT);
+               add_recur(&cal->cal, array, cal->comp, start, end, ICAL_VTODO_COMPONENT);
        icalarray_sort(array, ical_compare);
        todo_t *todos = to_todos(array);
        icalarray_free(array);
@@ -338,10 +350,10 @@ void ical_printr(icalcomponent *comp, int depth)
        }
 }
 
-void ical_test(void)
+void ical_test(char *path)
 {
        /* Load ical */
-       FILE *file = fopen("data/all.ics", "r");
+       FILE *file = fopen(path, "r");
        icalparser *parser = icalparser_new();
        icalparser_set_gen_data(parser, file);
        icalcomponent *comp = icalparser_parse(parser, (void*)fgets);
@@ -353,21 +365,21 @@ void ical_test(void)
 
        /* Find events */
        array = icalarray_new(sizeof(ical_inst), 1);
-       add_recur(array, comp, start, end, ICAL_VEVENT_COMPONENT);
+       add_recur(NULL, array, comp, start, end, ICAL_VEVENT_COMPONENT);
        icalarray_sort(array, ical_compare);
        event_t *events = to_events(array);
        icalarray_free(array);
 
        /* Find Todos */
        array = icalarray_new(sizeof(ical_inst), 1);
-       add_recur(array, comp, start, end, ICAL_VTODO_COMPONENT);
+       add_recur(NULL, array, comp, start, end, ICAL_VTODO_COMPONENT);
        icalarray_sort(array, ical_compare);
        todo_t *todos = to_todos(array);
        icalarray_free(array);
 
        /* Print */
-       //ical_printr(comp, 0);
-       //print_events(events);
+       ical_printr(comp, 0);
+       print_events(events);
        print_todos(todos);
 
        (void)print_events;