aboutsummaryrefslogtreecommitdiff
path: root/src/imap/mailbox_view.rs
diff options
context:
space:
mode:
authorQuentin Dufour <quentin@deuxfleurs.fr>2023-10-12 12:21:59 +0200
committerQuentin Dufour <quentin@deuxfleurs.fr>2023-10-12 12:21:59 +0200
commita1b7ca17c0badff4c983f3738531f3aa9b0fb977 (patch)
tree419f57fbd9c179c518fd57c0e3f31098305eb4b3 /src/imap/mailbox_view.rs
parentb444ef7ef3d6e0e57726130e3b89a676cdcd6245 (diff)
downloadaerogramme-a1b7ca17c0badff4c983f3738531f3aa9b0fb977.tar.gz
aerogramme-a1b7ca17c0badff4c983f3738531f3aa9b0fb977.zip
basic body ext testing + format
Diffstat (limited to 'src/imap/mailbox_view.rs')
-rw-r--r--src/imap/mailbox_view.rs486
1 files changed, 349 insertions, 137 deletions
diff --git a/src/imap/mailbox_view.rs b/src/imap/mailbox_view.rs
index ecc0208..99069e2 100644
--- a/src/imap/mailbox_view.rs
+++ b/src/imap/mailbox_view.rs
@@ -1,7 +1,7 @@
use std::borrow::Cow;
+use std::iter::zip;
use std::num::NonZeroU32;
use std::sync::Arc;
-use std::iter::zip;
use anyhow::{anyhow, bail, Error, Result};
use boitalettres::proto::res::body::Data as Body;
@@ -22,15 +22,13 @@ use imap_codec::types::response::{Code, Data, MessageAttribute, Status};
use imap_codec::types::sequence::{self, SequenceSet};
use eml_codec::{
- header,
- imf,
- part::{AnyPart, composite::Message},
+ header, imf, mime,
mime::r#type::Deductible,
- mime,
+ part::{composite::Message, AnyPart},
};
use crate::cryptoblob::Key;
-use crate::mail::mailbox::{Mailbox, MailMeta};
+use crate::mail::mailbox::{MailMeta, Mailbox};
use crate::mail::uidindex::{ImapUid, ImapUidvalidity, UidIndex};
use crate::mail::unique_ident::UniqueIdent;
@@ -44,26 +42,42 @@ const DEFAULT_FLAGS: [Flag; 5] = [
enum FetchedMail<'a> {
Partial(imf::Imf<'a>),
- Full(Message<'a>),
+ Full(AnyPart<'a>),
}
impl<'a> FetchedMail<'a> {
- fn as_full(&self) -> Result<&Message<'a>> {
+ fn new_from_message(msg: Message<'a>) -> Self {
+ FetchedMail::Full(AnyPart::Msg(msg))
+ }
+
+ /*fn new_from_header(hdr: imf::Imf<'a>) -> Self {
+ FetchedMail::Partial(hdr)
+ }*/
+
+ fn as_anypart(&self) -> Result<&AnyPart<'a>> {
match self {
FetchedMail::Full(x) => Ok(&x),
- _ => bail!("The full message must be fetched, not only its headers: it's a logic error"),
+ _ => bail!("The full message must be fetched, not only its headers"),
+ }
+ }
+
+ fn as_full(&self) -> Result<&Message<'a>> {
+ match self {
+ FetchedMail::Full(AnyPart::Msg(x)) => Ok(&x),
+ _ => bail!("The full message must be fetched, not only its headers AND it must be an AnyPart::Msg."),
}
}
fn imf(&self) -> &imf::Imf<'a> {
match self {
- FetchedMail::Full(x) => &x.imf,
+ FetchedMail::Full(AnyPart::Msg(x)) => &x.imf,
FetchedMail::Partial(x) => &x,
+ _ => panic!("Can't contain AnyPart that is not a message"),
}
}
}
pub struct AttributesProxy {
- attrs: Vec<FetchAttribute>
+ attrs: Vec<FetchAttribute>,
}
impl AttributesProxy {
fn new(attrs: &MacroOrFetchAttributes, is_uid_fetch: bool) -> Self {
@@ -112,8 +126,8 @@ pub struct MailView<'a> {
ids: &'a MailIdentifiers,
meta: &'a MailMeta,
flags: &'a Vec<String>,
- content: FetchedMail<'a>,
- add_seen: bool
+ content: FetchedMail<'a>,
+ add_seen: bool,
}
impl<'a> MailView<'a> {
@@ -123,7 +137,10 @@ impl<'a> MailView<'a> {
fn flags(&self) -> MessageAttribute {
MessageAttribute::Flags(
- self.flags.iter().filter_map(|f| string_to_flag(f)).collect(),
+ self.flags
+ .iter()
+ .filter_map(|f| string_to_flag(f))
+ .collect(),
)
}
@@ -132,18 +149,37 @@ impl<'a> MailView<'a> {
}
fn rfc_822_header(&self) -> MessageAttribute {
- MessageAttribute::Rfc822Header(NString(self.meta.headers.to_vec().try_into().ok().map(IString::Literal)))
+ MessageAttribute::Rfc822Header(NString(
+ self.meta
+ .headers
+ .to_vec()
+ .try_into()
+ .ok()
+ .map(IString::Literal),
+ ))
}
fn rfc_822_text(&self) -> Result<MessageAttribute> {
Ok(MessageAttribute::Rfc822Text(NString(
- self.content.as_full()?.raw_body.try_into().ok().map(IString::Literal),
+ self.content
+ .as_full()?
+ .raw_body
+ .try_into()
+ .ok()
+ .map(IString::Literal),
)))
}
fn rfc822(&self) -> Result<MessageAttribute> {
Ok(MessageAttribute::Rfc822(NString(
- self.content.as_full()?.raw_body.clone().try_into().ok().map(IString::Literal))))
+ self.content
+ .as_full()?
+ .raw_body
+ .clone()
+ .try_into()
+ .ok()
+ .map(IString::Literal),
+ )))
}
fn envelope(&self) -> MessageAttribute {
@@ -151,24 +187,29 @@ impl<'a> MailView<'a> {
}
fn body(&self) -> Result<MessageAttribute> {
- Ok(MessageAttribute::Body(
- build_imap_email_struct(self.content.as_full()?.child.as_ref())?,
- ))
+ Ok(MessageAttribute::Body(build_imap_email_struct(
+ self.content.as_full()?.child.as_ref(),
+ )?))
}
fn body_structure(&self) -> Result<MessageAttribute> {
- Ok(MessageAttribute::Body(
- build_imap_email_struct(self.content.as_full()?.child.as_ref())?,
- ))
+ Ok(MessageAttribute::Body(build_imap_email_struct(
+ self.content.as_full()?.child.as_ref(),
+ )?))
}
/// maps to BODY[<section>]<<partial>> and BODY.PEEK[<section>]<<partial>>
/// peek does not implicitly set the \Seen flag
/// eg. BODY[HEADER.FIELDS (DATE FROM)]
/// eg. BODY[]<0.2048>
- fn body_ext(&mut self, section: &Option<FetchSection>, partial: &Option<(u32, NonZeroU32)>, peek: &bool) -> Result<MessageAttribute> {
+ fn body_ext(
+ &mut self,
+ section: &Option<FetchSection>,
+ partial: &Option<(u32, NonZeroU32)>,
+ peek: &bool,
+ ) -> Result<MessageAttribute> {
// Extract message section
- let text = get_message_section(self.content.as_full()?, section)?;
+ let text = get_message_section(self.content.as_anypart()?, section)?;
let seen_flag = Flag::Seen.to_string();
if !peek && !self.flags.iter().any(|x| *x == seen_flag) {
@@ -186,29 +227,40 @@ impl<'a> MailView<'a> {
section: section.clone(),
origin,
data,
- })
-
+ });
}
fn internal_date(&self) -> Result<MessageAttribute> {
- let dt = Utc.fix().timestamp_opt(i64::try_from(self.meta.internaldate / 1000)?, 0).earliest().ok_or(anyhow!("Unable to parse internal date"))?;
+ let dt = Utc
+ .fix()
+ .timestamp_opt(i64::try_from(self.meta.internaldate / 1000)?, 0)
+ .earliest()
+ .ok_or(anyhow!("Unable to parse internal date"))?;
Ok(MessageAttribute::InternalDate(MyDateTime(dt)))
}
fn filter(&mut self, ap: &AttributesProxy) -> Result<Body> {
- let res_attrs = ap.attrs.iter().map(|attr| match attr {
- FetchAttribute::Uid => Ok(self.uid()),
- FetchAttribute::Flags => Ok(self.flags()),
- FetchAttribute::Rfc822Size => Ok(self.rfc_822_size()),
- FetchAttribute::Rfc822Header => Ok(self.rfc_822_header()),
- FetchAttribute::Rfc822Text => self.rfc_822_text(),
- FetchAttribute::Rfc822 => self.rfc822(),
- FetchAttribute::Envelope => Ok(self.envelope()),
- FetchAttribute::Body => self.body(),
- FetchAttribute::BodyStructure => self.body_structure(),
- FetchAttribute::BodyExt { section, partial, peek } => self.body_ext(section, partial, peek),
- FetchAttribute::InternalDate => self.internal_date(),
- }).collect::<Result<Vec<_>, _>>()?;
+ let res_attrs = ap
+ .attrs
+ .iter()
+ .map(|attr| match attr {
+ FetchAttribute::Uid => Ok(self.uid()),
+ FetchAttribute::Flags => Ok(self.flags()),
+ FetchAttribute::Rfc822Size => Ok(self.rfc_822_size()),
+ FetchAttribute::Rfc822Header => Ok(self.rfc_822_header()),
+ FetchAttribute::Rfc822Text => self.rfc_822_text(),
+ FetchAttribute::Rfc822 => self.rfc822(),
+ FetchAttribute::Envelope => Ok(self.envelope()),
+ FetchAttribute::Body => self.body(),
+ FetchAttribute::BodyStructure => self.body_structure(),
+ FetchAttribute::BodyExt {
+ section,
+ partial,
+ peek,
+ } => self.body_ext(section, partial, peek),
+ FetchAttribute::InternalDate => self.internal_date(),
+ })
+ .collect::<Result<Vec<_>, _>>()?;
Ok(Body::Data(Data::Fetch {
seq_or_uid: self.ids.i,
@@ -217,7 +269,10 @@ impl<'a> MailView<'a> {
}
}
-fn apply_partial<'a>(partial: &'_ Option<(u32, NonZeroU32)>, text: &'a [u8]) -> (&'a [u8], Option<u32>) {
+fn apply_partial<'a>(
+ partial: &'_ Option<(u32, NonZeroU32)>,
+ text: &'a [u8],
+) -> (&'a [u8], Option<u32>) {
match partial {
Some((begin, len)) => {
if *begin as usize > text.len() {
@@ -253,11 +308,11 @@ pub struct MailSelectionBuilder<'a> {
impl<'a> MailSelectionBuilder<'a> {
fn new(need_body: bool, mail_count: usize) -> Self {
- Self {
- mail_count,
+ Self {
+ mail_count,
need_body,
..MailSelectionBuilder::default()
- }
+ }
}
fn with_mail_identifiers(&mut self, mi: &'a [MailIdentifiers]) -> &mut Self {
@@ -269,7 +324,7 @@ impl<'a> MailSelectionBuilder<'a> {
self.meta = meta;
self
}
-
+
fn with_flags(&mut self, flags: &'a [&'a Vec<String>]) -> &mut Self {
self.flags = flags;
self
@@ -277,9 +332,14 @@ impl<'a> MailSelectionBuilder<'a> {
fn bodies_to_collect(&self) -> Vec<BodyIdentifier> {
if !self.need_body {
- return vec![]
+ return vec![];
}
- zip(self.mi, self.meta).map(|(mi, meta)| BodyIdentifier { msg_uuid: &mi.uuid, msg_key: &meta.message_key }).collect::<Vec<_>>()
+ zip(self.mi, self.meta)
+ .map(|(mi, meta)| BodyIdentifier {
+ msg_uuid: &mi.uuid,
+ msg_key: &meta.message_key,
+ })
+ .collect::<Vec<_>>()
}
fn with_bodies(&mut self, rbodies: &'a [Vec<u8>]) -> &mut Self {
@@ -292,22 +352,32 @@ impl<'a> MailSelectionBuilder<'a> {
if !self.need_body {
for m in self.meta.iter() {
- let (_, hdrs) = eml_codec::parse_imf(&m.headers).or(Err(anyhow!("Invalid mail headers")))?;
+ let (_, hdrs) =
+ eml_codec::parse_imf(&m.headers).or(Err(anyhow!("Invalid mail headers")))?;
bodies.push(FetchedMail::Partial(hdrs));
}
} else {
for rb in self.bodies.iter() {
let (_, p) = eml_codec::parse_message(&rb).or(Err(anyhow!("Invalid mail body")))?;
- bodies.push(FetchedMail::Full(p));
+ bodies.push(FetchedMail::new_from_message(p));
}
}
- if self.mi.len() != self.mail_count && self.meta.len() != self.mail_count || self.flags.len() != self.mail_count || bodies.len() != self.mail_count {
- return Err(anyhow!("Can't build a mail view selection as parts were not correctly registered into the builder."))
+ if self.mi.len() != self.mail_count && self.meta.len() != self.mail_count
+ || self.flags.len() != self.mail_count
+ || bodies.len() != self.mail_count
+ {
+ return Err(anyhow!("Can't build a mail view selection as parts were not correctly registered into the builder."));
}
Ok(zip(self.mi, zip(self.meta, zip(self.flags, bodies)))
- .map(|(ids, (meta, (flags, content)))| MailView { ids, meta, flags, content, add_seen: false })
+ .map(|(ids, (meta, (flags, content)))| MailView {
+ ids,
+ meta,
+ flags,
+ content,
+ add_seen: false,
+ })
.collect())
}
}
@@ -505,7 +575,6 @@ impl MailboxView {
attributes: &MacroOrFetchAttributes,
is_uid_fetch: &bool,
) -> Result<Vec<Body>> {
-
let ap = AttributesProxy::new(attributes, *is_uid_fetch);
// Prepare data
@@ -513,22 +582,38 @@ impl MailboxView {
let mail_count = mids.0.len();
let uuids = mids.uuids();
let meta = self.mailbox.fetch_meta(&uuids).await?;
- let flags = uuids.iter().map(|uuid| self.known_state.table.get(uuid).map(|(_uuid, f)| f).ok_or(anyhow!("missing email from the flag table"))).collect::<Result<Vec<_>, _>>()?;
+ let flags = uuids
+ .iter()
+ .map(|uuid| {
+ self.known_state
+ .table
+ .get(uuid)
+ .map(|(_uuid, f)| f)
+ .ok_or(anyhow!("missing email from the flag table"))
+ })
+ .collect::<Result<Vec<_>, _>>()?;
// Start filling data to build the view
let mut selection = MailSelectionBuilder::new(ap.need_body(), mail_count);
selection
- .with_mail_identifiers(&mids.0)
- .with_metadata(&meta)
- .with_flags(&flags);
-
+ .with_mail_identifiers(&mids.0)
+ .with_metadata(&meta)
+ .with_flags(&flags);
+
// Asynchronously fetch full bodies (if needed)
let btc = selection.bodies_to_collect();
- let future_bodies = btc.iter().map(|bi| async move {
- let body = self.mailbox.fetch_full(*bi.msg_uuid, bi.msg_key).await?;
- Ok::<_, anyhow::Error>(body)
- }).collect::<FuturesOrdered<_>>();
- let bodies = future_bodies.collect::<Vec<_>>().await.into_iter().collect::<Result<Vec<_>, _>>()?;
+ let future_bodies = btc
+ .iter()
+ .map(|bi| async move {
+ let body = self.mailbox.fetch_full(*bi.msg_uuid, bi.msg_key).await?;
+ Ok::<_, anyhow::Error>(body)
+ })
+ .collect::<FuturesOrdered<_>>();
+ let bodies = future_bodies
+ .collect::<Vec<_>>()
+ .await
+ .into_iter()
+ .collect::<Result<Vec<_>, _>>()?;
// Add bodies
selection.with_bodies(bodies.as_slice());
@@ -537,15 +622,26 @@ impl MailboxView {
let mut views = selection.build()?;
// Filter views to build the result
- let ret = views.iter_mut().filter_map(|mv| mv.filter(&ap).ok()).collect::<Vec<_>>();
+ let ret = views
+ .iter_mut()
+ .filter_map(|mv| mv.filter(&ap).ok())
+ .collect::<Vec<_>>();
// Register seen flags
- let future_flags = views.iter().filter(|mv| mv.add_seen).map(|mv| async move {
- let seen_flag = Flag::Seen.to_string();
- self.mailbox.add_flags(mv.ids.uuid, &[seen_flag]).await?;
- Ok::<_, anyhow::Error>(())
- }).collect::<FuturesOrdered<_>>();
- future_flags.collect::<Vec<_>>().await.into_iter().collect::<Result<_, _>>()?;
+ let future_flags = views
+ .iter()
+ .filter(|mv| mv.add_seen)
+ .map(|mv| async move {
+ let seen_flag = Flag::Seen.to_string();
+ self.mailbox.add_flags(mv.ids.uuid, &[seen_flag]).await?;
+ Ok::<_, anyhow::Error>(())
+ })
+ .collect::<FuturesOrdered<_>>();
+ future_flags
+ .collect::<Vec<_>>()
+ .await
+ .into_iter()
+ .collect::<Result<_, _>>()?;
Ok(ret)
}
@@ -583,7 +679,11 @@ impl MailboxView {
}
if let Some(mail) = mail_vec.get(i) {
if mail.0 == uid {
- mails.push(MailIdentifiers { i: NonZeroU32::try_from(i as u32 + 1).unwrap(), uid: mail.0, uuid: mail.1 });
+ mails.push(MailIdentifiers {
+ i: NonZeroU32::try_from(i as u32 + 1).unwrap(),
+ uid: mail.0,
+ uuid: mail.1,
+ });
}
} else {
break;
@@ -600,7 +700,11 @@ impl MailboxView {
for i in sequence_set.iter(iter_strat) {
if let Some(mail) = mail_vec.get(i.get() as usize - 1) {
- mails.push(MailIdentifiers { i, uid: mail.0, uuid: mail.1 });
+ mails.push(MailIdentifiers {
+ i,
+ uid: mail.0,
+ uuid: mail.1,
+ });
} else {
bail!("No such mail: {}", i);
}
@@ -759,26 +863,39 @@ fn message_envelope(msg: &imf::Imf) -> Envelope {
Envelope {
date: NString(
- msg.date.as_ref()
+ msg.date
+ .as_ref()
.map(|d| IString::try_from(d.to_rfc3339()).unwrap()),
),
subject: NString(
- msg.subject.as_ref()
+ msg.subject
+ .as_ref()
.map(|d| IString::try_from(d.to_string()).unwrap()),
),
- sender: msg.sender.as_ref().map(|v| vec![convert_mbx(v)]).unwrap_or(from.clone()),
+ sender: msg
+ .sender
+ .as_ref()
+ .map(|v| vec![convert_mbx(v)])
+ .unwrap_or(from.clone()),
reply_to: if msg.reply_to.is_empty() {
- from.clone()
- } else {
- convert_addresses(&msg.reply_to)
- },
+ from.clone()
+ } else {
+ convert_addresses(&msg.reply_to)
+ },
from,
to: convert_addresses(&msg.to),
cc: convert_addresses(&msg.cc),
bcc: convert_addresses(&msg.bcc),
- in_reply_to: NString(msg.in_reply_to.iter().next().map(|d| IString::try_from(d.to_string()).unwrap())),
+ in_reply_to: NString(
+ msg.in_reply_to
+ .iter()
+ .next()
+ .map(|d| IString::try_from(d.to_string()).unwrap()),
+ ),
message_id: NString(
- msg.msg_id.as_ref().map(|d| IString::try_from(d.to_string()).unwrap()),
+ msg.msg_id
+ .as_ref()
+ .map(|d| IString::try_from(d.to_string()).unwrap()),
),
}
}
@@ -788,20 +905,28 @@ fn convert_addresses(addrlist: &Vec<imf::address::AddressRef>) -> Vec<Address> {
for item in addrlist {
match item {
imf::address::AddressRef::Single(a) => acc.push(convert_mbx(a)),
- imf::address::AddressRef::Many(l) => acc.extend(l.participants.iter().map(convert_mbx))
+ imf::address::AddressRef::Many(l) => acc.extend(l.participants.iter().map(convert_mbx)),
}
}
- return acc
+ return acc;
}
fn convert_mbx(addr: &imf::mailbox::MailboxRef) -> Address {
Address::new(
- NString(addr.name.as_ref().map(|x| IString::try_from(x.to_string()).unwrap())),
+ NString(
+ addr.name
+ .as_ref()
+ .map(|x| IString::try_from(x.to_string()).unwrap()),
+ ),
// SMTP at-domain-list (source route) seems obsolete since at least 1991
// https://www.mhonarc.org/archive/html/ietf-822/1991-06/msg00060.html
NString(None),
- NString(Some(IString::try_from(addr.addrspec.local_part.to_string()).unwrap())),
- NString(Some(IString::try_from(addr.addrspec.domain.to_string()).unwrap())),
+ NString(Some(
+ IString::try_from(addr.addrspec.local_part.to_string()).unwrap(),
+ )),
+ NString(Some(
+ IString::try_from(addr.addrspec.domain.to_string()).unwrap(),
+ )),
)
}
@@ -824,10 +949,12 @@ fn build_imap_email_struct<'a>(part: &AnyPart<'a>) -> Result<BodyStructure> {
match part {
AnyPart::Mult(x) => {
let itype = &x.mime.interpreted_type;
- let subtype = IString::try_from(itype.subtype.to_string()).unwrap_or(unchecked_istring("alternative"));
+ let subtype = IString::try_from(itype.subtype.to_string())
+ .unwrap_or(unchecked_istring("alternative"));
Ok(BodyStructure::Multi {
- bodies: x.children
+ bodies: x
+ .children
.iter()
.filter_map(|inner| build_imap_email_struct(&inner).ok())
.collect(),
@@ -845,18 +972,19 @@ fn build_imap_email_struct<'a>(part: &AnyPart<'a>) -> Result<BodyStructure> {
AnyPart::Txt(x) => {
let mut basic = basic_fields(&x.mime.fields, x.body.len())?;
- // Get the interpreted content type, set it
+ // Get the interpreted content type, set it
let itype = match &x.mime.interpreted_type {
- Deductible::Inferred(v) | Deductible::Explicit(v) => v
+ Deductible::Inferred(v) | Deductible::Explicit(v) => v,
};
- let subtype = IString::try_from(itype.subtype.to_string()).unwrap_or(unchecked_istring("plain"));
+ let subtype =
+ IString::try_from(itype.subtype.to_string()).unwrap_or(unchecked_istring("plain"));
// Add charset to the list of parameters if we know it has been inferred as it will be
// missing from the parsed content.
if let Deductible::Inferred(charset) = &itype.charset {
basic.parameter_list.push((
unchecked_istring("charset"),
- IString::try_from(charset.to_string()).unwrap_or(unchecked_istring("us-ascii"))
+ IString::try_from(charset.to_string()).unwrap_or(unchecked_istring("us-ascii")),
));
}
@@ -874,15 +1002,21 @@ fn build_imap_email_struct<'a>(part: &AnyPart<'a>) -> Result<BodyStructure> {
AnyPart::Bin(x) => {
let basic = basic_fields(&x.mime.fields, x.body.len())?;
- let default = mime::r#type::NaiveType { main: &b"application"[..], sub: &b"octet-stream"[..], params: vec![] };
+ let default = mime::r#type::NaiveType {
+ main: &b"application"[..],
+ sub: &b"octet-stream"[..],
+ params: vec![],
+ };
let ct = x.mime.fields.ctype.as_ref().unwrap_or(&default);
- let type_ = IString::try_from(String::from_utf8_lossy(ct.main).to_string())
- .or(Err(anyhow!("Unable to build IString from given Content-Type type given")))?;
+ let type_ = IString::try_from(String::from_utf8_lossy(ct.main).to_string()).or(Err(
+ anyhow!("Unable to build IString from given Content-Type type given"),
+ ))?;
-
- let subtype = IString::try_from(String::from_utf8_lossy(ct.sub).to_string())
- .or(Err(anyhow!("Unable to build IString from given Content-Type subtype given")))?;
+ let subtype =
+ IString::try_from(String::from_utf8_lossy(ct.sub).to_string()).or(Err(anyhow!(
+ "Unable to build IString from given Content-Type subtype given"
+ )))?;
Ok(BodyStructure::Single {
body: FetchBody {
@@ -911,7 +1045,8 @@ fn build_imap_email_struct<'a>(part: &AnyPart<'a>) -> Result<BodyStructure> {
}
fn nol(input: &[u8]) -> u32 {
- input.iter()
+ input
+ .iter()
.filter(|x| **x == b'\n')
.count()
.try_into()
@@ -925,13 +1060,22 @@ fn unchecked_istring(s: &'static str) -> IString {
}
fn basic_fields(m: &mime::NaiveMIME, sz: usize) -> Result<BasicFields> {
- let parameter_list = m.ctype
+ let parameter_list = m
+ .ctype
.as_ref()
- .map(|x| x.params.iter()
- .map(|p| (IString::try_from(String::from_utf8_lossy(p.name).to_string()), IString::try_from(p.value.to_string())))
- .filter(|(k, v)| k.is_ok() && v.is_ok())
- .map(|(k, v)| (k.unwrap(), v.unwrap()))
- .collect())
+ .map(|x| {
+ x.params
+ .iter()
+ .map(|p| {
+ (
+ IString::try_from(String::from_utf8_lossy(p.name).to_string()),
+ IString::try_from(p.value.to_string()),
+ )
+ })
+ .filter(|(k, v)| k.is_ok() && v.is_ok())
+ .map(|(k, v)| (k.unwrap(), v.unwrap()))
+ .collect()
+ })
.unwrap_or(vec![]);
Ok(BasicFields {
@@ -939,17 +1083,18 @@ fn basic_fields(m: &mime::NaiveMIME, sz: usize) -> Result<BasicFields> {
id: NString(
m.id.as_ref()
.and_then(|ci| IString::try_from(ci.to_string()).ok()),
- ),
+ ),
description: NString(
- m.description.as_ref()
+ m.description
+ .as_ref()
.and_then(|cd| IString::try_from(cd.to_string()).ok()),
),
content_transfer_encoding: match m.transfer_encoding {
- mime::mechanism::Mechanism::_8Bit => unchecked_istring("8bit"),
- mime::mechanism::Mechanism::Binary => unchecked_istring("binary"),
- mime::mechanism::Mechanism::QuotedPrintable => unchecked_istring("quoted-printable"),
- mime::mechanism::Mechanism::Base64 => unchecked_istring("base64"),
- _ => unchecked_istring("7bit"),
+ mime::mechanism::Mechanism::_8Bit => unchecked_istring("8bit"),
+ mime::mechanism::Mechanism::Binary => unchecked_istring("binary"),
+ mime::mechanism::Mechanism::QuotedPrintable => unchecked_istring("quoted-printable"),
+ mime::mechanism::Mechanism::Base64 => unchecked_istring("base64"),
+ _ => unchecked_istring("7bit"),
},
// @FIXME we can't compute the size of the message currently...
size: u32::try_from(sz)?,
@@ -983,25 +1128,34 @@ fn basic_fields(m: &mime::NaiveMIME, sz: usize) -> Result<BasicFields> {
/// 4.2.2.2 TEXT/RICHTEXT
/// ```
fn get_message_section<'a>(
- parsed: &'a Message<'a>,
+ parsed: &'a AnyPart<'a>,
section: &Option<FetchSection>,
) -> Result<Cow<'a, [u8]>> {
+ let msg = parsed
+ .as_message()
+ .ok_or(anyhow!("Part must be a message"))?;
match section {
- Some(FetchSection::Text(None)) => {
- Ok(parsed.raw_body.into())
- }
+ Some(FetchSection::Text(None)) => Ok(msg.raw_body.into()),
Some(FetchSection::Text(Some(part))) => {
- map_subpart(parsed.child.as_ref(), part.0.as_slice(), |part_msg| {
- Ok(part_msg.as_message().ok_or(Error::msg("Not a message/rfc822 part while expected by request (xxx.TEXT)"))?
+ map_subpart(parsed, part.0.as_slice(), |part_msg| {
+ Ok(part_msg
+ .as_message()
+ .ok_or(Error::msg(
+ "Not a message/rfc822 part while expected by request (TEXT)",
+ ))?
.raw_body
.into())
})
}
Some(FetchSection::Header(part)) => map_subpart(
- parsed.child.as_ref(),
+ parsed,
part.as_ref().map(|p| p.0.as_slice()).unwrap_or(&[]),
|part_msg| {
- Ok(part_msg.as_message().ok_or(Error::msg("Not a message/rfc822 part while expected by request (xxx.TEXT)"))?
+ Ok(part_msg
+ .as_message()
+ .ok_or(Error::msg(
+ "Not a message/rfc822 part while expected by request (HEADER)",
+ ))?
.raw_headers
.into())
},
@@ -1019,9 +1173,8 @@ fn get_message_section<'a>(
})
.collect::<Vec<_>>();
-
map_subpart(
- parsed.child.as_ref(),
+ parsed,
part.as_ref().map(|p| p.0.as_slice()).unwrap_or(&[]),
|part_msg| {
let mut ret = vec![];
@@ -1030,11 +1183,7 @@ fn get_message_section<'a>(
header::Field::Good(header::Kv2(k, v)) => (k, v),
_ => continue,
};
- if fields
- .as_slice()
- .iter()
- .any(|x| (x == k) ^ invert)
- {
+ if fields.as_slice().iter().any(|x| (x == k) ^ invert) {
ret.extend(*k);
ret.extend(b": ");
ret.extend(*v);
@@ -1046,7 +1195,7 @@ fn get_message_section<'a>(
},
)
}
- Some(FetchSection::Part(part)) => map_subpart(parsed.child.as_ref(), part.0.as_slice(), |part| {
+ Some(FetchSection::Part(part)) => map_subpart(parsed, part.0.as_slice(), |part| {
let bytes = match &part {
AnyPart::Txt(p) => p.body,
AnyPart::Bin(p) => p.body,
@@ -1055,7 +1204,7 @@ fn get_message_section<'a>(
};
Ok(bytes.to_vec().into())
}),
- Some(FetchSection::Mime(part)) => map_subpart(parsed.child.as_ref(), part.0.as_slice(), |part| {
+ Some(FetchSection::Mime(part)) => map_subpart(parsed, part.0.as_slice(), |part| {
let bytes = match &part {
AnyPart::Txt(p) => p.mime.fields.raw,
AnyPart::Bin(p) => p.mime.fields.raw,
@@ -1064,11 +1213,11 @@ fn get_message_section<'a>(
};
Ok(bytes.to_vec().into())
}),
- None => Ok(parsed.raw_part.into()),
+ None => Ok(msg.raw_part.into()),
}
}
-/// Fetch a MIME SubPart
+/// Fetch a MIME SubPart
///
/// eg. FETCH BODY[4.2.2.1] -> [4, 2, 2, 1]
fn map_subpart<'a, F, R>(part: &AnyPart<'a>, path: &[NonZeroU32], f: F) -> Result<R>
@@ -1094,9 +1243,72 @@ where
#[cfg(test)]
mod tests {
use super::*;
+ use crate::cryptoblob;
+ use crate::mail::unique_ident;
use imap_codec::codec::Encode;
+ use imap_codec::types::fetch_attributes::Section;
use std::fs;
+ #[test]
+ fn mailview_body_ext() -> Result<()> {
+ let ap = AttributesProxy::new(
+ &MacroOrFetchAttributes::FetchAttributes(vec![FetchAttribute::BodyExt {
+ section: Some(Section::Header(None)),
+ partial: None,
+ peek: false,
+ }]),
+ false,
+ );
+
+ let flags = vec![];
+ let key = cryptoblob::gen_key();
+ let meta = MailMeta {
+ internaldate: 0u64,
+ headers: vec![],
+ message_key: key,
+ rfc822_size: 8usize,
+ };
+ let ids = MailIdentifiers {
+ i: NonZeroU32::MIN,
+ uid: NonZeroU32::MIN,
+ uuid: unique_ident::gen_ident(),
+ };
+ let rfc822 = b"Subject: hello\r\nFrom: a@a.a\r\nTo: b@b.b\r\nDate: Thu, 12 Oct 2023 08:45:28 +0000\r\n\r\nhello world";
+ let content = FetchedMail::new_from_message(eml_codec::parse_message(rfc822)?.1);
+
+ let mut mv = MailView {
+ ids: &ids,
+ content,
+ meta: &meta,
+ flags: &flags,
+ add_seen: false,
+ };
+ let res_body = mv.filter(&ap)?;
+
+ let fattr = match res_body {
+ Body::Data(Data::Fetch {
+ seq_or_uid: _seq,
+ attributes: attr,
+ }) => Ok(attr),
+ _ => Err(anyhow!("Not a fetch body")),
+ }?;
+
+ assert_eq!(fattr.len(), 1);
+
+ let (sec, _orig, _data) = match &fattr[0] {
+ MessageAttribute::BodyExt {
+ section,
+ origin,
+ data,
+ } => Ok((section, origin, data)),
+ _ => Err(anyhow!("not a body ext message attribute")),
+ }?;
+
+ assert_eq!(sec.as_ref().unwrap(), &Section::Header(None));
+
+ Ok(())
+ }
+
/// Future automated test. We use lossy utf8 conversion + lowercase everything,
/// so this test might allow invalid results. But at least it allows us to quickly test a
/// large variety of emails.
@@ -1113,7 +1325,6 @@ mod tests {
//"tests/emails/dxflrs/0005_mail-parser-readme",
"tests/emails/dxflrs/0006_single-mime",
"tests/emails/dxflrs/0007_raw_msg_in_rfc822",
-
/* *** (STRANGE) RFC *** */
//"tests/emails/rfc/000", // must return text/enriched, we return text/plain
//"tests/emails/rfc/001", // does not recognize the multipart/external-body, breaks the
@@ -1127,9 +1338,8 @@ mod tests {
//"tests/emails/thirdparty/001", // same
"tests/emails/thirdparty/002", // same
-
- /* *** LEGACY *** */
- //"tests/emails/legacy/000", // same issue with \r\r
+ /* *** LEGACY *** */
+ //"tests/emails/legacy/000", // same issue with \r\r
];
for pref in prefixes.iter() {
@@ -1139,7 +1349,9 @@ mod tests {
let message = eml_codec::parse_message(&txt).unwrap().1;
let mut resp = Vec::new();
- MessageAttribute::Body(build_imap_email_struct(&message.child)?).encode(&mut resp).unwrap();
+ MessageAttribute::Body(build_imap_email_struct(&message.child)?)
+ .encode(&mut resp)
+ .unwrap();
let resp_str = String::from_utf8_lossy(&resp).to_lowercase();