Cosmetics + now check that each ^"From " line follows an empty line.
[mymail.git] / mymail.c
index dcb44bf..eb329b1 100644 (file)
--- a/mymail.c
+++ b/mymail.c
 
 #define BUFFER_SIZE 65536
 
+char *db_filename;
+char *db_root_path;
+
+int paranoid;
+int action_index;
+
+/********************************************************************/
+
 enum {
-  ID_MAIL,
+  ID_MAIL = 0,
   ID_FROM,
   ID_DEST,
   ID_SUBJECT,
-  ID_FROMDEST,
+  ID_PARTICIPANT,
   MAX_ID
 };
 
@@ -64,21 +72,52 @@ static char *field_names[] = {
   "mail",
   "from",
   "dest",
-  "subj",
-  "fromdest"
+  "subject",
+  "part"
 };
 
+/********************************************************************/
+
+struct search_request {
+  int field_id;
+  int negation;
+  regex_t regexp;
+};
+
+/********************************************************************/
+
 struct parsable_field {
   int id;
   char *regexp_string;
   regex_t regexp;
 };
 
-char *db_filename;
-char *db_root_path;
+static struct parsable_field fields_to_parse[] = {
+  {
+    ID_FROM,
+    "^\\(From \\|[Ff][Rr][Oo][Mm]:\\|[R][r][E][e][P][p][L][l][Y][y]-[T][t][O][o]:\\)",
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
+  },
 
-int paranoid;
-int action_index;
+  {
+    ID_DEST,
+    "^\\([Tt][Oo]\\|[Cc][Cc]\\|[Bb][Cc][Cc]\\): ",
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
+  },
+
+  {
+    ID_SUBJECT,
+    "^[Ss][Uu][Bb][Jj][Ee][Cc][Tt]: ",
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
+  },
+
+};
+
+/********************************************************************/
+
+int xor(int a, int b) {
+  return (a && !b) || (!a && b);
+}
 
 char *segment_next_field(char *current) {
   while(*current && *current != ' ') current++;
@@ -141,13 +180,17 @@ int ignore_entry(const char *name) {
     (name[0] == '.' && name[1] != '/');
 }
 
-int mbox_line_match_search(int search_id, regex_t *search_regexp,
+int mbox_line_match_search(struct search_request *request,
                            int mbox_id, char *mbox_value) {
-  return search_id == mbox_id && regexec(search_regexp, mbox_value, 0, 0, 0) == 0;
+  return
+    (request->field_id == mbox_id ||
+     (request->field_id == ID_PARTICIPANT && (mbox_id == ID_FROM || mbox_id == ID_DEST)))
+    &&
+    regexec(&request->regexp, mbox_value, 0, 0, 0) == 0;
 }
 
 void search_in_db(int nb_search_patterns,
-                  int *search_ids, char **search_regexp_strings,
+                  struct search_request *search_requests,
                   FILE *db_file) {
   int hits[MAX_NB_SEARCH_PATTERNS];
   char raw_db_line[BUFFER_SIZE];
@@ -156,20 +199,8 @@ void search_in_db(int nb_search_patterns,
   unsigned long int current_position_in_mail;
   char *mbox_name, *mbox_value;
   int mbox_id;
-  regex_t search_regexps[MAX_NB_SEARCH_PATTERNS];
   int already_written, m, n;
-
-  for(n = 0; n < nb_search_patterns; n++) {
-    if(regcomp(&search_regexps[n],
-               search_regexp_strings[n],
-               REG_ICASE)) {
-      fprintf(stderr,
-              "mymail: Syntax error in regexp \"%s\" for field \"%s\".\n",
-              search_regexp_strings[n],
-              field_names[search_ids[n]]);
-      exit(EXIT_FAILURE);
-    }
-  }
+  int last_mbox_line_was_empty;
 
   current_position_in_mail = 0;
   already_written = 0;
@@ -184,7 +215,11 @@ void search_in_db(int nb_search_patterns,
       char *position_in_file_string;
       char *mail_filename;
 
-      for(n = 0; n < nb_search_patterns && hits[n]; n++);
+      for(n = 0; n < nb_search_patterns && xor(hits[n], search_requests[n].negation); n++);
+
+      /* for(n = 0; n < nb_search_patterns && */
+            /* ((hits[n] && !search_requests[n].negation) || */
+             /* (!hits[n] && search_requests[n].negation)); n++); */
 
       if(n == nb_search_patterns) {
         FILE *mail_file;
@@ -199,9 +234,12 @@ void search_in_db(int nb_search_patterns,
         fseek(mail_file, current_position_in_mail, SEEK_SET);
 
         if(fgets(raw_mbox_line, BUFFER_SIZE, mail_file)) {
+          last_mbox_line_was_empty = 1;
           printf("%s", raw_mbox_line);
-          while(fgets(raw_mbox_line, BUFFER_SIZE, mail_file) &&
-                strncmp(raw_mbox_line, "From ", 5)) {
+          while(1) {
+            if(!fgets(raw_mbox_line, BUFFER_SIZE, mail_file) ||
+               (last_mbox_line_was_empty && strncmp(raw_mbox_line, "From ", 5) == 0)) break;
+            last_mbox_line_was_empty = (raw_mbox_line[0] == '\n');
             printf("%s", raw_mbox_line);
           }
         }
@@ -228,20 +266,16 @@ void search_in_db(int nb_search_patterns,
         }
       }
       for(n = 0; n < nb_search_patterns; n++) {
-        hits[n] |= mbox_line_match_search(search_ids[n], &search_regexps[n],
+        hits[n] |= mbox_line_match_search(&search_requests[n],
                                           mbox_id, mbox_value);
       }
     }
   }
-
-  for(n = 0; n < nb_search_patterns; n++) {
-    regfree(&search_regexps[n]);
-  }
 }
 
 void recursive_search_in_db(const char *entry_name,
                             int nb_search_patterns,
-                            int *search_ids, char **search_regexp_strings) {
+                            struct search_request *search_requests) {
   DIR *dir;
   struct dirent *dir_e;
   struct stat sb;
@@ -264,7 +298,7 @@ void recursive_search_in_db(const char *entry_name,
         snprintf(subname, PATH_MAX, "%s/%s", entry_name, dir_e->d_name);
         recursive_search_in_db(subname,
                                nb_search_patterns,
-                               search_ids, search_regexp_strings);
+                               search_requests);
       }
     }
     closedir(dir);
@@ -297,8 +331,7 @@ void recursive_search_in_db(const char *entry_name,
         exit(EXIT_FAILURE);
       }
 
-      search_in_db(nb_search_patterns, search_ids, search_regexp_strings,
-                   db_file);
+      search_in_db(nb_search_patterns, search_requests, db_file);
 
       fclose(db_file);
     }
@@ -326,7 +359,7 @@ void index_mbox(const char *mbox_filename,
   char raw_mbox_line[BUFFER_SIZE], full_line[BUFFER_SIZE];
   char *end_of_full_line;
   FILE *file;
-  int in_header, new_header;
+  int in_header, new_header, last_mbox_line_was_empty;
   unsigned long int position_in_file;
 
   file = fopen(mbox_filename, "r");
@@ -343,9 +376,10 @@ void index_mbox(const char *mbox_filename,
   position_in_file = 0;
   end_of_full_line = 0;
   full_line[0] = '\0';
+  last_mbox_line_was_empty = 1;
 
   while(fgets(raw_mbox_line, BUFFER_SIZE, file)) {
-    if(strncmp(raw_mbox_line, "From ", 5) == 0) {
+    if(last_mbox_line_was_empty && strncmp(raw_mbox_line, "From ", 5) == 0) {
       if(in_header) {
         fprintf(stderr,
                 "Got a ^\"From \" in the header in %s:%lu.\n",
@@ -355,10 +389,12 @@ void index_mbox(const char *mbox_filename,
       }
       in_header = 1;
       new_header = 1;
-    } else if(strncmp(raw_mbox_line, "\n", 1) == 0) {
+    } else if(raw_mbox_line[0] == '\n') {
       if(in_header) { in_header = 0; }
     }
 
+    last_mbox_line_was_empty = (raw_mbox_line[0] == '\n');
+
     if(in_header) {
       if(new_header) {
         fprintf(db_file, "mail %lu %s\n", position_in_file, mbox_filename);
@@ -456,27 +492,6 @@ static struct option long_options[] = {
   { 0, 0, 0, 0 }
 };
 
-static struct parsable_field fields_to_parse[] = {
-  {
-    ID_FROM,
-    "^\\([Ff][Rr][Oo][Mm]:\\|From\\) *",
-    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
-  },
-
-  {
-    ID_DEST,
-    "^\\([Tt][Oo]\\|[Cc][Cc]\\|[Bb][Cc][Cc]\\): *",
-    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
-  },
-
-  {
-    ID_SUBJECT,
-    "^[Ss][Uu][Bb][Jj][Ee][Cc][Tt]: *",
-    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
-  },
-
-};
-
 /*********************************************************************/
 
 int main(int argc, char **argv) {
@@ -488,7 +503,7 @@ int main(int argc, char **argv) {
   char *search_pattern[MAX_NB_SEARCH_PATTERNS];
 
   /* for(f = 0; f < argc; f++) { */
-    /* printf("arg %d \"%s\"\n", f, argv[f]); */
+  /* printf("arg %d \"%s\"\n", f, argv[f]); */
   /* } */
 
   paranoid = 0;
@@ -618,28 +633,40 @@ int main(int argc, char **argv) {
   else {
 
     if(nb_search_patterns > 0) {
-      int search_ids[MAX_NB_SEARCH_PATTERNS];
-      char *search_regexp_strings[MAX_NB_SEARCH_PATTERNS];
+      struct search_request search_requests[MAX_NB_SEARCH_PATTERNS];
+      char *search_regexp_string;
       int m, n;
 
       for(n = 0; n < nb_search_patterns; n++) {
-        search_regexp_strings[n] = segment_next_field(search_pattern[n]);
-        search_ids[n] = -1;
-        for(m = 0; (m < MAX_ID) && search_ids[n] == -1; m++) {
+        search_regexp_string = segment_next_field(search_pattern[n]);
+
+        if(search_pattern[n][0] == '!') {
+          search_pattern[n]++;
+          search_requests[n].negation = 1;
+        } else {
+          search_requests[n].negation = 0;
+        }
+
+        search_requests[n].field_id = -1;
+        for(m = 0; (m < MAX_ID) && search_requests[n].field_id == -1; m++) {
           if(strncmp(field_names[m], search_pattern[n], strlen(search_pattern[n])) == 0) {
-            search_ids[n] = m;
+            search_requests[n].field_id = m;
           }
         }
-      }
 
-      if(!*search_regexp_strings) {
-        fprintf(stderr,
-                "Syntax error in the search pattern.\n");
-        exit(EXIT_FAILURE);
+        if(regcomp(&search_requests[n].regexp,
+                   search_regexp_string,
+                   REG_ICASE)) {
+          fprintf(stderr,
+                  "mymail: Syntax error in regexp \"%s\" for field \"%s\".\n",
+                  search_regexp_string,
+                  field_names[search_requests[n].field_id]);
+          exit(EXIT_FAILURE);
+        }
       }
 
       recursive_search_in_db(db_root_path,
-                             nb_search_patterns, search_ids, search_regexp_strings);
+                             nb_search_patterns, search_requests);
 
       for(n = 0; n < nb_search_patterns; n++) {
         free(search_pattern[n]);