imap_client/tasks/tasks/
list.rs

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
use imap_next::imap_types::{
    command::CommandBody,
    core::QuotedChar,
    flag::FlagNameAttribute,
    mailbox::{ListMailbox, Mailbox},
    response::{Data, StatusBody, StatusKind},
};

use super::TaskError;
use crate::tasks::Task;

#[derive(Clone, Debug)]
pub struct ListTask {
    mailbox: Mailbox<'static>,
    mailbox_wildcard: ListMailbox<'static>,
    output: Vec<(
        Mailbox<'static>,
        Option<QuotedChar>,
        Vec<FlagNameAttribute<'static>>,
    )>,
}

impl ListTask {
    pub fn new(mailbox: Mailbox<'static>, mailbox_wildcard: ListMailbox<'static>) -> Self {
        Self {
            mailbox,
            mailbox_wildcard,
            output: Vec::new(),
        }
    }
}

impl Task for ListTask {
    type Output = Result<
        Vec<(
            Mailbox<'static>,
            Option<QuotedChar>,
            Vec<FlagNameAttribute<'static>>,
        )>,
        TaskError,
    >;

    fn command_body(&self) -> CommandBody<'static> {
        CommandBody::List {
            reference: self.mailbox.clone(),
            mailbox_wildcard: self.mailbox_wildcard.clone(),
        }
    }

    fn process_data(&mut self, data: Data<'static>) -> Option<Data<'static>> {
        if let Data::List {
            items,
            delimiter,
            mailbox,
        } = data
        {
            self.output.push((mailbox, delimiter, items));
            None
        } else {
            Some(data)
        }
    }

    fn process_tagged(self, status_body: StatusBody<'static>) -> Self::Output {
        match status_body.kind {
            StatusKind::Ok => Ok(self.output),
            StatusKind::No => Err(TaskError::UnexpectedNoResponse(status_body)),
            StatusKind::Bad => Err(TaskError::UnexpectedBadResponse(status_body)),
        }
    }
}