use super::State; use crate::{appservice_server, server_server, ConduitResult, Database, Error, Ruma}; use ruma::{ api::{ appservice, client::{ error::ErrorKind, r0::alias::{create_alias, delete_alias, get_alias}, }, federation, }, RoomAliasId, }; #[cfg(feature = "conduit_bin")] use rocket::{delete, get, put}; #[cfg_attr( feature = "conduit_bin", put("/_matrix/client/r0/directory/room/<_>", data = "") )] pub async fn create_alias_route( db: State<'_, Database>, body: Ruma>, ) -> ConduitResult { if db.rooms.id_from_alias(&body.room_alias)?.is_some() { return Err(Error::Conflict("Alias already exists.")); } db.rooms .set_alias(&body.room_alias, Some(&body.room_id), &db.globals)?; db.flush().await?; Ok(create_alias::Response::new().into()) } #[cfg_attr( feature = "conduit_bin", delete("/_matrix/client/r0/directory/room/<_>", data = "") )] pub async fn delete_alias_route( db: State<'_, Database>, body: Ruma>, ) -> ConduitResult { db.rooms.set_alias(&body.room_alias, None, &db.globals)?; db.flush().await?; Ok(delete_alias::Response::new().into()) } #[cfg_attr( feature = "conduit_bin", get("/_matrix/client/r0/directory/room/<_>", data = "") )] pub async fn get_alias_route( db: State<'_, Database>, body: Ruma>, ) -> ConduitResult { get_alias_helper(&db, &body.room_alias).await } pub async fn get_alias_helper( db: &Database, room_alias: &RoomAliasId, ) -> ConduitResult { if room_alias.server_name() != db.globals.server_name() { let response = server_server::send_request( &db.globals, room_alias.server_name().to_owned(), federation::query::get_room_information::v1::Request { room_alias }, ) .await?; return Ok(get_alias::Response::new(response.room_id, response.servers).into()); } let mut room_id = None; match db.rooms.id_from_alias(&room_alias)? { Some(r) => room_id = Some(r), None => { for (_id, registration) in db.appservice.iter_all().filter_map(|r| r.ok()) { if appservice_server::send_request( &db.globals, registration, appservice::query::query_room_alias::v1::Request { room_alias }, ) .await .is_ok() { room_id = Some(db.rooms.id_from_alias(&room_alias)?.ok_or_else(|| { Error::bad_config("Appservice lied to us. Room does not exist.") })?); break; } } } }; let room_id = match room_id { Some(room_id) => room_id, None => { return Err(Error::BadRequest( ErrorKind::NotFound, "Room with alias not found.", )) } }; Ok(get_alias::Response::new(room_id, vec![db.globals.server_name().to_owned()]).into()) }