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
|
//! Transports for communicating with the docker daemon
extern crate hyper;
extern crate mime;
extern crate unix_socket;
use std::ops::DerefMut;
use hyper::Client;
use hyper::client;
use self::hyper::buffer::BufReader;
use self::hyper::http::{ parse_response };
use self::hyper::http::HttpReader::EofReader;
use self::hyper::header::ContentType;
use hyper::method::Method;
use self::mime::{ Attr, Mime, Value };
use self::mime::TopLevel::Application;
use self::mime::SubLevel::Json;
use std::io::{ Read, Result, Write };
use unix_socket::UnixStream;
#[doc(hidden)]
pub struct Body<'a> {
read: &'a mut Box<&'a mut Read>,
size: u64
}
impl<'a> Body<'a> {
/// Create a new body instance
pub fn new(read: &'a mut Box<&'a mut Read>, size: u64) -> Body<'a> {
Body { read: read, size: size }
}
}
/// Primary interface for communicating with docker daemon
pub trait Transport {
fn request(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<String>;
fn stream(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<Box<Read>>;
}
impl Transport for UnixStream {
fn request(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<String> {
let mut res = match self.stream(method, endpoint, body) {
Ok(r) => r,
Err(e) => panic!("failed request {:?}", e)
};
let mut body = String::new();
res.read_to_string(&mut body).map(|_| body)
}
fn stream(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<Box<Read>> {
let method_str = match method {
Method::Put => "PUT",
Method::Post => "POST",
Method::Delete => "DELETE",
_ => "GET"
};
let req = format!("{} {} HTTP/1.0\r\n\r\n", method_str, endpoint);
try!(self.write_all(req.as_bytes()));
// read the body -- https://github.com/hyperium/hyper/blob/06d072bca1b4af3507af370cbd0ca2ac8f64fc00/src/client/response.rs#L36-L74
let cloned = try!(self.try_clone());
let mut stream = BufReader::new(cloned);
let _ = parse_response(&mut stream).unwrap();
Ok(Box::new(EofReader(stream)))
}
}
impl Transport for (Client, String) {
fn request(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<String> {
let mut res = match self.stream(method, endpoint, body) {
Ok(r) => r,
Err(e) => panic!("failed request {:?}", e)
};
let mut body = String::new();
res.read_to_string(&mut body).map(|_| body)
}
fn stream(&mut self, method: Method, endpoint: &str, body: Option<Body>) -> Result<Box<Read>> {
let uri = format!("{}{}", self.1, endpoint);
let req = match method {
Method::Put => self.0.put(&uri[..]),
Method::Post => self.0.post(&uri[..]),
Method::Delete => self.0.delete(&uri[..]),
_ => self.0.get(&uri[..])
};
let embodied = match body {
Some(Body { read: r, size: l }) => {
let reader: &mut Read = *r.deref_mut();
let content_type: Mime = Mime(Application, Json, vec![(Attr::Charset, Value::Utf8)]);
req.header(ContentType(content_type)).body(client::Body::SizedBody(reader, l))
},
_ => req
};
let res = match embodied.send() {
Ok(r) => r,
Err(e) => panic!("failed request {:?}", e)
};
Ok(Box::new(res))
}
}
|