crappyrules
/
conduit
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

public rooms dir

ruma
timokoesters 6 years ago
parent
9214a62dde
commit
5ac45eac59
No known key found for this signature in database
GPG Key ID: 356E705610F626D5
2 changed files with 61 additions and 0 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 28
      src/data.rs
  2. 33
      src/main.rs

28
src/data.rs
Unescape View File

@ -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> { pub fn pdu_get(&self, event_id: &EventId) -> Option<RoomV3Pdu> {
self.db self.db
.eventid_pduid .eventid_pduid

33
src/main.rs
Unescape View File

@ -19,6 +19,7 @@ use ruma_client_api::{
UserInteractiveAuthenticationResponse, UserInteractiveAuthenticationResponse,
}, },
alias::get_alias, alias::get_alias,
directory::{self, get_public_rooms},
filter::{self, create_filter, get_filter}, filter::{self, create_filter, get_filter},
keys::get_keys, keys::get_keys,
membership::{join_room_by_id, join_room_by_id_or_alias}, 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( #[put(
"/_matrix/client/r0/rooms/<_room_id>/send/<_event_type>/<_txn_id>", "/_matrix/client/r0/rooms/<_room_id>/send/<_event_type>/<_txn_id>",
data = "<body>" data = "<body>"
@ -549,6 +581,7 @@ fn main() {
get_alias_route, get_alias_route,
join_room_by_id_route, join_room_by_id_route,
join_room_by_id_or_alias_route, join_room_by_id_or_alias_route,
get_public_rooms_route,
create_message_event_route, create_message_event_route,
create_state_event_for_key_route, create_state_event_for_key_route,
create_state_event_for_empty_key_route, create_state_event_for_empty_key_route,

Write Preview
Loading…
Cancel
Save