82 lines
2.3 KiB
Rust
82 lines
2.3 KiB
Rust
use std::{fs::File, io, io::Read};
|
|
|
|
use mailparse::{
|
|
addrparse_header, dateparse, parse_mail, MailHeaderMap, MailParseError, ParsedMail,
|
|
};
|
|
use sqlx::postgres::PgPool;
|
|
use thiserror::Error;
|
|
|
|
#[derive(Error, Debug)]
|
|
pub enum MailError {
|
|
#[error("missing from header")]
|
|
MissingFrom,
|
|
#[error("missing from header display name")]
|
|
MissingFromDisplayName,
|
|
#[error("missing subject header")]
|
|
MissingSubject,
|
|
#[error("missing html part")]
|
|
MissingHtmlPart,
|
|
#[error("missing message ID")]
|
|
MissingMessageId,
|
|
#[error("missing date")]
|
|
MissingDate,
|
|
#[error("DB error {0}")]
|
|
SqlxError(#[from] sqlx::Error),
|
|
#[error("IO error {0}")]
|
|
IOError(#[from] std::io::Error),
|
|
#[error("mail parse error {0}")]
|
|
MailParseError(#[from] MailParseError),
|
|
}
|
|
|
|
pub async fn read_mail_to_db(pool: &PgPool, path: &str) -> Result<(), MailError> {
|
|
let mut file = File::open(path)?;
|
|
let mut buffer = Vec::new();
|
|
file.read_to_end(&mut buffer)?;
|
|
let m = parse_mail(&buffer)?;
|
|
|
|
let subject = m
|
|
.headers
|
|
.get_first_value("subject")
|
|
.ok_or(MailError::MissingSubject)?;
|
|
|
|
let from = addrparse_header(
|
|
m.headers
|
|
.get_first_header("from")
|
|
.ok_or(MailError::MissingFrom)?,
|
|
)?;
|
|
let from = from.extract_single_info().ok_or(MailError::MissingFrom)?;
|
|
let name = from.display_name.ok_or(MailError::MissingFromDisplayName)?;
|
|
let slug = name.to_lowercase().replace(' ', "-");
|
|
let url = from.addr;
|
|
let message_id = m
|
|
.headers
|
|
.get_first_value("Message-ID")
|
|
.ok_or(MailError::MissingMessageId)?;
|
|
let uid = &message_id;
|
|
let feed_id = find_feed(&pool, &name, &slug, &url).await?;
|
|
let date = dateparse(
|
|
&m.headers
|
|
.get_first_value("Date")
|
|
.ok_or(MailError::MissingDate)?,
|
|
)?;
|
|
|
|
println!("Feed: {feed_id} Subject: {}", subject);
|
|
|
|
if let Some(m) = first_html(&m) {
|
|
let body = m.get_body()?;
|
|
add_post(&pool, &slug, &subject, &body, &message_id, date, &uid, &url).await?;
|
|
} else {
|
|
return Err(MailError::MissingHtmlPart.into());
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
fn first_html<'m>(m: &'m ParsedMail<'m>) -> Option<&'m ParsedMail<'m>> {
|
|
for ele in m.parts() {
|
|
if ele.ctype.mimetype == "text/html" {
|
|
return Some(ele);
|
|
}
|
|
}
|
|
None
|
|
}
|