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

//! -*- mode: rust; -*-
//!
//! This file is part of privatemail crate.
//! Copyright (c) 2021 Nyah Check
//! See LICENSE for licensing information.
//!
//! Authors:
//! - Nyah Check <hello@nyah.dev>
//! GPG signature verification.

//! Configuration struct for `PrivatEmail`
#![allow(clippy::style)]
use serde::Serialize;
use std::env;

/// Config object for `PrivatEmail`.
///
/// Implements [`serde::Deserialize`] and [`serde::Serialize`] and
/// can be composed with other consumer configs.
///  `PrivatEmailConfig`:
///  `from_email`: Original Recipient Email from Verified SES Domain
///  `to_email`: Recipient SES verified email address which receives the forwarded email
///  `black_list`: Black listed email addresses.
#[derive(Clone, Debug, PartialEq, Serialize)]
#[serde(default)]
pub struct PrivatEmailConfig {
    /// Original Recipient Email from Verified SES Domain
    pub from_email: String,

    /// Recipient email address that receives the forwarded SES email
    pub to_email: String,

    /// Black Listed email addresses
    #[serde(skip_serializing_if = "Option::is_none")]
    pub black_list: Option<Vec<String>>,
}

/// Default configuration for `PrivatEmailConfig`
impl Default for PrivatEmailConfig {
    fn default() -> Self {
        PrivatEmailConfig {
            from_email: String::from("hello@nyah.dev"),
            to_email: String::from("nyah@hey.com"),
            black_list: None,
        }
    }
}

/// Create a new `PrivatEmailConfig` client struct from environment variables.
impl PrivatEmailConfig {
    /// Create new PrivatEmailConfig struct from environment variables.
    pub fn new_from_env() -> Self {
        let b_list = env::var("BLACK_LIST").unwrap_or_default();
        let black_list = b_list
            .split(",")
            .map(|x| String::from(x.replace(" ", "")))
            .collect();

        PrivatEmailConfig {
            from_email: env::var("FROM_EMAIL")
                .unwrap_or_else(|_e| panic!("Invalid FROM_EMAIL")),
            to_email: env::var("TO_EMAIL")
                .unwrap_or_else(|_e| panic!("Invalid TO_EMAIL")),
            black_list: Some(black_list),
        }
    }

    /// Create a new `PrivatEmailConfig` struct
    pub fn new<F, T, B>(from_email: F, to_email: T, black_list: B) -> Self
    where
        F: ToString,
        T: ToString,
        B: ToString,
    {
        let b_list_vec = black_list.to_string();
        let b_list: Vec<String> = b_list_vec
            .split(",")
            .map(|x| String::from(x.replace(" ", "")))
            .collect();
        PrivatEmailConfig {
            from_email: from_email.to_string(),
            to_email: to_email.to_string(),
            black_list: Some(b_list),
        }
    }
}

/** Test module for PrivatEmailConfig struct */
#[cfg(test)]
mod tests {
    use super::*;
    use std::env;

    #[test]
    fn test_new_privatemail_config() {
        let new_config = PrivatEmailConfig::new(
            String::from("test_from"),
            String::from("test_to"),
            String::from("fake@email.t, second@fake.email"),
        );
        assert_eq!(new_config.from_email.contains("test_from"), true);
        assert_eq!(new_config.to_email.contains("test_to"), true);
        assert_eq!(
            new_config.black_list.unwrap(),
            ["fake@email.t", "second@fake.email"]
        );
    }
    #[test]
    fn test_default_privatemail_config() {
        let new_config = PrivatEmailConfig::default();
        assert_eq!(new_config.from_email.contains("hello@nyah.dev"), true);
        assert_eq!(new_config.to_email.contains("nyah@hey.com"), true);
        assert_eq!(new_config.black_list.is_none(), true);
    }

    #[test]
    fn test_new_from_env_privatemail_config() {
        env::set_var("FROM_EMAIL", "test_from");
        env::set_var("TO_EMAIL", "test_to");

        let new_config = PrivatEmailConfig::new_from_env();
        assert_eq!(new_config.from_email.contains("test_from"), true);
        assert_eq!(new_config.to_email.contains("test_to"), true);
        assert_eq!(new_config.black_list.unwrap(), [""]);
    }
}