studio/tuwunel
2
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Take stuff by ref in users service profile

Browse Source
This commit is contained in:
dasha_uwu
2026-01-26 13:26:17 +05:00
committed by Jason Volk
parent c76a66be28
commit 73eee626df
5 changed files with 128 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/api/client/profile.rs
View File

@@ -43,7 +43,7 @@ pub(crate) async fn set_displayname_route(
services
.users
.update_displayname(&body.user_id, body.displayname.clone(), &all_joined_rooms)
.update_displayname(&body.user_id, body.displayname.as_deref(), &all_joined_rooms)
.await;
// Presence update
@@ -87,13 +87,13 @@ pub(crate) async fn get_displayname_route(
services
.users
.set_displayname(&body.user_id, response.displayname.clone());
.set_displayname(&body.user_id, response.displayname.as_deref());
services
.users
.set_avatar_url(&body.user_id, response.avatar_url.clone());
.set_avatar_url(&body.user_id, response.avatar_url.as_deref());
services
.users
.set_blurhash(&body.user_id, response.blurhash.clone());
.set_blurhash(&body.user_id, response.blurhash.as_deref());
return Ok(get_display_name::v3::Response { displayname: response.displayname });
}
@@ -140,8 +140,8 @@ pub(crate) async fn set_avatar_url_route(
.users
.update_avatar_url(
&body.user_id,
body.avatar_url.clone(),
body.blurhash.clone(),
body.avatar_url.as_deref(),
body.blurhash.as_deref(),
&all_joined_rooms,
)
.await;
@@ -188,13 +188,13 @@ pub(crate) async fn get_avatar_url_route(
services
.users
.set_displayname(&body.user_id, response.displayname.clone());
.set_displayname(&body.user_id, response.displayname.as_deref());
services
.users
.set_avatar_url(&body.user_id, response.avatar_url.clone());
.set_avatar_url(&body.user_id, response.avatar_url.as_deref());
services
.users
.set_blurhash(&body.user_id, response.blurhash.clone());
.set_blurhash(&body.user_id, response.blurhash.as_deref());
return Ok(get_avatar_url::v3::Response {
avatar_url: response.avatar_url,
@@ -250,22 +250,22 @@ pub(crate) async fn get_profile_route(
services
.users
.set_displayname(&body.user_id, response.displayname.clone());
.set_displayname(&body.user_id, response.displayname.as_deref());
services
.users
.set_avatar_url(&body.user_id, response.avatar_url.clone());
.set_avatar_url(&body.user_id, response.avatar_url.as_deref());
services
.users
.set_blurhash(&body.user_id, response.blurhash.clone());
.set_blurhash(&body.user_id, response.blurhash.as_deref());
services
.users
.set_timezone(&body.user_id, response.tz.clone());
.set_timezone(&body.user_id, response.tz.as_deref());
for (profile_key, profile_key_value) in &response.custom_profile_fields {
services.users.set_profile_key(
&body.user_id,
profile_key,
Some(profile_key_value.clone()),
Some(profile_key_value),
);
}

6
src/api/client/session/sso.rs
View File

@@ -7,7 +7,7 @@ use base64::{Engine as _, engine::general_purpose::URL_SAFE_NO_PAD as b64};
use futures::{FutureExt, StreamExt, TryFutureExt, future::try_join};
use reqwest::header::{CONTENT_TYPE, HeaderValue};
use ruma::{
Mxc, OwnedRoomId, OwnedUserId, ServerName, UserId,
Mxc, OwnedMxcUri, OwnedRoomId, OwnedUserId, ServerName, UserId,
api::client::session::{sso_callback, sso_login, sso_login_with_provider},
};
use serde::{Deserialize, Serialize};
@@ -556,10 +556,10 @@ async fn set_avatar(
.collect()
.await;
let mxc_uri = mxc.to_string().into();
let mxc_uri: OwnedMxcUri = mxc.to_string().into();
services
.users
.update_avatar_url(user_id, Some(mxc_uri), None, &all_joined_rooms)
.update_avatar_url(user_id, Some(&mxc_uri), None, &all_joined_rooms)
.await;
Ok(())

184
src/api/client/unstable.rs
View File

@@ -16,7 +16,7 @@ use ruma::{
},
presence::PresenceState,
};
use tuwunel_core::{Err, Error, Result};
use tuwunel_core::{Err, Error, Result, err};
use crate::Ruma;
@@ -99,7 +99,7 @@ pub(crate) async fn set_timezone_key_route(
services
.users
.set_timezone(&body.user_id, body.tz.clone());
.set_timezone(&body.user_id, body.tz.as_deref());
// Presence update
services
@@ -129,42 +129,40 @@ pub(crate) async fn set_profile_field_route(
return Err!(Request(BadJson("Key names cannot be longer than 128 bytes")));
}
if body.value.field_name() == ProfileFieldName::DisplayName {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
match &body.value {
| ProfileFieldValue::DisplayName(displayname) => {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_displayname(
services
.users
.update_displayname(&body.user_id, Some(displayname), &all_joined_rooms)
.await;
},
| ProfileFieldValue::AvatarUrl(avatar_url) => {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_avatar_url(&body.user_id, Some(avatar_url), None, &all_joined_rooms)
.await;
},
| _ => {
services.users.set_profile_key(
&body.user_id,
Some(body.value.value().to_string()),
&all_joined_rooms,
)
.await;
} else if body.value.field_name() == ProfileFieldName::AvatarUrl {
let mxc = ruma::OwnedMxcUri::from(body.value.value().to_string());
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_avatar_url(&body.user_id, Some(mxc), None, &all_joined_rooms)
.await;
} else {
services.users.set_profile_key(
&body.user_id,
body.value.field_name().as_str(),
Some(body.value.value().into_owned()),
);
body.value.field_name().as_str(),
Some(&body.value.value()),
);
},
}
// Presence update
@@ -191,34 +189,38 @@ pub(crate) async fn delete_profile_field_route(
return Err!(Request(Forbidden("You cannot update the profile of another user")));
}
if body.field == ProfileFieldName::DisplayName {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
match body.field {
| ProfileFieldName::DisplayName => {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_displayname(&body.user_id, None, &all_joined_rooms)
.await;
} else if body.field == ProfileFieldName::AvatarUrl {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_displayname(&body.user_id, None, &all_joined_rooms)
.await;
},
| ProfileFieldName::AvatarUrl => {
let all_joined_rooms: Vec<OwnedRoomId> = services
.state_cache
.rooms_joined(&body.user_id)
.map(Into::into)
.collect()
.await;
services
.users
.update_avatar_url(&body.user_id, None, None, &all_joined_rooms)
.await;
} else {
services
.users
.set_profile_key(&body.user_id, body.field.as_str(), None);
services
.users
.update_avatar_url(&body.user_id, None, None, &all_joined_rooms)
.await;
},
| _ => {
services
.users
.set_profile_key(&body.user_id, body.field.as_str(), None);
},
}
// Presence update
@@ -262,19 +264,19 @@ pub(crate) async fn get_timezone_key_route(
services
.users
.set_displayname(&body.user_id, response.displayname.clone());
.set_displayname(&body.user_id, response.displayname.as_deref());
services
.users
.set_avatar_url(&body.user_id, response.avatar_url.clone());
.set_avatar_url(&body.user_id, response.avatar_url.as_deref());
services
.users
.set_blurhash(&body.user_id, response.blurhash.clone());
.set_blurhash(&body.user_id, response.blurhash.as_deref());
services
.users
.set_timezone(&body.user_id, response.tz.clone());
.set_timezone(&body.user_id, response.tz.as_deref());
return Ok(get_timezone_key::unstable::Response { tz: response.tz });
}
@@ -323,43 +325,34 @@ pub(crate) async fn get_profile_field_route(
services
.users
.set_displayname(&body.user_id, response.displayname.clone());
.set_displayname(&body.user_id, response.displayname.as_deref());
services
.users
.set_avatar_url(&body.user_id, response.avatar_url.clone());
.set_avatar_url(&body.user_id, response.avatar_url.as_deref());
services
.users
.set_blurhash(&body.user_id, response.blurhash.clone());
.set_blurhash(&body.user_id, response.blurhash.as_deref());
services
.users
.set_timezone(&body.user_id, response.tz.clone());
.set_timezone(&body.user_id, response.tz.as_deref());
let profile_key_value: Option<ProfileFieldValue> = match response
let value = response
.custom_profile_fields
.get(body.field.as_str())
{
| Some(value) => {
services.users.set_profile_key(
&body.user_id,
body.field.as_str(),
Some(value.clone()),
);
.ok_or_else(|| {
err!(Request(NotFound("The requested profile key does not exist.")))
})?;
Some(ProfileFieldValue::new(body.field.as_str(), value.clone())?)
},
| _ => {
return Err!(Request(NotFound("The requested profile key does not exist.")));
},
};
services
.users
.set_profile_key(&body.user_id, body.field.as_str(), Some(value));
if profile_key_value.is_none() {
return Err!(Request(NotFound("The requested profile key does not exist.")));
}
let profile_key_value = ProfileFieldValue::new(body.field.as_str(), value.clone())?;
return Ok(get_profile_field::v3::Response { value: profile_key_value });
return Ok(get_profile_field::v3::Response { value: Some(profile_key_value) });
}
}
@@ -369,20 +362,13 @@ pub(crate) async fn get_profile_field_route(
return Err!(Request(NotFound("Profile was not found.")));
}
let profile_key_value: Option<ProfileFieldValue> = match services
let value = services
.users
.profile_key(&body.user_id, body.field.as_str())
.await
{
| Ok(value) => Some(ProfileFieldValue::new(body.field.as_str(), value)?),
| _ => {
return Err!(Request(NotFound("The requested profile key does not exist.")));
},
};
.map_err(|_| err!(Request(NotFound("The requested profile key does not exist."))))?;
if profile_key_value.is_none() {
return Err!(Request(NotFound("The requested profile key does not exist.")));
}
let profile_key_value = ProfileFieldValue::new(body.field.as_str(), value)?;
Ok(get_profile_field::v3::Response { value: profile_key_value })
Ok(get_profile_field::v3::Response { value: Some(profile_key_value) })
}