rusty_s3/actions/multipart_upload/
abort.rs1use std::iter;
2use std::time::Duration;
3
4use jiff::Timestamp;
5use url::Url;
6
7use crate::actions::Method;
8use crate::actions::S3Action;
9use crate::signing::sign;
10use crate::sorting_iter::SortingIterator;
11use crate::{Bucket, Credentials, Map};
12
13#[allow(clippy::module_name_repetitions)]
21#[derive(Debug, Clone)]
22pub struct AbortMultipartUpload<'a> {
23 bucket: &'a Bucket,
24 credentials: Option<&'a Credentials>,
25 object: &'a str,
26
27 upload_id: &'a str,
28
29 query: Map<'a>,
30 headers: Map<'a>,
31}
32
33impl<'a> AbortMultipartUpload<'a> {
34 #[inline]
35 #[must_use]
36 pub const fn new(
37 bucket: &'a Bucket,
38 credentials: Option<&'a Credentials>,
39 object: &'a str,
40 upload_id: &'a str,
41 ) -> Self {
42 Self {
43 bucket,
44 credentials,
45 object,
46
47 upload_id,
48
49 query: Map::new(),
50 headers: Map::new(),
51 }
52 }
53}
54
55impl<'a> S3Action<'a> for AbortMultipartUpload<'a> {
56 const METHOD: Method = Method::Delete;
57
58 fn query_mut(&mut self) -> &mut Map<'a> {
59 &mut self.query
60 }
61
62 fn headers_mut(&mut self) -> &mut Map<'a> {
63 &mut self.headers
64 }
65
66 fn sign_with_time(&self, expires_in: Duration, time: &Timestamp) -> Url {
67 let url = self.bucket.object_url(self.object).unwrap();
68 let query = iter::once(("uploadId", self.upload_id));
69
70 match self.credentials {
71 Some(credentials) => sign(
72 time,
73 Self::METHOD,
74 url,
75 credentials.key(),
76 credentials.secret(),
77 credentials.token(),
78 self.bucket.region(),
79 expires_in.as_secs(),
80 SortingIterator::new(query, self.query.iter()),
81 self.headers.iter(),
82 ),
83 None => crate::signing::util::add_query_params(url, query),
84 }
85 }
86}
87
88#[cfg(test)]
89mod tests {
90 use pretty_assertions::assert_eq;
91
92 use super::*;
93 use crate::{Bucket, Credentials, UrlStyle};
94
95 #[test]
96 fn aws_example() {
97 let date = Timestamp::from_second(1369353600).unwrap();
99 let expires_in = Duration::from_secs(86400);
100
101 let endpoint = "https://s3.amazonaws.com".parse().unwrap();
102 let bucket = Bucket::new(
103 endpoint,
104 UrlStyle::VirtualHost,
105 "examplebucket",
106 "us-east-1",
107 )
108 .unwrap();
109 let credentials = Credentials::new(
110 "AKIAIOSFODNN7EXAMPLE",
111 "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY",
112 );
113
114 let action = AbortMultipartUpload::new(&bucket, Some(&credentials), "test.txt", "abcd");
115
116 let url = action.sign_with_time(expires_in, &date);
117 let expected = "https://examplebucket.s3.amazonaws.com/test.txt?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=AKIAIOSFODNN7EXAMPLE%2F20130524%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20130524T000000Z&X-Amz-Expires=86400&X-Amz-SignedHeaders=host&uploadId=abcd&X-Amz-Signature=7670bc768a7cdb5c276a9dddadeefdffb52061f94db6c14b4a9284fdc195bb59";
118
119 assert_eq!(expected, url.as_str());
120 }
121
122 #[test]
123 fn anonymous_custom_query() {
124 let expires_in = Duration::from_secs(86400);
125
126 let endpoint = "https://s3.amazonaws.com".parse().unwrap();
127 let bucket = Bucket::new(
128 endpoint,
129 UrlStyle::VirtualHost,
130 "examplebucket",
131 "us-east-1",
132 )
133 .unwrap();
134
135 let action = AbortMultipartUpload::new(&bucket, None, "test.txt", "abcd");
136 let url = action.sign(expires_in);
137 let expected = "https://examplebucket.s3.amazonaws.com/test.txt?uploadId=abcd";
138
139 assert_eq!(expected, url.as_str());
140 }
141}