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.
143 lines
4.4 KiB
143 lines
4.4 KiB
use crate::{database::DatabaseGuard, ConduitResult, Error, Ruma}; |
|
use ruma::{ |
|
api::client::{ |
|
error::ErrorKind, |
|
r0::{read_marker::set_read_marker, receipt::create_receipt}, |
|
}, |
|
events::{AnyEphemeralRoomEvent, EventType}, |
|
receipt::ReceiptType, |
|
MilliSecondsSinceUnixEpoch, |
|
}; |
|
use std::collections::BTreeMap; |
|
|
|
#[cfg(feature = "conduit_bin")] |
|
use rocket::post; |
|
|
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/read_markers` |
|
/// |
|
/// Sets different types of read markers. |
|
/// |
|
/// - Updates fully-read account data event to `fully_read` |
|
/// - If `read_receipt` is set: Update private marker and public read receipt EDU |
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
post("/_matrix/client/r0/rooms/<_>/read_markers", data = "<body>") |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn set_read_marker_route( |
|
db: DatabaseGuard, |
|
body: Ruma<set_read_marker::Request<'_>>, |
|
) -> ConduitResult<set_read_marker::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
let fully_read_event = ruma::events::fully_read::FullyReadEvent { |
|
content: ruma::events::fully_read::FullyReadEventContent { |
|
event_id: body.fully_read.clone(), |
|
}, |
|
}; |
|
db.account_data.update( |
|
Some(&body.room_id), |
|
&sender_user, |
|
EventType::FullyRead, |
|
&fully_read_event, |
|
&db.globals, |
|
)?; |
|
|
|
if let Some(event) = &body.read_receipt { |
|
db.rooms.edus.private_read_set( |
|
&body.room_id, |
|
&sender_user, |
|
db.rooms.get_pdu_count(event)?.ok_or(Error::BadRequest( |
|
ErrorKind::InvalidParam, |
|
"Event does not exist.", |
|
))?, |
|
&db.globals, |
|
)?; |
|
db.rooms |
|
.reset_notification_counts(&sender_user, &body.room_id)?; |
|
|
|
let mut user_receipts = BTreeMap::new(); |
|
user_receipts.insert( |
|
sender_user.clone(), |
|
ruma::events::receipt::Receipt { |
|
ts: Some(MilliSecondsSinceUnixEpoch::now()), |
|
}, |
|
); |
|
|
|
let mut receipts = BTreeMap::new(); |
|
receipts.insert(ReceiptType::Read, user_receipts); |
|
|
|
let mut receipt_content = BTreeMap::new(); |
|
receipt_content.insert(event.to_owned(), receipts); |
|
|
|
db.rooms.edus.readreceipt_update( |
|
&sender_user, |
|
&body.room_id, |
|
AnyEphemeralRoomEvent::Receipt(ruma::events::receipt::ReceiptEvent { |
|
content: ruma::events::receipt::ReceiptEventContent(receipt_content), |
|
room_id: body.room_id.clone(), |
|
}), |
|
&db.globals, |
|
)?; |
|
} |
|
|
|
db.flush()?; |
|
|
|
Ok(set_read_marker::Response {}.into()) |
|
} |
|
|
|
/// # `POST /_matrix/client/r0/rooms/{roomId}/receipt/{receiptType}/{eventId}` |
|
/// |
|
/// Sets private read marker and public read receipt EDU. |
|
#[cfg_attr( |
|
feature = "conduit_bin", |
|
post("/_matrix/client/r0/rooms/<_>/receipt/<_>/<_>", data = "<body>") |
|
)] |
|
#[tracing::instrument(skip(db, body))] |
|
pub async fn create_receipt_route( |
|
db: DatabaseGuard, |
|
body: Ruma<create_receipt::Request<'_>>, |
|
) -> ConduitResult<create_receipt::Response> { |
|
let sender_user = body.sender_user.as_ref().expect("user is authenticated"); |
|
|
|
db.rooms.edus.private_read_set( |
|
&body.room_id, |
|
&sender_user, |
|
db.rooms |
|
.get_pdu_count(&body.event_id)? |
|
.ok_or(Error::BadRequest( |
|
ErrorKind::InvalidParam, |
|
"Event does not exist.", |
|
))?, |
|
&db.globals, |
|
)?; |
|
db.rooms |
|
.reset_notification_counts(&sender_user, &body.room_id)?; |
|
|
|
let mut user_receipts = BTreeMap::new(); |
|
user_receipts.insert( |
|
sender_user.clone(), |
|
ruma::events::receipt::Receipt { |
|
ts: Some(MilliSecondsSinceUnixEpoch::now()), |
|
}, |
|
); |
|
let mut receipts = BTreeMap::new(); |
|
receipts.insert(ReceiptType::Read, user_receipts); |
|
|
|
let mut receipt_content = BTreeMap::new(); |
|
receipt_content.insert(body.event_id.to_owned(), receipts); |
|
|
|
db.rooms.edus.readreceipt_update( |
|
&sender_user, |
|
&body.room_id, |
|
AnyEphemeralRoomEvent::Receipt(ruma::events::receipt::ReceiptEvent { |
|
content: ruma::events::receipt::ReceiptEventContent(receipt_content), |
|
room_id: body.room_id.clone(), |
|
}), |
|
&db.globals, |
|
)?; |
|
|
|
db.flush()?; |
|
|
|
Ok(create_receipt::Response {}.into()) |
|
}
|
|
|