patron 0.2.1

A wrapper around the hyper.rs library to allow for targeted clients to specific remote APIs. This library should be useful on it's own or as a building block for specific remote API wrappers.
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149

use error;
use hyper;
use hyper_rustls;
use response;
use std;
use types;
use url;


#[derive(Debug, Clone)]
pub struct Request<'a> {
  client: std::sync::Arc<hyper::Client>,
  method: types::Method,
  url: url::Url<'a>,
  headers: hyper::header::Headers,
  body: String,
}


impl<'a> Request<'a> {
  pub fn new() -> Request<'a> {
    Request {
      client: std::sync::Arc::new(hyper::Client::new()),
      method: types::Method::Get,
      url: url::Url::new(),
      headers: hyper::header::Headers::new(),
      body: String::new(),
    }
  }


  pub fn get_scheme(&self) -> types::Scheme {
    self.url.scheme.clone()
  }


  pub fn set_url(&mut self, url: url::Url<'a>) -> &mut Request<'a> {
    let client = match url.scheme {
      types::Scheme::Http => hyper::Client::new(),
      types::Scheme::Https => {
        let tls = hyper_rustls::TlsClient::new();
        let conn = hyper::net::HttpsConnector::new(tls);

        hyper::Client::with_connector(conn)
      }
    };

    self.client = std::sync::Arc::new(client);
    self.url = url;

    self
  }


  pub fn set_method(&mut self, method: types::Method) -> &mut Request<'a> {
    self.method = method;
    self
  }


  pub fn set_headers(
    &mut self,
    headers: hyper::header::Headers,
  ) -> &mut Request<'a> {
    self.headers = headers;
    self
  }


  pub fn set_body<S>(&mut self, body: S) -> &mut Request<'a>
    where S: Into<String>
  {
    self.body = body.into();
    self
  }


  pub fn add_header<H>(&mut self, header: H) -> &mut Request<'a>
    where H: hyper::header::Header + hyper::header::HeaderFormat
  {
    self.headers.set(header);
    self
  }



  pub fn add_path<S>(&mut self, path: S) -> &mut Request<'a>
    where S: Into<String>
  {
    self.url.add_path(path.into());
    self
  }


  pub fn add_query_param<K, S>(&mut self, key: K, val: S) -> &mut Request<'a>
    where K: Into<String>,
          S: Into<std::borrow::Cow<'a, str>>
  {
    self.url.add_query_param(key, val);
    self
  }


  pub fn basic_auth<U, P>(&mut self, u: U, p: Option<P>) -> &mut Request<'a>
    where U: Into<String>,
          P: Into<String>
  {
    let auth = hyper::header::Basic {
      username: u.into(),
      password: p.map(|s| s.into()),
    };
    self.add_header(hyper::header::Authorization(auth))
  }


  pub fn set_bearer_token<T>(&mut self, token: T) -> &mut Request<'a>
    where T: Into<String>
  {
    let auth = hyper::header::Bearer { token: token.into() };

    self.add_header(hyper::header::Authorization(auth))
  }


  pub fn send(&self) -> std::result::Result<response::Response, error::Error> {

    let url = try!(self.url.to_url());

    let request = match self.method {
      types::Method::Get => self.client.get(url),
      types::Method::Post => self.client.post(url),
      types::Method::Put => self.client.put(url),
    };

    request
      .headers(self.headers.clone())
      .body(&self.body)
      .send()
      .map_err(error::Error::from)
      .map(response::Response::from)
      .and_then(
        |res| match res.ok {
          true => Ok(res),
          false => Err(error::Error::from(res)),
        },
      )

  }
}