use std::iter;
use std::time::Duration;
use md5::{Digest, Md5};
use serde::Serialize;
use time::OffsetDateTime;
use url::Url;
use crate::actions::Method;
use crate::actions::OSSAction;
use crate::signing::sign;
use crate::sorting_iter::SortingIterator;
use crate::{Bucket, Credentials, Map};
#[derive(Debug, Clone)]
pub struct DeleteObjects<'a, I> {
bucket: &'a Bucket,
credentials: Option<&'a Credentials>,
objects: I,
quiet: bool,
query: Map<'a>,
headers: Map<'a>,
}
impl<'a, I> DeleteObjects<'a, I> {
#[inline]
pub fn new(bucket: &'a Bucket, credentials: Option<&'a Credentials>, objects: I) -> Self {
Self {
bucket,
credentials,
objects,
quiet: false,
query: Map::new(),
headers: Map::new(),
}
}
pub fn quiet(&self) -> bool {
self.quiet
}
pub fn set_quiet(&mut self, quiet: bool) {
self.quiet = quiet;
}
}
#[derive(Debug, Clone, Default)]
pub struct ObjectIdentifier {
pub key: String,
pub version_id: Option<String>,
}
impl ObjectIdentifier {
pub fn new(key: String) -> Self {
Self {
key,
..Default::default()
}
}
}
impl<'a, I> DeleteObjects<'a, I>
where
I: Iterator<Item = &'a ObjectIdentifier>,
{
pub fn body_with_md5(self) -> (String, String) {
#[derive(Serialize)]
#[serde(rename = "Delete")]
struct DeleteSerde<'a> {
#[serde(rename = "Object")]
objects: Vec<Object<'a>>,
#[serde(rename = "Quiet")]
quiet: Option<bool>,
}
#[derive(Serialize)]
#[serde(rename = "Delete")]
struct Object<'a> {
#[serde(rename = "$value")]
nodes: Vec<Node<'a>>,
}
#[derive(Serialize)]
enum Node<'a> {
Key(&'a str),
VersionId(&'a str),
}
let objects: Vec<Object<'a>> = self
.objects
.map(|o| {
let mut nodes = vec![Node::Key(o.key.as_str())];
if let Some(ref version_id) = o.version_id {
nodes.push(Node::VersionId(version_id.as_str()));
}
Object { nodes }
})
.collect();
let req = DeleteSerde {
objects,
quiet: self.quiet.then(|| true),
};
let body = quick_xml::se::to_string(&req).unwrap();
let content_md5 = crate::base64::encode(Md5::digest(body.as_bytes()));
(body, content_md5)
}
}
impl<'a, I> OSSAction<'a> for DeleteObjects<'a, I>
where
I: Iterator<Item = &'a ObjectIdentifier>,
{
const METHOD: Method = Method::Post;
fn query_mut(&mut self) -> &mut Map<'a> {
&mut self.query
}
fn headers_mut(&mut self) -> &mut Map<'a> {
&mut self.headers
}
fn sign_with_time(&self, expires_in: Duration, time: &OffsetDateTime) -> Url {
let url = self.bucket.base_url().clone();
let query = SortingIterator::new(iter::once(("delete", "1")), self.query.iter());
match self.credentials {
Some(credentials) => sign(
time,
Method::Post,
url,
credentials.key(),
credentials.secret(),
credentials.token(),
self.bucket.region(),
expires_in.as_secs(),
query,
self.headers.iter(),
),
None => crate::signing::util::add_query_params(url, query),
}
}
}
#[cfg(test)]
mod tests {
use pretty_assertions::assert_eq;
use time::OffsetDateTime;
use crate::{Bucket, Credentials, UrlStyle};
use super::*;
#[test]
fn oss_example() {
let date = OffsetDateTime::from_unix_timestamp(1369353600).unwrap();
let expires_in = Duration::from_secs(86400);
let endpoint = "https://oss-cn-hangzhou.aliyuncs.com".parse().unwrap();
let bucket = Bucket::new(
endpoint,
UrlStyle::VirtualHost,
"examplebucket",
"cn-hangzhou",
)
.unwrap();
let credentials = Credentials::new(
"access_key_id",
"access_key_secret",
);
let objects = [
ObjectIdentifier {
key: "123".to_owned(),
..Default::default()
},
ObjectIdentifier {
key: "456".to_owned(),
version_id: Some("ver1234".to_owned()),
},
];
let action = DeleteObjects::new(&bucket, Some(&credentials), objects.iter());
let url = action.sign_with_time(expires_in, &date);
let expected = "https://examplebucket.oss-cn-hangzhou.aliyuncs.com/?delete=1&x-oss-additional-headers=host&x-oss-credential=access_key_id%2F20130524%2Fcn-hangzhou%2Foss%2Faliyun_v4_request&x-oss-date=20130524T000000Z&x-oss-expires=86400&x-oss-signature-version=OSS4-HMAC-SHA256&x-oss-signature=b2d1a8a3818834f42fd4b828f5a461d74dc396ffef3540f834ed872fba693cfe";
assert_eq!(expected, url.as_str());
}
#[test]
fn anonymous_custom_query() {
let expires_in = Duration::from_secs(86400);
let endpoint = "https://oss-cn-hangzhou.aliyuncs.com".parse().unwrap();
let bucket = Bucket::new(
endpoint,
UrlStyle::VirtualHost,
"examplebucket",
"cn-hangzhou",
)
.unwrap();
let objects = [
ObjectIdentifier {
key: "123".to_owned(),
..Default::default()
},
ObjectIdentifier {
key: "456".to_owned(),
version_id: Some("ver1234".to_owned()),
},
];
let action = DeleteObjects::new(&bucket, None, objects.iter());
let url = action.sign(expires_in);
let expected = "https://examplebucket.oss-cn-hangzhou.aliyuncs.com/?delete=1";
assert_eq!(expected, url.as_str());
}
}