email/email/envelope/flag/set/
imap.rs

1use async_trait::async_trait;
2use imap_client::imap_next::imap_types::sequence::{Sequence, SequenceSet};
3use tracing::{debug, info};
4use utf7_imap::encode_utf7_imap as encode_utf7;
5
6use super::{Flags, SetFlags};
7use crate::{envelope::Id, imap::ImapContext, AnyResult, Error};
8
9#[derive(Clone, Debug)]
10pub struct SetImapFlags {
11    ctx: ImapContext,
12}
13
14impl SetImapFlags {
15    pub fn new(ctx: &ImapContext) -> Self {
16        Self { ctx: ctx.clone() }
17    }
18
19    pub fn new_boxed(ctx: &ImapContext) -> Box<dyn SetFlags> {
20        Box::new(Self::new(ctx))
21    }
22
23    pub fn some_new_boxed(ctx: &ImapContext) -> Option<Box<dyn SetFlags>> {
24        Some(Self::new_boxed(ctx))
25    }
26}
27
28#[async_trait]
29impl SetFlags for SetImapFlags {
30    async fn set_flags(&self, folder: &str, id: &Id, flags: &Flags) -> AnyResult<()> {
31        info!("setting imap flag(s) {flags} to envelope {id} from folder {folder}");
32
33        let mut client = self.ctx.client().await;
34        let config = &client.account_config;
35
36        let folder = config.get_folder_alias(folder);
37        let folder_encoded = encode_utf7(folder.clone());
38        debug!("utf7 encoded folder: {folder_encoded}");
39
40        let uids: SequenceSet = match id {
41            Id::Single(id) => Sequence::try_from(id.as_str())
42                .map_err(Error::ParseSequenceError)?
43                .into(),
44            Id::Multiple(ids) => ids
45                .iter()
46                .filter_map(|id| {
47                    let seq = Sequence::try_from(id.as_str());
48
49                    if let Err(err) = &seq {
50                        debug!(?id, ?err, "skipping invalid sequence");
51                    }
52
53                    seq.ok()
54                })
55                .collect::<Vec<_>>()
56                .try_into()
57                .map_err(Error::ParseSequenceError)?,
58        };
59
60        client.select_mailbox(&folder_encoded).await?;
61        client.set_flags(uids, flags.to_imap_flags_iter()).await?;
62
63        Ok(())
64    }
65}
email/email/envelope/flag/set/imap.rs

email/email/envelope/flag/set/
imap.rs
