magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

make set avatar, displayname, and blurhash async and forgot another let _

Browse source 
Signed-off-by: strawberry <strawberry@puppygock.gay>
This commit is contained in:
strawberry 2024-01-14 01:42:37 -05:00 • committed by June
parent 9ec1532580
commit 149d22eef7
5 changed files with 49 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/api/client_server/account.rs
View file

@ -205,7 +205,8 @@ pub async fn register_route(body: Ruma<register::v3::Request>) -> Result<registe
services() services()
.users .users
.set_displayname(&user_id, Some(displayname.clone()))?; .set_displayname(&user_id, Some(displayname.clone()))
.await?;
// Initial account data // Initial account data
services().account_data.update( services().account_data.update(

36
src/api/client_server/profile.rs
View file

@ -27,7 +27,8 @@ pub async fn set_displayname_route(
services() services()
.users .users
.set_displayname(sender_user, body.displayname.clone())?; .set_displayname(sender_user, body.displayname.clone())
.await?;
// Send a new membership event and presence update into all joined rooms // Send a new membership event and presence update into all joined rooms
let all_rooms_joined: Vec<_> = services() let all_rooms_joined: Vec<_> = services()
@ -132,13 +133,16 @@ pub async fn get_displayname_route(
let _ = services().users.create(&body.user_id, None); let _ = services().users.create(&body.user_id, None);
let _ = services() let _ = services()
.users .users
.set_displayname(&body.user_id, response.displayname.clone()); .set_displayname(&body.user_id, response.displayname.clone())
.await;
let _ = services() let _ = services()
.users .users
.set_avatar_url(&body.user_id, response.avatar_url); .set_avatar_url(&body.user_id, response.avatar_url)
.await;
let _ = services() let _ = services()
.users .users
.set_blurhash(&body.user_id, response.blurhash); .set_blurhash(&body.user_id, response.blurhash)
.await;
return Ok(get_display_name::v3::Response { return Ok(get_display_name::v3::Response {
displayname: response.displayname, displayname: response.displayname,
@ -162,11 +166,13 @@ pub async fn set_avatar_url_route(
services() services()
.users .users
.set_avatar_url(sender_user, body.avatar_url.clone())?; .set_avatar_url(sender_user, body.avatar_url.clone())
.await?;
services() services()
.users .users
.set_blurhash(sender_user, body.blurhash.clone())?; .set_blurhash(sender_user, body.blurhash.clone())
.await?;
// Send a new membership event and presence update into all joined rooms // Send a new membership event and presence update into all joined rooms
let all_joined_rooms: Vec<_> = services() let all_joined_rooms: Vec<_> = services()
@ -271,13 +277,16 @@ pub async fn get_avatar_url_route(
let _ = services().users.create(&body.user_id, None); let _ = services().users.create(&body.user_id, None);
let _ = services() let _ = services()
.users .users
.set_displayname(&body.user_id, response.displayname); .set_displayname(&body.user_id, response.displayname)
.await;
let _ = services() let _ = services()
.users .users
.set_avatar_url(&body.user_id, response.avatar_url.clone()); .set_avatar_url(&body.user_id, response.avatar_url.clone())
.await;
let _ = services() let _ = services()
.users .users
.set_blurhash(&body.user_id, response.blurhash.clone()); .set_blurhash(&body.user_id, response.blurhash.clone())
.await;
return Ok(get_avatar_url::v3::Response { return Ok(get_avatar_url::v3::Response {
avatar_url: response.avatar_url, avatar_url: response.avatar_url,
@ -320,13 +329,16 @@ pub async fn get_profile_route(
let _ = services().users.create(&body.user_id, None); let _ = services().users.create(&body.user_id, None);
let _ = services() let _ = services()
.users .users
.set_displayname(&body.user_id, response.displayname.clone()); .set_displayname(&body.user_id, response.displayname.clone())
.await;
let _ = services() let _ = services()
.users .users
.set_avatar_url(&body.user_id, response.avatar_url.clone()); .set_avatar_url(&body.user_id, response.avatar_url.clone())
.await;
let _ = services() let _ = services()
.users .users
.set_blurhash(&body.user_id, response.blurhash.clone()); .set_blurhash(&body.user_id, response.blurhash.clone())
.await;
return Ok(get_profile::v3::Response { return Ok(get_profile::v3::Response {
displayname: response.displayname, displayname: response.displayname,

3
src/service/admin/mod.rs
View file

@ -600,7 +600,8 @@ impl Service {
services() services()
.users .users
.set_displayname(&user_id, Some(displayname))?; .set_displayname(&user_id, Some(displayname))
.await?;
// Initial account data // Initial account data
services().account_data.update( services().account_data.update(

15
src/service/rooms/state_cache/mod.rs
View file

@ -58,13 +58,18 @@ impl Service {
}, },
) )
.await?; .await?;
services() let _ = services()
.users .users
.set_displayname(user_id, response.displayname.clone())?; .set_displayname(user_id, response.displayname.clone())
services() .await;
let _ = services()
.users .users
.set_avatar_url(user_id, response.avatar_url)?; .set_avatar_url(user_id, response.avatar_url)
services().users.set_blurhash(user_id, response.blurhash)?; .await;
let _ = services()
.users
.set_blurhash(user_id, response.blurhash)
.await;
}; };
} }

14
src/service/users/mod.rs
View file

@ -314,7 +314,11 @@ impl Service {
} }
/// Sets a new displayname or removes it if displayname is None. You still need to nofify all rooms of this change. /// Sets a new displayname or removes it if displayname is None. You still need to nofify all rooms of this change.
pub fn set_displayname(&self, user_id: &UserId, displayname: Option<String>) -> Result<()> { pub async fn set_displayname(
&self,
user_id: &UserId,
displayname: Option<String>,
) -> Result<()> {
self.db.set_displayname(user_id, displayname) self.db.set_displayname(user_id, displayname)
} }
@ -324,7 +328,11 @@ impl Service {
} }
/// Sets a new avatar_url or removes it if avatar_url is None. /// Sets a new avatar_url or removes it if avatar_url is None.
pub fn set_avatar_url(&self, user_id: &UserId, avatar_url: Option<OwnedMxcUri>) -> Result<()> { pub async fn set_avatar_url(
&self,
user_id: &UserId,
avatar_url: Option<OwnedMxcUri>,
) -> Result<()> {
self.db.set_avatar_url(user_id, avatar_url) self.db.set_avatar_url(user_id, avatar_url)
} }
@ -334,7 +342,7 @@ impl Service {
} }
/// Sets a new avatar_url or removes it if avatar_url is None. /// Sets a new avatar_url or removes it if avatar_url is None.
pub fn set_blurhash(&self, user_id: &UserId, blurhash: Option<String>) -> Result<()> { pub async fn set_blurhash(&self, user_id: &UserId, blurhash: Option<String>) -> Result<()> {
self.db.set_blurhash(user_id, blurhash) self.db.set_blurhash(user_id, blurhash)
} }