use super::State; use crate::{server_server, ConduitResult, Database, Error, Ruma}; use ruma::{ api::{ 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 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)?; Ok(create_alias::Response::new().into()) } #[cfg_attr( feature = "conduit_bin", delete("/_matrix/client/r0/directory/room/<_>", data = "") )] pub fn delete_alias_route( db: State<'_, Database>, body: Ruma>, ) -> ConduitResult { db.rooms.set_alias(&body.room_alias, None, &db.globals)?; 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, room_alias.server_name().to_string(), federation::query::get_room_information::v1::Request { room_alias }, ) .await?; return Ok(get_alias::Response::new(response.room_id, response.servers).into()); } let room_id = db .rooms .id_from_alias(&room_alias)? .ok_or(Error::BadRequest( ErrorKind::NotFound, "Room with alias not found.", ))?; Ok(get_alias::Response::new(room_id, vec![db.globals.server_name().to_string()]).into()) }