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/config.rs
+++ b/src/client_server/config.rs
@ -0,0 +1,67 @@
+use super::State;
+use crate::{ConduitResult, Database, Error, Ruma};
+use ruma::{
+    api::client::{
+        error::ErrorKind,
+        r0::config::{get_global_account_data, set_global_account_data},
+    },
+    events::{custom::CustomEventContent, BasicEvent, EventType},
+    Raw,
+};
+use std::convert::TryFrom;
+
+#[cfg(feature = "conduit_bin")]
+use rocket::{get, put};
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    put("/_matrix/client/r0/user/<_>/account_data/<_>", data = "<body>")
+)]
+pub fn set_global_account_data_route(
+    db: State<'_, Database>,
+    body: Ruma<set_global_account_data::Request>,
+) -> ConduitResult<set_global_account_data::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let content = serde_json::from_str::<serde_json::Value>(body.data.get())
+        .map_err(|_| Error::BadRequest(ErrorKind::BadJson, "Data is invalid."))?;
+
+    let event_type = body.event_type.to_string();
+
+    db.account_data.update(
+        None,
+        sender_id,
+        event_type.clone().into(),
+        &BasicEvent {
+            content: CustomEventContent {
+                event_type,
+                json: content,
+            },
+        },
+        &db.globals,
+    )?;
+
+    Ok(set_global_account_data::Response.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get("/_matrix/client/r0/user/<_>/account_data/<_>", data = "<body>")
+)]
+pub fn get_global_account_data_route(
+    db: State<'_, Database>,
+    body: Ruma<get_global_account_data::Request>,
+) -> ConduitResult<get_global_account_data::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let data = db
+        .account_data
+        .get::<Raw<ruma::events::AnyBasicEvent>>(
+            None,
+            sender_id,
+            EventType::try_from(&body.event_type).expect("EventType::try_from can never fail"),
+        )?
+        .ok_or(Error::BadRequest(ErrorKind::NotFound, "Data not found."))?;
+
+    Ok(get_global_account_data::Response { account_data: data }.into())
+}