Conduit is a simple, fast and reliable chat server powered by Matrix https://conduit.rs
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
150 lines
4.4 KiB
150 lines
4.4 KiB
use super::State; |
|
use crate::{ConduitResult, Database, Error, Ruma}; |
|
use ruma::{ |
|
api::client::{ |
|
error::ErrorKind, |
|
r0::config::{ |
|
get_global_account_data, get_room_account_data, set_global_account_data, |
|
set_room_account_data, |
|
}, |
|
}, |
|
events::{AnyGlobalAccountDataEventContent, AnyRoomAccountDataEventContent}, |
|
serde::Raw, |
|
}; |
|
use serde::Deserialize; |
|
use serde_json::{json, value::RawValue as RawJsonValue}; |
|
|
|
#[cfg(feature = "conduit_bin")] |
|
use rocket::{get, put}; |
|
|
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
put("/_matrix/client/r0/user/<_>/account_data/<_>", data = "<body>") |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn set_global_account_data_route( |
|
db: State<'_, Database>, |
|
body: Ruma<set_global_account_data::Request<'_>>, |
|
) -> ConduitResult<set_global_account_data::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
let data = 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_user, |
|
event_type.clone().into(), |
|
&json!({ |
|
"type": event_type, |
|
"content": data, |
|
}), |
|
&db.globals, |
|
)?; |
|
|
|
db.flush().await?; |
|
|
|
Ok(set_global_account_data::Response.into()) |
|
} |
|
|
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
put( |
|
"/_matrix/client/r0/user/<_>/rooms/<_>/account_data/<_>", |
|
data = "<body>" |
|
) |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn set_room_account_data_route( |
|
db: State<'_, Database>, |
|
body: Ruma<set_room_account_data::Request<'_>>, |
|
) -> ConduitResult<set_room_account_data::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
let data = 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( |
|
Some(&body.room_id), |
|
sender_user, |
|
event_type.clone().into(), |
|
&json!({ |
|
"type": event_type, |
|
"content": data, |
|
}), |
|
&db.globals, |
|
)?; |
|
|
|
db.flush().await?; |
|
|
|
Ok(set_room_account_data::Response.into()) |
|
} |
|
|
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
get("/_matrix/client/r0/user/<_>/account_data/<_>", data = "<body>") |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn get_global_account_data_route( |
|
db: State<'_, Database>, |
|
body: Ruma<get_global_account_data::Request<'_>>, |
|
) -> ConduitResult<get_global_account_data::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
let event = db |
|
.account_data |
|
.get::<Box<RawJsonValue>>(None, sender_user, body.event_type.clone().into())? |
|
.ok_or(Error::BadRequest(ErrorKind::NotFound, "Data not found."))?; |
|
db.flush().await?; |
|
|
|
let account_data = serde_json::from_str::<ExtractGlobalEventContent>(event.get()) |
|
.map_err(|_| Error::bad_database("Invalid account data event in db."))? |
|
.content; |
|
|
|
Ok(get_global_account_data::Response { account_data }.into()) |
|
} |
|
|
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
get( |
|
"/_matrix/client/r0/user/<_>/rooms/<_>/account_data/<_>", |
|
data = "<body>" |
|
) |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn get_room_account_data_route( |
|
db: State<'_, Database>, |
|
body: Ruma<get_room_account_data::Request<'_>>, |
|
) -> ConduitResult<get_room_account_data::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
let event = db |
|
.account_data |
|
.get::<Box<RawJsonValue>>( |
|
Some(&body.room_id), |
|
sender_user, |
|
body.event_type.clone().into(), |
|
)? |
|
.ok_or(Error::BadRequest(ErrorKind::NotFound, "Data not found."))?; |
|
db.flush().await?; |
|
|
|
let account_data = serde_json::from_str::<ExtractRoomEventContent>(event.get()) |
|
.map_err(|_| Error::bad_database("Invalid account data event in db."))? |
|
.content; |
|
|
|
Ok(get_room_account_data::Response { account_data }.into()) |
|
} |
|
|
|
#[derive(Deserialize)] |
|
struct ExtractRoomEventContent { |
|
content: Raw<AnyRoomAccountDataEventContent>, |
|
} |
|
|
|
#[derive(Deserialize)] |
|
struct ExtractGlobalEventContent { |
|
content: Raw<AnyGlobalAccountDataEventContent>, |
|
}
|
|
|