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
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use delivery_result::DeliveryResult;
use email_format::rfc5322::headers::Bcc;
use email_format::rfc5322::types::{Address, GroupList, Mailbox};
use email_format::Email;
use error::Error;
use lettre::{EmailAddress, SendableEmail, Envelope};
use message_status::InternalMessageStatus;
use recipient_status::InternalRecipientStatus;
use uuid::Uuid;

/// An email, prepared for delivery.
#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct PreparedEmail {
    pub to: Vec<String>,
    pub from: String,
    pub message_id: String,
    pub message: Vec<u8>,
}

impl PreparedEmail {
    pub fn as_sendable_email(&self) -> Result<SendableEmail, failure::Error> {
        let to: Result<Vec<EmailAddress>, failure::Error> =
            self.to.iter().map(|s| EmailAddress::new(s.clone())).collect();
        let to = to?;

        Ok(SendableEmail::new(
            Envelope::new(
                Some(EmailAddress::new(self.from.clone())?),
                to)?,
            self.message_id.clone(),
            self.message.clone()
        ))
    }
}

pub fn prepare_email(
    mut email: Email,
    helo_name: &str,
) -> Result<(PreparedEmail, InternalMessageStatus), Error> {
    let recipients = determine_recipients(&email);

    // Blind the Bcc
    email.clear_bcc();

    let message_id = match email.get_message_id() {
        Some(mid) => format!("{}@{}", mid.0.id_left, mid.0.id_right),
        None => {
            // Generate message-id
            let message_id = format!("{}@{}", Uuid::new_v4().hyphenated().to_string(), helo_name);
            email.set_message_id(&*format!("<{}>", message_id))?;
            message_id
        }
    };

    let prepared_email = PreparedEmail {
        to: recipients
            .iter()
            .map(|r| r.smtp_email_addr.clone())
            .collect(),
        from: format!("{}", email.get_from().0),
        message_id: message_id.clone(),
        message: format!("{}", email).into_bytes(),
    };

    // Verify that lettre::SendableEmail will not give us errors later on
    // down the track
    let _ = ::lettre::EmailAddress::new(prepared_email.from.clone())?;
    prepared_email.to.iter()
        .try_for_each(|s| ::lettre::EmailAddress::new(s.clone()).map(|_|()))?;

    let internal_message_status = InternalMessageStatus {
        message_id: message_id,
        recipients: recipients,
        attempts_remaining: 3,
    };

    Ok((prepared_email, internal_message_status))
}

fn determine_recipients(email: &Email) -> Vec<InternalRecipientStatus> {
    let mut addresses: Vec<Address> = Vec::new();

    if let Some(to) = email.get_to() {
        addresses.extend((to.0).0);
    }
    if let Some(cc) = email.get_cc() {
        addresses.extend((cc.0).0);
    }
    if let Some(bcc) = email.get_bcc() {
        if let Bcc::AddressList(al) = bcc {
            addresses.extend(al.0);
        }
    }

    addresses.dedup();

    let mut recipients: Vec<InternalRecipientStatus> = Vec::new();

    for address in addresses {
        match address {
            Address::Mailbox(mb) => {
                recipients.push(recipient_from_mailbox(mb));
            }
            Address::Group(grp) => {
                if let Some(gl) = grp.group_list {
                    match gl {
                        GroupList::MailboxList(mbl) => {
                            for mb in mbl.0 {
                                recipients.push(recipient_from_mailbox(mb));
                            }
                        }
                        GroupList::CFWS(_) => continue,
                    }
                }
            }
        }
    }

    recipients
}

fn recipient_from_mailbox(mb: Mailbox) -> InternalRecipientStatus {
    let (email_addr, smtp_email_addr, domain) = match mb {
        Mailbox::NameAddr(na) => (
            format!("{}", na),
            format!("{}", na.angle_addr.addr_spec),
            format!("{}", na.angle_addr.addr_spec.domain),
        ),
        Mailbox::AddrSpec(ads) => (
            format!("{}", ads),
            format!("{}", ads),
            format!("{}", ads.domain),
        ),
    };

    InternalRecipientStatus {
        email_addr: email_addr.trim().to_owned(),
        smtp_email_addr: smtp_email_addr.trim().to_owned(),
        domain: domain.trim().to_owned(),
        mx_servers: None, // To be determined later by a worker task
        current_mx: 0,
        result: DeliveryResult::Queued,
    }
}