| //! HTTP response types. |
| //! |
| //! This module contains structs related to HTTP responses, notably the |
| //! `Response` type itself as well as a builder to create responses. Typically |
| //! you'll import the `http::Response` type rather than reaching into this |
| //! module itself. |
| //! |
| //! # Examples |
| //! |
| //! Creating a `Response` to return |
| //! |
| //! ``` |
| //! use http::{Request, Response, StatusCode}; |
| //! |
| //! fn respond_to(req: Request<()>) -> http::Result<Response<()>> { |
| //! let mut builder = Response::builder() |
| //! .header("Foo", "Bar") |
| //! .status(StatusCode::OK); |
| //! |
| //! if req.headers().contains_key("Another-Header") { |
| //! builder = builder.header("Another-Header", "Ack"); |
| //! } |
| //! |
| //! builder.body(()) |
| //! } |
| //! ``` |
| //! |
| //! A simple 404 handler |
| //! |
| //! ``` |
| //! use http::{Request, Response, StatusCode}; |
| //! |
| //! fn not_found(_req: Request<()>) -> http::Result<Response<()>> { |
| //! Response::builder() |
| //! .status(StatusCode::NOT_FOUND) |
| //! .body(()) |
| //! } |
| //! ``` |
| //! |
| //! Or otherwise inspecting the result of a request: |
| //! |
| //! ```no_run |
| //! use http::{Request, Response}; |
| //! |
| //! fn get(url: &str) -> http::Result<Response<()>> { |
| //! // ... |
| //! # panic!() |
| //! } |
| //! |
| //! let response = get("https://www.rust-lang.org/").unwrap(); |
| //! |
| //! if !response.status().is_success() { |
| //! panic!("failed to get a successful response status!"); |
| //! } |
| //! |
| //! if let Some(date) = response.headers().get("Date") { |
| //! // we've got a `Date` header! |
| //! } |
| //! |
| //! let body = response.body(); |
| //! // ... |
| //! ``` |
| |
| use std::any::Any; |
| use std::convert::TryFrom; |
| use std::fmt; |
| |
| use crate::header::{HeaderMap, HeaderName, HeaderValue}; |
| use crate::status::StatusCode; |
| use crate::version::Version; |
| use crate::{Extensions, Result}; |
| |
| /// Represents an HTTP response |
| /// |
| /// An HTTP response consists of a head and a potentially optional body. The body |
| /// component is generic, enabling arbitrary types to represent the HTTP body. |
| /// For example, the body could be `Vec<u8>`, a `Stream` of byte chunks, or a |
| /// value that has been deserialized. |
| /// |
| /// Typically you'll work with responses on the client side as the result of |
| /// sending a `Request` and on the server you'll be generating a `Response` to |
| /// send back to the client. |
| /// |
| /// # Examples |
| /// |
| /// Creating a `Response` to return |
| /// |
| /// ``` |
| /// use http::{Request, Response, StatusCode}; |
| /// |
| /// fn respond_to(req: Request<()>) -> http::Result<Response<()>> { |
| /// let mut builder = Response::builder() |
| /// .header("Foo", "Bar") |
| /// .status(StatusCode::OK); |
| /// |
| /// if req.headers().contains_key("Another-Header") { |
| /// builder = builder.header("Another-Header", "Ack"); |
| /// } |
| /// |
| /// builder.body(()) |
| /// } |
| /// ``` |
| /// |
| /// A simple 404 handler |
| /// |
| /// ``` |
| /// use http::{Request, Response, StatusCode}; |
| /// |
| /// fn not_found(_req: Request<()>) -> http::Result<Response<()>> { |
| /// Response::builder() |
| /// .status(StatusCode::NOT_FOUND) |
| /// .body(()) |
| /// } |
| /// ``` |
| /// |
| /// Or otherwise inspecting the result of a request: |
| /// |
| /// ```no_run |
| /// use http::{Request, Response}; |
| /// |
| /// fn get(url: &str) -> http::Result<Response<()>> { |
| /// // ... |
| /// # panic!() |
| /// } |
| /// |
| /// let response = get("https://www.rust-lang.org/").unwrap(); |
| /// |
| /// if !response.status().is_success() { |
| /// panic!("failed to get a successful response status!"); |
| /// } |
| /// |
| /// if let Some(date) = response.headers().get("Date") { |
| /// // we've got a `Date` header! |
| /// } |
| /// |
| /// let body = response.body(); |
| /// // ... |
| /// ``` |
| /// |
| /// Deserialize a response of bytes via json: |
| /// |
| /// ``` |
| /// # extern crate serde; |
| /// # extern crate serde_json; |
| /// # extern crate http; |
| /// use http::Response; |
| /// use serde::de; |
| /// |
| /// fn deserialize<T>(res: Response<Vec<u8>>) -> serde_json::Result<Response<T>> |
| /// where for<'de> T: de::Deserialize<'de>, |
| /// { |
| /// let (parts, body) = res.into_parts(); |
| /// let body = serde_json::from_slice(&body)?; |
| /// Ok(Response::from_parts(parts, body)) |
| /// } |
| /// # |
| /// # fn main() {} |
| /// ``` |
| /// |
| /// Or alternatively, serialize the body of a response to json |
| /// |
| /// ``` |
| /// # extern crate serde; |
| /// # extern crate serde_json; |
| /// # extern crate http; |
| /// use http::Response; |
| /// use serde::ser; |
| /// |
| /// fn serialize<T>(res: Response<T>) -> serde_json::Result<Response<Vec<u8>>> |
| /// where T: ser::Serialize, |
| /// { |
| /// let (parts, body) = res.into_parts(); |
| /// let body = serde_json::to_vec(&body)?; |
| /// Ok(Response::from_parts(parts, body)) |
| /// } |
| /// # |
| /// # fn main() {} |
| /// ``` |
| pub struct Response<T> { |
| head: Parts, |
| body: T, |
| } |
| |
| /// Component parts of an HTTP `Response` |
| /// |
| /// The HTTP response head consists of a status, version, and a set of |
| /// header fields. |
| pub struct Parts { |
| /// The response's status |
| pub status: StatusCode, |
| |
| /// The response's version |
| pub version: Version, |
| |
| /// The response's headers |
| pub headers: HeaderMap<HeaderValue>, |
| |
| /// The response's extensions |
| pub extensions: Extensions, |
| |
| _priv: (), |
| } |
| |
| /// An HTTP response builder |
| /// |
| /// This type can be used to construct an instance of `Response` through a |
| /// builder-like pattern. |
| #[derive(Debug)] |
| pub struct Builder { |
| inner: Result<Parts>, |
| } |
| |
| impl Response<()> { |
| /// Creates a new builder-style object to manufacture a `Response` |
| /// |
| /// This method returns an instance of `Builder` which can be used to |
| /// create a `Response`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response = Response::builder() |
| /// .status(200) |
| /// .header("X-Custom-Foo", "Bar") |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| #[inline] |
| pub fn builder() -> Builder { |
| Builder::new() |
| } |
| } |
| |
| impl<T> Response<T> { |
| /// Creates a new blank `Response` with the body |
| /// |
| /// The component ports of this response will be set to their default, e.g. |
| /// the ok status, no headers, etc. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response = Response::new("hello world"); |
| /// |
| /// assert_eq!(response.status(), StatusCode::OK); |
| /// assert_eq!(*response.body(), "hello world"); |
| /// ``` |
| #[inline] |
| pub fn new(body: T) -> Response<T> { |
| Response { |
| head: Parts::new(), |
| body: body, |
| } |
| } |
| |
| /// Creates a new `Response` with the given head and body |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response = Response::new("hello world"); |
| /// let (mut parts, body) = response.into_parts(); |
| /// |
| /// parts.status = StatusCode::BAD_REQUEST; |
| /// let response = Response::from_parts(parts, body); |
| /// |
| /// assert_eq!(response.status(), StatusCode::BAD_REQUEST); |
| /// assert_eq!(*response.body(), "hello world"); |
| /// ``` |
| #[inline] |
| pub fn from_parts(parts: Parts, body: T) -> Response<T> { |
| Response { |
| head: parts, |
| body: body, |
| } |
| } |
| |
| /// Returns the `StatusCode`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<()> = Response::default(); |
| /// assert_eq!(response.status(), StatusCode::OK); |
| /// ``` |
| #[inline] |
| pub fn status(&self) -> StatusCode { |
| self.head.status |
| } |
| |
| /// Returns a mutable reference to the associated `StatusCode`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let mut response: Response<()> = Response::default(); |
| /// *response.status_mut() = StatusCode::CREATED; |
| /// assert_eq!(response.status(), StatusCode::CREATED); |
| /// ``` |
| #[inline] |
| pub fn status_mut(&mut self) -> &mut StatusCode { |
| &mut self.head.status |
| } |
| |
| /// Returns a reference to the associated version. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<()> = Response::default(); |
| /// assert_eq!(response.version(), Version::HTTP_11); |
| /// ``` |
| #[inline] |
| pub fn version(&self) -> Version { |
| self.head.version |
| } |
| |
| /// Returns a mutable reference to the associated version. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let mut response: Response<()> = Response::default(); |
| /// *response.version_mut() = Version::HTTP_2; |
| /// assert_eq!(response.version(), Version::HTTP_2); |
| /// ``` |
| #[inline] |
| pub fn version_mut(&mut self) -> &mut Version { |
| &mut self.head.version |
| } |
| |
| /// Returns a reference to the associated header field map. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<()> = Response::default(); |
| /// assert!(response.headers().is_empty()); |
| /// ``` |
| #[inline] |
| pub fn headers(&self) -> &HeaderMap<HeaderValue> { |
| &self.head.headers |
| } |
| |
| /// Returns a mutable reference to the associated header field map. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// # use http::header::*; |
| /// let mut response: Response<()> = Response::default(); |
| /// response.headers_mut().insert(HOST, HeaderValue::from_static("world")); |
| /// assert!(!response.headers().is_empty()); |
| /// ``` |
| #[inline] |
| pub fn headers_mut(&mut self) -> &mut HeaderMap<HeaderValue> { |
| &mut self.head.headers |
| } |
| |
| /// Returns a reference to the associated extensions. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<()> = Response::default(); |
| /// assert!(response.extensions().get::<i32>().is_none()); |
| /// ``` |
| #[inline] |
| pub fn extensions(&self) -> &Extensions { |
| &self.head.extensions |
| } |
| |
| /// Returns a mutable reference to the associated extensions. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// # use http::header::*; |
| /// let mut response: Response<()> = Response::default(); |
| /// response.extensions_mut().insert("hello"); |
| /// assert_eq!(response.extensions().get(), Some(&"hello")); |
| /// ``` |
| #[inline] |
| pub fn extensions_mut(&mut self) -> &mut Extensions { |
| &mut self.head.extensions |
| } |
| |
| /// Returns a reference to the associated HTTP body. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<String> = Response::default(); |
| /// assert!(response.body().is_empty()); |
| /// ``` |
| #[inline] |
| pub fn body(&self) -> &T { |
| &self.body |
| } |
| |
| /// Returns a mutable reference to the associated HTTP body. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let mut response: Response<String> = Response::default(); |
| /// response.body_mut().push_str("hello world"); |
| /// assert!(!response.body().is_empty()); |
| /// ``` |
| #[inline] |
| pub fn body_mut(&mut self) -> &mut T { |
| &mut self.body |
| } |
| |
| /// Consumes the response, returning just the body. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::Response; |
| /// let response = Response::new(10); |
| /// let body = response.into_body(); |
| /// assert_eq!(body, 10); |
| /// ``` |
| #[inline] |
| pub fn into_body(self) -> T { |
| self.body |
| } |
| |
| /// Consumes the response returning the head and body parts. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response: Response<()> = Response::default(); |
| /// let (parts, body) = response.into_parts(); |
| /// assert_eq!(parts.status, StatusCode::OK); |
| /// ``` |
| #[inline] |
| pub fn into_parts(self) -> (Parts, T) { |
| (self.head, self.body) |
| } |
| |
| /// Consumes the response returning a new response with body mapped to the |
| /// return type of the passed in function. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// let response = Response::builder().body("some string").unwrap(); |
| /// let mapped_response: Response<&[u8]> = response.map(|b| { |
| /// assert_eq!(b, "some string"); |
| /// b.as_bytes() |
| /// }); |
| /// assert_eq!(mapped_response.body(), &"some string".as_bytes()); |
| /// ``` |
| #[inline] |
| pub fn map<F, U>(self, f: F) -> Response<U> |
| where |
| F: FnOnce(T) -> U, |
| { |
| Response { |
| body: f(self.body), |
| head: self.head, |
| } |
| } |
| } |
| |
| impl<T: Default> Default for Response<T> { |
| #[inline] |
| fn default() -> Response<T> { |
| Response::new(T::default()) |
| } |
| } |
| |
| impl<T: fmt::Debug> fmt::Debug for Response<T> { |
| fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
| f.debug_struct("Response") |
| .field("status", &self.status()) |
| .field("version", &self.version()) |
| .field("headers", self.headers()) |
| // omits Extensions because not useful |
| .field("body", self.body()) |
| .finish() |
| } |
| } |
| |
| impl Parts { |
| /// Creates a new default instance of `Parts` |
| fn new() -> Parts { |
| Parts { |
| status: StatusCode::default(), |
| version: Version::default(), |
| headers: HeaderMap::default(), |
| extensions: Extensions::default(), |
| _priv: (), |
| } |
| } |
| } |
| |
| impl fmt::Debug for Parts { |
| fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { |
| f.debug_struct("Parts") |
| .field("status", &self.status) |
| .field("version", &self.version) |
| .field("headers", &self.headers) |
| // omits Extensions because not useful |
| // omits _priv because not useful |
| .finish() |
| } |
| } |
| |
| impl Builder { |
| /// Creates a new default instance of `Builder` to construct either a |
| /// `Head` or a `Response`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// |
| /// let response = response::Builder::new() |
| /// .status(200) |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| #[inline] |
| pub fn new() -> Builder { |
| Builder::default() |
| } |
| |
| /// Set the HTTP status for this response. |
| /// |
| /// This function will configure the HTTP status code of the `Response` that |
| /// will be returned from `Builder::build`. |
| /// |
| /// By default this is `200`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// |
| /// let response = Response::builder() |
| /// .status(200) |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| pub fn status<T>(self, status: T) -> Builder |
| where |
| StatusCode: TryFrom<T>, |
| <StatusCode as TryFrom<T>>::Error: Into<crate::Error>, |
| { |
| self.and_then(move |mut head| { |
| head.status = TryFrom::try_from(status).map_err(Into::into)?; |
| Ok(head) |
| }) |
| } |
| |
| /// Set the HTTP version for this response. |
| /// |
| /// This function will configure the HTTP version of the `Response` that |
| /// will be returned from `Builder::build`. |
| /// |
| /// By default this is HTTP/1.1 |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// |
| /// let response = Response::builder() |
| /// .version(Version::HTTP_2) |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| pub fn version(self, version: Version) -> Builder { |
| self.and_then(move |mut head| { |
| head.version = version; |
| Ok(head) |
| }) |
| } |
| |
| /// Appends a header to this response builder. |
| /// |
| /// This function will append the provided key/value as a header to the |
| /// internal `HeaderMap` being constructed. Essentially this is equivalent |
| /// to calling `HeaderMap::append`. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// # use http::header::HeaderValue; |
| /// |
| /// let response = Response::builder() |
| /// .header("Content-Type", "text/html") |
| /// .header("X-Custom-Foo", "bar") |
| /// .header("content-length", 0) |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| pub fn header<K, V>(self, key: K, value: V) -> Builder |
| where |
| HeaderName: TryFrom<K>, |
| <HeaderName as TryFrom<K>>::Error: Into<crate::Error>, |
| HeaderValue: TryFrom<V>, |
| <HeaderValue as TryFrom<V>>::Error: Into<crate::Error>, |
| { |
| self.and_then(move |mut head| { |
| let name = <HeaderName as TryFrom<K>>::try_from(key).map_err(Into::into)?; |
| let value = <HeaderValue as TryFrom<V>>::try_from(value).map_err(Into::into)?; |
| head.headers.append(name, value); |
| Ok(head) |
| }) |
| } |
| |
| /// Get header on this response builder. |
| /// |
| /// When builder has error returns None. |
| /// |
| /// # Example |
| /// |
| /// ``` |
| /// # use http::Response; |
| /// # use http::header::HeaderValue; |
| /// let res = Response::builder() |
| /// .header("Accept", "text/html") |
| /// .header("X-Custom-Foo", "bar"); |
| /// let headers = res.headers_ref().unwrap(); |
| /// assert_eq!( headers["Accept"], "text/html" ); |
| /// assert_eq!( headers["X-Custom-Foo"], "bar" ); |
| /// ``` |
| pub fn headers_ref(&self) -> Option<&HeaderMap<HeaderValue>> { |
| self.inner.as_ref().ok().map(|h| &h.headers) |
| } |
| |
| /// Get header on this response builder. |
| /// when builder has error returns None |
| /// |
| /// # Example |
| /// |
| /// ``` |
| /// # use http::*; |
| /// # use http::header::HeaderValue; |
| /// # use http::response::Builder; |
| /// let mut res = Response::builder(); |
| /// { |
| /// let headers = res.headers_mut().unwrap(); |
| /// headers.insert("Accept", HeaderValue::from_static("text/html")); |
| /// headers.insert("X-Custom-Foo", HeaderValue::from_static("bar")); |
| /// } |
| /// let headers = res.headers_ref().unwrap(); |
| /// assert_eq!( headers["Accept"], "text/html" ); |
| /// assert_eq!( headers["X-Custom-Foo"], "bar" ); |
| /// ``` |
| pub fn headers_mut(&mut self) -> Option<&mut HeaderMap<HeaderValue>> { |
| self.inner.as_mut().ok().map(|h| &mut h.headers) |
| } |
| |
| /// Adds an extension to this builder |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// |
| /// let response = Response::builder() |
| /// .extension("My Extension") |
| /// .body(()) |
| /// .unwrap(); |
| /// |
| /// assert_eq!(response.extensions().get::<&'static str>(), |
| /// Some(&"My Extension")); |
| /// ``` |
| pub fn extension<T>(self, extension: T) -> Builder |
| where |
| T: Any + Send + Sync + 'static, |
| { |
| self.and_then(move |mut head| { |
| head.extensions.insert(extension); |
| Ok(head) |
| }) |
| } |
| |
| /// Get a reference to the extensions for this response builder. |
| /// |
| /// If the builder has an error, this returns `None`. |
| /// |
| /// # Example |
| /// |
| /// ``` |
| /// # use http::Response; |
| /// let res = Response::builder().extension("My Extension").extension(5u32); |
| /// let extensions = res.extensions_ref().unwrap(); |
| /// assert_eq!(extensions.get::<&'static str>(), Some(&"My Extension")); |
| /// assert_eq!(extensions.get::<u32>(), Some(&5u32)); |
| /// ``` |
| pub fn extensions_ref(&self) -> Option<&Extensions> { |
| self.inner.as_ref().ok().map(|h| &h.extensions) |
| } |
| |
| /// Get a mutable reference to the extensions for this response builder. |
| /// |
| /// If the builder has an error, this returns `None`. |
| /// |
| /// # Example |
| /// |
| /// ``` |
| /// # use http::Response; |
| /// let mut res = Response::builder().extension("My Extension"); |
| /// let mut extensions = res.extensions_mut().unwrap(); |
| /// assert_eq!(extensions.get::<&'static str>(), Some(&"My Extension")); |
| /// extensions.insert(5u32); |
| /// assert_eq!(extensions.get::<u32>(), Some(&5u32)); |
| /// ``` |
| pub fn extensions_mut(&mut self) -> Option<&mut Extensions> { |
| self.inner.as_mut().ok().map(|h| &mut h.extensions) |
| } |
| |
| /// "Consumes" this builder, using the provided `body` to return a |
| /// constructed `Response`. |
| /// |
| /// # Errors |
| /// |
| /// This function may return an error if any previously configured argument |
| /// failed to parse or get converted to the internal representation. For |
| /// example if an invalid `head` was specified via `header("Foo", |
| /// "Bar\r\n")` the error will be returned when this function is called |
| /// rather than when `header` was called. |
| /// |
| /// # Examples |
| /// |
| /// ``` |
| /// # use http::*; |
| /// |
| /// let response = Response::builder() |
| /// .body(()) |
| /// .unwrap(); |
| /// ``` |
| pub fn body<T>(self, body: T) -> Result<Response<T>> { |
| self.inner.map(move |head| { |
| Response { |
| head, |
| body, |
| } |
| }) |
| } |
| |
| // private |
| |
| fn and_then<F>(self, func: F) -> Self |
| where |
| F: FnOnce(Parts) -> Result<Parts> |
| { |
| Builder { |
| inner: self.inner.and_then(func), |
| } |
| } |
| } |
| |
| impl Default for Builder { |
| #[inline] |
| fn default() -> Builder { |
| Builder { |
| inner: Ok(Parts::new()), |
| } |
| } |
| } |
| |
| #[cfg(test)] |
| mod tests { |
| use super::*; |
| |
| #[test] |
| fn it_can_map_a_body_from_one_type_to_another() { |
| let response = Response::builder().body("some string").unwrap(); |
| let mapped_response = response.map(|s| { |
| assert_eq!(s, "some string"); |
| 123u32 |
| }); |
| assert_eq!(mapped_response.body(), &123u32); |
| } |
| } |