awsm-env 0.0.3

A lightweight utility for syncing AWS Secrets Manager secrets to environment variables.
Documentation 
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

use std::collections::HashMap;

pub struct OutputEntry<'a>(pub &'a str, pub &'a str);

/// By implementing `Formatter` a type provides a way to
/// format [`OutputEntry`]s
pub trait Formatter<'a, I: IntoIterator<Item = OutputEntry<'a>>> {
    fn format(&self, entries: I) -> String;
}

/// Formats environment entries into `.env` format using [`EnvOutput::format`]
pub struct EnvFormatter {}

impl EnvFormatter {
    pub fn new() -> Self {
        EnvFormatter {}
    }
}

impl<'a, I: IntoIterator<Item = OutputEntry<'a>>> Formatter<'a, I> for EnvFormatter {
    /// Formats environment entries into `.env` format
    fn format(&self, entries: I) -> String {
        let mut output = String::new();

        for entry in entries {
            output.push_str(&format!("{}={}\n", entry.0, entry.1));
        }

        output
    }
}

/// Formats environment entries into shell variable export commands using [`ShellOutput::format`]
pub struct ShellFormatter {}

impl ShellFormatter {
    pub fn new() -> Self {
        ShellFormatter {}
    }
}

impl<'a, I: IntoIterator<Item = OutputEntry<'a>>> Formatter<'a, I> for ShellFormatter {
    /// Formats environment entries into shell variable export commands
    fn format(&self, entries: I) -> String {
        let mut output = String::new();

        for entry in entries {
            output.push_str(&format!("export {}={}\n", entry.0, entry.1));
        }

        output
    }
}

/// Formats environment entries into JSON using [`JsonOutput::format`]
pub struct JsonFormatter {}

impl JsonFormatter {
    pub fn new() -> Self {
        JsonFormatter {}
    }
}

impl<'a, I: IntoIterator<Item = OutputEntry<'a>>> Formatter<'a, I> for JsonFormatter {
    /// Formats environment entries into JSON of the form `{"KEY": "value"}`
    fn format(&self, entries: I) -> String {
        let mut output = HashMap::new();

        for entry in entries {
            output.insert(entry.0, entry.1);
        }

        serde_json::to_string(&output).expect("HashMap should be serialized to JSON") + "\n"
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_env_output() {
        let input = vec![OutputEntry("KEY1", "value1"), OutputEntry("KEY2", "value2")];

        let formatter = EnvFormatter::new();
        let result = formatter.format(input);
        assert_eq!(result, "KEY1=value1\nKEY2=value2\n")
    }

    #[test]
    fn test_shell_output() {
        let input = vec![OutputEntry("KEY1", "value1"), OutputEntry("KEY2", "value2")];

        let formatter = ShellFormatter::new();
        let result = formatter.format(input);
        assert_eq!(result, "export KEY1=value1\nexport KEY2=value2\n")
    }

    #[test]
    fn test_json_output() {
        let input = vec![OutputEntry("KEY1", "value1"), OutputEntry("KEY2", "value2")];

        let formatter = JsonFormatter::new();
        let result = formatter.format(input);

        let mut expected = HashMap::new();

        expected.insert("KEY1", "value1");
        expected.insert("KEY2", "value2");

        assert_eq!(
            serde_json::from_str::<serde_json::Value>(&result).unwrap(),
            serde_json::to_value(expected).unwrap()
        )
    }
}