summaryrefslogtreecommitdiff
path: root/src/api/client_server/profile.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/api/client_server/profile.rs')
-rw-r--r--src/api/client_server/profile.rs325
1 files changed, 325 insertions, 0 deletions
diff --git a/src/api/client_server/profile.rs b/src/api/client_server/profile.rs
new file mode 100644
index 0000000..6400e89
--- /dev/null
+++ b/src/api/client_server/profile.rs
@@ -0,0 +1,325 @@
+use crate::{service::pdu::PduBuilder, services, utils, Error, Result, Ruma};
+use ruma::{
+ api::{
+ client::{
+ error::ErrorKind,
+ profile::{
+ get_avatar_url, get_display_name, get_profile, set_avatar_url, set_display_name,
+ },
+ },
+ federation::{self, query::get_profile_information::v1::ProfileField},
+ },
+ events::{room::member::RoomMemberEventContent, RoomEventType, StateEventType},
+};
+use serde_json::value::to_raw_value;
+use std::sync::Arc;
+
+/// # `PUT /_matrix/client/r0/profile/{userId}/displayname`
+///
+/// Updates the displayname.
+///
+/// - Also makes sure other users receive the update using presence EDUs
+pub async fn set_displayname_route(
+ body: Ruma<set_display_name::v3::Request>,
+) -> Result<set_display_name::v3::Response> {
+ let sender_user = body.sender_user.as_ref().expect("user is authenticated");
+
+ services()
+ .users
+ .set_displayname(sender_user, body.displayname.clone())?;
+
+ // Send a new membership event and presence update into all joined rooms
+ let all_rooms_joined: Vec<_> = services()
+ .rooms
+ .state_cache
+ .rooms_joined(sender_user)
+ .filter_map(|r| r.ok())
+ .map(|room_id| {
+ Ok::<_, Error>((
+ PduBuilder {
+ event_type: RoomEventType::RoomMember,
+ content: to_raw_value(&RoomMemberEventContent {
+ displayname: body.displayname.clone(),
+ ..serde_json::from_str(
+ services()
+ .rooms
+ .state_accessor
+ .room_state_get(
+ &room_id,
+ &StateEventType::RoomMember,
+ sender_user.as_str(),
+ )?
+ .ok_or_else(|| {
+ Error::bad_database(
+ "Tried to send displayname update for user not in the \
+ room.",
+ )
+ })?
+ .content
+ .get(),
+ )
+ .map_err(|_| Error::bad_database("Database contains invalid PDU."))?
+ })
+ .expect("event is valid, we just created it"),
+ unsigned: None,
+ state_key: Some(sender_user.to_string()),
+ redacts: None,
+ },
+ room_id,
+ ))
+ })
+ .filter_map(|r| r.ok())
+ .collect();
+
+ for (pdu_builder, room_id) in all_rooms_joined {
+ let mutex_state = Arc::clone(
+ services()
+ .globals
+ .roomid_mutex_state
+ .write()
+ .unwrap()
+ .entry(room_id.clone())
+ .or_default(),
+ );
+ let state_lock = mutex_state.lock().await;
+
+ let _ = services().rooms.timeline.build_and_append_pdu(
+ pdu_builder,
+ sender_user,
+ &room_id,
+ &state_lock,
+ );
+
+ // Presence update
+ services().rooms.edus.presence.update_presence(
+ sender_user,
+ &room_id,
+ ruma::events::presence::PresenceEvent {
+ content: ruma::events::presence::PresenceEventContent {
+ avatar_url: services().users.avatar_url(sender_user)?,
+ currently_active: None,
+ displayname: services().users.displayname(sender_user)?,
+ last_active_ago: Some(
+ utils::millis_since_unix_epoch()
+ .try_into()
+ .expect("time is valid"),
+ ),
+ presence: ruma::presence::PresenceState::Online,
+ status_msg: None,
+ },
+ sender: sender_user.clone(),
+ },
+ )?;
+ }
+
+ Ok(set_display_name::v3::Response {})
+}
+
+/// # `GET /_matrix/client/r0/profile/{userId}/displayname`
+///
+/// Returns the displayname of the user.
+///
+/// - If user is on another server: Fetches displayname over federation
+pub async fn get_displayname_route(
+ body: Ruma<get_display_name::v3::Request>,
+) -> Result<get_display_name::v3::Response> {
+ if body.user_id.server_name() != services().globals.server_name() {
+ let response = services()
+ .sending
+ .send_federation_request(
+ body.user_id.server_name(),
+ federation::query::get_profile_information::v1::Request {
+ user_id: body.user_id.clone(),
+ field: Some(ProfileField::DisplayName),
+ },
+ )
+ .await?;
+
+ return Ok(get_display_name::v3::Response {
+ displayname: response.displayname,
+ });
+ }
+
+ Ok(get_display_name::v3::Response {
+ displayname: services().users.displayname(&body.user_id)?,
+ })
+}
+
+/// # `PUT /_matrix/client/r0/profile/{userId}/avatar_url`
+///
+/// Updates the avatar_url and blurhash.
+///
+/// - Also makes sure other users receive the update using presence EDUs
+pub async fn set_avatar_url_route(
+ body: Ruma<set_avatar_url::v3::Request>,
+) -> Result<set_avatar_url::v3::Response> {
+ let sender_user = body.sender_user.as_ref().expect("user is authenticated");
+
+ services()
+ .users
+ .set_avatar_url(sender_user, body.avatar_url.clone())?;
+
+ services()
+ .users
+ .set_blurhash(sender_user, body.blurhash.clone())?;
+
+ // Send a new membership event and presence update into all joined rooms
+ let all_joined_rooms: Vec<_> = services()
+ .rooms
+ .state_cache
+ .rooms_joined(sender_user)
+ .filter_map(|r| r.ok())
+ .map(|room_id| {
+ Ok::<_, Error>((
+ PduBuilder {
+ event_type: RoomEventType::RoomMember,
+ content: to_raw_value(&RoomMemberEventContent {
+ avatar_url: body.avatar_url.clone(),
+ ..serde_json::from_str(
+ services()
+ .rooms
+ .state_accessor
+ .room_state_get(
+ &room_id,
+ &StateEventType::RoomMember,
+ sender_user.as_str(),
+ )?
+ .ok_or_else(|| {
+ Error::bad_database(
+ "Tried to send displayname update for user not in the \
+ room.",
+ )
+ })?
+ .content
+ .get(),
+ )
+ .map_err(|_| Error::bad_database("Database contains invalid PDU."))?
+ })
+ .expect("event is valid, we just created it"),
+ unsigned: None,
+ state_key: Some(sender_user.to_string()),
+ redacts: None,
+ },
+ room_id,
+ ))
+ })
+ .filter_map(|r| r.ok())
+ .collect();
+
+ for (pdu_builder, room_id) in all_joined_rooms {
+ let mutex_state = Arc::clone(
+ services()
+ .globals
+ .roomid_mutex_state
+ .write()
+ .unwrap()
+ .entry(room_id.clone())
+ .or_default(),
+ );
+ let state_lock = mutex_state.lock().await;
+
+ let _ = services().rooms.timeline.build_and_append_pdu(
+ pdu_builder,
+ sender_user,
+ &room_id,
+ &state_lock,
+ );
+
+ // Presence update
+ services().rooms.edus.presence.update_presence(
+ sender_user,
+ &room_id,
+ ruma::events::presence::PresenceEvent {
+ content: ruma::events::presence::PresenceEventContent {
+ avatar_url: services().users.avatar_url(sender_user)?,
+ currently_active: None,
+ displayname: services().users.displayname(sender_user)?,
+ last_active_ago: Some(
+ utils::millis_since_unix_epoch()
+ .try_into()
+ .expect("time is valid"),
+ ),
+ presence: ruma::presence::PresenceState::Online,
+ status_msg: None,
+ },
+ sender: sender_user.clone(),
+ },
+ )?;
+ }
+
+ Ok(set_avatar_url::v3::Response {})
+}
+
+/// # `GET /_matrix/client/r0/profile/{userId}/avatar_url`
+///
+/// Returns the avatar_url and blurhash of the user.
+///
+/// - If user is on another server: Fetches avatar_url and blurhash over federation
+pub async fn get_avatar_url_route(
+ body: Ruma<get_avatar_url::v3::Request>,
+) -> Result<get_avatar_url::v3::Response> {
+ if body.user_id.server_name() != services().globals.server_name() {
+ let response = services()
+ .sending
+ .send_federation_request(
+ body.user_id.server_name(),
+ federation::query::get_profile_information::v1::Request {
+ user_id: body.user_id.clone(),
+ field: Some(ProfileField::AvatarUrl),
+ },
+ )
+ .await?;
+
+ return Ok(get_avatar_url::v3::Response {
+ avatar_url: response.avatar_url,
+ blurhash: response.blurhash,
+ });
+ }
+
+ Ok(get_avatar_url::v3::Response {
+ avatar_url: services().users.avatar_url(&body.user_id)?,
+ blurhash: services().users.blurhash(&body.user_id)?,
+ })
+}
+
+/// # `GET /_matrix/client/r0/profile/{userId}`
+///
+/// Returns the displayname, avatar_url and blurhash of the user.
+///
+/// - If user is on another server: Fetches profile over federation
+pub async fn get_profile_route(
+ body: Ruma<get_profile::v3::Request>,
+) -> Result<get_profile::v3::Response> {
+ if body.user_id.server_name() != services().globals.server_name() {
+ let response = services()
+ .sending
+ .send_federation_request(
+ body.user_id.server_name(),
+ federation::query::get_profile_information::v1::Request {
+ user_id: body.user_id.clone(),
+ field: None,
+ },
+ )
+ .await?;
+
+ return Ok(get_profile::v3::Response {
+ displayname: response.displayname,
+ avatar_url: response.avatar_url,
+ blurhash: response.blurhash,
+ });
+ }
+
+ if !services().users.exists(&body.user_id)? {
+ // Return 404 if this user doesn't exist
+ return Err(Error::BadRequest(
+ ErrorKind::NotFound,
+ "Profile was not found.",
+ ));
+ }
+
+ Ok(get_profile::v3::Response {
+ avatar_url: services().users.avatar_url(&body.user_id)?,
+ blurhash: services().users.blurhash(&body.user_id)?,
+ displayname: services().users.displayname(&body.user_id)?,
+ })
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-23 11:31:16 +0000