tuwunel_api/server/
state.rs

1use std::{borrow::Borrow, iter::once};
2
3use axum::extract::State;
4use futures::{FutureExt, StreamExt, TryFutureExt, TryStreamExt, future::try_join};
5use ruma::{OwnedEventId, api::federation::event::get_room_state};
6use tuwunel_core::{
7	Result, at, err,
8	utils::stream::{IterStream, TryBroadbandExt},
9};
10
11use super::{AccessCheck, utils::require_event_in_room};
12use crate::Ruma;
13
14/// # `GET /_matrix/federation/v1/state/{roomId}`
15///
16/// Retrieves a snapshot of a room's state at a given event.
17pub(crate) async fn get_room_state_route(
18	State(services): State<crate::State>,
19	body: Ruma<get_room_state::v1::Request>,
20) -> Result<get_room_state::v1::Response> {
21	let access_check = AccessCheck {
22		services: &services,
23		origin: body.origin(),
24		room_id: &body.room_id,
25		event_id: None,
26	};
27
28	access_check.check().await?;
29
30	require_event_in_room(&services, &body.event_id, &body.room_id).await?;
31
32	let shortstatehash = services
33		.state
34		.pdu_shortstatehash(&body.event_id)
35		.await
36		.map_err(|_| err!(Request(NotFound("PDU state not found."))))?;
37
38	let state_ids = services
39		.state_accessor
40		.state_full_ids(shortstatehash)
41		.map(at!(1))
42		.collect::<Vec<OwnedEventId>>()
43		.map(Ok);
44
45	let room_version = services.state.get_room_version(&body.room_id);
46
47	let (room_version, state_ids) = try_join(room_version, state_ids).await?;
48
49	let into_federation_format = |pdu| {
50		services
51			.federation
52			.format_pdu_into(pdu, Some(&room_version))
53			.map(Ok)
54	};
55
56	let auth_chain = services
57		.auth_chain
58		.event_ids_iter(&body.room_id, &room_version, once(body.event_id.borrow()))
59		.broad_and_then(async |id| {
60			services
61				.timeline
62				.get_pdu_json(&id)
63				.and_then(into_federation_format)
64				.await
65		})
66		.try_collect();
67
68	let pdus = state_ids
69		.iter()
70		.try_stream()
71		.broad_and_then(|id| {
72			services
73				.timeline
74				.get_pdu_json(id)
75				.and_then(into_federation_format)
76		})
77		.try_collect();
78
79	let (auth_chain, pdus) = try_join(auth_chain, pdus).await?;
80
81	Ok(get_room_state::v1::Response { auth_chain, pdus })
82}
tuwunel_api/server/state.rs

tuwunel_api/server/
state.rs
