aboutsummaryrefslogtreecommitdiff
path: root/src/imap/mailbox_view.rs
diff options
context:
space:
mode:
authorAlex Auvolat <alex@adnab.me>2022-06-29 17:58:31 +0200
committerAlex Auvolat <alex@adnab.me>2022-06-29 17:58:31 +0200
commita8d0e4a994daca39f9619cddf2847c1a7820c040 (patch)
tree329dc1919fb0c7fc6c6531e9f53afbde7f83a542 /src/imap/mailbox_view.rs
parent733c59ed8060f8903d723635ed2e7c13ccca7974 (diff)
downloadaerogramme-a8d0e4a994daca39f9619cddf2847c1a7820c040.tar.gz
aerogramme-a8d0e4a994daca39f9619cddf2847c1a7820c040.zip
Implement IDLE in selected state
Diffstat (limited to 'src/imap/mailbox_view.rs')
-rw-r--r--src/imap/mailbox_view.rs104
1 files changed, 91 insertions, 13 deletions
diff --git a/src/imap/mailbox_view.rs b/src/imap/mailbox_view.rs
index 76ca80c..6066528 100644
--- a/src/imap/mailbox_view.rs
+++ b/src/imap/mailbox_view.rs
@@ -1,10 +1,11 @@
+use std::num::NonZeroU32;
use std::sync::Arc;
use anyhow::{Error, Result};
use boitalettres::proto::res::body::Data as Body;
use imap_codec::types::core::Atom;
use imap_codec::types::flag::Flag;
-use imap_codec::types::response::{Code, Data, Status};
+use imap_codec::types::response::{Code, Data, MessageAttribute, Status};
use crate::mail::mailbox::Mailbox;
use crate::mail::uidindex::UidIndex;
@@ -58,6 +59,79 @@ impl MailboxView {
Ok((new_view, data))
}
+ /// Looks up state changes in the mailbox and produces a set of IMAP
+ /// responses describing the new state.
+ pub async fn update(&mut self) -> Result<Vec<Body>> {
+ self.mailbox.sync().await?;
+ // TODO THIS IS JUST A TEST REMOVE LATER
+ self.mailbox.test().await?;
+
+ let new_view = MailboxView {
+ mailbox: self.mailbox.clone(),
+ known_state: self.mailbox.current_uid_index().await,
+ };
+
+ let mut data = Vec::<Body>::new();
+
+ 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()?);
+ } else {
+ // Calculate diff between two mailbox states
+ // See example in IMAP RFC in section on NOOP command:
+ // we want to produce something like this:
+ // C: a047 NOOP
+ // S: * 22 EXPUNGE
+ // S: * 23 EXISTS
+ // S: * 14 FETCH (UID 1305 FLAGS (\Seen \Deleted))
+ // S: a047 OK Noop completed
+ // In other words:
+ // - notify client of expunged mails
+ // - if new mails arrived, notify client of number of existing mails
+ // - if flags changed for existing mails, tell client
+
+ // - notify client of expunged mails
+ let mut n_expunge = 0;
+ for (i, (uid, uuid)) in self.known_state.idx_by_uid.iter().enumerate() {
+ if !new_view.known_state.table.contains_key(uuid) {
+ data.push(Body::Data(Data::Expunge(
+ NonZeroU32::try_from((i + 1 - n_expunge) as u32).unwrap(),
+ )));
+ n_expunge += 1;
+ }
+ }
+
+ // - 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()?);
+ }
+
+ // - if flags changed for existing mails, tell client
+ for (i, (uid, uuid)) in new_view.known_state.idx_by_uid.iter().enumerate() {
+ let old_mail = self.known_state.table.get(uuid);
+ let new_mail = new_view.known_state.table.get(uuid);
+ if old_mail.is_some() && old_mail != new_mail {
+ if let Some((uid, flags)) = new_mail {
+ data.push(Body::Data(Data::Fetch {
+ seq_or_uid: NonZeroU32::try_from((i + 1) as u32).unwrap(),
+ attributes: vec![
+ MessageAttribute::Uid((*uid).try_into().unwrap()),
+ MessageAttribute::Flags(
+ flags.iter().filter_map(|f| string_to_flag(f)).collect(),
+ ),
+ ],
+ }));
+ }
+ }
+ }
+ }
+
+ *self = new_view;
+ Ok(data)
+ }
+
// ----
/// Produce an OK [UIDVALIDITY _] message corresponding to `known_state`
@@ -128,18 +202,7 @@ impl MailboxView {
.known_state
.idx_by_flag
.flags()
- .map(|f| match f.chars().next() {
- Some('\\') => None,
- Some('$') if f == "$unseen" => None,
- Some(_) => match Atom::try_from(f.clone()) {
- Err(_) => {
- tracing::error!(flag=%f, "Unable to encode flag as IMAP atom");
- None
- }
- Ok(a) => Some(Flag::Keyword(a)),
- },
- None => None,
- })
+ .map(|f| string_to_flag(f))
.flatten()
.collect();
for f in DEFAULT_FLAGS.iter() {
@@ -158,3 +221,18 @@ impl MailboxView {
Ok(ret)
}
}
+
+fn string_to_flag(f: &str) -> Option<Flag> {
+ match f.chars().next() {
+ Some('\\') => None,
+ Some('$') if f == "$unseen" => None,
+ Some(_) => match Atom::try_from(f.clone()) {
+ Err(_) => {
+ tracing::error!(flag=%f, "Unable to encode flag as IMAP atom");
+ None
+ }
+ Ok(a) => Some(Flag::Keyword(a)),
+ },
+ None => None,
+ }
+}