use anyhow::{anyhow, Error, Result}; use boitalettres::proto::Response; use imap_codec::types::command::CommandBody; use imap_codec::types::core::{Atom, Tag}; use imap_codec::types::flag::Flag; 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; use crate::imap::flow; use crate::imap::session::InnerContext; use crate::mailbox::Mailbox; const DEFAULT_FLAGS: [Flag; 5] = [ Flag::Seen, Flag::Answered, Flag::Flagged, Flag::Deleted, Flag::Draft, ]; pub async fn dispatch<'a>( inner: InnerContext<'a>, user: &'a flow::User, ) -> Result<(Response, flow::Transition)> { let ctx = StateContext { user, tag: &inner.req.tag, inner, }; match &ctx.inner.req.body { CommandBody::Lsub { reference, mailbox_wildcard, } => ctx.lsub(reference, mailbox_wildcard).await, CommandBody::List { reference, mailbox_wildcard, } => ctx.list(reference, mailbox_wildcard).await, CommandBody::Select { mailbox } => ctx.select(mailbox).await, _ => anonymous::dispatch(ctx.inner).await, } } // --- PRIVATE --- struct StateContext<'a> { inner: InnerContext<'a>, user: &'a flow::User, tag: &'a Tag, } impl<'a> StateContext<'a> { async fn lsub( &self, reference: &MailboxCodec, mailbox_wildcard: &ListMailbox, ) -> Result<(Response, flow::Transition)> { Ok(( vec![ImapRes::Status( Status::bad(Some(self.tag.clone()), None, "Not implemented").map_err(Error::msg)?, )], flow::Transition::No, )) } async fn list( &self, reference: &MailboxCodec, mailbox_wildcard: &ListMailbox, ) -> Result<(Response, flow::Transition)> { Ok(( vec![ImapRes::Status( Status::bad(Some(self.tag.clone()), None, "Not implemented").map_err(Error::msg)?, )], flow::Transition::No, )) } /* * 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 mut mb = Mailbox::new(&self.user.creds, name.clone())?; tracing::info!(username=%self.user.name, mailbox=%name, "mailbox.selected"); let sum = mb.summary().await?; tracing::trace!(summary=%sum, "mailbox.summary"); let r_unseen = Status::ok( None, Some(Code::Unseen( std::num::NonZeroU32::new(1).ok_or(anyhow!("Invalid message identifier"))?, )), "First unseen UID", ) .map_err(Error::msg)?; let mut res = Vec::::new(); res.push(ImapRes::Data(Data::Exists(sum.exists))); res.push(ImapRes::Data(Data::Recent(sum.recent))); let mut flags: Vec = 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); res.push(ImapRes::Data(Data::Flags(flags.clone()))); let uid_validity = Status::ok( None, Some(Code::UidValidity(sum.validity)), "UIDs valid" ) .map_err(Error::msg)?; res.push(ImapRes::Status(uid_validity)); let next_uid = Status::ok( None, Some(Code::UidNext(sum.next)), "Predict next UID" ).map_err(Error::msg)?; res.push(ImapRes::Status(next_uid)); if let Some(unseen) = sum.unseen { let status_unseen = Status::ok( None, Some(Code::Unseen(unseen.clone())), "First unseen UID", ) .map_err(Error::msg)?; res.push(ImapRes::Status(status_unseen)); } flags.push(Flag::Permanent); let permanent_flags = Status::ok( None, Some(Code::PermanentFlags(flags)), "Flags permitted", ).map_err(Error::msg)?; res.push(ImapRes::Status(permanent_flags)); let last = Status::ok( Some(self.tag.clone()), Some(Code::ReadWrite), "Select completed", ).map_err(Error::msg)?; res.push(ImapRes::Status(last)); let tr = flow::Transition::Select(mb); Ok((res, tr)) } }