use crate::s3::client::DEFAULT_REGION;
use crate::s3::client::MinioClient;
use crate::s3::error::ValidationErr;
use crate::s3::header_constants::*;
use crate::s3::multimap_ext::{Multimap, MultimapExt};
use crate::s3::response::CreateBucketResponse;
use crate::s3::segmented_bytes::SegmentedBytes;
use crate::s3::types::{BucketName, Region, S3Api, S3Request, ToS3Request};
use http::Method;
use std::sync::Arc;
use typed_builder::TypedBuilder;
#[derive(Clone, Debug, TypedBuilder)]
pub struct CreateBucket {
#[builder(!default)] client: MinioClient,
#[builder(default, setter(into))]
extra_headers: Option<Multimap>,
#[builder(default, setter(into))]
extra_query_params: Option<Multimap>,
#[builder(default, setter(into))]
region: Option<Region>,
#[builder(!default, setter(into))] bucket: BucketName,
#[builder(default = false)]
object_lock: bool,
}
impl S3Api for CreateBucket {
type S3Response = CreateBucketResponse;
}
pub type CreateBucketBldr = CreateBucketBuilder<((MinioClient,), (), (), (), (BucketName,), ())>;
impl ToS3Request for CreateBucket {
fn to_s3request(self) -> Result<S3Request, ValidationErr> {
let region1: Option<&str> = self.region.as_ref().map(|r| r.as_str());
let region2: Option<&str> = self.client.get_region_from_url();
let region_str: String = match (region1, region2) {
(None, None) => DEFAULT_REGION.as_str().to_string(),
(Some(_), None) => self.region.unwrap().to_string(),
(None, Some(v)) => v.to_string(),
(Some(r1), Some(r2)) if r1 == r2 => self.region.unwrap().to_string(), (Some(r1), Some(r2)) => {
return Err(ValidationErr::RegionMismatch {
bucket_region: Region::new(r1)?,
region: Region::new(r2)?,
});
}
};
let mut headers: Multimap = self.extra_headers.unwrap_or_default();
if self.object_lock {
headers.add(X_AMZ_BUCKET_OBJECT_LOCK_ENABLED, "true");
}
let data: String = if region_str == DEFAULT_REGION.as_str() {
String::new()
} else {
format!(
"<CreateBucketConfiguration><LocationConstraint>{region_str}</LocationConstraint></CreateBucketConfiguration>",
)
};
let body: Option<Arc<SegmentedBytes>> = match data.is_empty() {
true => None,
false => Some(Arc::new(SegmentedBytes::from(data))),
};
Ok(S3Request::builder()
.client(self.client)
.method(Method::PUT)
.region(Some(Region::new(region_str)?))
.bucket(self.bucket)
.query_params(self.extra_query_params.unwrap_or_default())
.headers(headers)
.body(body)
.build())
}
}