From 96fb1e701b42eff6d4fac218017667af1e633269 Mon Sep 17 00:00:00 2001
From: Hatter Jiang <jht5945@gmail.com>
Date: Mon, 25 Nov 2019 00:44:47 +0800
Subject: [PATCH] add OSSClient

---
 src/main.rs | 68 ++++++++++++++++++++++++++++++-----------------------
 1 file changed, 39 insertions(+), 29 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 08ce719..1dc6417 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -12,46 +12,56 @@ use crypto::{
 
 pub const OSS_VERB_GET: &str = "GET";
 
-// TODO impl
-pub struct OSSConfig {
+// https://help.aliyun.com/document_detail/31952.html
+pub struct OSSClient {
     pub endpoint: String,
     pub access_key_id: String,
     pub access_key_secret: String,
 }
 
+impl OSSClient {
+    pub fn new(endpoint: &str, access_key_id: &str, access_key_secret: &str) -> OSSClient {
+        OSSClient {
+            endpoint: endpoint.to_string(),
+            access_key_id: access_key_id.to_string(),
+            access_key_secret: access_key_secret.to_string(),
+        }
+    }
+
+    pub fn generate_signed_url(&self, verb: &str, bucket_name: &str, key: &str, expire_in_seconds: u64, is_https: bool) -> String {
+        let mut signed_url = String::new();
+        signed_url.push_str(if is_https { "https://" } else { "http://" });
+        signed_url.push_str(bucket_name);
+        signed_url.push_str(".");
+        signed_url.push_str(self.endpoint.as_str());
+        signed_url.push_str("/");
+        signed_url.push_str(key);
+    
+        let current_secs = get_current_secs();
+        let expire_secs = current_secs + expire_in_seconds;
+    
+        signed_url.push_str("?Expires=");
+        signed_url.push_str(expire_secs.to_string().as_str());
+        signed_url.push_str("&OSSAccessKeyId=");
+        signed_url.push_str(&urlencoding::encode(self.access_key_id.as_str()));
+        signed_url.push_str("&Signature=");
+    
+        let to_be_signed = get_to_be_signed(verb, expire_secs, bucket_name, key);
+        let signature = to_base64(calc_hmac_sha1(self.access_key_secret.as_bytes(), to_be_signed.as_bytes()));
+        signed_url.push_str(&urlencoding::encode(signature.as_str()));
+    
+        signed_url
+    }
+}
+
 fn main() {
     println!("Hello, world!");
 
     println!("{}", SystemTime::now().duration_since(SystemTime::UNIX_EPOCH).unwrap().as_secs());
 
-    println!("{}", generate_oss_signed_url("shanghai.aliyuncs.com", "aaa", "bbb", "aaaa", "b.txt", 1_u64, true));
-}
+    let oss_client = OSSClient::new("shanghai.aliyuncs.com", "aaa", "bbb");
 
-// https://help.aliyun.com/document_detail/31952.html
-fn generate_oss_signed_url(endpoint: &str, access_key_id: &str, access_key_secret: &str,
-                            bucket_name: &str, key: &str, expire_in_seconds: u64, is_https: bool) -> String {
-    let mut signed_url = String::new();
-    signed_url.push_str(if is_https { "https://" } else { "http://" });
-    signed_url.push_str(bucket_name);
-    signed_url.push_str(".");
-    signed_url.push_str(endpoint);
-    signed_url.push_str("/");
-    signed_url.push_str(key);
-
-    let current_secs = get_current_secs();
-    let expire_secs = current_secs + expire_in_seconds;
-
-    signed_url.push_str("?Expires=");
-    signed_url.push_str(expire_secs.to_string().as_str());
-    signed_url.push_str("&OSSAccessKeyId=");
-    signed_url.push_str(&urlencoding::encode(access_key_id));
-    signed_url.push_str("&Signature=");
-
-    let to_be_signed = get_to_be_signed(OSS_VERB_GET, expire_secs, bucket_name, key);
-    let signature = to_base64(calc_hmac_sha1(access_key_secret.as_bytes(), to_be_signed.as_bytes()));
-    signed_url.push_str(&urlencoding::encode(signature.as_str()));
-
-    signed_url
+    println!("{}", oss_client.generate_signed_url(OSS_VERB_GET, "aaaa", "b.txt", 1_u64, true));
 }
 
 fn to_base64(mac_result: MacResult) -> String {