|
|
|
@ -1,7 +1,7 @@ |
|
|
|
use rocket::{ |
|
|
|
use rocket::{ |
|
|
|
data::{FromDataSimple, Outcome}, |
|
|
|
data::{Data, FromData, FromDataFuture, Transform, Transformed, TransformFuture}, |
|
|
|
http::Status, |
|
|
|
http::Status, |
|
|
|
response::Responder, |
|
|
|
response::{self, Responder}, |
|
|
|
Outcome::*, |
|
|
|
Outcome::*, |
|
|
|
Request, State, |
|
|
|
Request, State, |
|
|
|
}; |
|
|
|
}; |
|
|
|
@ -13,9 +13,10 @@ use ruma_client_api::error::Error; |
|
|
|
use ruma_identifiers::UserId; |
|
|
|
use ruma_identifiers::UserId; |
|
|
|
use std::{ |
|
|
|
use std::{ |
|
|
|
convert::{TryFrom, TryInto}, |
|
|
|
convert::{TryFrom, TryInto}, |
|
|
|
io::{Cursor, Read}, |
|
|
|
io::Cursor, |
|
|
|
ops::Deref, |
|
|
|
ops::Deref, |
|
|
|
}; |
|
|
|
}; |
|
|
|
|
|
|
|
use tokio::io::AsyncReadExt; |
|
|
|
|
|
|
|
|
|
|
|
const MESSAGE_LIMIT: u64 = 65535; |
|
|
|
const MESSAGE_LIMIT: u64 = 65535; |
|
|
|
|
|
|
|
|
|
|
|
@ -27,7 +28,7 @@ pub struct Ruma<T: Outgoing> { |
|
|
|
pub json_body: serde_json::Value, |
|
|
|
pub json_body: serde_json::Value, |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
impl<T: Endpoint> FromDataSimple for Ruma<T> |
|
|
|
impl<'a, T: Endpoint> FromData<'a> for Ruma<T> |
|
|
|
where |
|
|
|
where |
|
|
|
// We need to duplicate Endpoint's where clauses because the compiler is not smart enough yet.
|
|
|
|
// We need to duplicate Endpoint's where clauses because the compiler is not smart enough yet.
|
|
|
|
// See https://github.com/rust-lang/rust/issues/54149
|
|
|
|
// See https://github.com/rust-lang/rust/issues/54149
|
|
|
|
@ -38,10 +39,22 @@ where |
|
|
|
>, |
|
|
|
>, |
|
|
|
{ |
|
|
|
{ |
|
|
|
type Error = (); // TODO: Better error handling
|
|
|
|
type Error = (); // TODO: Better error handling
|
|
|
|
|
|
|
|
type Owned = Data; |
|
|
|
|
|
|
|
type Borrowed = Self::Owned; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fn transform<'r>(_req: &'r Request, data: Data) -> TransformFuture<'r, Self::Owned, Self::Error> { |
|
|
|
|
|
|
|
Box::pin(async move { Transform::Owned(Success(data)) }) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fn from_data( |
|
|
|
|
|
|
|
request: &'a Request, |
|
|
|
|
|
|
|
outcome: Transformed<'a, Self>, |
|
|
|
|
|
|
|
) -> FromDataFuture<'a, Self, Self::Error> { |
|
|
|
|
|
|
|
Box::pin(async move { |
|
|
|
|
|
|
|
let data = rocket::try_outcome!(outcome.owned()); |
|
|
|
|
|
|
|
|
|
|
|
fn from_data(request: &Request, data: rocket::Data) -> Outcome<Self, Self::Error> { |
|
|
|
|
|
|
|
let user_id = if T::METADATA.requires_authentication { |
|
|
|
let user_id = if T::METADATA.requires_authentication { |
|
|
|
let data = request.guard::<State<crate::Data>>().unwrap(); |
|
|
|
let data = request.guard::<State<crate::Data>>().await.unwrap(); |
|
|
|
|
|
|
|
|
|
|
|
// Get token from header or query value
|
|
|
|
// Get token from header or query value
|
|
|
|
let token = match request |
|
|
|
let token = match request |
|
|
|
@ -74,7 +87,7 @@ where |
|
|
|
|
|
|
|
|
|
|
|
let mut handle = data.open().take(MESSAGE_LIMIT); |
|
|
|
let mut handle = data.open().take(MESSAGE_LIMIT); |
|
|
|
let mut body = Vec::new(); |
|
|
|
let mut body = Vec::new(); |
|
|
|
handle.read_to_end(&mut body).unwrap(); |
|
|
|
handle.read_to_end(&mut body).await.unwrap(); |
|
|
|
|
|
|
|
|
|
|
|
let http_request = http_request.body(body.clone()).unwrap(); |
|
|
|
let http_request = http_request.body(body.clone()).unwrap(); |
|
|
|
log::info!("{:?}", http_request); |
|
|
|
log::info!("{:?}", http_request); |
|
|
|
@ -95,6 +108,7 @@ where |
|
|
|
Failure((Status::InternalServerError, ())) |
|
|
|
Failure((Status::InternalServerError, ())) |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
}) |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
@ -108,7 +122,9 @@ impl<T: Outgoing> Deref for Ruma<T> { |
|
|
|
|
|
|
|
|
|
|
|
/// This struct converts ruma responses into rocket http responses.
|
|
|
|
/// This struct converts ruma responses into rocket http responses.
|
|
|
|
pub struct MatrixResult<T>(pub std::result::Result<T, Error>); |
|
|
|
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> { |
|
|
|
|
|
|
|
|
|
|
|
impl<T: TryInto<http::Response<Vec<u8>>>> TryInto<http::Response<Vec<u8>>> for MatrixResult<T> |
|
|
|
|
|
|
|
{ |
|
|
|
type Error = T::Error; |
|
|
|
type Error = T::Error; |
|
|
|
|
|
|
|
|
|
|
|
fn try_into(self) -> Result<http::Response<Vec<u8>>, T::Error> { |
|
|
|
fn try_into(self) -> Result<http::Response<Vec<u8>>, T::Error> { |
|
|
|
@ -119,13 +135,14 @@ impl<T: TryInto<http::Response<Vec<u8>>>> TryInto<http::Response<Vec<u8>>> for M |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
impl<'r, T: TryInto<http::Response<Vec<u8>>>> Responder<'r> for MatrixResult<T> { |
|
|
|
#[rocket::async_trait] |
|
|
|
fn respond_to(self, _: &Request) -> rocket::response::Result<'r> { |
|
|
|
impl<'r, T: Send + TryInto<http::Response<Vec<u8>>>> Responder<'r> for MatrixResult<T> where T::Error: Send{ |
|
|
|
|
|
|
|
async fn respond_to(self, _: &'r Request<'_>) -> response::Result<'r> { |
|
|
|
let http_response: Result<http::Response<_>, _> = self.try_into(); |
|
|
|
let http_response: Result<http::Response<_>, _> = self.try_into(); |
|
|
|
match http_response { |
|
|
|
match http_response { |
|
|
|
Ok(http_response) => { |
|
|
|
Ok(http_response) => { |
|
|
|
let mut response = rocket::response::Response::build(); |
|
|
|
let mut response = rocket::response::Response::build(); |
|
|
|
response.sized_body(Cursor::new(http_response.body().clone())); |
|
|
|
response.sized_body(Cursor::new(http_response.body().clone())).await; |
|
|
|
|
|
|
|
|
|
|
|
for header in http_response.headers() { |
|
|
|
for header in http_response.headers() { |
|
|
|
response |
|
|
|
response |
|
|
|
|