Break things into modules

2026-08-29 00:00:18 +02:00 · 2021-05-30 13:24:03 +02:00
parent e1fa8d4fed
commit f4268471b6
7 changed files with 848 additions and 784 deletions
@@ -0,0 +1,79 @@
+use crate::{Body, Error};
+use bytes::Bytes;
+use http::{header, HeaderValue, Response};
+use serde::Serialize;
+
+pub trait IntoResponse<B> {
+    fn into_response(self) -> Result<Response<B>, Error>;
+}
+
+impl<B> IntoResponse<B> for Response<B> {
+    fn into_response(self) -> Result<Response<B>, Error> {
+        Ok(self)
+    }
+}
+
+impl IntoResponse<Body> for &'static str {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for String {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for Bytes {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for &'static [u8] {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for Vec<u8> {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for std::borrow::Cow<'static, str> {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+impl IntoResponse<Body> for std::borrow::Cow<'static, [u8]> {
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        Ok(Response::new(Body::from(self)))
+    }
+}
+
+pub struct Json<T>(pub T);
+
+impl<T> IntoResponse<Body> for Json<T>
+where
+    T: Serialize,
+{
+    fn into_response(self) -> Result<Response<Body>, Error> {
+        let bytes = serde_json::to_vec(&self.0).map_err(Error::SerializeResponseBody)?;
+        let len = bytes.len();
+        let mut res = Response::new(Body::from(bytes));
+
+        res.headers_mut().insert(
+            header::CONTENT_TYPE,
+            HeaderValue::from_static("application/json"),
+        );
+
+        res.headers_mut()
+            .insert(header::CONTENT_LENGTH, HeaderValue::from(len));
+
+        Ok(res)
+    }
+}