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::PutObjectLegalHoldResponse;
use crate::s3::segmented_bytes::SegmentedBytes;
use crate::s3::types::{BucketName, ObjectKey, Region, S3Api, S3Request, ToS3Request, VersionId};
use crate::s3::utils::{insert, md5sum_hash};
use bytes::Bytes;
use http::Method;
use std::sync::Arc;
use typed_builder::TypedBuilder;
#[derive(Clone, Debug, TypedBuilder)]
pub struct PutObjectLegalHold {
#[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(setter(into), !default)]
bucket: BucketName,
#[builder(setter(into), !default)]
object: ObjectKey,
#[builder(default, setter(into))]
version_id: Option<VersionId>,
#[builder(default, setter(into))]
legal_hold: Option<bool>,
}
pub type PutObjectLegalHoldBldr = PutObjectLegalHoldBuilder<(
(MinioClient,),
(),
(),
(),
(BucketName,),
(ObjectKey,),
(),
(Option<bool>,),
)>;
impl S3Api for PutObjectLegalHold {
type S3Response = PutObjectLegalHoldResponse;
}
impl ToS3Request for PutObjectLegalHold {
fn to_s3request(self) -> Result<S3Request, ValidationErr> {
let mut headers: Multimap = self.extra_headers.unwrap_or_default();
let mut query_params: Multimap = insert(self.extra_query_params, "legal-hold");
query_params.add_version(self.version_id);
let payload: &str = match self.legal_hold {
Some(true) => "<LegalHold><Status>ON</Status></LegalHold>",
_ => "<LegalHold><Status>OFF</Status></LegalHold>",
};
let bytes: Bytes = Bytes::from(payload);
headers.add(CONTENT_MD5, md5sum_hash(bytes.as_ref()));
let body = Arc::new(SegmentedBytes::from(bytes));
Ok(S3Request::builder()
.client(self.client)
.method(Method::PUT)
.region(self.region)
.bucket(self.bucket)
.query_params(query_params)
.headers(headers)
.object(self.object)
.body(body)
.build())
}
}