conduit/src/ruma_wrapper.rs

use rocket::{
    data::{FromDataSimple, Outcome},
    http::Status,
    response::Responder,
    Outcome::*,
    Request, State,
};
use ruma_api::{
    error::{FromHttpRequestError, FromHttpResponseError},
    Endpoint, Outgoing,
};
use ruma_client_api::error::Error;
use ruma_identifiers::UserId;
use std::{
    convert::{TryFrom, TryInto},
    io::{Cursor, Read},
    ops::Deref,
};

const MESSAGE_LIMIT: u64 = 65535;

/// This struct converts rocket requests into ruma structs by converting them into http requests
/// first.
pub struct Ruma<T: Outgoing> {
    body: T::Incoming,
    pub user_id: Option<UserId>,
    pub json_body: serde_json::Value,
}

impl<T: Endpoint> FromDataSimple for Ruma<T>
where
    // 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
    <T as Outgoing>::Incoming: TryFrom<http::Request<Vec<u8>>, Error = FromHttpRequestError>,
    <T::Response as Outgoing>::Incoming: TryFrom<
        http::Response<Vec<u8>>,
        Error = FromHttpResponseError<<T as Endpoint>::ResponseError>,
    >,
{
    type Error = (); // TODO: Better error handling

    fn from_data(request: &Request, data: rocket::Data) -> Outcome<Self, Self::Error> {
        let user_id = if T::METADATA.requires_authentication {
            let data = request.guard::<State<crate::Data>>().unwrap();

            // Get token from header or query value
            let token = match request
                .headers()
                .get_one("Authorization")
                .map(|s| s.to_owned())
                .or_else(|| request.get_query_value("access_token").and_then(|r| r.ok()))
            {
                // TODO: M_MISSING_TOKEN
                None => return Failure((Status::Unauthorized, ())),
                Some(token) => token,
            };

            // Check if token is valid
            match data.user_from_token(&token) {
                // TODO: M_UNKNOWN_TOKEN
                None => return Failure((Status::Unauthorized, ())),
                Some(user_id) => Some(user_id),
            }
        } else {
            None
        };

        let mut http_request = http::Request::builder()
            .uri(request.uri().to_string())
            .method(&*request.method().to_string());
        for header in request.headers().iter() {
            http_request = http_request.header(header.name.as_str(), &*header.value);
        }

        let mut handle = data.open().take(MESSAGE_LIMIT);
        let mut body = Vec::new();
        handle.read_to_end(&mut body).unwrap();

        let http_request = http_request.body(body.clone()).unwrap();
        log::info!("{:?}", http_request);

        match T::Incoming::try_from(http_request) {
            Ok(t) => Success(Ruma {
                body: t,
                user_id,
                // TODO: Can we avoid parsing it again?
                json_body: serde_json::from_slice(&body)
                    .expect("Ruma already parsed it successfuly"),
            }),
            Err(e) => {
                log::error!("{:?}", e);
                Failure((Status::InternalServerError, ()))
            }
        }
    }
}

impl<T: Outgoing> Deref for Ruma<T> {
    type Target = T::Incoming;

    fn deref(&self) -> &Self::Target {
        &self.body
    }
}

/// This struct converts ruma responses into rocket http responses.
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> {
    fn respond_to(self, _: &Request) -> rocket::response::Result<'r> {
        let http_response: Result<http::Response<_>, _> = self.try_into();
        match http_response {
            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.raw_header("Access-Control-Allow-Origin", "*");
                response.raw_header(
                    "Access-Control-Allow-Methods",
                    "GET, POST, PUT, DELETE, OPTIONS",
                );
                response.raw_header(
                    "Access-Control-Allow-Headers",
                    "Origin, X-Requested-With, Content-Type, Accept, Authorization",
                );
                response.ok()
            }
            Err(_) => Err(Status::InternalServerError),
        }
    }
}