320 lines
11 KiB
Rust
320 lines
11 KiB
Rust
use std::collections::BTreeMap;
|
|
use std::sync::Arc;
|
|
|
|
use anyhow::{anyhow, Result};
|
|
use boitalettres::proto::res::body::Data as Body;
|
|
use boitalettres::proto::{Request, Response};
|
|
use imap_codec::types::command::{CommandBody, StatusAttribute};
|
|
use imap_codec::types::flag::FlagNameAttribute;
|
|
use imap_codec::types::mailbox::{ListMailbox, Mailbox as MailboxCodec};
|
|
use imap_codec::types::response::{Code, Data, StatusAttributeValue};
|
|
|
|
use crate::imap::command::anonymous;
|
|
use crate::imap::flow;
|
|
use crate::imap::mailbox_view::MailboxView;
|
|
|
|
use crate::mail::user::{User, INBOX, MAILBOX_HIERARCHY_DELIMITER};
|
|
|
|
pub struct AuthenticatedContext<'a> {
|
|
pub req: &'a Request,
|
|
pub user: &'a Arc<User>,
|
|
}
|
|
|
|
pub async fn dispatch<'a>(ctx: AuthenticatedContext<'a>) -> Result<(Response, flow::Transition)> {
|
|
match &ctx.req.command.body {
|
|
CommandBody::Create { mailbox } => ctx.create(mailbox).await,
|
|
CommandBody::Delete { mailbox } => ctx.delete(mailbox).await,
|
|
CommandBody::Rename {
|
|
mailbox,
|
|
new_mailbox,
|
|
} => ctx.rename(mailbox, new_mailbox).await,
|
|
CommandBody::Lsub {
|
|
reference,
|
|
mailbox_wildcard,
|
|
} => ctx.list(reference, mailbox_wildcard, true).await,
|
|
CommandBody::List {
|
|
reference,
|
|
mailbox_wildcard,
|
|
} => ctx.list(reference, mailbox_wildcard, false).await,
|
|
CommandBody::Status {
|
|
mailbox,
|
|
attributes,
|
|
} => ctx.status(mailbox, attributes).await,
|
|
CommandBody::Subscribe { mailbox } => ctx.subscribe(mailbox).await,
|
|
CommandBody::Unsubscribe { mailbox } => ctx.unsubscribe(mailbox).await,
|
|
CommandBody::Select { mailbox } => ctx.select(mailbox).await,
|
|
CommandBody::Examine { mailbox } => ctx.examine(mailbox).await,
|
|
_ => {
|
|
let ctx = anonymous::AnonymousContext {
|
|
req: ctx.req,
|
|
login_provider: None,
|
|
};
|
|
anonymous::dispatch(ctx).await
|
|
}
|
|
}
|
|
}
|
|
|
|
// --- PRIVATE ---
|
|
|
|
impl<'a> AuthenticatedContext<'a> {
|
|
async fn create(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
if name == INBOX {
|
|
return Ok((
|
|
Response::bad("Cannot create INBOX")?,
|
|
flow::Transition::None,
|
|
));
|
|
}
|
|
|
|
match self.user.create_mailbox(&name).await {
|
|
Ok(()) => Ok((Response::ok("CREATE complete")?, flow::Transition::None)),
|
|
Err(e) => Ok((Response::no(&e.to_string())?, flow::Transition::None)),
|
|
}
|
|
}
|
|
|
|
async fn delete(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
match self.user.delete_mailbox(&name).await {
|
|
Ok(()) => Ok((Response::ok("DELETE complete")?, flow::Transition::None)),
|
|
Err(e) => Ok((Response::no(&e.to_string())?, flow::Transition::None)),
|
|
}
|
|
}
|
|
|
|
async fn rename(
|
|
self,
|
|
mailbox: &MailboxCodec,
|
|
new_mailbox: &MailboxCodec,
|
|
) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
let new_name = String::try_from(new_mailbox.clone())?;
|
|
|
|
match self.user.rename_mailbox(&name, &new_name).await {
|
|
Ok(()) => Ok((Response::ok("RENAME complete")?, flow::Transition::None)),
|
|
Err(e) => Ok((Response::no(&e.to_string())?, flow::Transition::None)),
|
|
}
|
|
}
|
|
|
|
async fn list(
|
|
self,
|
|
reference: &MailboxCodec,
|
|
mailbox_wildcard: &ListMailbox,
|
|
is_lsub: bool,
|
|
) -> Result<(Response, flow::Transition)> {
|
|
let reference = String::try_from(reference.clone())?;
|
|
if !reference.is_empty() {
|
|
return Ok((
|
|
Response::bad("References not supported")?,
|
|
flow::Transition::None,
|
|
));
|
|
}
|
|
|
|
let mailboxes = self.user.list_mailboxes().await?;
|
|
let mut vmailboxes = BTreeMap::new();
|
|
for mb in mailboxes.iter() {
|
|
for (i, _) in mb.match_indices(MAILBOX_HIERARCHY_DELIMITER) {
|
|
if i > 0 {
|
|
let smb = &mb[..i];
|
|
if !vmailboxes.contains_key(&smb) {
|
|
vmailboxes.insert(smb, false);
|
|
}
|
|
}
|
|
}
|
|
vmailboxes.insert(mb, true);
|
|
}
|
|
|
|
let wildcard = String::try_from(mailbox_wildcard.clone())?;
|
|
let wildcard_pat = globset::Glob::new(&wildcard)?.compile_matcher();
|
|
|
|
let mut ret = vec![];
|
|
for (mb, is_real) in vmailboxes.iter() {
|
|
if wildcard_pat.is_match(mb) {
|
|
let mailbox = mb
|
|
.to_string()
|
|
.try_into()
|
|
.map_err(|_| anyhow!("invalid mailbox name"))?;
|
|
let mut items = vec![];
|
|
if !*is_real {
|
|
items.push(FlagNameAttribute::Noselect);
|
|
}
|
|
if is_lsub {
|
|
items.push(FlagNameAttribute::Extension(
|
|
"\\Subscribed".try_into().unwrap(),
|
|
));
|
|
ret.push(Data::Lsub {
|
|
items,
|
|
delimiter: Some(MAILBOX_HIERARCHY_DELIMITER),
|
|
mailbox,
|
|
});
|
|
} else {
|
|
ret.push(Data::List {
|
|
items,
|
|
delimiter: Some(MAILBOX_HIERARCHY_DELIMITER),
|
|
mailbox,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok((
|
|
Response::ok("LIST completed")?.with_body(ret),
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
|
|
async fn status(
|
|
self,
|
|
mailbox: &MailboxCodec,
|
|
attributes: &[StatusAttribute],
|
|
) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
let mb_opt = self.user.open_mailbox(&name).await?;
|
|
let mb = match mb_opt {
|
|
Some(mb) => mb,
|
|
None => {
|
|
return Ok((
|
|
Response::no("Mailbox does not exist")?,
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
};
|
|
|
|
let (view, _data) = MailboxView::new(mb).await?;
|
|
|
|
let mut ret_attrs = vec![];
|
|
for attr in attributes.iter() {
|
|
ret_attrs.push(match attr {
|
|
StatusAttribute::Messages => StatusAttributeValue::Messages(view.exists()?),
|
|
StatusAttribute::Unseen => {
|
|
StatusAttributeValue::Unseen(view.unseen().map(|x| x.get()).unwrap_or(0))
|
|
}
|
|
StatusAttribute::Recent => StatusAttributeValue::Recent(view.recent()?),
|
|
StatusAttribute::UidNext => StatusAttributeValue::UidNext(view.uidnext()),
|
|
StatusAttribute::UidValidity => {
|
|
StatusAttributeValue::UidValidity(view.uidvalidity())
|
|
}
|
|
});
|
|
}
|
|
|
|
let data = vec![Body::Data(Data::Status {
|
|
mailbox: mailbox.clone(),
|
|
attributes: ret_attrs,
|
|
})];
|
|
|
|
Ok((
|
|
Response::ok("STATUS completed")?.with_body(data),
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
|
|
async fn subscribe(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
if self.user.has_mailbox(&name).await? {
|
|
Ok((Response::ok("SUBSCRIBE complete")?, flow::Transition::None))
|
|
} else {
|
|
Ok((
|
|
Response::bad(&format!("Mailbox {} does not exist", name))?,
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
}
|
|
|
|
async fn unsubscribe(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
if self.user.has_mailbox(&name).await? {
|
|
Ok((
|
|
Response::bad(&format!(
|
|
"Cannot unsubscribe from mailbox {}: not supported by Aerogramme",
|
|
name
|
|
))?,
|
|
flow::Transition::None,
|
|
))
|
|
} else {
|
|
Ok((
|
|
Response::bad(&format!("Mailbox {} does not exist", name))?,
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
}
|
|
|
|
/*
|
|
* TRACE BEGIN ---
|
|
|
|
|
|
Example: C: A142 SELECT INBOX
|
|
S: * 172 EXISTS
|
|
S: * 1 RECENT
|
|
S: * OK [UNSEEN 12] Message 12 is first unseen
|
|
S: * OK [UIDVALIDITY 3857529045] UIDs valid
|
|
S: * OK [UIDNEXT 4392] Predicted next UID
|
|
S: * FLAGS (\Answered \Flagged \Deleted \Seen \Draft)
|
|
S: * OK [PERMANENTFLAGS (\Deleted \Seen \*)] Limited
|
|
S: A142 OK [READ-WRITE] SELECT completed
|
|
|
|
--- a mailbox with no unseen message -> no unseen entry
|
|
|
|
20 select "INBOX.achats"
|
|
* FLAGS (\Answered \Flagged \Deleted \Seen \Draft $Forwarded JUNK $label1)
|
|
* OK [PERMANENTFLAGS (\Answered \Flagged \Deleted \Seen \Draft $Forwarded JUNK $label1 \*)] Flags permitted.
|
|
* 88 EXISTS
|
|
* 0 RECENT
|
|
* OK [UIDVALIDITY 1347986788] UIDs valid
|
|
* OK [UIDNEXT 91] Predicted next UID
|
|
* OK [HIGHESTMODSEQ 72] Highest
|
|
20 OK [READ-WRITE] Select completed (0.001 + 0.000 secs).
|
|
|
|
* TRACE END ---
|
|
*/
|
|
async fn select(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
let mb_opt = self.user.open_mailbox(&name).await?;
|
|
let mb = match mb_opt {
|
|
Some(mb) => mb,
|
|
None => {
|
|
return Ok((
|
|
Response::no("Mailbox does not exist")?,
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
};
|
|
tracing::info!(username=%self.user.username, mailbox=%name, "mailbox.selected");
|
|
|
|
let (mb, data) = MailboxView::new(mb).await?;
|
|
|
|
Ok((
|
|
Response::ok("Select completed")?
|
|
.with_extra_code(Code::ReadWrite)
|
|
.with_body(data),
|
|
flow::Transition::Select(mb),
|
|
))
|
|
}
|
|
|
|
async fn examine(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
let name = String::try_from(mailbox.clone())?;
|
|
|
|
let mb_opt = self.user.open_mailbox(&name).await?;
|
|
let mb = match mb_opt {
|
|
Some(mb) => mb,
|
|
None => {
|
|
return Ok((
|
|
Response::no("Mailbox does not exist")?,
|
|
flow::Transition::None,
|
|
))
|
|
}
|
|
};
|
|
tracing::info!(username=%self.user.username, mailbox=%name, "mailbox.examined");
|
|
|
|
let (mb, data) = MailboxView::new(mb).await?;
|
|
|
|
Ok((
|
|
Response::ok("Examine completed")?
|
|
.with_extra_code(Code::ReadOnly)
|
|
.with_body(data),
|
|
flow::Transition::Examine(mb),
|
|
))
|
|
}
|
|
}
|