This commit is contained in:
2025-11-08 02:03:52 +01:00
parent 9bf1fc1983
commit d24d2c9249
6 changed files with 524 additions and 1027 deletions
+26 -30
View File
@@ -1,38 +1,34 @@
use anyhow::Result;
use aws_config::meta::region::RegionProviderChain;
use aws_credential_types::Credentials;
use aws_sdk_s3::{
config::{Builder as S3ConfigBuilder, Region},
Client as S3Client,
};
use anyhow::{anyhow, Context, Result};
use s3::{bucket::Bucket, creds::Credentials, region::Region};
use crate::config::AppConfig;
pub async fn build_client(config: &AppConfig) -> Result<S3Client> {
let region = Region::new(config.aws_region.clone());
let region_provider = RegionProviderChain::first_try(Some(region))
.or_default_provider()
.or_else("us-east-1");
pub fn build_bucket(config: &AppConfig) -> Result<Bucket> {
let region = if let Some(endpoint) = &config.aws_endpoint_url {
Region::Custom {
region: config.aws_region.clone(),
endpoint: endpoint.clone(),
}
} else {
config
.aws_region
.parse::<Region>()
.context("invalid AWS region")?
};
#[allow(deprecated)]
let mut loader = aws_config::from_env().region(region_provider);
if let Some(endpoint) = &config.aws_endpoint_url {
loader = loader.endpoint_url(endpoint);
}
if let (Some(access_key), Some(secret_key)) = (
config.aws_access_key_id.clone(),
config.aws_secret_access_key.clone(),
let credentials = if let (Some(access_key), Some(secret_key)) = (
config.aws_access_key_id.as_deref(),
config.aws_secret_access_key.as_deref(),
) {
let credentials = Credentials::new(access_key, secret_key, None, None, "static");
loader = loader.credentials_provider(credentials);
}
Credentials::new(Some(access_key), Some(secret_key), None, None, None)
.context("failed to create static AWS credentials")?
} else {
Credentials::default().context("failed to load AWS credentials")?
};
let base_config = loader.load().await;
let s3_config = S3ConfigBuilder::from(&base_config)
.force_path_style(true)
.build();
let bucket = Bucket::new(&config.s3_bucket, region, credentials)
.map_err(|err| anyhow!("failed to create S3 bucket client: {err}"))?;
let bucket = bucket.with_path_style();
Ok(S3Client::from_conf(s3_config))
Ok(*bucket)
}