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

use lettre::smtp::authentication::Credentials;
use lettre::stub::StubTransport;
use lettre::{SendableEmail, SmtpClient, Transport};
use lettre_email::EmailBuilder;
use std::str;

#[derive(Clone)]
pub struct Mailer {
    pub from_address: String,
    pub smtp_server: String,
    pub smtp_username: String,
    pub smtp_password: String,
    pub actually_send: bool,
}

impl Mailer {
    pub fn new() -> Mailer {
        Mailer::check_environment_variables();

        let from_address: String =
            std::env::var("SMTP_FROM_ADDRESS").unwrap_or("create-rust-app@localhost".to_string());
        let smtp_server: String = std::env::var("SMTP_SERVER").unwrap_or("".to_string());
        let smtp_username: String = std::env::var("SMTP_USERNAME").unwrap_or("".to_string());
        let smtp_password: String = std::env::var("SMTP_PASSWORD").unwrap_or("".to_string());
        let actually_send: bool = std::env::var("SEND_MAIL")
            .unwrap_or("false".to_string())
            .eq_ignore_ascii_case("true");

        return Mailer {
            from_address: from_address,
            smtp_server: smtp_server,
            smtp_username: smtp_username,
            smtp_password: smtp_password,
            actually_send: actually_send,
        };
    }

    pub fn check_environment_variables() {
        if std::env::var("SECRET_KEY").is_err() {
            panic!("No SECRET_KEY environment variable set!");
        }

        if std::env::var("DATABASE_URL").is_err() {
            panic!("No DATABASE_URL environment variable set!");
        }

        if std::env::var("SMTP_FROM_ADDRESS").is_err() {
            println!(
                "Note: SMTP_FROM_ADDRESS environment variable is not set, no mail will be sent!"
            );
        }

        if std::env::var("SMTP_SERVER").is_err() {
            println!("Note: SMTP_SERVER environment variable is not set, no mail will be sent!");
        }

        if std::env::var("SMTP_USERNAME").is_err() {
            println!("Note: SMTP_USERNAME environment variable is not set, no mail will be sent!");
        }

        if std::env::var("SMTP_PASSWORD").is_err() {
            println!("Note: SMTP_PASSWORD environment variable is not set, no mail will be sent!");
        }

        if std::env::var("SEND_MAIL").is_err() || !std::env::var("SEND_MAIL").unwrap().eq("true") {
            println!(
                "Note: SEND_MAIL environment variale is not set to \"true\", no mail will be sent!"
            );
        }
    }

    pub fn send(&self, to: &str, subject: &str, text: &str, html: &str) {
        let email = EmailBuilder::new()
            .to(to)
            .from(self.from_address.as_ref())
            .subject(subject)
            .text(text)
            .html(html)
            .build()
            .unwrap();

        let sendable_email: SendableEmail = email.clone().into();

        if self.actually_send {
            let mut mailer = SmtpClient::new_simple(self.smtp_server.as_str())
                .unwrap()
                .credentials(Credentials::new(
                    self.smtp_username.to_string(),
                    self.smtp_password.to_string(),
                ))
                .transport();

            let result = mailer.send(sendable_email);
            println!(
                r#"====================
Sent email {:#?}
--------------------
to: {:?}
from: {}
message:
{}
===================="#,
                result, to, self.from_address, text
            );
        } else {
            let mut mailer = StubTransport::new_positive();
            let result = mailer.send(sendable_email);
            println!(
                r#"====================
Sent email {:#?}
--------------------
to: {:?}
from: {}
message:
{}
===================="#,
                result, to, self.from_address, text
            );
        }
    }
}