rusty-cat 0.2.2

Async HTTP client for resumable file upload and download.
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

use reqwest::header::{HeaderMap, ACCEPT, RANGE};
use reqwest::Url;

use super::constants::DEFAULT_RANGE_ACCEPT;
use super::signing::{header_value, param_error, signed_headers};
use crate::{BreakpointDownload, DownloadHeadCtx, DownloadRangeGetCtx, MeowError, TransferTask};

/// Aliyun OSS direct range download protocol using AccessKey signing.
#[derive(Clone)]
pub struct AliOssDirectDownload {
    bucket: String,
    access_key_id: String,
    access_key_secret: String,
    region: String,
}

impl AliOssDirectDownload {
    pub fn new(
        bucket: impl Into<String>,
        access_key_id: impl Into<String>,
        access_key_secret: impl Into<String>,
        region: impl Into<String>,
    ) -> Self {
        Self {
            bucket: bucket.into(),
            access_key_id: access_key_id.into(),
            access_key_secret: access_key_secret.into(),
            region: region.into(),
        }
    }

    fn object_canonical_uri_from_task_url(&self, task: &TransferTask) -> Result<String, MeowError> {
        let url = Url::parse(task.url()).map_err(param_error)?;
        Ok(format!("/{}{}", self.bucket, url.path()))
    }

    fn apply_signed_headers(
        &self,
        task: &TransferTask,
        method: &str,
        base: &mut HeaderMap,
    ) -> Result<(), MeowError> {
        let headers = signed_headers(
            method,
            self.object_canonical_uri_from_task_url(task)?.as_str(),
            None,
            &[],
            None,
            self.access_key_id.as_str(),
            self.access_key_secret.as_str(),
            self.region.as_str(),
        )?;
        for (k, v) in headers {
            if let Some(k) = k {
                base.insert(k, v);
            }
        }
        Ok(())
    }
}

impl BreakpointDownload for AliOssDirectDownload {
    fn merge_head_headers(&self, ctx: DownloadHeadCtx<'_>) -> Result<(), MeowError> {
        self.apply_signed_headers(ctx.task, "HEAD", ctx.base)
    }

    fn merge_range_get_headers(&self, ctx: DownloadRangeGetCtx<'_>) -> Result<(), MeowError> {
        ctx.base.insert(RANGE, header_value(ctx.range_value)?);
        if !ctx.base.contains_key(ACCEPT) {
            ctx.base.insert(ACCEPT, header_value(DEFAULT_RANGE_ACCEPT)?);
        }
        self.apply_signed_headers(ctx.task, "GET", ctx.base)
    }
}