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 v2::*;

/// Configuration for a `Client`.
#[derive(Debug)]
pub struct Config {
    config: client::Config<hyper_rustls::HttpsConnector, hyper::Body>,
    handle: reactor::Handle,
    index: String,
    insecure_registry: bool,
    user_agent: Option<String>,
    username: Option<String>,
    password: Option<String>,
}

impl Config {
    /// Initialize `Config` with default values.
    pub fn default(handle: &reactor::Handle) -> Self {
        Self {
            config: hyper::client::Client::configure()
                .connector(hyper_rustls::HttpsConnector::new(4, handle)),
            handle: handle.clone(),
            index: "registry-1.docker.io".into(),
            insecure_registry: false,
            user_agent: Some(::USER_AGENT.to_owned()),
            username: None,
            password: None,
        }
    }

    /// Set registry service to use (vhost or IP).
    pub fn registry(mut self, reg: &str) -> Self {
        self.index = reg.to_owned();
        self
    }

    /// Whether to use an insecure HTTP connection to the registry.
    pub fn insecure_registry(mut self, insecure: bool) -> Self {
        self.insecure_registry = insecure;
        self
    }


    /// Set the user-agent to be used for registry authentication.
    pub fn user_agent(mut self, user_agent: Option<String>) -> Self {
        self.user_agent = user_agent;
        self
    }

    /// Set the username to be used for registry authentication.
    pub fn username(mut self, user: Option<String>) -> Self {
        self.username = user;
        self
    }

    /// Set the password to be used for registry authentication.
    pub fn password(mut self, password: Option<String>) -> Self {
        self.password = password;
        self
    }

    /// Read credentials from a JSON config file
    pub fn read_credentials<T: ::std::io::Read>(mut self, reader: T) -> Self {
        if let Ok(creds) = ::get_credentials(reader, &self.index) {
            self.username = creds.0;
            self.password = creds.1;
        };
        self
    }

    /// Return a `Client` to interact with a v2 registry.
    pub fn build(self) -> Result<Client> {
        let hclient = self.config.build(&self.handle);
        let base = match self.insecure_registry {
            false => "https://".to_string() + &self.index,
            true => "http://".to_string() + &self.index,
        };
        trace!("Built client for {:?}: endpoint {:?} - user {:?}",
               self.index,
               base,
               self.username);
        let creds = match (self.username, self.password) {
            (None, None) => None,
            (u, p) => Some((u.unwrap_or("".into()), p.unwrap_or("".into()))),
        };
        let c = Client {
            base_url: base,
            credentials: creds,
            hclient: hclient,
            index: self.index,
            user_agent: self.user_agent,
            token: None,
        };
        return Ok(c);
    }
}