Now uses the environment variable MYMAIL_DB_FILE.

[mymail.git] / mymail.c

diff --git a/mymail.c b/mymail.c
index c21be1c..001d14d 100644 (file)
--- a/mymail.c
+++ b/mymail.c
@@ -46,7 +46,7 @@
 
 #define VERSION "0.1"
 
-#define BUFFER_SIZE 16384
+#define BUFFER_SIZE 65536
 
 struct parsable_field {
   char *name;
@@ -138,7 +138,7 @@ void search_in_db(const char *search_name, const char *search_regexp_string,
         FILE *mail_file;
         mail_file = fopen(current_mail_filename, "r");
         if(!mail_file) {
-          fprintf(stderr, "mymail: Can not open `%s'.\n", current_mail_filename);
+          fprintf(stderr, "mymail: Can not open '%s'.\n", current_mail_filename);
           exit(EXIT_FAILURE);
         }
         fseek(mail_file, current_position_in_mail, SEEK_SET);
@@ -160,10 +160,24 @@ void search_in_db(const char *search_name, const char *search_regexp_string,
 
 /*********************************************************************/
 
+void index_one_mbox_line(int nb_fields_to_parse, struct parsable_field *fields_to_parse,
+                         char *raw_line, FILE *db_file) {
+  regmatch_t matches;
+  int f;
+  for(f = 0; f < nb_fields_to_parse; f++) {
+    if(regexec(&fields_to_parse[f].regexp, raw_line, 1, &matches, 0) == 0) {
+      fprintf(db_file, "%s %s\n",
+              fields_to_parse[f].name,
+              raw_line + matches.rm_eo);
+    }
+  }
+}
+
 void index_mbox(const char *input_filename,
                 int nb_fields_to_parse, struct parsable_field *fields_to_parse,
                 FILE *db_file) {
-  char raw_line[BUFFER_SIZE];
+  char raw_line[BUFFER_SIZE], full_line[BUFFER_SIZE];
+  char *end_of_full_line;
   FILE *file;
   int in_header, new_header;
   unsigned long int position_in_file;
@@ -171,7 +185,7 @@ void index_mbox(const char *input_filename,
   file = fopen(input_filename, "r");
 
   if(!file) {
-    fprintf(stderr, "mymail: Can not open `%s'.\n", input_filename);
+    fprintf(stderr, "mymail: Can not open '%s'.\n", input_filename);
     if(paranoid) { exit(EXIT_FAILURE); }
     return;
   }
@@ -180,6 +194,7 @@ void index_mbox(const char *input_filename,
   new_header = 0;
 
   position_in_file = 0;
+  end_of_full_line = 0;
 
   while(fgets(raw_line, BUFFER_SIZE, file)) {
     if(strncmp(raw_line, "From ", 5) == 0) {
@@ -196,20 +211,46 @@ void index_mbox(const char *input_filename,
       if(in_header) { in_header = 0; }
     }
 
+    /* printf("PARSE %d %s", in_header, raw_line); */
+
     if(in_header) {
-      int f;
-      regmatch_t matches;
       if(new_header) {
         fprintf(db_file, "mail %lu %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(db_file, "%s %s",
-                  fields_to_parse[f].name,
-                  raw_line + matches.rm_eo);
+
+      if(raw_line[0] == ' ' || raw_line[0] == '\t') {
+        char *start = raw_line;
+        while(*start == ' ' || *start == '\t') start++;
+        *(end_of_full_line++) = ' ';
+        strcpy(end_of_full_line, start);
+        while(*end_of_full_line && *end_of_full_line != '\n') {
+          end_of_full_line++;
         }
+        *end_of_full_line = '\0';
       }
+
+      else {
+        /* if(!((raw_line[0] >= 'a' && raw_line[0] <= 'z') || */
+             /* (raw_line[0] >= 'A' && raw_line[0] <= 'Z'))) { */
+          /* fprintf(stderr, */
+                  /* "Header line syntax error %s:%lu.\n", */
+                  /* input_filename, position_in_file); */
+          /* fprintf(stderr, "%s", raw_line); */
+        /* } */
+
+        if(full_line[0]) {
+          index_one_mbox_line(nb_fields_to_parse, fields_to_parse, full_line, db_file);
+        }
+
+        end_of_full_line = full_line;
+        strcpy(end_of_full_line, raw_line);
+        while(*end_of_full_line && *end_of_full_line != '\n') {
+          end_of_full_line++;
+        }
+        *end_of_full_line = '\0';
+      }
+
     }
 
     position_in_file += strlen(raw_line);
@@ -267,7 +308,7 @@ enum {
 
 static struct option long_options[] = {
   { "help", no_argument, 0, 'h' },
-  { "db-prefix", 1, 0, 'p' },
+  { "db-file", 1, 0, 'p' },
   { "search-pattern", 1, 0, 's' },
   { "index", 0, 0, 'i' },
   { 0, 0, 0, 0 }
@@ -314,6 +355,7 @@ int main(int argc, char **argv) {
 
     case 'p':
       db_filename = strdup(optarg);
+      printf("db_filename=\"%s\"\n", db_filename);
       break;
 
     case 's':
@@ -331,7 +373,9 @@ int main(int argc, char **argv) {
   }
 
   if(!db_filename) {
-    db_filename = strdup("/tmp/mymail");
+    char *default_db_filename = getenv("MYMAIL_DB_FILE");
+    if(!default_db_filename) { default_db_filename = "/tmp/mymail.db"; }
+    db_filename = strdup(default_db_filename);
   }
 
   if(error) {