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

use crate::oss;

use self::builders::{GetBucketRefererBuilder, PutBucketRefererBuilder};

pub mod builders {
    use crate::oss::{
        self,
        api::{self, ApiResponseFrom},
        entities::referer::RefererConfiguration,
        http,
    };

    #[derive(Debug)]
    pub struct PutBucketRefererBuilder<'a> {
        client: &'a oss::Client<'a>,
        config: RefererConfiguration,
    }

    impl<'a> PutBucketRefererBuilder<'a> {
        pub fn new(cilent: &'a oss::Client) -> Self {
            Self {
                client: cilent,
                config: RefererConfiguration::default(),
            }
        }

        pub fn with_config(mut self, value: RefererConfiguration) -> Self {
            self.config = value;
            self
        }

        fn config(&self) -> String {
            quick_xml::se::to_string(&self.config).unwrap()
        }

        pub async fn execute(&self) -> api::ApiResult {
            let res = format!("/{}/?{}", self.client.bucket(), "referer");
            let url = format!("{}/?{}", self.client.base_url(), "referer");
            let config = self.config();
            let data = oss::Bytes::from(config);

            let resp = self
                .client
                .request
                .task()
                .with_url(&url)
                .with_resource(&res)
                .with_method(http::Method::PUT)
                .with_body(data)
                .execute()
                .await?;

            Ok(ApiResponseFrom(resp).to_empty().await)
        }
    }

    pub struct GetBucketRefererBuilder<'a> {
        client: &'a oss::Client<'a>,
    }

    impl<'a> GetBucketRefererBuilder<'a> {
        pub fn new(cilent: &'a oss::Client) -> Self {
            Self { client: cilent }
        }

        pub async fn execute(&self) -> api::ApiResult<RefererConfiguration> {
            let res = format!("/{}/?{}", self.client.options.bucket, "referer");
            let url = format!("{}?{}", self.client.options.base_url(), "referer");

            let resp = self
                .client
                .request
                .task()
                .with_url(&url)
                .with_resource(&res)
                .execute()
                .await?;
            Ok(ApiResponseFrom(resp).to_type().await)
        }
    }
}

/// # 防盗链`Referer``
#[allow(non_snake_case)]
impl<'a> oss::Client<'a> {
    /// 调用PutBucketReferer接口设置存储空间`Bucket`级别的Referer访问白名单以及黑名单
    ///
    /// - [official docs](https://help.aliyun.com/zh/oss/developer-reference/putbucketreferer)
    /// - [xtoss example](https://github.com/isme-sun/xt_oss/blob/main/examples/api_bucket_referer_put.rs)
    pub fn PutBucketReferer(&self) -> PutBucketRefererBuilder {
        PutBucketRefererBuilder::new(self)
    }

    /// GetBucketReferer接口用于查看存储空间`Bucket`的防盗链`Referer`相关配置。
    ///
    /// - [official docs](https://help.aliyun.com/zh/oss/developer-reference/getbucketreferer)
    /// - [xtoss example](https://github.com/isme-sun/xt_oss/blob/main/examples/api_bucket_referer_get.rs)
    pub fn GetBucketReferer(&self) -> GetBucketRefererBuilder {
        GetBucketRefererBuilder::new(self)
    }
}