2020-07-30 16:14:47 +00:00
|
|
|
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>,
|
2020-09-08 15:32:03 +00:00
|
|
|
body: Ruma<get_device::Request<'_>>,
|
2020-07-30 16:14:47 +00:00
|
|
|
_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>,
|
2020-09-08 15:32:03 +00:00
|
|
|
body: Ruma<update_device::Request<'_>>,
|
2020-07-30 16:14:47 +00:00
|
|
|
_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>,
|
2020-09-08 15:32:03 +00:00
|
|
|
body: Ruma<delete_device::Request<'_>>,
|
2020-07-30 16:14:47 +00:00
|
|
|
_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>,
|
2020-09-08 15:32:03 +00:00
|
|
|
body: Ruma<delete_devices::Request<'_>>,
|
2020-07-30 16:14:47 +00:00
|
|
|
) -> 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())
|
|
|
|
|
}
|
