use std::path::Path;
use std::fs::OpenOptions;
use std::io::Read;
use failure::Error;
use failure::Fallible as Result;
use failure::ResultExt;
pub(crate) fn get_message_header_at_key<P: AsRef<Path>, K: AsRef<str>>(p: P, k: K) -> Result<String> {
let buffer = {
let mut buffer = Vec::with_capacity(4096); OpenOptions::new()
.read(true)
.open(p.as_ref())?
.read_to_end(&mut buffer)?;
buffer
};
::mailparse::parse_mail(&buffer)
.context(format_err!("Cannot parse Email {}", p.as_ref().display()))?
.headers
.into_iter()
.filter_map(|hdr| match hdr.get_key().context("Cannot get key from mail header").map_err(Error::from) {
Ok(key) => {
let lower_key = key.to_lowercase();
trace!("Test: {} == {}", lower_key, k.as_ref());
if lower_key == k.as_ref() {
Some(Ok(hdr))
} else {
None
}
},
Err(e) => Some(Err(e))
})
.next()
.ok_or_else(|| format_err!("'{}' not found in {}", k.as_ref(), p.as_ref().display()))?
.and_then(|hdr| hdr.get_value().context("Cannot get value from mail header").map_err(Error::from))
}
pub(crate) fn get_message_id_for_mailfile<P: AsRef<Path>>(p: P) -> Result<String> {
get_message_header_at_key(p, "message-id").map(strip_message_delimiters)
}
pub(crate) fn strip_message_delimiters<ID: AsRef<str>>(id: ID) -> String {
let len = id.as_ref().len();
id.as_ref()
.chars()
.enumerate()
.filter(|(idx, chr)| !(*idx == 0 && *chr == '<' || *idx == len - 1 && *chr == '>'))
.map(|tpl| tpl.1)
.collect()
}
pub fn get_mail_text_content<P: AsRef<Path>>(p: P) -> Result<String> {
::mailparse::parse_mail(::std::fs::read_to_string(p.as_ref())?.as_bytes())
.context(format_err!("Cannot parse Email {}", p.as_ref().display()))?
.get_body()
.context("Cannot get body of mail")
.map_err(Error::from)
}