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

pub mod config;

use anyhow::Context;
use config::MailerConfig;
pub use lettre::message::*;
pub use lettre::AsyncTransport;
pub use lettre::Message;
use lettre::{transport::smtp::authentication::Credentials, Tokio1Executor};
use spring_boot::async_trait;
use spring_boot::config::Configurable;
use spring_boot::{app::AppBuilder, error::Result, plugin::Plugin};

pub type Mailer = lettre::AsyncSmtpTransport<Tokio1Executor>;

#[derive(Configurable)]
#[config_prefix = "mail"]
pub struct MailPlugin;

#[async_trait]
impl Plugin for MailPlugin {
    async fn build(&self, app: &mut AppBuilder) {
        let config = app
            .get_config::<MailerConfig>(self)
            .expect("mail plugin config load failed");

        let mailer = Self::build_mailer(&config).expect("build mail plugin failed");

        app.add_component(mailer);
    }
}

impl MailPlugin {
    fn build_mailer(config: &MailerConfig) -> Result<Mailer> {
        let mut email_builder = if config.secure {
            Mailer::starttls_relay(&config.host)
                .with_context(|| format!("build mailer failed: {}", config.host))?
                .port(config.port)
        } else {
            Mailer::builder_dangerous(&config.host).port(config.port)
        };

        if let Some(auth) = config.auth.as_ref() {
            let credentials = Credentials::new(auth.user.clone(), auth.password.clone());
            email_builder = email_builder.credentials(credentials);
        }

        Ok(email_builder.build())
    }
}