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/context.rs
+++ b/src/client_server/context.rs
@ -0,0 +1,92 @@
+use super::State;
+use crate::{ConduitResult, Database, Error, Ruma};
+use ruma::api::client::{error::ErrorKind, r0::context::get_context};
+use std::convert::TryFrom;
+
+#[cfg(feature = "conduit_bin")]
+use rocket::get;
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get("/_matrix/client/r0/rooms/<_>/context/<_>", data = "<body>")
+)]
+pub fn get_context_route(
+    db: State<'_, Database>,
+    body: Ruma<get_context::Request>,
+) -> ConduitResult<get_context::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 base_event = db
+        .rooms
+        .get_pdu(&body.event_id)?
+        .ok_or(Error::BadRequest(
+            ErrorKind::NotFound,
+            "Base event not found.",
+        ))?
+        .to_room_event();
+
+    let base_token = db
+        .rooms
+        .get_pdu_count(&body.event_id)?
+        .expect("event still exists");
+
+    let events_before = db
+        .rooms
+        .pdus_until(&sender_id, &body.room_id, base_token)
+        .take(
+            u32::try_from(body.limit).map_err(|_| {
+                Error::BadRequest(ErrorKind::InvalidParam, "Limit value is invalid.")
+            })? as usize
+                / 2,
+        )
+        .filter_map(|r| r.ok()) // Remove buggy events
+        .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<_>>();
+
+    let events_after = db
+        .rooms
+        .pdus_after(&sender_id, &body.room_id, base_token)
+        .take(
+            u32::try_from(body.limit).map_err(|_| {
+                Error::BadRequest(ErrorKind::InvalidParam, "Limit value is invalid.")
+            })? as usize
+                / 2,
+        )
+        .filter_map(|r| r.ok()) // Remove buggy events
+        .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_context::Response {
+        start: start_token,
+        end: end_token,
+        events_before,
+        event: Some(base_event),
+        events_after,
+        state: db // TODO: State at event
+            .rooms
+            .room_state_full(&body.room_id)?
+            .values()
+            .map(|pdu| pdu.to_state_event())
+            .collect(),
+    }
+    .into())
+}