aerogramme/src/imap/command/authenticated.rs

156 lines
5 KiB
Rust
Raw Normal View History

2022-06-29 10:52:58 +00:00
use anyhow::{Error, Result};
2022-06-29 10:50:44 +00:00
use boitalettres::proto::{res::body::Data as Body, Request, Response};
2022-06-20 16:09:20 +00:00
use imap_codec::types::command::CommandBody;
2022-06-28 07:34:24 +00:00
use imap_codec::types::core::Atom;
2022-06-27 09:40:45 +00:00
use imap_codec::types::flag::Flag;
2022-06-20 16:09:20 +00:00
use imap_codec::types::mailbox::{ListMailbox, Mailbox as MailboxCodec};
2022-06-29 10:52:58 +00:00
use imap_codec::types::response::{Code, Data, Status};
2022-06-20 16:09:20 +00:00
use crate::imap::command::anonymous;
2022-06-22 14:52:38 +00:00
use crate::imap::flow;
2022-06-29 10:52:58 +00:00
2022-06-29 11:16:58 +00:00
use crate::mail::mailbox::Mailbox;
use crate::mail::user::User;
2022-06-20 16:09:20 +00:00
2022-06-27 09:40:45 +00:00
const DEFAULT_FLAGS: [Flag; 5] = [
2022-06-28 08:49:28 +00:00
Flag::Seen,
Flag::Answered,
Flag::Flagged,
Flag::Deleted,
Flag::Draft,
2022-06-27 09:40:45 +00:00
];
2022-06-29 10:50:44 +00:00
pub struct AuthenticatedContext<'a> {
pub req: &'a Request,
2022-06-29 11:16:58 +00:00
pub user: &'a User,
2022-06-29 10:50:44 +00:00
}
2022-06-20 16:09:20 +00:00
2022-06-29 10:50:44 +00:00
pub async fn dispatch<'a>(ctx: AuthenticatedContext<'a>) -> Result<(Response, flow::Transition)> {
match &ctx.req.command.body {
2022-06-22 15:26:52 +00:00
CommandBody::Lsub {
reference,
mailbox_wildcard,
} => ctx.lsub(reference, mailbox_wildcard).await,
CommandBody::List {
reference,
mailbox_wildcard,
} => ctx.list(reference, mailbox_wildcard).await,
2022-06-22 15:25:04 +00:00
CommandBody::Select { mailbox } => ctx.select(mailbox).await,
2022-06-29 10:50:44 +00:00
_ => {
let ctx = anonymous::AnonymousContext {
req: ctx.req,
login_provider: None,
};
anonymous::dispatch(ctx).await
}
2022-06-20 16:09:20 +00:00
}
2022-06-22 14:52:38 +00:00
}
2022-06-20 16:09:20 +00:00
// --- PRIVATE ---
2022-06-29 10:50:44 +00:00
impl<'a> AuthenticatedContext<'a> {
2022-06-20 16:09:20 +00:00
async fn lsub(
2022-06-29 10:50:44 +00:00
self,
2022-06-29 10:52:58 +00:00
_reference: &MailboxCodec,
_mailbox_wildcard: &ListMailbox,
2022-06-22 15:26:52 +00:00
) -> Result<(Response, flow::Transition)> {
2022-06-29 10:50:44 +00:00
Ok((Response::bad("Not implemented")?, flow::Transition::None))
2022-06-17 16:39:36 +00:00
}
2022-06-20 16:09:20 +00:00
async fn list(
2022-06-29 10:50:44 +00:00
self,
2022-06-29 10:52:58 +00:00
_reference: &MailboxCodec,
_mailbox_wildcard: &ListMailbox,
2022-06-22 15:26:52 +00:00
) -> Result<(Response, flow::Transition)> {
2022-06-29 10:50:44 +00:00
Ok((Response::bad("Not implemented")?, flow::Transition::None))
2022-06-17 16:39:36 +00:00
}
/*
2022-06-22 15:26:52 +00:00
* TRACE BEGIN ---
2022-06-20 16:09:20 +00:00
2022-06-22 15:26:52 +00:00
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
2022-06-20 16:09:20 +00:00
2022-06-23 12:41:10 +00:00
--- 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).
2022-06-22 15:26:52 +00:00
* TRACE END ---
*/
2022-06-29 10:50:44 +00:00
async fn select(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
2022-06-22 15:25:04 +00:00
let name = String::try_from(mailbox.clone())?;
2022-06-17 16:39:36 +00:00
2022-06-29 11:41:05 +00:00
let mut mb = self.user.open_mailbox(name)?;
2022-06-29 11:16:58 +00:00
tracing::info!(username=%self.user.username, mailbox=%name, "mailbox.selected");
2022-06-17 16:39:36 +00:00
let sum = mb.summary().await?;
tracing::trace!(summary=%sum, "mailbox.summary");
2022-06-28 07:34:24 +00:00
let mut res = Vec::<Body>::new();
2022-06-23 12:41:10 +00:00
2022-06-28 07:34:24 +00:00
res.push(Body::Data(Data::Exists(sum.exists)));
2022-06-24 08:27:26 +00:00
2022-06-28 07:34:24 +00:00
res.push(Body::Data(Data::Recent(sum.recent)));
2022-06-24 08:27:26 +00:00
2022-06-27 09:40:45 +00:00
let mut flags: Vec<Flag> = sum.flags.map(|f| match f.chars().next() {
Some('\\') => None,
Some('$') if f == "$unseen" => None,
Some(_) => match Atom::try_from(f.clone()) {
Err(_) => {
2022-06-29 11:16:58 +00:00
tracing::error!(username=%self.user.username, mailbox=%name, flag=%f, "Unable to encode flag as IMAP atom");
2022-06-27 09:40:45 +00:00
None
},
Ok(a) => Some(Flag::Keyword(a)),
},
None => None,
}).flatten().collect();
flags.extend_from_slice(&DEFAULT_FLAGS);
2022-06-28 07:34:24 +00:00
res.push(Body::Data(Data::Flags(flags.clone())));
2022-06-24 08:27:26 +00:00
2022-06-28 08:49:28 +00:00
let uid_validity = Status::ok(None, Some(Code::UidValidity(sum.validity)), "UIDs valid")
2022-06-24 08:27:26 +00:00
.map_err(Error::msg)?;
2022-06-28 07:34:24 +00:00
res.push(Body::Status(uid_validity));
2022-06-24 08:27:26 +00:00
2022-06-28 08:49:28 +00:00
let next_uid = Status::ok(None, Some(Code::UidNext(sum.next)), "Predict next UID")
.map_err(Error::msg)?;
2022-06-28 07:34:24 +00:00
res.push(Body::Status(next_uid));
2022-06-24 08:27:26 +00:00
if let Some(unseen) = sum.unseen {
2022-06-28 08:49:28 +00:00
let status_unseen =
Status::ok(None, Some(Code::Unseen(unseen.clone())), "First unseen UID")
.map_err(Error::msg)?;
2022-06-28 07:34:24 +00:00
res.push(Body::Status(status_unseen));
2022-06-24 08:27:26 +00:00
}
2022-06-27 09:40:45 +00:00
flags.push(Flag::Permanent);
2022-06-28 08:49:28 +00:00
let permanent_flags =
Status::ok(None, Some(Code::PermanentFlags(flags)), "Flags permitted")
.map_err(Error::msg)?;
2022-06-28 07:34:24 +00:00
res.push(Body::Status(permanent_flags));
2022-06-24 08:27:26 +00:00
2022-06-28 07:34:24 +00:00
Ok((
2022-06-28 13:38:06 +00:00
Response::ok("Select completed")?
.with_extra_code(Code::ReadWrite)
.with_body(res),
2022-06-28 07:34:24 +00:00
flow::Transition::Select(mb),
))
2022-06-17 16:39:36 +00:00
}
2022-06-20 16:09:20 +00:00
}