split s2s into units

Signed-off-by: Jason Volk <jason@zemos.net>

src/api/server/query.rs

@@ -0,0 +1,102 @@
+use get_profile_information::v1::ProfileField;
+use rand::seq::SliceRandom;
+use ruma::{
+	api::{
+		client::error::ErrorKind,
+		federation::query::{get_profile_information, get_room_information},
+	},
+	OwnedServerName,
+};
+
+use crate::{service::server_is_ours, services, Error, Result, Ruma};
+
+/// # `GET /_matrix/federation/v1/query/directory`
+///
+/// Resolve a room alias to a room id.
+pub(crate) async fn get_room_information_route(
+	body: Ruma<get_room_information::v1::Request>,
+) -> Result<get_room_information::v1::Response> {
+	let room_id = services()
+		.rooms
+		.alias
+		.resolve_local_alias(&body.room_alias)?
+		.ok_or_else(|| Error::BadRequest(ErrorKind::NotFound, "Room alias not found."))?;
+
+	let mut servers: Vec<OwnedServerName> = services()
+		.rooms
+		.state_cache
+		.room_servers(&room_id)
+		.filter_map(Result::ok)
+		.collect();
+
+	servers.sort_unstable();
+	servers.dedup();
+
+	servers.shuffle(&mut rand::thread_rng());
+
+	// insert our server as the very first choice if in list
+	if let Some(server_index) = servers
+		.iter()
+		.position(|server| server == services().globals.server_name())
+	{
+		servers.swap_remove(server_index);
+		servers.insert(0, services().globals.server_name().to_owned());
+	}
+
+	Ok(get_room_information::v1::Response {
+		room_id,
+		servers,
+	})
+}
+
+/// # `GET /_matrix/federation/v1/query/profile`
+///
+///
+/// Gets information on a profile.
+pub(crate) async fn get_profile_information_route(
+	body: Ruma<get_profile_information::v1::Request>,
+) -> Result<get_profile_information::v1::Response> {
+	if !services()
+		.globals
+		.allow_profile_lookup_federation_requests()
+	{
+		return Err(Error::BadRequest(
+			ErrorKind::forbidden(),
+			"Profile lookup over federation is not allowed on this homeserver.",
+		));
+	}
+
+	if !server_is_ours(body.user_id.server_name()) {
+		return Err(Error::BadRequest(
+			ErrorKind::InvalidParam,
+			"User does not belong to this server.",
+		));
+	}
+
+	let mut displayname = None;
+	let mut avatar_url = None;
+	let mut blurhash = None;
+
+	match &body.field {
+		Some(ProfileField::DisplayName) => {
+			displayname = services().users.displayname(&body.user_id)?;
+		},
+		Some(ProfileField::AvatarUrl) => {
+			avatar_url = services().users.avatar_url(&body.user_id)?;
+			blurhash = services().users.blurhash(&body.user_id)?;
+		},
+		// TODO: what to do with custom
+		Some(_) => {},
+		None => {
+			displayname = services().users.displayname(&body.user_id)?;
+			avatar_url = services().users.avatar_url(&body.user_id)?;
+			blurhash = services().users.blurhash(&body.user_id)?;
+		},
+	}
+
+	Ok(get_profile_information::v1::Response {
+		displayname,
+		avatar_url,
+		blurhash,
+	})
+}