aboutsummaryrefslogtreecommitdiff
path: root/src/imap
diff options
context:
space:
mode:
Diffstat (limited to 'src/imap')
-rw-r--r--src/imap/command/authenticated.rs75
-rw-r--r--src/imap/mailbox_view.rs76
2 files changed, 110 insertions, 41 deletions
diff --git a/src/imap/command/authenticated.rs b/src/imap/command/authenticated.rs
index e7198ee..bc9fddf 100644
--- a/src/imap/command/authenticated.rs
+++ b/src/imap/command/authenticated.rs
@@ -2,11 +2,12 @@ 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};
+use imap_codec::types::response::{Code, Data, StatusAttributeValue};
use crate::imap::command::anonymous;
use crate::imap::flow;
@@ -165,26 +166,78 @@ impl<'a> AuthenticatedContext<'a> {
async fn status(
self,
mailbox: &MailboxCodec,
- _attributes: &[StatusAttribute],
+ attributes: &[StatusAttribute],
) -> Result<(Response, flow::Transition)> {
- let _name = String::try_from(mailbox.clone())?;
+ 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::bad("Not implemented")?, flow::Transition::None))
+ 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())?;
+ let name = String::try_from(mailbox.clone())?;
- Ok((Response::bad("Not implemented")?, flow::Transition::None))
+ 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())?;
+ let name = String::try_from(mailbox.clone())?;
- Ok((
- Response::bad("Aerogramme does not support unsubscribing from a mailbox")?,
- flow::Transition::None,
- ))
+ 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,
+ ))
+ }
}
/*
diff --git a/src/imap/mailbox_view.rs b/src/imap/mailbox_view.rs
index 0adba18..db6f490 100644
--- a/src/imap/mailbox_view.rs
+++ b/src/imap/mailbox_view.rs
@@ -19,7 +19,7 @@ use imap_codec::types::sequence::{self, SequenceSet};
use mail_parser::*;
use crate::mail::mailbox::Mailbox;
-use crate::mail::uidindex::UidIndex;
+use crate::mail::uidindex::{ImapUid, ImapUidvalidity, UidIndex};
const DEFAULT_FLAGS: [Flag; 5] = [
Flag::Seen,
@@ -55,12 +55,12 @@ impl MailboxView {
};
let mut data = Vec::<Body>::new();
- data.push(new_view.exists()?);
- data.push(new_view.recent()?);
- data.extend(new_view.flags()?.into_iter());
- data.push(new_view.uidvalidity()?);
- data.push(new_view.uidnext()?);
- if let Some(unseen) = new_view.unseen()? {
+ data.push(new_view.exists_status()?);
+ data.push(new_view.recent_status()?);
+ data.extend(new_view.flags_status()?.into_iter());
+ data.push(new_view.uidvalidity_status()?);
+ data.push(new_view.uidnext_status()?);
+ if let Some(unseen) = new_view.unseen_status()? {
data.push(unseen);
}
@@ -87,9 +87,9 @@ impl MailboxView {
if new_view.known_state.uidvalidity != self.known_state.uidvalidity {
// TODO: do we want to push less/more info than this?
- data.push(new_view.uidvalidity()?);
- data.push(new_view.exists()?);
- data.push(new_view.uidnext()?);
+ data.push(new_view.uidvalidity_status()?);
+ data.push(new_view.exists_status()?);
+ data.push(new_view.uidnext_status()?);
} else {
// Calculate diff between two mailbox states
// See example in IMAP RFC in section on NOOP command:
@@ -117,7 +117,7 @@ impl MailboxView {
// - if new mails arrived, notify client of number of existing mails
if new_view.known_state.table.len() != self.known_state.table.len() - n_expunge {
- data.push(new_view.exists()?);
+ data.push(new_view.exists_status()?);
}
// - if flags changed for existing mails, tell client
@@ -315,37 +315,39 @@ impl MailboxView {
// ----
/// Produce an OK [UIDVALIDITY _] message corresponding to `known_state`
- fn uidvalidity(&self) -> Result<Body> {
+ fn uidvalidity_status(&self) -> Result<Body> {
let uid_validity = Status::ok(
None,
- Some(Code::UidValidity(self.known_state.uidvalidity)),
+ Some(Code::UidValidity(self.uidvalidity())),
"UIDs valid",
)
.map_err(Error::msg)?;
Ok(Body::Status(uid_validity))
}
+ pub(crate) fn uidvalidity(&self) -> ImapUidvalidity {
+ self.known_state.uidvalidity
+ }
+
/// Produce an OK [UIDNEXT _] message corresponding to `known_state`
- fn uidnext(&self) -> Result<Body> {
+ fn uidnext_status(&self) -> Result<Body> {
let next_uid = Status::ok(
None,
- Some(Code::UidNext(self.known_state.uidnext)),
+ Some(Code::UidNext(self.uidnext())),
"Predict next UID",
)
.map_err(Error::msg)?;
Ok(Body::Status(next_uid))
}
+ pub(crate) fn uidnext(&self) -> ImapUid {
+ self.known_state.uidnext
+ }
+
/// Produces an UNSEEN message (if relevant) corresponding to the
/// first unseen message id in `known_state`
- fn unseen(&self) -> Result<Option<Body>> {
- let unseen = self
- .known_state
- .idx_by_flag
- .get(&"$unseen".to_string())
- .and_then(|os| os.get_min())
- .cloned();
- if let Some(unseen) = unseen {
+ fn unseen_status(&self) -> Result<Option<Body>> {
+ if let Some(unseen) = self.unseen() {
let status_unseen =
Status::ok(None, Some(Code::Unseen(unseen.clone())), "First unseen UID")
.map_err(Error::msg)?;
@@ -355,29 +357,43 @@ impl MailboxView {
}
}
+ pub(crate) fn unseen(&self) -> Option<ImapUid> {
+ self.known_state
+ .idx_by_flag
+ .get(&"$unseen".to_string())
+ .and_then(|os| os.get_min())
+ .cloned()
+ }
+
/// Produce an EXISTS message corresponding to the number of mails
/// in `known_state`
- fn exists(&self) -> Result<Body> {
- let exists = u32::try_from(self.known_state.idx_by_uid.len())?;
- Ok(Body::Data(Data::Exists(exists)))
+ fn exists_status(&self) -> Result<Body> {
+ Ok(Body::Data(Data::Exists(self.exists()?)))
+ }
+
+ pub(crate) fn exists(&self) -> Result<u32> {
+ Ok(u32::try_from(self.known_state.idx_by_uid.len())?)
}
/// Produce a RECENT message corresponding to the number of
/// recent mails in `known_state`
- fn recent(&self) -> Result<Body> {
+ fn recent_status(&self) -> Result<Body> {
+ Ok(Body::Data(Data::Recent(self.recent()?)))
+ }
+
+ pub(crate) fn recent(&self) -> Result<u32> {
let recent = self
.known_state
.idx_by_flag
.get(&"\\Recent".to_string())
.map(|os| os.len())
.unwrap_or(0);
- let recent = u32::try_from(recent)?;
- Ok(Body::Data(Data::Recent(recent)))
+ Ok(u32::try_from(recent)?)
}
/// Produce a FLAGS and a PERMANENTFLAGS message that indicates
/// the flags that are in `known_state` + default flags
- fn flags(&self) -> Result<Vec<Body>> {
+ fn flags_status(&self) -> Result<Vec<Body>> {
let mut flags: Vec<Flag> = self
.known_state
.idx_by_flag
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-16 05:04:55 +0000