refactor: put endpoints into modules

2020-07-30 18:14:47 +02:00 · 2020-07-30 18:14:47 +02:00 · c1c62b7eb4
commit c1c62b7eb4
parent 485a24398b
33 changed files with 4153 additions and 3848 deletions
--- a/src/client_server/message.rs
+++ b/src/client_server/message.rs
@ -0,0 +1,128 @@
+use super::State;
+use crate::{pdu::PduBuilder, ConduitResult, Database, Error, Ruma};
+use ruma::api::client::{
+    error::ErrorKind,
+    r0::message::{create_message_event, get_message_events},
+};
+use std::convert::TryInto;
+
+#[cfg(feature = "conduit_bin")]
+use rocket::{get, put};
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    put("/_matrix/client/r0/rooms/<_>/send/<_>/<_>", data = "<body>")
+)]
+pub fn create_message_event_route(
+    db: State<'_, Database>,
+    body: Ruma<create_message_event::Request>,
+) -> ConduitResult<create_message_event::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let mut unsigned = serde_json::Map::new();
+    unsigned.insert("transaction_id".to_owned(), body.txn_id.clone().into());
+
+    let event_id = db.rooms.append_pdu(
+        PduBuilder {
+            room_id: body.room_id.clone(),
+            sender: sender_id.clone(),
+            event_type: body.event_type.clone(),
+            content: serde_json::from_str(
+                body.json_body
+                    .ok_or(Error::BadRequest(ErrorKind::BadJson, "Invalid JSON body."))?
+                    .get(),
+            )
+            .map_err(|_| Error::BadRequest(ErrorKind::BadJson, "Invalid JSON body."))?,
+            unsigned: Some(unsigned),
+            state_key: None,
+            redacts: None,
+        },
+        &db.globals,
+        &db.account_data,
+    )?;
+
+    Ok(create_message_event::Response { event_id }.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get("/_matrix/client/r0/rooms/<_>/messages", data = "<body>")
+)]
+pub fn get_message_events_route(
+    db: State<'_, Database>,
+    body: Ruma<get_message_events::Request>,
+) -> ConduitResult<get_message_events::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    if !db.rooms.is_joined(sender_id, &body.room_id)? {
+        return Err(Error::BadRequest(
+            ErrorKind::Forbidden,
+            "You don't have permission to view this room.",
+        ));
+    }
+
+    let from = body
+        .from
+        .clone()
+        .parse()
+        .map_err(|_| Error::BadRequest(ErrorKind::InvalidParam, "Invalid `from` value."))?;
+
+    let to = body.to.as_ref().map(|t| t.parse());
+
+    // Use limit or else 10
+    let limit = body
+        .limit
+        .try_into()
+        .map_or(Ok::<_, Error>(10_usize), |l: u32| Ok(l as usize))?;
+
+    match body.dir {
+        get_message_events::Direction::Forward => {
+            let events_after = db
+                .rooms
+                .pdus_after(&sender_id, &body.room_id, from)
+                .take(limit)
+                .filter_map(|r| r.ok()) // Filter out buggy events
+                .take_while(|&(k, _)| Some(Ok(k)) != to) // Stop at `to`
+                .collect::<Vec<_>>();
+
+            let end_token = events_after.last().map(|(count, _)| count.to_string());
+
+            let events_after = events_after
+                .into_iter()
+                .map(|(_, pdu)| pdu.to_room_event())
+                .collect::<Vec<_>>();
+
+            Ok(get_message_events::Response {
+                start: Some(body.from.clone()),
+                end: end_token,
+                chunk: events_after,
+                state: Vec::new(),
+            }
+            .into())
+        }
+        get_message_events::Direction::Backward => {
+            let events_before = db
+                .rooms
+                .pdus_until(&sender_id, &body.room_id, from)
+                .take(limit)
+                .filter_map(|r| r.ok()) // Filter out buggy events
+                .take_while(|&(k, _)| Some(Ok(k)) != to) // Stop at `to`
+                .collect::<Vec<_>>();
+
+            let start_token = events_before.last().map(|(count, _)| count.to_string());
+
+            let events_before = events_before
+                .into_iter()
+                .map(|(_, pdu)| pdu.to_room_event())
+                .collect::<Vec<_>>();
+
+            Ok(get_message_events::Response {
+                start: Some(body.from.clone()),
+                end: start_token,
+                chunk: events_before,
+                state: Vec::new(),
+            }
+            .into())
+        }
+    }
+}