Debugged and cleaned up the time-related queries.
[mymail.git] / mymail.c
index fd200a1..ef6e5c1 100644 (file)
--- a/mymail.c
+++ b/mymail.c
@@ -61,7 +61,6 @@ regex_t leading_from_line_regexp;
 
 int paranoid;
 int quiet;
-char *default_search_field;
 int ignore_dot_files;
 
 /********************************************************************/
@@ -153,8 +152,8 @@ int xor(int a, int b) {
   return (a && !b) || (!a && b);
 }
 
-char *parse_token(char *token_buffer, size_t token_buffer_size,
-                  char separator, char *string) {
+const char *parse_token(char *token_buffer, size_t token_buffer_size,
+                        char separator, const char *string) {
   char *u = token_buffer;
   while(u < token_buffer + token_buffer_size - 1 && *string &&
         *string != separator) {
@@ -184,6 +183,7 @@ char *default_value(char *current_value,
 
 FILE *safe_fopen(const char *path, const char *mode, const char *comment) {
   FILE *result = fopen(path, mode);
+  /* printf("Opening '%s' with mode \"%s\"\n", path, mode); */
   if(result) {
     return result;
   } else {
@@ -234,18 +234,6 @@ void print_usage(FILE *out) {
 
 /*********************************************************************/
 
-time_t time_for_past_day(int day) {
-  time_t t;
-  struct tm *tm;
-  int delta_day;
-  t = time(0);
-  tm = localtime(&t);
-  delta_day = (7 + tm->tm_wday - day) % 7 + 1;
-  return t - delta_day * 3600 * 24 + tm->tm_sec + 60 * tm->tm_min + 3600 * tm->tm_hour;
-}
-
-/*********************************************************************/
-
 int ignore_entry(const char *name) {
   return
     strcmp(name, ".") == 0 ||
@@ -255,27 +243,16 @@ int ignore_entry(const char *name) {
 
 int is_a_leading_from_line(char *mbox_line) {
   return
-
-    /*
-
-      The mbox man page in qmail documentation states:
-
-       > The reader should not attempt to take advantage of the fact
-       > that every From_ line (past the beginning of the file) is
-       > preceded by a blank line.
-
-    */
-
     strncmp(mbox_line, "From ", 5) == 0 &&
     regexec(&leading_from_line_regexp, mbox_line, 0, 0, 0) == 0;
 }
 
 int mbox_line_match_search(struct search_condition *condition,
-                           int mbox_id, char *mbox_value) {
+                           int mbox_id, const char *mbox_value) {
 
   if(condition->field_id == ID_INTERVAL) {
     if(mbox_id == ID_LEADING_LINE) {
-      char *c;
+      const char *c;
       time_t t;
       struct tm tm;
 
@@ -388,7 +365,8 @@ int search_in_db(const char *db_filename,
   char raw_db_line[BUFFER_SIZE];
   char current_mail_filename[PATH_MAX + 1];
   unsigned long int current_position_in_mail;
-  char mbox_name[TOKEN_BUFFER_SIZE], *mbox_value;
+  char mbox_name[TOKEN_BUFFER_SIZE];
+  const char *mbox_value;
   int mbox_id;
   int already_written, m, n;
   int nb_body_conditions, nb_fulfilled_body_conditions;
@@ -527,6 +505,8 @@ int recursive_search_in_db(const char *entry_name, regex_t *db_filename_regexp,
     exit(EXIT_FAILURE);
   }
 
+  /* printf("recursive_search_in_db %s\n", entry_name); */
+
   dir = opendir(entry_name);
 
   if(dir) {
@@ -556,10 +536,11 @@ int recursive_search_in_db(const char *entry_name, regex_t *db_filename_regexp,
 
 /*********************************************************************/
 
-void index_one_mbox_line(int nb_fields_to_parse, struct parsable_field *fields_to_parse,
+void index_one_mbox_line(unsigned int nb_fields_to_parse,
+                         struct parsable_field *fields_to_parse,
                          char *raw_mbox_line, FILE *db_file) {
   regmatch_t matches;
-  int f;
+  unsigned int f;
   for(f = 0; f < nb_fields_to_parse; f++) {
     if(regexec(&fields_to_parse[f].regexp, raw_mbox_line, 1, &matches, 0) == 0) {
       fprintf(db_file, "%s %s\n",
@@ -710,12 +691,53 @@ static struct option long_options[] = {
   { 0, 0, 0, 0 }
 };
 
+struct time_criterion {
+  char *label;
+  int start_hour, end_hour;
+  int past_week_day;
+};
+
 /*********************************************************************/
 
-void init_condition(struct search_condition *condition, char *full_string) {
+static struct time_criterion time_criteria[] = {
+
+  { "8h",         8,       -1, -1 },
+  { "today",     24,       -1, -1 },
+  { "24h",       24,       -1, -1 },
+  { "week",      24 *   7, -1, -1 },
+  { "month",     24 *  31, -1, -1 },
+  { "year",      24 * 365, -1, -1 },
+
+  { "yesterday", 48,       24, -1 },
+
+  { "monday",    -1,       -1,  1 },
+  { "tuesday",   -1,       -1,  2 },
+  { "wednesday", -1,       -1,  3 },
+  { "thursday",  -1,       -1,  4 },
+  { "friday",    -1,       -1,  5 },
+  { "saturday",  -1,       -1,  6 },
+  { "sunday",    -1,       -1,  7 },
+
+};
+
+/*********************************************************************/
+
+time_t time_for_past_day(int day) {
+  time_t t;
+  struct tm *tm;
+  int delta_day;
+  t = time(0);
+  tm = localtime(&t);
+  delta_day = (7 + tm->tm_wday - day) % 7;
+  if(delta_day == 0) { delta_day = 7; }
+  return t - (delta_day * 3600 * 24 + tm->tm_sec + 60 * tm->tm_min + 3600 * tm->tm_hour);
+}
+
+void init_condition(struct search_condition *condition, const char *full_string,
+                    const char *default_search_field) {
   char full_search_field[TOKEN_BUFFER_SIZE], *search_field;
-  int m;
-  char *string;
+  unsigned int k, m;
+  const char *string;
 
   string = parse_token(full_search_field, TOKEN_BUFFER_SIZE, ' ', full_string);
   search_field = full_search_field;
@@ -727,96 +749,31 @@ void init_condition(struct search_condition *condition, char *full_string) {
     condition->negation = 0;
   }
 
-  /* Recently */
+  condition->field_id = -1;
 
-  if(strcmp(search_field, "8h") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 3600 * 8;
-    condition->interval_stop = 0;
-  }
-
-  else if(strcmp(search_field, "24h") == 0 ||
-          strcmp(search_field, "today") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 3600 * 24;
-    condition->interval_stop = 0;
-  }
-
-  else if(strcmp(search_field, "week") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 3600 * 24 * 7;
-    condition->interval_stop = 0;
-  }
-
-  else if(strcmp(search_field, "month") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 3600 * 24 * 31;
-    condition->interval_stop = 0;
-  }
-
-  else if(strcmp(search_field, "year") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 3600 * 24 * 365;
-    condition->interval_stop = 0;
-  }
-
-  /* Yesterday */
-
-  else if(strcmp(search_field, "yesterday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time(0) - 2 * 3600 * 24;
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  /* Week days */
-
-  else if(strcmp(search_field, "monday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(1);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  else if(strcmp(search_field, "tuesday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(2);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  else if(strcmp(search_field, "wednesday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(3);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  else if(strcmp(search_field, "thursday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(4);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  else if(strcmp(search_field, "friday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(5);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
-
-  else if(strcmp(search_field, "saturday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(6);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
-  }
+  /* Recently */
 
-  else if(strcmp(search_field, "sunday") == 0) {
-    condition->field_id = ID_INTERVAL;
-    condition->interval_start = time_for_past_day(7);
-    condition->interval_stop = condition->interval_start + 3600 * 24;
+  for(k = 0; k < sizeof(time_criteria) / sizeof(struct time_criterion); k++) {
+    if(strcmp(time_criteria[k].label, search_field) == 0) {
+      condition->field_id = ID_INTERVAL;
+      if(time_criteria[k].past_week_day < 0) {
+        condition->interval_start = time(0) - 3600 * time_criteria[k].start_hour;
+        if(time_criteria[k].end_hour >= 0) {
+          condition->interval_stop = time(0) - 3600 * time_criteria[k].end_hour;
+        } else {
+          condition->interval_stop = 0;
+        }
+      } else {
+        condition->interval_start = time_for_past_day(time_criteria[k].past_week_day);
+        condition->interval_stop = condition->interval_start + 3600 * 24;
+      }
+      break;
+    }
   }
 
-  else {
-
-    /* header-related conditions */
+  if(condition->field_id == -1) {
 
-    condition->field_id = -1;
+    /* No time condition matched, look for the search fields */
 
     for(m = 0; (m < MAX_ID) && condition->field_id == -1; m++) {
       if(strncmp(field_names[m], search_field, strlen(search_field)) == 0) {
@@ -824,6 +781,8 @@ void init_condition(struct search_condition *condition, char *full_string) {
       }
     }
 
+    /* None match, if there is a default search field, re-run the search with it */
+
     if(condition->field_id == -1) {
       if(default_search_field) {
         for(m = 0; (m < MAX_ID) && condition->field_id == -1; m++) {
@@ -866,18 +825,20 @@ void free_condition(struct search_condition *condition) {
 /*********************************************************************/
 
 int main(int argc, char **argv) {
-  char *db_filename;
-  char *db_filename_regexp_string;
-  char *db_root_path;
-  char *db_filename_list;
-  char *mbox_filename_regexp_string;
+  char *db_filename = 0;
+  char *db_filename_regexp_string = 0;
+  char *db_root_path = 0;
+  char *db_filename_list = 0;
+  char *mbox_filename_regexp_string = 0;
+  char *default_search_field;
   char output_filename[PATH_MAX + 1];
-  int action_index;
+  int action_index = 0;
   int error = 0, show_help = 0;
-  const int nb_fields_to_parse = sizeof(fields_to_parse) / sizeof(struct parsable_field);
+  const unsigned int nb_fields_to_parse =
+    sizeof(fields_to_parse) / sizeof(struct parsable_field);
   char c;
-  int f, n;
-  int nb_search_conditions;
+  unsigned int f, n;
+  unsigned int nb_search_conditions;
   struct search_condition search_conditions[MAX_NB_SEARCH_CONDITIONS];
 
   if(regcomp(&leading_from_line_regexp, LEADING_FROM_LINE_REGEXP_STRING, 0)) {
@@ -887,21 +848,16 @@ int main(int argc, char **argv) {
   }
 
   paranoid = 0;
-  action_index = 0;
-  db_filename = 0;
-  db_filename_regexp_string = 0;
-  db_root_path = 0;
-  db_filename_list = 0;
-  mbox_filename_regexp_string = 0;
   quiet = 0;
   default_search_field = 0;
   ignore_dot_files = 1;
+  strncpy(output_filename, "", PATH_MAX);
 
   setlocale(LC_ALL, "");
 
   nb_search_conditions = 0;
 
-  while ((c = getopt_long(argc, argv, "hvqip:s:d:r:l:o:a:",
+  while ((c = getopt_long(argc, argv, "hvqip:s:d:r:l:o:a:m:",
                           long_options, NULL)) != -1) {
 
     switch(c) {
@@ -971,7 +927,7 @@ int main(int argc, char **argv) {
         fprintf(stderr, "mymail: Too many search patterns.\n");
         exit(EXIT_FAILURE);
       }
-      init_condition(&search_conditions[nb_search_conditions], optarg);
+      init_condition(&search_conditions[nb_search_conditions], optarg, default_search_field);
       nb_search_conditions++;
       break;
 
@@ -994,7 +950,7 @@ int main(int argc, char **argv) {
 
   db_filename_regexp_string = default_value(db_filename_regexp_string,
                                             "MYMAIL_DB_FILE",
-                                            "^mymail.db$");
+                                            "\\.db$");
 
   db_root_path = default_value(db_root_path,
                                "MYMAIL_DB_ROOT",
@@ -1006,7 +962,7 @@ int main(int argc, char **argv) {
 
   mbox_filename_regexp_string = default_value(mbox_filename_regexp_string,
                                               "MYMAIL_MBOX_PATTERN",
-                                              "mbox$");
+                                              0);
 
   /* Start the processing */
 
@@ -1116,7 +1072,7 @@ int main(int argc, char **argv) {
 
       if(db_filename_list) {
         char db_filename[PATH_MAX + 1];
-        char *s;
+        const char *s;
 
         s = db_filename_list;