awsm-env 0.2.5

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

use crate::provider::{Provider, ResolvedSecret};
use anyhow::{Result, anyhow};
use itertools::Itertools;

/// Fetches secrets from AWS Secrets Manager
pub struct AwsSecretsManagerProvider {
    client: aws_sdk_secretsmanager::Client,
}

impl AwsSecretsManagerProvider {
    pub async fn new() -> Self {
        let config = aws_config::load_from_env().await;
        let client = aws_sdk_secretsmanager::Client::new(&config);

        Self { client }
    }
}

impl Provider for AwsSecretsManagerProvider {
    // All the expects are because the AWS SDK isn't idiomatic
    async fn provide_secrets(&self, ids: Vec<String>) -> Result<Vec<ResolvedSecret>> {
        let mut result = Vec::new();

        for chunk in &ids.into_iter().chunks(20) {
            let secrets = self
                .client
                .batch_get_secret_value()
                .set_secret_id_list(Some(chunk.collect()))
                .send()
                .await?;

            let first_error = secrets.errors.and_then(|errors| {
                errors
                    .into_iter()
                    .filter(|e| e.error_code() != Some("ResourceNotFoundException"))
                    .next()
            });

            if let Some(error) = first_error {
                return Err(anyhow!("Failed to fetch secrets: {:?}", error));
            }

            result.extend(
                secrets
                    .secret_values
                    .expect("should have secrets if there were no ResourceNotFound errors")
                    .into_iter()
                    .map(|s| ResolvedSecret {
                        id: s.name.expect("should have a name"),
                        secret: s.secret_string.expect("should have a secret string"),
                    }),
            );
        }

        Ok(result)
    }
}