use super::*;
use crate::utils::{
datetime,
parsec::{CRLF, LF},
};
impl MboxFormat {
#[allow(clippy::too_many_arguments)]
pub fn append(
&self,
writer: &mut dyn std::io::Write,
input: &[u8],
envelope_from: Option<&Address>,
delivery_date: Option<crate::UnixTimestamp>,
(flags, tags): (Flag, Vec<&str>),
metadata_format: MboxMetadata,
is_empty: bool,
crlf: bool,
) -> Result<()> {
if tags.iter().any(|t| t.contains(' ')) {
return Err(Error::new("mbox tags/keywords can't contain spaces"));
}
let line_ending: &'static [u8] = if crlf { CRLF } else { LF };
if !is_empty {
writer.write_all(line_ending)?;
writer.write_all(line_ending)?;
}
writer.write_all(b"From ")?;
if let Some(from) = envelope_from {
writer.write_all(from.address_spec_raw())?;
} else {
write!(writer, "{}", uuid::Uuid::nil().as_simple())?;
}
writer.write_all(b" ")?;
writer.write_all(
datetime::timestamp_to_string(
delivery_date.unwrap_or_else(datetime::now),
Some(datetime::formats::ASCTIME_FMT),
true,
)
.trim()
.as_bytes(),
)?;
writer.write_all(line_ending)?;
let (mut headers, body) = parser::mail(input)?;
headers.retain(|(header_name, _)| {
header_name != HeaderName::STATUS
&& header_name != HeaderName::X_STATUS
&& header_name != HeaderName::X_KEYWORDS
&& header_name != HeaderName::CONTENT_LENGTH
});
let write_header_val_fn = |writer: &mut dyn std::io::Write, bytes: &[u8]| {
let mut i = 0;
if crlf {
while i < bytes.len() {
if bytes[i..].starts_with(CRLF) {
writer.write_all(CRLF)?;
i += 2;
continue;
} else if bytes[i] == b'\n' {
writer.write_all(CRLF)?;
} else {
writer.write_all(&[bytes[i]])?;
}
i += 1;
}
} else {
while i < bytes.len() {
if bytes[i..].starts_with(CRLF) {
writer.write_all(LF)?;
i += 2;
} else {
writer.write_all(&[bytes[i]])?;
i += 1;
}
}
}
Ok::<(), Error>(())
};
let write_metadata_fn = |writer: &mut dyn std::io::Write| match metadata_format {
MboxMetadata::CClient => {
for (h, v) in {
if flags.is_seen() {
Some((HeaderName::STATUS, "R".into()))
} else {
None
}
.into_iter()
.chain(
if !flags.is_flagged()
&& !flags.is_replied()
&& !flags.is_draft()
&& !flags.is_trashed()
{
None
} else {
Some((
HeaderName::X_STATUS,
format!(
"{flagged}{replied}{draft}{trashed}",
flagged = if flags.is_flagged() { "F" } else { "" },
replied = if flags.is_replied() { "A" } else { "" },
draft = if flags.is_draft() { "T" } else { "" },
trashed = if flags.is_trashed() { "D" } else { "" }
),
))
},
)
.chain(if tags.is_empty() {
None
} else {
Some((HeaderName::X_KEYWORDS, tags.as_slice().join(" ")))
})
} {
writer.write_fmt(format_args!("{h}: "))?;
writer.write_all(v.as_bytes())?;
writer.write_all(line_ending)?;
}
Ok::<(), Error>(())
}
MboxMetadata::None => Ok(()),
};
let body_len = {
let mut len = body.len();
if crlf {
let stray_lfs = body.iter().filter(|b| **b == b'\n').count()
- body.windows(CRLF.len()).filter(|w| w == &CRLF).count();
len += stray_lfs;
} else {
let crlfs = body.windows(CRLF.len()).filter(|w| w == &CRLF).count();
len -= crlfs;
}
len
};
match self {
Self::MboxO | Self::MboxRd => {
Err(Error::new("Not implemented.").set_kind(ErrorKind::NotImplemented))
}
Self::MboxCl => {
let len = (body_len
+ body
.windows(b"\nFrom ".len())
.filter(|w| w == b"\nFrom ")
.count()
+ if body.starts_with(b"From ") { 1 } else { 0 })
.to_string();
for (h, v) in headers
.into_iter()
.chain(Some((HeaderName::CONTENT_LENGTH, len.as_bytes())))
{
writer.write_fmt(format_args!("{h}: "))?;
write_header_val_fn(writer, v)?;
writer.write_all(line_ending)?;
}
write_metadata_fn(writer)?;
writer.write_all(line_ending)?;
if body.starts_with(b"From ") {
writer.write_all(b">")?;
}
let mut i = 0;
if crlf {
while i < body.len() {
if body[i..].starts_with(CRLF) {
writer.write_all(CRLF)?;
if body[i..].starts_with(b"\r\nFrom ") {
writer.write_all(b">")?;
}
i += 2;
} else if body[i] == b'\n' {
writer.write_all(CRLF)?;
if body[i..].starts_with(b"\nFrom ") {
writer.write_all(b">")?;
}
i += 1;
} else {
writer.write_all(&[body[i]])?;
i += 1;
}
}
} else {
while i < body.len() {
if body[i..].starts_with(CRLF) {
writer.write_all(LF)?;
if body[i..].starts_with(b"\r\nFrom ") {
writer.write_all(b">")?;
}
i += 2;
} else {
writer.write_all(&[body[i]])?;
if body[i..].starts_with(b"\nFrom ") {
writer.write_all(b">")?;
}
i += 1;
}
}
}
Ok(())
}
Self::MboxCl2 => {
let len = body_len.to_string();
for (h, v) in headers
.into_iter()
.chain(Some((HeaderName::CONTENT_LENGTH, len.as_bytes())))
{
writer.write_fmt(format_args!("{h}: "))?;
write_header_val_fn(writer, v)?;
writer.write_all(line_ending)?;
}
write_metadata_fn(writer)?;
writer.write_all(line_ending)?;
let mut i = 0;
if crlf {
while i < body.len() {
if body[i..].starts_with(CRLF) {
writer.write_all(CRLF)?;
i += 2;
continue;
} else if body[i] == b'\n' {
writer.write_all(CRLF)?;
} else {
writer.write_all(&[body[i]])?;
}
i += 1;
}
} else {
while i < body.len() {
if body[i..].starts_with(CRLF) {
writer.write_all(LF)?;
i += 2;
} else {
writer.write_all(&[body[i]])?;
i += 1;
}
}
}
Ok(())
}
}
}
}