aerogramme/src/imap/command/authenticated.rs

use anyhow::{Result, Error, anyhow};
use boitalettres::proto::Response;
use imap_codec::types::command::CommandBody;
use imap_codec::types::core::Tag;
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::session::InnerContext;
use crate::imap::flow;
use crate::mailbox::Mailbox;

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.clone(), mailbox_wildcard.clone()).await,
        CommandBody::List { reference, mailbox_wildcard, } => ctx.list(reference.clone(), mailbox_wildcard.clone()).await,
        CommandBody::Select { mailbox } => ctx.select(mailbox.clone()).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

     * TRACE END ---
     */
    async fn select(&self, mailbox: MailboxCodec) -> Result<(Response, flow::Transition)> {
        let name = String::try_from(mailbox)?;

        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 body = vec![Data::Exists(sum.exists.try_into()?), Data::Recent(0)];

        let tr = flow::Transition::Select(mb);

        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 r_permanentflags = Status::ok(None, Some(Code::

        Ok((vec![
           ImapRes::Data(Data::Exists(0)),
           ImapRes::Data(Data::Recent(0)),
           ImapRes::Data(Data::Flags(vec![])),
           /*ImapRes::Status(),
             ImapRes::Status(),
             ImapRes::Status(),*/
           ImapRes::Status(Status::ok(
               Some(self.tag.clone()),
               Some(Code::ReadWrite),
               "Select completed",
               ).map_err(Error::msg)?),
        ], tr))
    }
}