downlowd 0.1.1

Download files with automatic retries and resumes.
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 http::{HeaderMap, HeaderName, HeaderValue, header::IntoHeaderName};

use crate::Error;

/// Insert a header into the header map, replacing any existing headers of the same name.
pub fn insert_header<K, V>(headers: &mut HeaderMap, key: K, value: V) -> Result<(), Error>
where
    K: IntoHeaderName,
    HeaderValue: TryFrom<V>,
    <HeaderValue as TryFrom<V>>::Error: Into<http::Error>,
{
    match value.try_into() {
        Ok(v) => {
            headers.insert(key, v);
        }
        Err(e) => {
            return Err(Error::InvalidHeader {
                cause: e.into().to_string(),
            });
        }
    };

    Ok(())
}

/// Append a header to a `HeaderMap`.
pub fn append_header<K, V>(headers: &mut HeaderMap, key: K, value: V) -> Result<(), Error>
where
    K: IntoHeaderName,
    HeaderValue: TryFrom<V>,
    <HeaderValue as TryFrom<V>>::Error: Into<http::Error>,
{
    match value.try_into() {
        Ok(v) => {
            headers.append(key, v);
        }
        Err(e) => {
            return Err(Error::InvalidHeader {
                cause: e.into().to_string(),
            });
        }
    };

    Ok(())
}

/// Append all values from `new_headers` to `headers`.
pub fn append_all_headers<T>(headers: &mut HeaderMap<T>, new_headers: HeaderMap<T>) {
    let mut key = HeaderName::from_static("x-placeholder");
    for (k, value) in new_headers.into_iter() {
        if let Some(k) = k {
            key = k;
        }
        headers.append(&key, value);
    }
}

#[cfg(test)]
mod tests {
    use http::HeaderValue;

    use super::*;

    #[test]
    fn should_insert_headers() {
        let mut headers = HeaderMap::new();
        insert_header(&mut headers, "a", "a").unwrap();
        insert_header(&mut headers, "a", "b").unwrap();
        assert_eq!(headers.get("a").unwrap(), "b");
    }

    #[test]
    fn should_append_headers() {
        let mut headers = HeaderMap::new();
        headers.insert("a", HeaderValue::from_static("a"));

        let mut new_headers = HeaderMap::new();
        new_headers.insert("b", HeaderValue::from_static("b"));

        append_all_headers(&mut headers, new_headers);

        assert_eq!(headers.get("a").unwrap(), "a");
        assert_eq!(headers.get("b").unwrap(), "b");
    }

    #[test]
    fn should_not_remove_existing_headers() {
        let mut headers = HeaderMap::new();
        headers.insert("a", HeaderValue::from_static("a"));

        let mut new_headers = HeaderMap::new();
        new_headers.insert("a", HeaderValue::from_static("b"));

        append_all_headers(&mut headers, new_headers);

        let mut iter = headers.get_all("a").iter();
        assert_eq!(iter.next().unwrap(), "a");
        assert_eq!(iter.next().unwrap(), "b");
        assert_eq!(iter.next(), None);
    }
}