2020-05-01 14:48:11 +00:00
|
|
|
use gotham::hyper::{Body, StatusCode};
|
|
|
|
|
use mime::{Mime, APPLICATION_JSON};
|
|
|
|
|
|
|
|
|
|
/// A response, used to create the final gotham response from.
|
2020-05-03 18:21:50 +02:00
|
|
|
#[derive(Debug)]
|
2020-05-01 14:48:11 +00:00
|
|
|
pub struct Response
|
|
|
|
|
{
|
|
|
|
|
pub status : StatusCode,
|
|
|
|
|
pub body : Body,
|
|
|
|
|
pub mime : Option<Mime>
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Response
|
|
|
|
|
{
|
|
|
|
|
/// Create a new `Response` from raw data.
|
|
|
|
|
pub fn new<B : Into<Body>>(status : StatusCode, body : B, mime : Option<Mime>) -> Self
|
|
|
|
|
{
|
|
|
|
|
Self {
|
|
|
|
|
status,
|
|
|
|
|
body: body.into(),
|
|
|
|
|
mime
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Create a `Response` with mime type json from already serialized data.
|
|
|
|
|
pub fn json<B : Into<Body>>(status : StatusCode, body : B) -> Self
|
|
|
|
|
{
|
|
|
|
|
Self {
|
|
|
|
|
status,
|
|
|
|
|
body: body.into(),
|
|
|
|
|
mime: Some(APPLICATION_JSON)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Create a _204 No Content_ `Response`.
|
|
|
|
|
pub fn no_content() -> Self
|
|
|
|
|
{
|
|
|
|
|
Self {
|
|
|
|
|
status: StatusCode::NO_CONTENT,
|
|
|
|
|
body: Body::empty(),
|
|
|
|
|
mime: None
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Create an empty _403 Forbidden_ `Response`.
|
|
|
|
|
pub fn forbidden() -> Self
|
|
|
|
|
{
|
|
|
|
|
Self {
|
|
|
|
|
status: StatusCode::FORBIDDEN,
|
|
|
|
|
body: Body::empty(),
|
|
|
|
|
mime: None
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
|
pub(crate) fn full_body(mut self) -> Result<Vec<u8>, <Body as gotham::hyper::body::HttpBody>::Error>
|
|
|
|
|
{
|
|
|
|
|
use futures_executor::block_on;
|
|
|
|
|
use gotham::hyper::body::to_bytes;
|
|
|
|
|
|
|
|
|
|
let bytes : &[u8] = &block_on(to_bytes(&mut self.body))?;
|
|
|
|
|
Ok(bytes.to_vec())
|
|
|
|
|
}
|
|
|
|
|
}
|
