Basic Web Push support.

src/routes/channels/message_query_stale.rs

@@ -8,7 +8,7 @@ use serde::{Deserialize, Serialize};
 
 #[derive(Serialize, Deserialize)]
 pub struct Options {
-    ids: Vec<String>
+    ids: Vec<String>,
 }
 
 #[post("/<target>/messages/stale", data = "<data>")]
@@ -35,7 +35,7 @@ pub async fn req(user: User, target: Ref, data: Json<Options>) -> Result<JsonVal
                 },
                 "channel": target.id()
             },
-            None
+            None,
         )
         .await
         .map_err(|_| Error::DatabaseError {
@@ -47,12 +47,11 @@ pub async fn req(user: User, target: Ref, data: Json<Options>) -> Result<JsonVal
     let mut found_ids = vec![];
     while let Some(result) = cursor.next().await {
         if let Ok(doc) = result {
-            let msg = from_document::<Message>(doc)
-                .map_err(|_| Error::DatabaseError {
-                    operation: "from_document",
-                    with: "message",
-                })?;
-            
+            let msg = from_document::<Message>(doc).map_err(|_| Error::DatabaseError {
+                operation: "from_document",
+                with: "message",
+            })?;
+
             found_ids.push(msg.id.clone());
             if msg.edited.is_some() {
                 updated.push(msg);

src/routes/channels/message_send.rs

@@ -1,7 +1,10 @@
 use crate::database::*;
 use crate::util::result::{Error, Result};
 
-use mongodb::{bson::{doc, from_document}, options::FindOneOptions};
+use mongodb::{
+    bson::{doc, from_document},
+    options::FindOneOptions,
+};
 use rocket_contrib::json::{Json, JsonValue};
 use serde::{Deserialize, Serialize};
 use ulid::Ulid;
@@ -41,7 +44,7 @@ pub async fn req(user: User, target: Ref, message: Json<Data>) -> Result<JsonVal
             },
             FindOneOptions::builder()
                 .projection(doc! { "_id": 1 })
-                .build()
+                .build(),
         )
         .await
         .map_err(|_| Error::DatabaseError {
@@ -64,33 +67,40 @@ pub async fn req(user: User, target: Ref, message: Json<Data>) -> Result<JsonVal
                         "$exists": false
                     }
                 },
-                None
+                None,
             )
             .await
             .map_err(|_| Error::DatabaseError {
                 operation: "find_one",
                 with: "attachment",
-            })? {
-            let attachment = from_document::<File>(doc)
-                .map_err(|_| Error::DatabaseError { operation: "from_document", with: "attachment" })?;
+            })?
+        {
+            let attachment = from_document::<File>(doc).map_err(|_| Error::DatabaseError {
+                operation: "from_document",
+                with: "attachment",
+            })?;
 
-            attachments.update_one(
-                doc! {
-                    "_id": &attachment.id
-                },
-                doc! {
-                    "$set": {
-                        "message_id": &id
-                    }
-                },
-                None
-            )
-            .await
-            .map_err(|_| Error::DatabaseError { operation: "update_one", with: "attachment" })?;
+            attachments
+                .update_one(
+                    doc! {
+                        "_id": &attachment.id
+                    },
+                    doc! {
+                        "$set": {
+                            "message_id": &id
+                        }
+                    },
+                    None,
+                )
+                .await
+                .map_err(|_| Error::DatabaseError {
+                    operation: "update_one",
+                    with: "attachment",
+                })?;
 
             Some(attachment)
         } else {
-            return Err(Error::UnknownAttachment)
+            return Err(Error::UnknownAttachment);
         }
     } else {
         None