2025-04-20 14:57:01 +00:00 · 2020-05-05 23:18:05 +02:00 · 2020-05-05 23:18:05 +02:00 · 5587ded60d
commit 5587ded60d
parent 52679ad29d
45 changed files with 58 additions and 67 deletions
--- a/src/response.rs
+++ b/src/response.rs
@ -0,0 +1,64 @@
+use gotham::hyper::{Body, StatusCode};
+use mime::{Mime, APPLICATION_JSON};
+
+/// A response, used to create the final gotham response from.
+#[derive(Debug)]
+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())
+	}
+}