91 lines
2.4 KiB
Rust
91 lines
2.4 KiB
Rust
use std::time::Duration;
|
|
|
|
use axum::extract::State;
|
|
use ruma::api::client::{
|
|
error::ErrorKind,
|
|
presence::{get_presence, set_presence},
|
|
};
|
|
|
|
use crate::{Error, Result, Ruma};
|
|
|
|
/// # `PUT /_matrix/client/r0/presence/{userId}/status`
|
|
///
|
|
/// Sets the presence state of the sender user.
|
|
pub(crate) async fn set_presence_route(
|
|
State(services): State<crate::State>, 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");
|
|
if sender_user != &body.user_id && body.appservice_info.is_none() {
|
|
return Err(Error::BadRequest(
|
|
ErrorKind::InvalidParam,
|
|
"Not allowed to set presence of other users",
|
|
));
|
|
}
|
|
|
|
services
|
|
.presence
|
|
.set_presence(sender_user, &body.presence, 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(crate) async fn get_presence_route(
|
|
State(services): State<crate::State>, 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()])?
|
|
{
|
|
if let Some(presence) = services.presence.get_presence(&body.user_id)? {
|
|
presence_event = Some(presence);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if let Some(presence) = presence_event {
|
|
let status_msg = if presence
|
|
.content
|
|
.status_msg
|
|
.as_ref()
|
|
.is_some_and(String::is_empty)
|
|
{
|
|
None
|
|
} else {
|
|
presence.content.status_msg
|
|
};
|
|
|
|
Ok(get_presence::v3::Response {
|
|
// TODO: Should ruma just use the presenceeventcontent type here?
|
|
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",
|
|
))
|
|
}
|
|
}
|