Add group icons / profile backgrounds.

src/routes/channels/edit_channel.rs

@@ -15,14 +15,17 @@ pub struct Data {
     #[validate(length(min = 0, max = 1024))]
     #[serde(skip_serializing_if = "Option::is_none")]
     description: Option<String>,
+    #[validate(length(min = 1, max = 128))]
+    icon: Option<String>,
 }
 
 #[patch("/<target>", data = "<info>")]
 pub async fn req(user: User, target: Ref, info: Json<Data>) -> Result<()> {
+    let info = info.into_inner();
     info.validate()
         .map_err(|error| Error::FailedValidation { error })?;
 
-    if info.name.is_none() && info.description.is_none() {
+    if info.name.is_none() && info.description.is_none() && info.icon.is_none() {
         return Ok(());
     }
 
@@ -37,11 +40,34 @@ pub async fn req(user: User, target: Ref, info: Json<Data>) -> Result<()> {
     }
 
     match &target {
-        Channel::Group { id, .. } => {
+        Channel::Group { id, icon, .. } => {
+            let mut set = doc! {};
+            if let Some(name) = &info.name {
+                set.insert("name", name);
+            }
+
+            if let Some(description) = info.description {
+                set.insert("description", description);
+            }
+
+            let mut remove_icon = false;
+            if let Some(attachment_id) = info.icon {
+                let attachment = File::find_and_use(&attachment_id, "icons", "object", &user.id).await?;
+                set.insert(
+                    "icon",
+                    to_document(&attachment).map_err(|_| Error::DatabaseError {
+                        operation: "to_document",
+                        with: "attachment",
+                    })?,
+                );
+    
+                remove_icon = true;
+            }
+
             get_collection("channels")
             .update_one(
                 doc! { "_id": &id },
-                doc! { "$set": to_document(&info.0).map_err(|_| Error::DatabaseError { operation: "to_document", with: "data" })? },
+                doc! { "$set": &set },
                 None
             )
             .await
@@ -49,18 +75,18 @@ pub async fn req(user: User, target: Ref, info: Json<Data>) -> Result<()> {
 
             ClientboundNotification::ChannelUpdate {
                 id: id.clone(),
-                data: json!(info.0),
+                data: json!(set),
             }
             .publish(id.clone())
             .await
             .ok();
 
-            if let Some(name) = &info.name {
+            if let Some(name) = info.name {
                 Message::create(
                     "00000000000000000000000000".to_string(),
                     id.clone(),
                     Content::SystemMessage(SystemMessage::ChannelRenamed {
-                        name: name.clone(),
+                        name,
                         by: user.id,
                     }),
                 )
@@ -69,6 +95,12 @@ pub async fn req(user: User, target: Ref, info: Json<Data>) -> Result<()> {
                 .ok();
             }
 
+            if remove_icon {
+                if let Some(old_icon) = icon {
+                    old_icon.delete().await?;
+                }
+            }
+
             Ok(())
         }
         _ => Err(Error::InvalidOperation),

src/routes/channels/group_create.rs

@@ -70,6 +70,7 @@ pub async fn req(user: User, info: Json<Data>) -> Result<JsonValue> {
             .unwrap_or_else(|| "A group.".to_string()),
         owner: user.id,
         recipients: set.into_iter().collect::<Vec<String>>(),
+        icon: None,
         last_message: None,
     };