2022-06-22 15:26:52 +00:00
|
|
|
use anyhow::{anyhow, Error, Result};
|
2022-06-28 08:49:28 +00:00
|
|
|
use boitalettres::proto::{res::body::Data as Body, 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};
|
|
|
|
|
use imap_codec::types::response::{Code, Data, Response as ImapRes, Status};
|
|
|
|
|
|
|
|
|
|
use crate::imap::command::anonymous;
|
2022-06-22 14:52:38 +00:00
|
|
|
use crate::imap::flow;
|
2022-06-22 15:26:52 +00:00
|
|
|
use crate::imap::session::InnerContext;
|
2022-06-27 14:56:20 +00:00
|
|
|
use crate::mail::Mailbox;
|
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-22 15:26:52 +00:00
|
|
|
pub async fn dispatch<'a>(
|
|
|
|
|
inner: InnerContext<'a>,
|
|
|
|
|
user: &'a flow::User,
|
|
|
|
|
) -> Result<(Response, flow::Transition)> {
|
2022-06-28 07:34:24 +00:00
|
|
|
let ctx = StateContext { user, inner };
|
2022-06-20 16:09:20 +00:00
|
|
|
|
2022-06-28 07:34:24 +00:00
|
|
|
match &ctx.inner.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-20 16:09:20 +00:00
|
|
|
_ => anonymous::dispatch(ctx.inner).await,
|
|
|
|
|
}
|
2022-06-22 14:52:38 +00:00
|
|
|
}
|
2022-06-20 16:09:20 +00:00
|
|
|
|
|
|
|
|
// --- PRIVATE ---
|
|
|
|
|
|
|
|
|
|
struct StateContext<'a> {
|
2022-06-22 14:52:38 +00:00
|
|
|
inner: InnerContext<'a>,
|
|
|
|
|
user: &'a flow::User,
|
2022-06-17 16:39:36 +00:00
|
|
|
}
|
|
|
|
|
|
2022-06-20 16:09:20 +00:00
|
|
|
impl<'a> StateContext<'a> {
|
|
|
|
|
async fn lsub(
|
2022-06-17 16:39:36 +00:00
|
|
|
&self,
|
2022-06-22 15:25:04 +00:00
|
|
|
reference: &MailboxCodec,
|
|
|
|
|
mailbox_wildcard: &ListMailbox,
|
2022-06-22 15:26:52 +00:00
|
|
|
) -> Result<(Response, flow::Transition)> {
|
2022-06-28 08:49:28 +00:00
|
|
|
Ok((Response::bad("Not implemented")?, flow::Transition::No))
|
2022-06-17 16:39:36 +00:00
|
|
|
}
|
|
|
|
|
|
2022-06-20 16:09:20 +00:00
|
|
|
async fn list(
|
2022-06-17 16:39:36 +00:00
|
|
|
&self,
|
2022-06-22 15:25:04 +00:00
|
|
|
reference: &MailboxCodec,
|
|
|
|
|
mailbox_wildcard: &ListMailbox,
|
2022-06-22 15:26:52 +00:00
|
|
|
) -> Result<(Response, flow::Transition)> {
|
2022-06-28 08:49:28 +00:00
|
|
|
Ok((Response::bad("Not implemented")?, flow::Transition::No))
|
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-22 15:25:04 +00:00
|
|
|
async fn select(&self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
|
|
|
|
|
let name = String::try_from(mailbox.clone())?;
|
2022-06-17 16:39:36 +00:00
|
|
|
|
2022-06-22 12:58:57 +00:00
|
|
|
let mut mb = Mailbox::new(&self.user.creds, name.clone())?;
|
2022-06-20 16:09:20 +00:00
|
|
|
tracing::info!(username=%self.user.name, 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(_) => {
|
|
|
|
|
tracing::error!(username=%self.user.name, mailbox=%name, flag=%f, "Unable to encode flag as IMAP atom");
|
|
|
|
|
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((
|
|
|
|
|
Response::ok("Select completed")?.with_body(res),
|
|
|
|
|
flow::Transition::Select(mb),
|
|
|
|
|
))
|
2022-06-17 16:39:36 +00:00
|
|
|
}
|
2022-06-20 16:09:20 +00:00
|
|
|
}
|
