diff --git a/src/data.rs b/src/data.rs
index a9a0ba5c..84a93df8 100644
--- a/src/data.rs
+++ b/src/data.rs
@@ -150,6 +150,34 @@ impl Data {
}
}
+ pub fn rooms_all(&self) -> Vec<RoomId> {
+ let mut room_ids = self
+ .db
+ .roomid_pduleaves
+ .iter_all()
+ .keys()
+ .map(|key| {
+ serde_json::from_slice(
+ &key.unwrap()
+ .iter()
+ .copied()
+ .take_while(|&x| x != 0xff)
+ .collect::<Vec<_>>(),
+ )
+ .unwrap()
+ })
+ .collect::<Vec<_>>();
+ room_ids.dedup();
+ room_ids
+ }
+
+ pub fn room_users(&self, room_id: &RoomId) -> u32 {
+ self.db
+ .roomid_userids
+ .get_iter(room_id.to_string().as_bytes())
+ .count() as u32
+ }
+
pub fn pdu_get(&self, event_id: &EventId) -> Option<RoomV3Pdu> {
self.db
.eventid_pduid
diff --git a/src/main.rs b/src/main.rs
index 6beb5abc..a0eb1372 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -19,6 +19,7 @@ use ruma_client_api::{
UserInteractiveAuthenticationResponse,
},
alias::get_alias,
+ directory::{self, get_public_rooms},
filter::{self, create_filter, get_filter},
keys::get_keys,
membership::{join_room_by_id, join_room_by_id_or_alias},
@@ -387,6 +388,37 @@ fn join_room_by_id_or_alias_route(
}
}
+#[get("/_matrix/client/r0/publicRooms", data = "<body>")]
+fn get_public_rooms_route(
+ data: State<Data>,
+ body: Ruma<get_public_rooms::Request>,
+) -> MatrixResult<get_public_rooms::Response> {
+ let chunk = data
+ .rooms_all()
+ .into_iter()
+ .map(|room_id| directory::PublicRoomsChunk {
+ aliases: None,
+ canonical_alias: None,
+ name: None,
+ num_joined_members: data.room_users(&room_id).into(),
+ room_id,
+ topic: None,
+ world_readable: false,
+ guest_can_join: true,
+ avatar_url: None,
+ })
+ .collect::<Vec<_>>();
+
+ let total_room_count_estimate = (chunk.len() as u32).into();
+
+ MatrixResult(Ok(get_public_rooms::Response {
+ chunk: chunk,
+ prev_batch: None,
+ next_batch: None,
+ total_room_count_estimate: Some(total_room_count_estimate),
+ }))
+}
+
#[put(
"/_matrix/client/r0/rooms/<_room_id>/send/<_event_type>/<_txn_id>",
data = "<body>"
@@ -549,6 +581,7 @@ fn main() {
get_alias_route,
join_room_by_id_route,
join_room_by_id_or_alias_route,
+ get_public_rooms_route,
create_message_event_route,
create_state_event_for_key_route,
create_state_event_for_empty_key_route,