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

//! Some types used to configure a `Client` instance.

use std::time::Duration;

/// Configures some parameters for a `Client` instance.
#[derive(Clone, Debug)]
pub struct ClientConfig {
    /// Enable auto gzip decompression checking the `ContentEncoding` response header.
    ///
    /// Default is enabled.
    pub gzip: bool,

    /// `RedirectPolicy` for this client.
    ///
    /// Default will follow up to 10 redirects.
    pub redirect: RedirectPolicy,

    /// Enable or disable automatic setting of the `Referer` header.
    ///
    /// Default is true.
    ///
    /// TODO: Not actually implemented yet.
    pub referer: bool,

    /// Timeout for both the read and write operations of a client.
    pub timeout: Option<Duration>,
}

impl Default for ClientConfig {
    fn default() -> Self {
        ClientConfig {
            gzip: true,
            redirect: RedirectPolicy::default(),
            referer: true,
            timeout: None,
        }
    }
}

impl ClientConfig {
    pub fn new() -> Self {
        Self::default()
    }

    // TODO implement builder pattern
}

/// Specifies how to hande redirects.
#[derive(Clone, Debug)]
pub enum RedirectPolicy {
    Limit(usize),
    None,
}

impl Default for RedirectPolicy {
    fn default() -> Self {
        RedirectPolicy::Limit(10)
    }
}

impl From<RedirectPolicy> for ::reqwest::RedirectPolicy {
    fn from(p: RedirectPolicy) -> Self {
        match p {
            RedirectPolicy::Limit(n) => ::reqwest::RedirectPolicy::limited(n),
            RedirectPolicy::None => ::reqwest::RedirectPolicy::none(),
        }
    }
}