1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192

//! bucket相关接口

use crate::client::Client;

use crate::request::{Request, Response};
use reqwest::Body;

use crate::acl::AclHeader;
use std::collections::HashMap;

#[async_trait::async_trait]
pub trait Bucket {
    /// 在指定账号下创建一个存储桶
    /// 创建存储桶时，如果没有指定访问权限，则默认使用私有读写（private）权限。
    async fn put_bucket(&self, acl_header: Option<&AclHeader>) -> Response;

    /// 用于删除指定的存储桶。该 API 的请求者需要对存储桶有写入权限。
    async fn delete_bucket(&self) -> Response;

    /// 可以列出该存储桶内的部分或者全部对象。该 API 的请求者需要对存储桶有读取权限。
    async fn list_objects(
        &self,
        prefix: &str,
        delimiter: &str,
        encoding_type: &str,
        marker: &str,
        max_keys: i32,
    ) -> Response;

    /// 检查bucket状态
    async fn check_bucket(&self) -> Response;

    /// 写入存储桶的访问控制列表（ACL)
    async fn put_bucket_acl(&self, acl_header: &AclHeader) -> Response;
}

#[async_trait::async_trait]
impl Bucket for Client {
    /// 创建一个存储桶
    /// 见[官网文档](https://cloud.tencent.com/document/product/436/7738)
    /// # Examples
    /// ```
    /// use rust_qcos::client::Client;
    /// use rust_qcos::bucket::Bucket;
    /// use rust_qcos::acl::{AclHeader, BucketAcl};
    /// async {
    /// let mut acl_header = AclHeader::new();
    /// acl_header.insert_bucket_x_cos_acl(BucketAcl::PublicRead);
    /// let client = Client::new("foo", "bar", "qcloudtest-1256650966", "ap-guangzhou");
    /// let res = client.put_bucket(Some(&acl_header)).await;
    /// assert!(res.error_message.contains("403"));
    /// };
    /// ```
    async fn put_bucket(&self, acl_header: Option<&AclHeader>) -> Response {
        let headers = self.get_headers_with_auth("put", "/", acl_header, None, None);
        let resp = Request::put(
            self.get_full_url_from_path("/").as_str(),
            None,
            Some(&headers),
            None,
            None,
            None as Option<Body>,
        )
        .await;
        self.make_response(resp)
    }
    /// 删除指定的存储桶。该 API 的请求者需要对存储桶有写入权限。
    /// 见[官网文档](https://cloud.tencent.com/document/product/436/7732)
    /// # Examples
    /// ```
    /// use rust_qcos::client::Client;
    /// use rust_qcos::bucket::Bucket;
    /// async {
    /// let client = Client::new("foo", "bar", "qcloudtest-1256650966", "ap-guangzhou");
    /// let res = client.delete_bucket().await;
    /// assert!(res.error_message.contains("403"));
    /// };
    /// ```
    async fn delete_bucket(&self) -> Response {
        let headers = self.get_headers_with_auth("delete", "/", None, None, None);
        let resp = Request::delete(
            self.get_full_url_from_path("/").as_str(),
            None,
            Some(&headers),
            None,
            None,
        )
        .await;
        self.make_response(resp)
    }
    /// 列出该存储桶内的部分或者全部对象。该 API 的请求者需要对存储桶有读取权限。
    /// 见[官网文档](https://cloud.tencent.com/document/product/436/7734)
    /// # Examples
    /// ```
    /// use rust_qcos::client::Client;
    /// use rust_qcos::bucket::Bucket;
    /// async {
    /// let client = Client::new("foo", "bar", "qcloudtest-1256650966", "ap-guangzhou");
    /// let res = client.list_objects("prefix", "", "", "/", 100).await;
    /// assert!(res.error_message.contains("403"));
    /// };
    /// ```
    async fn list_objects(
        &self,
        prefix: &str,
        delimiter: &str,
        encoding_type: &str,
        marker: &str,
        max_keys: i32,
    ) -> Response {
        let mut query = HashMap::new();
        if !prefix.is_empty() {
            query.insert("prefix".to_string(), prefix.to_string());
        }
        if !delimiter.is_empty() {
            query.insert("delimiter".to_string(), delimiter.to_string());
        }
        if !encoding_type.is_empty() {
            query.insert("encoding-type".to_string(), encoding_type.to_string());
        }
        if !marker.is_empty() {
            query.insert("marker".to_string(), marker.to_string());
        }
        if max_keys <= 1000 && max_keys > 0 {
            query.insert("max-keys".to_string(), max_keys.to_string());
        }
        let headers = self.get_headers_with_auth("get", "/", None, None, Some(&query));
        let resp = Request::get(
            self.get_full_url_from_path("/").as_str(),
            Some(&query),
            Some(&headers),
        )
        .await;
        self.make_response(resp)
    }

    /// 确认该存储桶是否存在，是否有权限访问
    /// [官网文档](https://cloud.tencent.com/document/product/436/7735)
    /// 存储桶存在且有读取权限，返回 `SUCCESS`
    /// 无存储桶读取权限，返回 `ErrNo::STATUS`, error_message包含403。
    /// 存储桶不存在，返回 `ErrNo::STATUS`, error_message包含404。
    /// # Examples
    /// ```
    /// use rust_qcos::client::Client;
    /// use rust_qcos::bucket::Bucket;
    /// async {
    /// let client = Client::new("foo", "bar", "qcloudtest-1256650966", "ap-guangzhou");
    /// let res = client.check_bucket().await;
    /// assert!(res.error_message.contains("403"));
    /// };
    /// ```
    async fn check_bucket(&self) -> Response {
        let headers = self.get_headers_with_auth("head", "/", None, None, None);
        let resp = Request::head(
            self.get_full_url_from_path("/").as_str(),
            None,
            Some(&headers),
        )
        .await;
        self.make_response(resp)
    }
    /// 写入存储桶的访问控制列表
    /// [官网文档](https://cloud.tencent.com/document/product/436/7737)
    /// # Examples
    /// ```
    /// use rust_qcos::client::Client;
    /// use rust_qcos::bucket::Bucket;
    /// use rust_qcos::acl::{AclHeader, BucketAcl};
    /// async {
    /// let mut acl_header = AclHeader::new();
    /// acl_header.insert_bucket_x_cos_acl(BucketAcl::PublicRead);
    /// let client = Client::new("foo", "bar", "qcloudtest-1256650966", "ap-guangzhou");
    /// let res = client.put_bucket(Some(&acl_header)).await;
    /// assert!(res.error_message.contains("403"));
    /// };
    /// ```
    async fn put_bucket_acl(&self, acl_header: &AclHeader) -> Response {
        let mut query = HashMap::new();
        query.insert("acl".to_string(), "".to_string());
        let headers = self.get_headers_with_auth("put", "/", Some(acl_header), None, Some(&query));
        let resp = Request::put(
            self.get_full_url_from_path("/").as_str(),
            Some(&query),
            Some(&headers),
            None,
            None,
            None as Option<Body>,
        )
        .await;
        self.make_response(resp)
    }
}