Files
reqwest-impersonate/src/client.rs
Sean McArthur 9bfdcec052 init
2016-06-30 17:23:51 -07:00

88 lines
1.8 KiB
Rust

use std::io::{self, Read};
use hyper::{Method, StatusCode, HttpVersion, Url};
use hyper::header::Headers;
use super::sync;
pub struct Client {
inner: sync::Client,
}
impl Client {
pub fn new() -> Client {
Client {
inner: sync::Client::new(),
}
}
pub fn get(&self, url: &str) -> RequestBuilder {
self.request(Method::Get, Url::parse(url).unwrap())
}
pub fn request(&self, method: Method, url: Url) -> RequestBuilder {
debug!("request {:?} \"{}\"", method, url);
RequestBuilder {
client: self,
method: method,
url: url,
version: HttpVersion::Http11,
headers: Headers::new(),
body: None,
}
}
}
pub struct RequestBuilder<'a> {
client: &'a Client,
method: Method,
url: Url,
version: HttpVersion,
headers: Headers,
body: Option<()>,
}
impl<'a> RequestBuilder<'a> {
pub fn body(mut self, body: ()) -> RequestBuilder<'a> {
self.body = Some(body);
self
}
pub fn send(mut self) -> Result<Response, String> {
self.headers.set(::hyper::header::ContentLength(0));
let req = try!(self.client.inner.request(self.method, self.url, self.version, self.headers));
let res = try!(req.end().map_err(|e| format!("RequestError: end: {}", e)));
Ok(Response {
inner: res
})
}
}
pub struct Response {
inner: sync::Response,
}
impl Response {
pub fn status(&self) -> &StatusCode {
&self.inner.status
}
pub fn headers(&self) -> &Headers {
&self.inner.headers
}
pub fn version(&self) -> &HttpVersion {
&self.inner.version
}
}
impl Read for Response {
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
self.inner.read(buf)
}
}