feat: implement edit user / username

src/routes/users/change_username.rs

@@ -18,9 +18,10 @@ pub struct Data {
 pub async fn req(
     db: &State<Database>,
     account: Account,
-    user: User,
+    mut user: User,
     data: Json<Data>,
 ) -> Result<EmptyResponse> {
+    let data = data.into_inner();
     account.verify_password(&data.password).map_err(|_| Error::InvalidCredentials)?;
-    user.update_username(db, &data.username).await
+    user.update_username(db, data.username).await.map(|_| EmptyResponse)
 }

src/routes/users/edit_user.rs

@@ -1,7 +1,10 @@
-use revolt_quark::{EmptyResponse, Result, models::user::FieldsUser};
+use revolt_quark::models::File;
+use revolt_quark::{Error, Result, Database};
+use revolt_quark::models::user::{FieldsUser, User, PartialUser};
 
-use mongodb::bson::{doc, to_document};
+use mongodb::bson::doc;
 use revolt_quark::models::user::UserStatus;
+use rocket::State;
 use rocket::serde::json::Json;
 use serde::{Deserialize, Serialize};
 use validator::Validate;
@@ -18,16 +21,80 @@ pub struct UserProfileData {
 
 #[derive(Validate, Serialize, Deserialize)]
 pub struct Data {
-    // #[validate]
+    #[validate]
     status: Option<UserStatus>,
     #[validate]
     profile: Option<UserProfileData>,
     #[validate(length(min = 1, max = 128))]
     avatar: Option<String>,
-    remove: Option<FieldsUser>,
+    #[validate(length(min = 1))]
+    remove: Option<Vec<FieldsUser>>,
 }
 
-#[patch("/<_ignore_id>", data = "<data>")]
-pub async fn req(/*user: UserRef,*/ data: Json<Data>, _ignore_id: String) -> Result<EmptyResponse> {
-    todo!()
+#[patch("/@me", data = "<data>")]
+pub async fn req(db: &State<Database>, mut user: User, data: Json<Data>) -> Result<Json<User>> {
+    let data = data.into_inner();
+    data.validate().map_err(|error| Error::FailedValidation { error })?;
+
+    if data.status.is_none() && data.profile.is_none() && data.avatar.is_none() && data.remove.is_none() {
+        return Ok(Json(user))
+    }
+
+    if let Some(fields) = &data.remove {
+        if fields.contains(&FieldsUser::Avatar) {
+            if let Some(_avatar) = &user.avatar {
+                // TODO: delete the avatar
+            }
+        }
+
+        if fields.contains(&FieldsUser::ProfileBackground) {
+            if let Some(profile) = &user.profile {
+                if let Some(_background) = &profile.background {
+                    // TODO: delete the background
+                }
+            }
+        }
+
+        for field in fields {
+            user.remove(field);
+        }
+    }
+
+    let mut partial: PartialUser = Default::default();
+
+    if let Some(avatar) = data.avatar {
+        partial.avatar = Some(File::use_avatar(db, &avatar, &user.id).await?);
+        user.avatar = partial.avatar.clone();
+    }
+
+    if let Some(status) = data.status {
+        let mut new_status = user.status.take().unwrap_or_default();
+        if let Some(text) = status.text {
+            new_status.text = Some(text);
+        }
+
+        if let Some(presence) = status.presence {
+            new_status.presence = Some(presence);
+        }
+
+        partial.status = Some(new_status);
+        user.status = partial.status.clone();
+    }
+
+    if let Some(profile) = data.profile {
+        let mut new_profile = user.profile.take().unwrap_or_default();
+        if let Some(content) = profile.content {
+            new_profile.content = Some(content);
+        }
+
+        if let Some(background) = profile.background {
+            new_profile.background = Some(File::use_background(db, &background, &user.id).await?);
+        }
+
+        partial.profile = Some(new_profile);
+        user.profile = partial.profile.clone();
+    }
+
+    db.update_user(&user.id, &partial, data.remove.unwrap_or_default()).await?;
+    Ok(Json(user))
 }