continuwuity/src/api/client_server/presence.rs

use std::time::Duration;

use ruma::api::client::{
	error::ErrorKind,
	presence::{get_presence, set_presence},
};

use crate::{services, Error, Result, Ruma};

/// # `PUT /_matrix/client/r0/presence/{userId}/status`
///
/// Sets the presence state of the sender user.
pub async fn set_presence_route(body: Ruma<set_presence::v3::Request>) -> Result<set_presence::v3::Response> {
	if !services().globals.allow_local_presence() {
		return Err(Error::BadRequest(ErrorKind::Forbidden, "Presence is disabled on this server"));
	}

	let sender_user = body.sender_user.as_ref().expect("user is authenticated");
	for room_id in services().rooms.state_cache.rooms_joined(sender_user) {
		let room_id = room_id?;

		services().rooms.edus.presence.set_presence(
			&room_id,
			sender_user,
			body.presence.clone(),
			None,
			None,
			body.status_msg.clone(),
		)?;
	}

	Ok(set_presence::v3::Response {})
}

/// # `GET /_matrix/client/r0/presence/{userId}/status`
///
/// Gets the presence state of the given user.
///
/// - Only works if you share a room with the user
pub async fn get_presence_route(body: Ruma<get_presence::v3::Request>) -> Result<get_presence::v3::Response> {
	if !services().globals.allow_local_presence() {
		return Err(Error::BadRequest(ErrorKind::Forbidden, "Presence is disabled on this server"));
	}

	let sender_user = body.sender_user.as_ref().expect("user is authenticated");

	let mut presence_event = None;

	for room_id in services().rooms.user.get_shared_rooms(vec![sender_user.clone(), body.user_id.clone()])? {
		let room_id = room_id?;

		if let Some(presence) = services().rooms.edus.presence.get_presence(&room_id, sender_user)? {
			presence_event = Some(presence);
			break;
		}
	}

	if let Some(presence) = presence_event {
		Ok(get_presence::v3::Response {
			// TODO: Should ruma just use the presenceeventcontent type here?
			status_msg: presence.content.status_msg,
			currently_active: presence.content.currently_active,
			last_active_ago: presence.content.last_active_ago.map(|millis| Duration::from_millis(millis.into())),
			presence: presence.content.presence,
		})
	} else {
		Err(Error::BadRequest(
			ErrorKind::NotFound,
			"Presence state for this user was not found",
		))
	}
}