1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63

//! Module dedicated to Maildir email envelope flags.
//!
//! This module contains flag-related mapping functions from the
//! [maildirpp] crate types.

use log::debug;
use maildirpp::MailEntry;
use thiserror::Error;

use crate::Result;

use super::{Flag, Flags};

#[derive(Debug, Error)]
pub enum Error {
    #[error("cannot parse maildir flag char {0}")]
    ParseFlagError(char),
}

impl Flag {
    pub fn try_from_mdir_char(c: char) -> Result<Self> {
        match c {
            'r' | 'R' => Ok(Flag::Answered),
            's' | 'S' => Ok(Flag::Seen),
            't' | 'T' => Ok(Flag::Deleted),
            'd' | 'D' => Ok(Flag::Draft),
            'f' | 'F' => Ok(Flag::Flagged),
            unknown => Err(Error::ParseFlagError(unknown).into()),
        }
    }

    pub fn to_opt_mdir_char(&self) -> Option<char> {
        match self {
            Flag::Answered => Some('R'),
            Flag::Seen => Some('S'),
            Flag::Deleted => Some('T'),
            Flag::Draft => Some('D'),
            Flag::Flagged => Some('F'),
            _ => None,
        }
    }
}

impl Flags {
    pub fn from_mdir_entry(entry: &MailEntry) -> Self {
        entry
            .flags()
            .chars()
            .filter_map(|c| match Flag::try_from_mdir_char(c) {
                Ok(flag) => Some(flag),
                Err(err) => {
                    debug!("cannot parse maildir flag char {c}, skipping it: {err}");
                    debug!("{err:?}");
                    None
                }
            })
            .collect()
    }

    pub fn to_mdir_string(&self) -> String {
        String::from_iter(self.iter().filter_map(|flag| flag.to_opt_mdir_char()))
    }
}