aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/imap/command/authenticated.rs131
-rw-r--r--src/mail/user.rs51
2 files changed, 145 insertions, 37 deletions
diff --git a/src/imap/command/authenticated.rs b/src/imap/command/authenticated.rs
index 0b34223..f46dfb4 100644
--- a/src/imap/command/authenticated.rs
+++ b/src/imap/command/authenticated.rs
@@ -1,16 +1,18 @@
+use std::collections::BTreeMap;
use std::sync::Arc;
-use anyhow::Result;
+use anyhow::{anyhow, Result};
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;
+use imap_codec::types::response::{Code, Data};
use crate::imap::command::anonymous;
use crate::imap::flow;
use crate::imap::mailbox_view::MailboxView;
-use crate::mail::user::User;
+use crate::mail::user::{User, INBOX, MAILBOX_HIERARCHY_DELIMITER};
pub struct AuthenticatedContext<'a> {
pub req: &'a Request,
@@ -28,15 +30,17 @@ pub async fn dispatch<'a>(ctx: AuthenticatedContext<'a>) -> Result<(Response, fl
CommandBody::Lsub {
reference,
mailbox_wildcard,
- } => ctx.lsub(reference, mailbox_wildcard).await,
+ } => ctx.list(reference, mailbox_wildcard, true).await,
CommandBody::List {
reference,
mailbox_wildcard,
- } => ctx.list(reference, mailbox_wildcard).await,
+ } => 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,
_ => {
@@ -53,11 +57,28 @@ pub async fn dispatch<'a>(ctx: AuthenticatedContext<'a>) -> Result<(Response, fl
impl<'a> AuthenticatedContext<'a> {
async fn create(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ 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)> {
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ 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(
@@ -65,23 +86,80 @@ impl<'a> AuthenticatedContext<'a> {
mailbox: &MailboxCodec,
new_mailbox: &MailboxCodec,
) -> Result<(Response, flow::Transition)> {
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
- }
+ let name = String::try_from(mailbox.clone())?;
+ let new_name = String::try_from(new_mailbox.clone())?;
- async fn lsub(
- self,
- _reference: &MailboxCodec,
- _mailbox_wildcard: &ListMailbox,
- ) -> Result<(Response, flow::Transition)> {
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ 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,
+ reference: &MailboxCodec,
+ mailbox_wildcard: &ListMailbox,
+ is_lsub: bool,
) -> Result<(Response, flow::Transition)> {
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ 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![];
+ if !*is_real {
+ items.push(FlagNameAttribute::Noselect);
+ }
+ if is_lsub {
+ items.push(FlagNameAttribute::Extension(
+ "\\Subscribed".try_into().unwrap(),
+ ));
+ 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(
@@ -89,9 +167,26 @@ impl<'a> AuthenticatedContext<'a> {
mailbox: &MailboxCodec,
attributes: &[StatusAttribute],
) -> Result<(Response, flow::Transition)> {
+ let name = String::try_from(mailbox.clone())?;
+
Ok((Response::bad("Not implemented")?, flow::Transition::None))
}
+ async fn subscribe(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
+ let name = String::try_from(mailbox.clone())?;
+
+ Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ }
+
+ async fn unsubscribe(self, mailbox: &MailboxCodec) -> Result<(Response, flow::Transition)> {
+ let name = String::try_from(mailbox.clone())?;
+
+ Ok((
+ Response::bad("Aerogramme does not support unsubscribing from a mailbox")?,
+ flow::Transition::None,
+ ))
+ }
+
/*
* TRACE BEGIN ---
diff --git a/src/mail/user.rs b/src/mail/user.rs
index c2d1d85..aeec630 100644
--- a/src/mail/user.rs
+++ b/src/mail/user.rs
@@ -15,7 +15,7 @@ use crate::mail::uidindex::ImapUidvalidity;
use crate::mail::unique_ident::{gen_ident, UniqueIdent};
use crate::time::now_msec;
-const MAILBOX_HIERARCHY_DELIMITER: &str = "/";
+pub const MAILBOX_HIERARCHY_DELIMITER: char = '.';
/// INBOX is the only mailbox that must always exist.
/// It is created automatically when the account is created.
@@ -25,7 +25,7 @@ const MAILBOX_HIERARCHY_DELIMITER: &str = "/";
/// In our implementation, we indeed move the underlying mailbox
/// to the new name (i.e. the new name has the same id as the previous
/// INBOX), and we create a new empty mailbox for INBOX.
-const INBOX: &str = "INBOX";
+pub const INBOX: &str = "INBOX";
const MAILBOX_LIST_PK: &str = "mailboxes";
const MAILBOX_LIST_SK: &str = "list";
@@ -94,20 +94,24 @@ impl User {
/// Creates a new mailbox in the user's IMAP namespace.
pub async fn create_mailbox(&self, name: &str) -> Result<()> {
let (mut list, ct) = self.load_mailbox_list().await?;
- match self.create_mailbox_internal(&mut list, ct, name).await? {
+ match self.mblist_create_mailbox(&mut list, ct, name).await? {
CreatedMailbox::Created(_, _) => Ok(()),
CreatedMailbox::Existed(_, _) => Err(anyhow!("Mailbox {} already exists", name)),
}
}
/// Deletes a mailbox in the user's IMAP namespace.
- pub fn delete_mailbox(&self, _name: &str) -> Result<()> {
- unimplemented!()
+ pub async fn delete_mailbox(&self, _name: &str) -> Result<()> {
+ bail!("Deleting mailboxes not implemented yet")
}
/// Renames a mailbox in the user's IMAP namespace.
- pub fn rename_mailbox(&self, _old_name: &str, _new_name: &str) -> Result<()> {
- unimplemented!()
+ pub async fn rename_mailbox(&self, old_name: &str, new_name: &str) -> Result<()> {
+ if old_name == INBOX {
+ bail!("Renaming INBOX not implemented yet")
+ } else {
+ bail!("Renaming not implemented yet")
+ }
}
// ---- Internal user & mailbox management ----
@@ -180,22 +184,31 @@ impl User {
}
};
+ self.ensure_inbox_exists(&mut list, &ct).await?;
+
+ Ok((list, ct))
+ }
+
+ async fn ensure_inbox_exists(
+ &self,
+ list: &mut MailboxList,
+ ct: &Option<CausalityToken>,
+ ) -> Result<()> {
// If INBOX doesn't exist, create a new mailbox with that name
// and save new mailbox list.
// Also, ensure that the mpsc::watch that keeps track of the
// inbox id is up-to-date.
- let (inbox_id, inbox_uidvalidity) = match self
- .create_mailbox_internal(&mut list, ct.clone(), INBOX)
- .await?
- {
- CreatedMailbox::Created(i, v) => (i, v),
- CreatedMailbox::Existed(i, v) => (i, v),
- };
- self.tx_inbox_id
- .send(Some((inbox_id, inbox_uidvalidity)))
- .unwrap();
+ let (inbox_id, inbox_uidvalidity) =
+ match self.mblist_create_mailbox(list, ct.clone(), INBOX).await? {
+ CreatedMailbox::Created(i, v) => (i, v),
+ CreatedMailbox::Existed(i, v) => (i, v),
+ };
+ let inbox_id = Some((inbox_id, inbox_uidvalidity));
+ if *self.tx_inbox_id.borrow() != inbox_id {
+ self.tx_inbox_id.send(inbox_id).unwrap();
+ }
- Ok((list, ct))
+ Ok(())
}
async fn save_mailbox_list(
@@ -210,7 +223,7 @@ impl User {
Ok(())
}
- async fn create_mailbox_internal(
+ async fn mblist_create_mailbox(
&self,
list: &mut MailboxList,
ct: Option<CausalityToken>,