conduit/src/client_server/to_device.rs

73 lines
2.5 KiB
Rust
Raw Normal View History

2020-07-30 16:14:47 +00:00
use super::State;
use crate::{ConduitResult, Database, Error, Ruma};
use ruma::api::client::{
error::ErrorKind,
r0::to_device::{self, send_event_to_device},
};
#[cfg(feature = "conduit_bin")]
use rocket::put;
#[cfg_attr(
feature = "conduit_bin",
put("/_matrix/client/r0/sendToDevice/<_>/<_>", data = "<body>")
)]
pub async fn send_event_to_device_route(
2020-07-30 16:14:47 +00:00
db: State<'_, Database>,
2020-09-08 15:32:03 +00:00
body: Ruma<send_event_to_device::Request<'_>>,
2020-07-30 16:14:47 +00:00
) -> ConduitResult<send_event_to_device::Response> {
let sender_user = body.sender_user.as_ref().expect("user is authenticated");
let sender_device = body.sender_device.as_deref();
2020-08-25 11:24:38 +00:00
// Check if this is a new transaction id
if db
.transaction_ids
.existing_txnid(sender_user, sender_device, &body.txn_id)?
2020-08-25 11:24:38 +00:00
.is_some()
{
return Ok(send_event_to_device::Response.into());
}
2020-07-30 16:14:47 +00:00
for (target_user_id, map) in &body.messages {
for (target_device_id_maybe, event) in map {
match target_device_id_maybe {
to_device::DeviceIdOrAllDevices::DeviceId(target_device_id) => {
db.users.add_to_device_event(
sender_user,
2020-07-30 16:14:47 +00:00
&target_user_id,
&target_device_id,
&body.event_type,
serde_json::from_str(event.get()).map_err(|_| {
Error::BadRequest(ErrorKind::InvalidParam, "Event is invalid")
})?,
&db.globals,
)?
}
to_device::DeviceIdOrAllDevices::AllDevices => {
for target_device_id in db.users.all_device_ids(&target_user_id) {
db.users.add_to_device_event(
sender_user,
2020-07-30 16:14:47 +00:00
&target_user_id,
&target_device_id?,
&body.event_type,
serde_json::from_str(event.get()).map_err(|_| {
Error::BadRequest(ErrorKind::InvalidParam, "Event is invalid")
})?,
&db.globals,
)?;
}
}
}
}
}
2020-08-25 11:24:38 +00:00
// Save transaction id with empty data
db.transaction_ids
.add_txnid(sender_user, sender_device, &body.txn_id, &[])?;
2020-08-25 11:24:38 +00:00
db.flush().await?;
2020-07-30 16:14:47 +00:00
Ok(send_event_to_device::Response.into())
}