ve-tos-rust-sdk 2.5.0

volcengine offical tos rust sdk
Documentation 
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

/*
 * Copyright (2024) Volcengine
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
use std::env;
use std::fmt::Debug;
use std::marker::PhantomData;

pub trait Credentials {
    fn ak(&self) -> &str;
    fn sk(&self) -> &str;
    fn security_token(&self) -> &str;

    fn new(ak: impl Into<String>, sk: impl Into<String>, security_token: impl Into<String>) -> Self;
}


pub trait CredentialsProvider<C> where C: Credentials {
    fn credentials(&self) -> &C;

    fn new(c: C) -> Self;
}


#[derive(Debug, Clone, PartialEq, Default)]
pub struct CommonCredentialsProvider<C> {
    pub(crate) credentials: C,
}

impl<C> CredentialsProvider<C> for CommonCredentialsProvider<C> where C: Credentials {
    fn credentials(&self) -> &C {
        &self.credentials
    }

    fn new(c: C) -> Self {
        CommonCredentialsProvider {
            credentials: c,
        }
    }
}


#[derive(Debug, Clone, PartialEq, Default)]
pub struct CommonCredentials {
    pub(crate) ak: String,
    pub(crate) sk: String,
    pub(crate) security_token: String,
}

impl Credentials for CommonCredentials {
    fn ak(&self) -> &str {
        &self.ak
    }

    fn sk(&self) -> &str {
        &self.sk
    }

    fn security_token(&self) -> &str {
        &self.security_token
    }

    fn new(ak: impl Into<String>, sk: impl Into<String>, security_token: impl Into<String>) -> Self {
        CommonCredentials {
            ak: ak.into(),
            sk: sk.into(),
            security_token: security_token.into(),
        }
    }
}


pub(crate) struct StaticCredentialsProvider<P, C> {
    pub(crate) p: PhantomData<P>,
    pub(crate) c: PhantomData<C>,
}

impl<P, C> StaticCredentialsProvider<P, C> where P: CredentialsProvider<C> + Debug + Default,
                                                 C: Credentials + Debug + Default {
    pub(crate) fn new(ak: impl Into<String>, sk: impl Into<String>, security_token: impl Into<String>) -> P {
        let mut ak = ak.into().trim().to_owned();
        let mut sk = sk.into().trim().to_owned();
        let mut security_token = security_token.into().trim().to_owned();
        if ak == "" {
            sk = "".to_owned();
            security_token = "".to_owned();
        } else if sk == "" {
            ak = "".to_owned();
            security_token = "".to_owned();
        }

        P::new(C::new(ak, sk, security_token))
    }
}

pub(crate) struct EnvCredentialsProvider<P, C> {
    pub(crate) p: PhantomData<P>,
    pub(crate) c: PhantomData<C>,
}

impl<P, C> EnvCredentialsProvider<P, C> where P: CredentialsProvider<C> + Debug + Default,
                                              C: Credentials + Debug + Default {
    pub(crate) fn new() -> P {
        let mut c = C::default();
        if let Ok(ak) = env::var("TOS_ACCESS_KEY") {
            if let Ok(sk) = env::var("TOS_SECRET_KEY") {
                if let Ok(security_token) = env::var("TOS_SECURITY_TOKEN") {
                    c = C::new(ak, sk, security_token)
                } else {
                    c = C::new(ak, sk, "".to_owned())
                }
            }
        }
        P::new(c)
    }
}