Works, with a remaining bug in the indexing of multi-line field values.
[mymail.git] / mymail.c
index f657306..6388365 100644 (file)
--- a/mymail.c
+++ b/mymail.c
  *
  */
 
-/*
-
-  To use it as a super-history-search for bash:
-  mymail --bash <(history)
-
-*/
-
 #define _GNU_SOURCE
 
 #include <stdio.h>
@@ -47,10 +40,24 @@ struct parsable_field {
   char *name;
   char *regexp_string;
   regex_t regexp;
-  FILE *db_file;
 };
 
-char *db_filename_prefix;
+char *db_filename;
+char *search_pattern;
+
+int action_index;
+
+char *segment_next_field(char *current) {
+  while(*current && *current != ' ') current++;
+  *current = '\0'; current++;
+  while(*current && *current == ' ') current++;
+  return current;
+}
+
+void remove_eof(char *c) {
+  while(*c && *c != '\n' && *c != '\r') c++;
+  *c = '\0';
+}
 
 /********************************************************************/
 
@@ -76,11 +83,77 @@ void usage(FILE *out) {
   fprintf(out, "\n");
 }
 
+/*********************************************************************/
+
+void search_in_db(const char *search_name, const char *search_regexp_string,
+                  FILE *db_file) {
+  char raw_line[BUFFER_SIZE];
+  char current_mail_filename[BUFFER_SIZE];
+  unsigned int current_position_in_mail;
+  char *name, *value;
+  regex_t regexp;
+  int already_written;
+
+  if(regcomp(&regexp,
+             search_regexp_string,
+             REG_ICASE)) {
+    fprintf(stderr,
+            "mymail: Syntax error in regexp \"%s\" for field \"%s\".\n",
+            search_regexp_string,
+            search_name);
+    exit(EXIT_FAILURE);
+  }
+
+  current_position_in_mail = 0;
+  already_written = 0;
+
+  while(fgets(raw_line, BUFFER_SIZE, db_file)) {
+    name = raw_line;
+    value = segment_next_field(raw_line);
+
+    /* printf("LINE [%s] %s", name, value); */
+
+    if(strcmp("mail", name) == 0) {
+      char *position_in_file_string = value;
+      char *mail_filename = segment_next_field(value);
+      current_position_in_mail = atoi(position_in_file_string);
+      strcpy(current_mail_filename, mail_filename);
+      remove_eof(current_mail_filename);
+      /* printf("READING [%s]\n", current_mail_filename); */
+      already_written = 0;
+    } else if(!already_written) {
+      if(strcmp(search_name, name) == 0 && regexec(&regexp, value, 0, 0, 0) == 0) {
+        FILE *mail_file;
+        /* printf("%s:%u\n", current_mail_filename, current_position_in_mail); */
+        mail_file = fopen(current_mail_filename, "r");
+        if(!mail_file) {
+          fprintf(stderr, "mymail: Can not open `%s'.\n", current_mail_filename);
+          exit(EXIT_FAILURE);
+        }
+        fseek(mail_file, current_position_in_mail, SEEK_SET);
+        if(fgets(raw_line, BUFFER_SIZE, mail_file)) {
+          printf("%s", raw_line);
+          while(fgets(raw_line, BUFFER_SIZE, mail_file) &&
+                strncmp(raw_line, "From ", 5)) {
+            printf("%s", raw_line);
+          }
+        }
+        fclose(mail_file);
+        already_written = 1;
+      }
+    }
+  }
+}
+
+
+/*********************************************************************/
+
 void read_file(const char *input_filename,
-               int nb_fields_to_parse, struct parsable_field *fields_to_parse) {
+               int nb_fields_to_parse, struct parsable_field *fields_to_parse,
+               FILE *db_file) {
   char raw_line[BUFFER_SIZE];
   FILE *file;
-  int in_header;
+  int in_header, new_header;
   unsigned int position_in_file;
 
   file = fopen(input_filename, "r");
@@ -91,6 +164,7 @@ void read_file(const char *input_filename,
   }
 
   in_header = 0;
+  new_header = 0;
 
   position_in_file = 0;
 
@@ -104,23 +178,28 @@ void read_file(const char *input_filename,
         exit(EXIT_FAILURE);
       }
       in_header = 1;
+      new_header = 1;
     } else if(strncmp(raw_line, "\n", 1) == 0) {
       if(in_header) { in_header = 0; }
     }
 
     /* if(in_header) { */
-      /* printf("LINE.H %s", raw_line); */
+    /* printf("LINE.H %s", raw_line); */
     /* } else { */
-      /* printf("LINE.B %s", raw_line); */
+    /* printf("LINE.B %s", raw_line); */
     /* } */
 
     if(in_header) {
       int f;
       regmatch_t matches;
+      if(new_header) {
+        fprintf(db_file, "mail %u %s\n", position_in_file, input_filename);
+        new_header = 0;
+      }
       for(f = 0; f < nb_fields_to_parse; f++) {
         if(regexec(&fields_to_parse[f].regexp, raw_line, 1, &matches, 0) == 0) {
-          fprintf(fields_to_parse[f].db_file, "%s:%d %s",
-                  input_filename, position_in_file,
+          fprintf(db_file, "%s %s",
+                  fields_to_parse[f].name,
                   raw_line + matches.rm_eo);
         }
       }
@@ -140,7 +219,8 @@ int ignore_entry(const char *name) {
 }
 
 void process_entry(const char *dir_name,
-                   int nb_fields_to_parse, struct parsable_field *fields_to_parse) {
+                   int nb_fields_to_parse, struct parsable_field *fields_to_parse,
+                   FILE *db_file) {
   DIR *dir;
   struct dirent *dir_e;
   struct stat sb;
@@ -166,14 +246,14 @@ void process_entry(const char *dir_name,
     while((dir_e = readdir(dir))) {
       if(!ignore_entry(dir_e->d_name)) {
         snprintf(subname, PATH_MAX, "%s/%s", dir_name, dir_e->d_name);
-        process_entry(subname, nb_fields_to_parse, fields_to_parse);
+        process_entry(subname, nb_fields_to_parse, fields_to_parse, db_file);
       }
     }
     closedir(dir);
   } else {
     if(S_ISREG(sb.st_mode)) {
       /* printf("Processing regular file '%s'.\n", dir_name); */
-      read_file(dir_name, nb_fields_to_parse, fields_to_parse);
+      read_file(dir_name, nb_fields_to_parse, fields_to_parse, db_file);
     }
   }
 }
@@ -190,20 +270,22 @@ enum
 static struct option long_options[] = {
   { "help", no_argument, 0, 'h' },
   { "db-prefix", 1, 0, 'p' },
+  { "search-pattern", 1, 0, 's' },
+  { "index", 0, 0, 'i' },
   { 0, 0, 0, 0 }
 };
 
 static struct parsable_field fields_to_parse[] = {
   {
     "from",
-    "^[Ff][Rr][Oo][Mm]: *",
-    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }, 0
+    "^\\([Ff][Rr][Oo][Mm]:\\|From\\) *",
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
   },
 
   {
     "dest",
     "^\\([Tt][Oo]\\|[Cc][Cc]\\|[Bb][Cc][Cc]\\): *",
-    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }, 0
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }
   },
 };
 
@@ -213,9 +295,12 @@ int main(int argc, char **argv) {
   char c;
   int f;
 
+  action_index = 0;
+  search_pattern = 0;
+
   setlocale(LC_ALL, "");
 
-  while ((c = getopt_long(argc, argv, "hp:",
+  while ((c = getopt_long(argc, argv, "hip:s:",
                           long_options, NULL)) != -1) {
 
     switch(c) {
@@ -224,8 +309,21 @@ int main(int argc, char **argv) {
       show_help = 1;
       break;
 
+    case 'i':
+      action_index = 1;
+      break;
+
     case 'p':
-      db_filename_prefix = strdup(optarg);
+      db_filename = strdup(optarg);
+      break;
+
+    case 's':
+      if(search_pattern) {
+        fprintf(stderr,
+                "mymail: Search pattern already defined.\n");
+        exit(EXIT_FAILURE);
+      }
+      search_pattern = strdup(optarg);
       break;
 
     default:
@@ -234,8 +332,8 @@ int main(int argc, char **argv) {
     }
   }
 
-  if(!db_filename_prefix) {
-    db_filename_prefix = strdup("/tmp/mymail_");
+  if(!db_filename) {
+    db_filename = strdup("/tmp/mymail");
   }
 
   if(error) {
@@ -248,11 +346,9 @@ int main(int argc, char **argv) {
     exit(EXIT_SUCCESS);
   }
 
-  for(f = 0; f < nb_fields_to_parse; f++) {
-    char db_filename[BUFFER_SIZE];
-    sprintf(db_filename, "%s%s", db_filename_prefix, fields_to_parse[f].name);
-    fields_to_parse[f].db_file = fopen(db_filename, "w");
-    if(!fields_to_parse[f].db_file) {
+  if(action_index) {
+    FILE *db_file = fopen(db_filename, "w");
+    if(!db_file) {
       fprintf(stderr,
               "mymail: Can not open \"%s\" for writing: %s\n",
               db_filename,
@@ -260,28 +356,64 @@ int main(int argc, char **argv) {
       exit(EXIT_FAILURE);
     }
 
-    printf("Initialized %s.\n", db_filename);
+    for(f = 0; f < nb_fields_to_parse; f++) {
+      if(regcomp(&fields_to_parse[f].regexp,
+                 fields_to_parse[f].regexp_string,
+                 REG_ICASE)) {
+        fprintf(stderr,
+                "mymail: Syntax error in regexp \"%s\" for field \"%s\".\n",
+                fields_to_parse[f].regexp_string,
+                fields_to_parse[f].name);
+        exit(EXIT_FAILURE);
+      }
+    }
 
-    if(regcomp(&fields_to_parse[f].regexp,
-               fields_to_parse[f].regexp_string,
-               REG_ICASE)) {
-      fprintf(stderr,
-              "mymail: Syntax error in regexp \"%s\" for field \"%s\".\n",
-              fields_to_parse[f].regexp_string,
-              fields_to_parse[f].name);
-      exit(EXIT_FAILURE);
+    while(optind < argc) {
+      process_entry(argv[optind],
+                    nb_fields_to_parse, fields_to_parse, db_file);
+      optind++;
     }
-  }
 
-  while(optind < argc) {
-    process_entry(argv[optind],
-                  nb_fields_to_parse, fields_to_parse);
-    optind++;
+    fclose(db_file);
+
+    for(f = 0; f < nb_fields_to_parse; f++) {
+      regfree(&fields_to_parse[f].regexp);
+    }
   }
 
-  for(f = 0; f < nb_fields_to_parse; f++) {
-    fclose(fields_to_parse[f].db_file);
-    regfree(&fields_to_parse[f].regexp);
+  else {
+    if(search_pattern) {
+      FILE *db_file;
+      char *search_name;
+      char *search_regexp_string;
+      search_name = search_pattern;
+      search_regexp_string = segment_next_field(search_pattern);
+      if(!*search_regexp_string) {
+        fprintf(stderr,
+                "Syntax error in the search pattern.\n");
+        exit(EXIT_FAILURE);
+      }
+
+      /* printf("Starting search in %s for field \"%s\" matching \"%s\".\n", */
+             /* db_filename, */
+             /* search_name, */
+             /* search_regexp_string); */
+
+      db_file = fopen(db_filename, "r");
+
+      if(!db_file) {
+        fprintf(stderr,
+                "mymail: Can not open \"%s\" for reading: %s\n",
+                db_filename,
+                strerror(errno));
+        exit(EXIT_FAILURE);
+      }
+
+      search_in_db(search_name, search_regexp_string, db_file);
+
+      fclose(db_file);
+      free(search_pattern);
+    }
   }
 
   exit(EXIT_SUCCESS);