aerogramme/src/imap/command/authenticated.rs

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![FlagNameAttribute::Extension(
                    "Subscribed".try_into().unwrap(),
                )];
                if !*is_real {
                    items.push(FlagNameAttribute::Noselect);
                }
                if is_lsub {
                    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),
        ))
    }
}