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
// Copyright Materialize, Inc. and contributors. All rights reserved.
//
// Use of this software is governed by the Business Source License
// included in the LICENSE file.
//
// As of the Change Date specified in that file, in accordance with
// the Business Source License, use of this software will be governed
// by the Apache License, Version 2.0.
//! AWS configuration.
use aws_config::ConfigLoader;
use aws_smithy_http::endpoint::Endpoint;
use aws_types::credentials::{Credentials, CredentialsError, ProvideCredentials};
use aws_types::region::Region;
use aws_types::sdk_config::SdkConfig;
/// Service agnostic configuration for AWS.
///
/// This wraps the upstream [`SdkConfig`] type to allow additionally configuring
/// a global endpoint (e.g., LocalStack) to be used for all services.
// TODO(benesch): this entire type can get deleted if awslabs/aws-sdk-rust#396
// gets resolved.
#[derive(Debug, Clone)]
pub struct AwsConfig {
endpoint: Option<Endpoint>,
inner: SdkConfig,
}
impl AwsConfig {
/// Creates a configuration from a [`ConfigLoader`].
pub async fn from_loader(loader: ConfigLoader) -> AwsConfig {
AwsConfig {
endpoint: None,
inner: loader.load().await,
}
}
/// Loads the default configuration from the environment.
pub async fn load_from_env() -> AwsConfig {
AwsConfig::from_loader(aws_config::from_env()).await
}
/// Returns the inner [`SdkConfig`] object.
pub fn inner(&self) -> &SdkConfig {
&self.inner
}
/// Returns the currently configured endpoint, if any.
///
/// This is a convenience method for `config.inner().region()`.
pub fn region(&self) -> Option<&Region> {
self.inner.region()
}
/// Sets the endpoint.
pub fn set_endpoint(&mut self, endpoint: Endpoint) {
self.endpoint = Some(endpoint);
}
/// Returns the currently configured endpoint, if any.
pub fn endpoint(&self) -> Option<&Endpoint> {
self.endpoint.as_ref()
}
/// Acquires credentials using the configured credentials provider.
pub async fn provide_credentials(&self) -> Result<Credentials, CredentialsError> {
match self.inner().credentials_provider() {
None => Err(CredentialsError::not_loaded(
"no credentials provider configured",
)),
Some(provider) => provider.provide_credentials().await,
}
}
}