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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use imap_proto::types::Quota as QuotaRef;
use imap_proto::types::QuotaResource as QuotaResourceRef;
use imap_proto::types::QuotaResourceName as QuotaResourceNameRef;
use imap_proto::types::QuotaRoot as QuotaRootRef;

/// <https://tools.ietf.org/html/rfc2087#section-3>
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
pub enum QuotaResourceName {
    /// Sum of messages' RFC822.SIZE, in units of 1024 octets
    Storage,
    /// Number of messages
    Message,
    /// A different/custom resource
    Atom(String),
}

impl<'a> From<QuotaResourceNameRef<'a>> for QuotaResourceName {
    fn from(name: QuotaResourceNameRef<'_>) -> Self {
        match name {
            QuotaResourceNameRef::Message => QuotaResourceName::Message,
            QuotaResourceNameRef::Storage => QuotaResourceName::Storage,
            QuotaResourceNameRef::Atom(v) => QuotaResourceName::Atom(v.to_string()),
        }
    }
}

/// 5.1. QUOTA Response (<https://tools.ietf.org/html/rfc2087#section-5.1>)
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
pub struct QuotaResource {
    /// name of the resource
    pub name: QuotaResourceName,
    /// current usage of the resource
    pub usage: u64,
    /// resource limit
    pub limit: u64,
}

impl<'a> From<QuotaResourceRef<'a>> for QuotaResource {
    fn from(resource: QuotaResourceRef<'_>) -> Self {
        Self {
            name: resource.name.into(),
            usage: resource.usage,
            limit: resource.limit,
        }
    }
}

impl QuotaResource {
    /// Returns the usage percentage of a QuotaResource.
    pub fn get_usage_percentage(&self) -> u64 {
        self.usage.saturating_mul(100) / self.limit
    }
}

/// 5.1. QUOTA Response (<https://tools.ietf.org/html/rfc2087#section-5.1>)
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
pub struct Quota {
    /// quota root name
    pub root_name: String,
    /// quota resources for this quota
    pub resources: Vec<QuotaResource>,
}

impl<'a> From<QuotaRef<'a>> for Quota {
    fn from(quota: QuotaRef<'_>) -> Self {
        Self {
            root_name: quota.root_name.to_string(),
            resources: quota.resources.iter().map(|r| r.clone().into()).collect(),
        }
    }
}

/// 5.2. QUOTAROOT Response (<https://tools.ietf.org/html/rfc2087#section-5.2>)
#[derive(Debug, Eq, PartialEq, Hash, Clone)]
pub struct QuotaRoot {
    /// mailbox name
    pub mailbox_name: String,
    /// zero or more quota root names
    pub quota_root_names: Vec<String>,
}

impl<'a> From<QuotaRootRef<'a>> for QuotaRoot {
    fn from(root: QuotaRootRef<'_>) -> Self {
        Self {
            mailbox_name: root.mailbox_name.to_string(),
            quota_root_names: root
                .quota_root_names
                .iter()
                .map(|n| n.to_string())
                .collect(),
        }
    }
}