1
0
Fork 0
mirror of https://gitlab.com/famedly/conduit.git synced 2024-11-10 01:01:06 +01:00
conduit/src/ruma_wrapper.rs

88 lines
2.6 KiB
Rust
Raw Normal View History

2020-02-15 22:42:21 +01:00
use {
rocket::data::{FromDataSimple, Outcome},
rocket::http::Status,
rocket::response::Responder,
rocket::Request,
rocket::{Data, Outcome::*},
2020-02-18 22:07:57 +01:00
ruma_client_api::error::Error,
std::fmt::Debug,
2020-02-15 22:42:21 +01:00
std::ops::Deref,
std::{
convert::{TryFrom, TryInto},
io::{Cursor, Read},
},
};
const MESSAGE_LIMIT: u64 = 65535;
pub struct Ruma<T>(pub T);
2020-02-18 22:07:57 +01:00
impl<T: TryFrom<http::Request<Vec<u8>>>> FromDataSimple for Ruma<T>
where
T::Error: Debug,
{
2020-02-15 22:42:21 +01:00
type Error = ();
fn from_data(request: &Request, data: Data) -> Outcome<Self, Self::Error> {
2020-02-18 22:07:57 +01:00
let mut http_request = http::Request::builder()
.uri(request.uri().to_string())
.method(&*request.method().to_string());
2020-02-15 22:42:21 +01:00
for header in request.headers().iter() {
http_request = http_request.header(header.name.as_str(), &*header.value);
}
2020-02-18 22:07:57 +01:00
let mut handle = data.open().take(MESSAGE_LIMIT);
let mut body = Vec::new();
handle.read_to_end(&mut body).unwrap();
2020-02-15 22:42:21 +01:00
let http_request = http_request.body(body).unwrap();
2020-02-18 22:07:57 +01:00
log::info!("{:?}", http_request);
2020-02-15 22:42:21 +01:00
match T::try_from(http_request) {
Ok(r) => Success(Ruma(r)),
2020-02-18 22:07:57 +01:00
Err(e) => {
log::error!("{:?}", e);
Failure((Status::InternalServerError, ()))
}
2020-02-15 22:42:21 +01:00
}
}
}
impl<T> Deref for Ruma<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
&self.0
}
}
2020-02-18 22:07:57 +01:00
pub struct MatrixResult<T>(pub std::result::Result<T, Error>);
impl<T: TryInto<http::Response<Vec<u8>>>> TryInto<http::Response<Vec<u8>>> for MatrixResult<T> {
type Error = T::Error;
fn try_into(self) -> Result<http::Response<Vec<u8>>, T::Error> {
match self.0 {
Ok(t) => t.try_into(),
Err(e) => Ok(e.into()),
}
}
}
impl<'r, T: TryInto<http::Response<Vec<u8>>>> Responder<'r> for MatrixResult<T> {
2020-02-15 22:42:21 +01:00
fn respond_to(self, _: &Request) -> rocket::response::Result<'r> {
2020-02-18 22:07:57 +01:00
let http_response: Result<http::Response<_>, _> = self.try_into();
match http_response {
2020-02-15 22:42:21 +01:00
Ok(http_response) => {
let mut response = rocket::response::Response::build();
response.sized_body(Cursor::new(http_response.body().clone()));
for header in http_response.headers() {
response
.raw_header(header.0.to_string(), header.1.to_str().unwrap().to_owned());
}
response.ok()
}
Err(_) => Err(Status::InternalServerError),
}
}
}