feature: use a better and more reliable way to change the message-id.

Author: maeln

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "eml-replicator"
-version = "0.1.2"
+version = "0.1.3"
 authors = ["Maël Naccache Tüfekçi <contact@maeln.com>"]
 edition = "2018"
 license = "CECILL-2.1"
@@ -17,4 +17,6 @@ native-tls = "0.2"
 clap = "2.33"
 walkdir = "2"
 indicatif = "0.15"
-rand = "0.8"
+rand = "0.8"
+regex = "1"
+lazy_static = "1.4"

Cargo.toml

@@ -5,7 +5,7 @@ This tool read all the EML (RFC822 / RFC2822) in a directory and copy them in a
 Usage:
 
 ```
-eml-replicator 0.1.2
+eml-replicator 0.1.3
 Maël Naccache Tüfekçi
 A tool that read EML files and copy them to a IMAP mailbox.
 

README.md

@@ -1,4 +1,7 @@
+#[macro_use]
+extern crate lazy_static;
 use indicatif::ProgressStyle;
+use regex::Regex;
 use std::fs::{self};
 use std::io::{Error, ErrorKind};
 use std::path::{Path, PathBuf};
@@ -45,13 +48,18 @@ fn list_eml_file(
 }
 
 fn randomize_message_id(eml: &String) -> Result<String, String> {
+    lazy_static! {
+        static ref MID_RE: Regex = Regex::new(r"(?imu)^message-id:.+$").unwrap();
+    }
+
     let mut new_eml = String::new();
 
-    let header_pos = eml.find("Message-ID:");
+    let header_pos = MID_RE.find(eml);
     if header_pos.is_none() {
         return Err("Could not find Message-ID in the EML.".to_string());
     }
-    let (fpart, lpart) = eml.split_at(header_pos.unwrap());
+
+    let (fpart, lpart) = eml.split_at(header_pos.unwrap().start());
     new_eml.push_str(fpart);
 
     let (_mid, rest) = lpart.split_at(lpart.find('\n').expect("Malformed Message-ID."));
@@ -95,6 +103,7 @@ impl Config {
         let recursive = matches.is_present("recursive");
         let symlink = matches.is_present("symlink");
         let random_id = matches.is_present("random-message-id");
+
         Config {
             server,
             port,
@@ -192,6 +201,10 @@ fn main() {
         println!("- {}", path.to_str().unwrap_or(""));
     }
 
+    if conf.random_id {
+        println!("Randomizing Message-IDs.")
+    }
+
     let tls = native_tls::TlsConnector::builder().build().unwrap();
     let client = imap::connect((conf.server.clone(), conf.port), conf.server, &tls).unwrap();
     let mut session = client.login(conf.login, conf.password).unwrap();
@@ -204,10 +217,17 @@ fn main() {
     for eml in &emls_files {
         let rfc822 = fs::read_to_string(eml).expect("Failed to read eml file.");
         if conf.random_id {
-            let randomize_id = randomize_message_id(&rfc822).unwrap();
-            session
-                .append(&conf.folder, &randomize_id)
-                .expect("Could not copy eml file to inbox.");
+            let randomize_id = randomize_message_id(&rfc822);
+            if randomize_id.is_err() {
+                println!(
+                    "Could not find Message-ID for file {}, skipping.",
+                    eml.to_string_lossy()
+                );
+            } else {
+                session
+                    .append(&conf.folder, &randomize_id.unwrap())
+                    .expect("Could not copy eml file to inbox.");
+            }
         } else {
             session
                 .append(&conf.folder, &rfc822)