84 lines
1.6 KiB
Rust
84 lines
1.6 KiB
Rust
use axum::extract::State;
|
|
use futures::TryStreamExt;
|
|
use ruma::api::client::room::initial_sync::v3::{PaginationChunk, Request, Response};
|
|
use tuwunel_core::{
|
|
Err, PduEvent, Result, at,
|
|
utils::{BoolExt, stream::TryTools},
|
|
};
|
|
|
|
use crate::Ruma;
|
|
|
|
const LIMIT_MAX: usize = 100;
|
|
|
|
pub(crate) async fn room_initial_sync_route(
|
|
State(services): State<crate::State>,
|
|
body: Ruma<Request>,
|
|
) -> Result<Response> {
|
|
let room_id = &body.room_id;
|
|
|
|
if !services
|
|
.rooms
|
|
.state_accessor
|
|
.user_can_see_state_events(body.sender_user(), room_id)
|
|
.await
|
|
{
|
|
return Err!(Request(Forbidden("No room preview available.")));
|
|
}
|
|
|
|
let limit = LIMIT_MAX;
|
|
let events: Vec<_> = services
|
|
.rooms
|
|
.timeline
|
|
.pdus_rev(None, room_id, None)
|
|
.try_take(limit)
|
|
.try_collect()
|
|
.await?;
|
|
|
|
let state: Vec<_> = services
|
|
.rooms
|
|
.state_accessor
|
|
.room_state_full_pdus(room_id)
|
|
.map_ok(PduEvent::into_state_event)
|
|
.try_collect()
|
|
.await?;
|
|
|
|
let messages = PaginationChunk {
|
|
start: events
|
|
.last()
|
|
.map(at!(0))
|
|
.as_ref()
|
|
.map(ToString::to_string),
|
|
|
|
end: events
|
|
.first()
|
|
.map(at!(0))
|
|
.as_ref()
|
|
.map(ToString::to_string)
|
|
.unwrap_or_default(),
|
|
|
|
chunk: events
|
|
.into_iter()
|
|
.map(at!(1))
|
|
.map(PduEvent::into_room_event)
|
|
.collect(),
|
|
};
|
|
|
|
Ok(Response {
|
|
room_id: room_id.to_owned(),
|
|
account_data: None,
|
|
state: state.into(),
|
|
messages: messages.chunk.is_empty().or_some(messages),
|
|
visibility: services
|
|
.rooms
|
|
.directory
|
|
.visibility(room_id)
|
|
.await
|
|
.into(),
|
|
membership: services
|
|
.rooms
|
|
.state_cache
|
|
.user_membership(body.sender_user(), room_id)
|
|
.await,
|
|
})
|
|
}
|