split s2s into units

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

src/api/server/user.rs

@@ -0,0 +1,108 @@
+use ruma::api::{
+	client::error::ErrorKind,
+	federation::{
+		device::get_devices::{self, v1::UserDevice},
+		keys::{claim_keys, get_keys},
+	},
+};
+
+use crate::{
+	client_server::{claim_keys_helper, get_keys_helper},
+	service::user_is_local,
+	services, Error, Result, Ruma,
+};
+
+/// # `GET /_matrix/federation/v1/user/devices/{userId}`
+///
+/// Gets information on all devices of the user.
+pub(crate) async fn get_devices_route(body: Ruma<get_devices::v1::Request>) -> Result<get_devices::v1::Response> {
+	if !user_is_local(&body.user_id) {
+		return Err(Error::BadRequest(
+			ErrorKind::InvalidParam,
+			"Tried to access user from other server.",
+		));
+	}
+
+	let origin = body.origin.as_ref().expect("server is authenticated");
+
+	Ok(get_devices::v1::Response {
+		user_id: body.user_id.clone(),
+		stream_id: services()
+			.users
+			.get_devicelist_version(&body.user_id)?
+			.unwrap_or(0)
+			.try_into()
+			.expect("version will not grow that large"),
+		devices: services()
+			.users
+			.all_devices_metadata(&body.user_id)
+			.filter_map(Result::ok)
+			.filter_map(|metadata| {
+				let device_id_string = metadata.device_id.as_str().to_owned();
+				let device_display_name = if services().globals.allow_device_name_federation() {
+					metadata.display_name
+				} else {
+					Some(device_id_string)
+				};
+				Some(UserDevice {
+					keys: services()
+						.users
+						.get_device_keys(&body.user_id, &metadata.device_id)
+						.ok()??,
+					device_id: metadata.device_id,
+					device_display_name,
+				})
+			})
+			.collect(),
+		master_key: services()
+			.users
+			.get_master_key(None, &body.user_id, &|u| u.server_name() == origin)?,
+		self_signing_key: services()
+			.users
+			.get_self_signing_key(None, &body.user_id, &|u| u.server_name() == origin)?,
+	})
+}
+
+/// # `POST /_matrix/federation/v1/user/keys/query`
+///
+/// Gets devices and identity keys for the given users.
+pub(crate) async fn get_keys_route(body: Ruma<get_keys::v1::Request>) -> Result<get_keys::v1::Response> {
+	if body.device_keys.iter().any(|(u, _)| !user_is_local(u)) {
+		return Err(Error::BadRequest(
+			ErrorKind::InvalidParam,
+			"User does not belong to this server.",
+		));
+	}
+
+	let result = get_keys_helper(
+		None,
+		&body.device_keys,
+		|u| Some(u.server_name()) == body.origin.as_deref(),
+		services().globals.allow_device_name_federation(),
+	)
+	.await?;
+
+	Ok(get_keys::v1::Response {
+		device_keys: result.device_keys,
+		master_keys: result.master_keys,
+		self_signing_keys: result.self_signing_keys,
+	})
+}
+
+/// # `POST /_matrix/federation/v1/user/keys/claim`
+///
+/// Claims one-time keys.
+pub(crate) async fn claim_keys_route(body: Ruma<claim_keys::v1::Request>) -> Result<claim_keys::v1::Response> {
+	if body.one_time_keys.iter().any(|(u, _)| !user_is_local(u)) {
+		return Err(Error::BadRequest(
+			ErrorKind::InvalidParam,
+			"Tried to access user from other server.",
+		));
+	}
+
+	let result = claim_keys_helper(&body.one_time_keys).await?;
+
+	Ok(claim_keys::v1::Response {
+		one_time_keys: result.one_time_keys,
+	})
+}