Update.
[mymail.git] / mymail.c
index 241d635..dc7c58c 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>
 #include <stdlib.h>
-#include <unistd.h>
 #include <string.h>
 #include <errno.h>
-#include <ncurses.h>
 #include <fcntl.h>
-#include <sys/ioctl.h>
-#include <termios.h>
-#include <regex.h>
 #include <locale.h>
 #include <getopt.h>
 #include <limits.h>
+#include <dirent.h>
+#include <regex.h>
 
-#define VERSION "1.1.7"
+#define VERSION "0.1"
 
 #define BUFFER_SIZE 16384
 
+struct parsable_field {
+  char *name;
+  char *regexp_string;
+  regex_t regexp;
+  FILE *db_file;
+};
+
+char *db_filename_prefix;
+
 /********************************************************************/
 
 /* malloc with error checking.  */
@@ -63,7 +62,6 @@ void *safe_malloc(size_t n) {
 /*********************************************************************/
 
 void usage(FILE *out) {
-
   fprintf(out, "mymail version %s (%s)\n", VERSION, UNAME);
   fprintf(out, "Written by Francois Fleuret <francois@fleuret.org>.\n");
   fprintf(out, "\n");
@@ -72,12 +70,11 @@ void usage(FILE *out) {
 }
 
 void read_file(const char *input_filename,
-               int nb_lines_max, int *nb_lines) {
-
+               int nb_fields_to_parse, struct parsable_field *fields_to_parse) {
   char raw_line[BUFFER_SIZE];
-  char *s;
   FILE *file;
-  int l;
+  int in_header;
+  unsigned int position_in_file;
 
   file = fopen(input_filename, "r");
 
@@ -86,18 +83,94 @@ void read_file(const char *input_filename,
     exit(EXIT_FAILURE);
   }
 
-  while(*nb_lines < nb_lines_max && fgets(raw_line, BUFFER_SIZE, file)) {
-    l = strlen(raw_line);
-    fgets(raw_line + l, BUFFER_SIZE - l, file);
-    for(s = raw_line + strlen(raw_line) - 1; s > raw_line && *s == '\n'; s--) {
-      *s = '\0';
+  in_header = 0;
+
+  position_in_file = 0;
+
+  while(fgets(raw_line, BUFFER_SIZE, file)) {
+    if(strncmp(raw_line, "From ", 5) == 0) {
+      if(in_header) {
+        fprintf(stderr,
+                "Got a 'From ' in the header in %s:%u.\n",
+                input_filename, position_in_file);
+        fprintf(stderr, "%s", raw_line);
+        exit(EXIT_FAILURE);
+      }
+      in_header = 1;
+    } else if(strncmp(raw_line, "\n", 1) == 0) {
+      if(in_header) { in_header = 0; }
     }
-    /* store_line(hash_table, raw_line, nb_lines, lines); */
+
+    /* if(in_header) { */
+      /* printf("LINE.H %s", raw_line); */
+    /* } else { */
+      /* printf("LINE.B %s", raw_line); */
+    /* } */
+
+    if(in_header) {
+      int f;
+      regmatch_t matches;
+      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,
+                  raw_line + matches.rm_eo);
+        }
+      }
+    }
+
+    position_in_file += strlen(raw_line);
   }
 
   fclose(file);
 }
 
+int ignore_entry(const char *name) {
+  return
+    strcmp(name, ".") == 0 ||
+    strcmp(name, "..") == 0 ||
+    (name[0] == '.' && name[1] != '/');
+}
+
+void process_entry(const char *dir_name,
+                   int nb_fields_to_parse, struct parsable_field *fields_to_parse) {
+  DIR *dir;
+  struct dirent *dir_e;
+  struct stat sb;
+  char subname[PATH_MAX + 1];
+
+  if(lstat(dir_name, &sb) != 0) {
+    fprintf(stderr,
+            "mymail: Can not stat \"%s\": %s\n",
+            dir_name,
+            strerror(errno));
+    exit(EXIT_FAILURE);
+  } else {
+  }
+
+  if(S_ISLNK(sb.st_mode)) {
+    return;
+  }
+
+  dir = opendir(dir_name);
+
+  if(dir) {
+    printf("Processing directory '%s'.\n", 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);
+      }
+    }
+    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);
+    }
+  }
+}
+
 /*********************************************************************/
 
 /* For long options that have no equivalent short option, use a
@@ -109,16 +182,33 @@ enum
 
 static struct option long_options[] = {
   { "help", no_argument, 0, 'h' },
+  { "db-prefix", 1, 0, 'p' },
   { 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
+  },
+
+  {
+    "dest",
+    "^\\([Tt][Oo]\\|[Cc][Cc]\\|[Bb][Cc][Cc]\\): *",
+    { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 }, 0
+  },
+};
+
 int main(int argc, char **argv) {
   int error = 0, show_help = 0;
+  const int nb_fields_to_parse = sizeof(fields_to_parse) / sizeof(struct parsable_field);
   char c;
+  int f;
 
   setlocale(LC_ALL, "");
 
-  while ((c = getopt_long(argc, argv, "o:s:x:vwmqf:ibzdeajyunt:r:l:c:-h",
+  while ((c = getopt_long(argc, argv, "hp:",
                           long_options, NULL)) != -1) {
 
     switch(c) {
@@ -127,12 +217,20 @@ int main(int argc, char **argv) {
       show_help = 1;
       break;
 
+    case 'p':
+      db_filename_prefix = strdup(optarg);
+      break;
+
     default:
       error = 1;
       break;
     }
   }
 
+  if(!db_filename_prefix) {
+    db_filename_prefix = strdup("/tmp/mymail_");
+  }
+
   if(error) {
     usage(stderr);
     exit(EXIT_FAILURE);
@@ -143,5 +241,41 @@ 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) {
+      fprintf(stderr,
+              "mymail: Can not open \"%s\" for writing: %s\n",
+              db_filename,
+              strerror(errno));
+      exit(EXIT_FAILURE);
+    }
+
+    printf("Initialized %s.\n", db_filename);
+
+    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);
+    optind++;
+  }
+
+  for(f = 0; f < nb_fields_to_parse; f++) {
+    fclose(fields_to_parse[f].db_file);
+    regfree(&fields_to_parse[f].regexp);
+  }
+
   exit(EXIT_SUCCESS);
 }