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

use crate::upload;
use crate::utils::auth::Auth;
use crate::utils::qiniu_err::{QiniuErr, QiniuErrCode};
use crate::utils::verification::crc32;

use reqwest::blocking::multipart::{Form, Part};
use reqwest::blocking::Client;

pub struct DataUploader {
    auth: Auth,
}

impl DataUploader {
    pub fn new(auth: Auth) -> Self {
        Self { auth }
    }

    pub fn upload(
        &self,
        bucket: &str,
        key: &str,
        expired_seconds: u64,
        data: &'static [u8],
    ) -> Result<(), QiniuErr> {
        let form = self.build_form(bucket, key, expired_seconds, data)?;
        self.upload_form(form)
    }

    fn build_form(
        &self,
        bucket: &str,
        key: &str,
        expired_seconds: u64,
        data: &'static [u8],
    ) -> Result<Form, QiniuErr> {
        let token = self.auth.upload_token(bucket, key, expired_seconds);
        let crc = crc32(data);
        let file_part = self.build_form_file_part(key, upload::UPLOAD_MIME_TYPE, data)?;

        let form = Form::new()
            .text("token", token)
            .text("key", key.to_string())
            .text("ctc32", crc.to_string())
            .part("file", file_part);

        Ok(form)
    }

    fn build_form_file_part(
        &self,
        filename: &str,
        mime_type: &str,
        data: &'static [u8],
    ) -> Result<Part, QiniuErr> {
        let result = Part::bytes(data)
            .file_name(filename.to_string())
            .mime_str(mime_type);
        match result {
            Ok(file_part) => Ok(file_part),
            Err(err) => Err(QiniuErr {
                message: err.to_string(),
                code: QiniuErrCode::Inval,
            }),
        }
    }

    fn upload_form(&self, form: Form) -> Result<(), QiniuErr> {
        let result = Client::new()
            .post(upload::UPLOAD_URL)
            .multipart(form)
            .send();
        if let Err(err) = result {
            return Err(QiniuErr {
                message: err.to_string(),
                code: QiniuErrCode::Inval,
            });
        }

        let status_code = result.unwrap().status();
        if status_code.is_success() {
            Ok(())
        } else if status_code.is_server_error() {
            let message = status_code.as_str().to_string();
            Err(QiniuErr {
                message,
                code: QiniuErrCode::BadResponse,
            })
        } else {
            Err(QiniuErr {
                message: "".to_string(),
                code: QiniuErrCode::Unknown,
            })
        }
    }
}