feat(messaging): add sendable embeds (#119)

src/routes/channels/message_edit.rs

@@ -1,8 +1,9 @@
 use crate::database::*;
 use crate::util::result::{Error, Result, EmptyResponse};
+use crate::routes::channels::message_send::SendableEmbed;
 
 use chrono::Utc;
-use mongodb::bson::{doc, Bson, DateTime, Document};
+use mongodb::bson::{Bson, Document, doc, to_document};
 use rocket::serde::json::Json;
 use serde::{Deserialize, Serialize};
 use validator::Validate;
@@ -10,7 +11,9 @@ use validator::Validate;
 #[derive(Validate, Serialize, Deserialize)]
 pub struct Data {
     #[validate(length(min = 1, max = 2000))]
-    content: String,
+    content: Option<String>,
+    #[validate(length(min = 0, max = 10))]
+    embeds: Option<Vec<SendableEmbed>>
 }
 
 #[patch("/<target>/messages/<msg>", data = "<edit>")]
@@ -35,26 +38,48 @@ pub async fn req(user: User, target: Ref, msg: Ref, edit: Json<Data>) -> Result<
     }
 
     let edited = Utc::now();
-    let mut set = doc! {
-        "content": &edit.content,
-        "edited": Bson::DateTime(edited)
-    };
+    let mut set = doc! { "edited": Bson::DateTime(edited) };
+    let mut unset = doc! {};
+    let mut update = json!({ "edited": Bson::DateTime(edited) });
 
-    message.content = Content::Text(edit.content.clone());
-    let mut update = json!({ "content": edit.content, "edited": DateTime(edited) });
+    if let Some(new_content) = &edit.content {
+        set.insert("content", new_content.clone());
+        update.as_object_mut().unwrap().insert("content".to_string(), json!(new_content.clone()));
+        message.content = Content::Text(new_content.clone());
+    }
+
+    let mut new_embeds: Vec<Embed> = vec![];
 
     if let Some(embeds) = &message.embeds {
-        let new_embeds: Vec<Document> = vec![];
-
         for embed in embeds {
             match embed {
-                Embed::Website(_) | Embed::Image(_) | Embed::None => {} // Otherwise push to new_embeds.
+                Embed::Text(embed) => new_embeds.push(Embed::Text(embed.clone())),
+                _ => {}
             }
         }
+    }
+
+    if let Some(edited_embeds) = &edit.embeds {
+        new_embeds.clear();
+
+        for embed in edited_embeds {
+            new_embeds.push(embed.clone().into_embed(message.id.clone()).await?);
+        }
+    }
+
+    if new_embeds.len() > 0 {
+        let embed_docs: Vec<Document> = new_embeds
+            .clone()
+            .into_iter()
+            .map(|embed| to_document(&embed).unwrap())
+            .collect();
 
         let obj = update.as_object_mut().unwrap();
-        obj.insert("embeds".to_string(), json!(new_embeds));
-        set.insert("embeds", new_embeds);
+        obj.insert("embeds".to_string(), json!(embed_docs));
+        set.insert("embeds", embed_docs);
+        message.embeds = Some(new_embeds)
+    } else if edit.embeds.is_some() {
+        unset.insert("embeds", 1 as u32);
     }
 
     get_collection("messages")
@@ -63,7 +88,8 @@ pub async fn req(user: User, target: Ref, msg: Ref, edit: Json<Data>) -> Result<
                 "_id": &message.id
             },
             doc! {
-                "$set": set
+                "$set": set,
+                "$unset": unset
             },
             None,
         )