Post-handle hook, partial updates and group delete.

src/routes/channels/delete_channel.rs

@@ -1,5 +1,5 @@
-use crate::database::*;
 use crate::util::result::{Error, Result};
+use crate::{database::*, notifications::events::ClientboundNotification};
 
 use mongodb::bson::doc;
 
@@ -12,7 +12,7 @@ pub async fn req(user: User, target: Ref) -> Result<()> {
         Err(Error::LabelMe)?
     }
 
-    match target {
+    match &target {
         Channel::SavedMessages { .. } => Err(Error::NoEffect),
         Channel::DirectMessage { .. } => {
             get_collection("channels")
@@ -35,6 +35,77 @@ pub async fn req(user: User, target: Ref) -> Result<()> {
 
             Ok(())
         }
-        _ => unimplemented!(),
+        Channel::Group {
+            id,
+            owner,
+            recipients,
+            ..
+        } => {
+            if &user.id == owner {
+                if let Some(new_owner) = recipients.iter().find(|x| *x != &user.id) {
+                    get_collection("channels")
+                        .update_one(
+                            doc! {
+                                "_id": &id
+                            },
+                            doc! {
+                                "$set": {
+                                    "owner": new_owner
+                                },
+                                "$pull": {
+                                    "recipients": &user.id
+                                }
+                            },
+                            None,
+                        )
+                        .await
+                        .map_err(|_| Error::DatabaseError {
+                            operation: "update_one",
+                            with: "channel",
+                        })?;
+
+                    target.publish_update(json!({ "owner": new_owner })).await?;
+                } else {
+                    return target.delete().await
+                }
+            } else {
+                get_collection("channels")
+                    .update_one(
+                        doc! {
+                            "_id": &id
+                        },
+                        doc! {
+                            "$pull": {
+                                "recipients": &user.id
+                            }
+                        },
+                        None,
+                    )
+                    .await
+                    .map_err(|_| Error::DatabaseError {
+                        operation: "update_one",
+                        with: "channel",
+                    })?;
+            }
+
+            ClientboundNotification::ChannelGroupLeave {
+                id: id.clone(),
+                user: user.id.clone(),
+            }
+            .publish(id.clone())
+            .await
+            .ok();
+
+            Message::create(
+                "00000000000000000000000000".to_string(),
+                id.clone(),
+                format!("<@{}> left the group.", user.id),
+            )
+            .publish()
+            .await
+            .ok();
+
+            Ok(())
+        }
     }
 }

src/routes/channels/message_delete.rs

@@ -20,20 +20,5 @@ pub async fn req(user: User, target: Ref, msg: Ref) -> Result<()> {
         }
     }
 
-    get_collection("messages")
-        .delete_one(
-            doc! {
-                "_id": &message.id
-            },
-            None,
-        )
-        .await
-        .map_err(|_| Error::DatabaseError {
-            operation: "delete_one",
-            with: "message",
-        })?;
-
-    message.publish_delete().await?;
-
-    Ok(())
+    message.delete().await
 }

src/routes/channels/message_edit.rs

@@ -25,7 +25,7 @@ pub async fn req(user: User, target: Ref, msg: Ref, edit: Json<Data>) -> Result<
         Err(Error::LabelMe)?
     }
 
-    let mut message = msg.fetch_message().await?;
+    let message = msg.fetch_message().await?;
     if message.author != user.id {
         Err(Error::CannotEditMessage)?
     }
@@ -50,9 +50,5 @@ pub async fn req(user: User, target: Ref, msg: Ref, edit: Json<Data>) -> Result<
             with: "message",
         })?;
 
-    message.content = edit.content.clone();
-    message.edited = Some(DateTime(edited));
-    message.publish_edit().await?;
-
-    Ok(())
+    message.publish_update(json!({ "content": edit.content, "edited": DateTime(edited) })).await
 }