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/device.rs
+++ b/src/client_server/device.rs
@ -0,0 +1,170 @@
+use super::State;
+use crate::{utils, ConduitResult, Database, Error, Ruma};
+use ruma::api::client::{
+    error::ErrorKind,
+    r0::{
+        device::{self, delete_device, delete_devices, get_device, get_devices, update_device},
+        uiaa::{AuthFlow, UiaaInfo},
+    },
+};
+
+use super::SESSION_ID_LENGTH;
+#[cfg(feature = "conduit_bin")]
+use rocket::{delete, get, post, put};
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get("/_matrix/client/r0/devices", data = "<body>")
+)]
+pub fn get_devices_route(
+    db: State<'_, Database>,
+    body: Ruma<get_devices::Request>,
+) -> ConduitResult<get_devices::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let devices = db
+        .users
+        .all_devices_metadata(sender_id)
+        .filter_map(|r| r.ok()) // Filter out buggy devices
+        .collect::<Vec<device::Device>>();
+
+    Ok(get_devices::Response { devices }.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    get("/_matrix/client/r0/devices/<_device_id>", data = "<body>")
+)]
+pub fn get_device_route(
+    db: State<'_, Database>,
+    body: Ruma<get_device::Request>,
+    _device_id: String,
+) -> ConduitResult<get_device::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let device = db
+        .users
+        .get_device_metadata(&sender_id, &body.body.device_id)?
+        .ok_or(Error::BadRequest(ErrorKind::NotFound, "Device not found."))?;
+
+    Ok(get_device::Response { device }.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    put("/_matrix/client/r0/devices/<_device_id>", data = "<body>")
+)]
+pub fn update_device_route(
+    db: State<'_, Database>,
+    body: Ruma<update_device::Request>,
+    _device_id: String,
+) -> ConduitResult<update_device::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+
+    let mut device = db
+        .users
+        .get_device_metadata(&sender_id, &body.body.device_id)?
+        .ok_or(Error::BadRequest(ErrorKind::NotFound, "Device not found."))?;
+
+    device.display_name = body.display_name.clone();
+
+    db.users
+        .update_device_metadata(&sender_id, &body.body.device_id, &device)?;
+
+    Ok(update_device::Response.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    delete("/_matrix/client/r0/devices/<_device_id>", data = "<body>")
+)]
+pub fn delete_device_route(
+    db: State<'_, Database>,
+    body: Ruma<delete_device::Request>,
+    _device_id: String,
+) -> ConduitResult<delete_device::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+    let device_id = body.device_id.as_ref().expect("user is authenticated");
+
+    // UIAA
+    let mut uiaainfo = UiaaInfo {
+        flows: vec![AuthFlow {
+            stages: vec!["m.login.password".to_owned()],
+        }],
+        completed: Vec::new(),
+        params: Default::default(),
+        session: None,
+        auth_error: None,
+    };
+
+    if let Some(auth) = &body.auth {
+        let (worked, uiaainfo) = db.uiaa.try_auth(
+            &sender_id,
+            &device_id,
+            auth,
+            &uiaainfo,
+            &db.users,
+            &db.globals,
+        )?;
+        if !worked {
+            return Err(Error::Uiaa(uiaainfo));
+        }
+    // Success!
+    } else {
+        uiaainfo.session = Some(utils::random_string(SESSION_ID_LENGTH));
+        db.uiaa.create(&sender_id, &device_id, &uiaainfo)?;
+        return Err(Error::Uiaa(uiaainfo));
+    }
+
+    db.users.remove_device(&sender_id, &body.body.device_id)?;
+
+    Ok(delete_device::Response.into())
+}
+
+#[cfg_attr(
+    feature = "conduit_bin",
+    post("/_matrix/client/r0/delete_devices", data = "<body>")
+)]
+pub fn delete_devices_route(
+    db: State<'_, Database>,
+    body: Ruma<delete_devices::Request>,
+) -> ConduitResult<delete_devices::Response> {
+    let sender_id = body.sender_id.as_ref().expect("user is authenticated");
+    let device_id = body.device_id.as_ref().expect("user is authenticated");
+
+    // UIAA
+    let mut uiaainfo = UiaaInfo {
+        flows: vec![AuthFlow {
+            stages: vec!["m.login.password".to_owned()],
+        }],
+        completed: Vec::new(),
+        params: Default::default(),
+        session: None,
+        auth_error: None,
+    };
+
+    if let Some(auth) = &body.auth {
+        let (worked, uiaainfo) = db.uiaa.try_auth(
+            &sender_id,
+            &device_id,
+            auth,
+            &uiaainfo,
+            &db.users,
+            &db.globals,
+        )?;
+        if !worked {
+            return Err(Error::Uiaa(uiaainfo));
+        }
+    // Success!
+    } else {
+        uiaainfo.session = Some(utils::random_string(SESSION_ID_LENGTH));
+        db.uiaa.create(&sender_id, &device_id, &uiaainfo)?;
+        return Err(Error::Uiaa(uiaainfo));
+    }
+
+    for device_id in &body.devices {
+        db.users.remove_device(&sender_id, &device_id)?
+    }
+
+    Ok(delete_devices::Response.into())
+}