fix: fixes for full mongo implementation

src/main.rs

@@ -121,7 +121,10 @@ async fn launch_web() {
         };
     }
 
-    let db = DatabaseInfo::MongoDb("mongodb://localhost").connect().await.unwrap();
+    let db = DatabaseInfo::MongoDb("mongodb://localhost")
+        .connect()
+        .await
+        .unwrap();
     db.migrate_database().await.unwrap();
 
     let mongo_db = mongodb::Client::with_uri_str("mongodb://localhost")

src/routes/bots/edit.rs

@@ -83,7 +83,7 @@ pub async fn edit_bot(db: &Db, user: User, target: Ref, data: Json<Data>) -> Res
 
     db.update_bot(&bot.id, &partial, remove.unwrap_or_else(Vec::new))
         .await?;
-    
+
     bot.apply_options(partial);
     Ok(Json(bot))
 }

src/routes/channels/channel_delete.rs

@@ -3,7 +3,6 @@ use revolt_quark::{
     perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result,
 };
 
-
 #[delete("/<target>")]
 pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> {
     let channel = target.as_channel(db).await?;

src/routes/channels/group_add_member.rs

@@ -3,7 +3,6 @@ use revolt_quark::{
     perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result,
 };
 
-
 #[put("/<target>/recipients/<member>")]
 pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
     let channel = target.as_channel(db).await?;

src/routes/channels/group_remove_member.rs

@@ -3,7 +3,6 @@ use revolt_quark::{
     Db, EmptyResponse, Error, Ref, Result,
 };
 
-
 #[delete("/<target>/recipients/<member>")]
 pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
     let channel = target.as_channel(db).await?;

src/routes/channels/message_delete.rs

@@ -1,6 +1,5 @@
 use revolt_quark::{models::User, perms, ChannelPermission, Db, EmptyResponse, Error, Ref, Result};
 
-
 #[delete("/<target>/messages/<msg>")]
 pub async fn req(db: &Db, user: User, target: Ref, msg: Ref) -> Result<EmptyResponse> {
     let message = msg.as_message(db).await?;

src/routes/channels/message_edit.rs

@@ -74,6 +74,6 @@ pub async fn req(
     message.embeds = Some(new_embeds);
     partial.embeds = message.embeds.clone();
 
-    db.update_message(&message.id, &partial, vec![]).await?;
+    db.update_message(&message.id, &partial).await?;
     Ok(Json(message))
 }

src/routes/invites/invite_join.rs

@@ -1,4 +1,7 @@
-use revolt_quark::{Error, Result, Ref, models::{User, Invite}, Db};
+use revolt_quark::{
+    models::{Invite, User},
+    Db, Error, Ref, Result,
+};
 
 use rocket::serde::json::Value;
 
@@ -7,32 +10,34 @@ use crate::util::variables::MAX_SERVER_COUNT;
 #[post("/<target>")]
 pub async fn req(db: &Db, user: User, target: Ref) -> Result<Value> {
     if user.bot.is_some() {
-        return Err(Error::IsBot)
+        return Err(Error::IsBot);
     }
 
     if !user.can_acquire_server(db).await? {
         return Err(Error::TooManyServers {
-            max: *MAX_SERVER_COUNT
-        })
+            max: *MAX_SERVER_COUNT,
+        });
     }
 
     let invite = target.as_invite(db).await?;
     match &invite {
-        Invite::Server { channel, server, .. } => {
+        Invite::Server {
+            channel, server, ..
+        } => {
             let server = db.fetch_server(server).await?;
             if db.fetch_ban(&server.id, &user.id).await.is_ok() {
-                return Err(Error::Banned)
+                return Err(Error::Banned);
             }
 
             let channel = db.fetch_channel(channel).await?;
             db.insert_member(&server.id, &user.id).await?;
-            
+
             Ok(json!({
                 "type": "Server",
                 "channel": channel,
                 "server": server
             }))
         }
-        _ => unreachable!()
+        _ => unreachable!(),
     }
 }

src/routes/servers/ban_list.rs

@@ -1,4 +1,4 @@
-use revolt_quark::models::{File, User, ServerBan};
+use revolt_quark::models::{File, ServerBan, User};
 use revolt_quark::{perms, Db, Error, Ref, Result, ServerPermission};
 
 use rocket::serde::json::Json;

src/routes/servers/member_remove.rs

@@ -1,6 +1,5 @@
 use revolt_quark::{models::User, perms, Db, EmptyResponse, Error, Ref, Result, ServerPermission};
 
-
 #[delete("/<target>/members/<member>")]
 pub async fn req(db: &Db, user: User, target: Ref, member: Ref) -> Result<EmptyResponse> {
     let server = target.as_server(db).await?;

src/routes/servers/server_delete.rs

@@ -1,6 +1,5 @@
 use revolt_quark::{models::User, Db, EmptyResponse, Ref, Result};
 
-
 #[delete("/<target>")]
 pub async fn req(db: &Db, user: User, target: Ref) -> Result<EmptyResponse> {
     let server = target.as_server(db).await?;

src/routes/users/block_user.rs

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
 use rocket::serde::json::Json;
 use rocket::State;
 
-#[put("/<username>/block")]
-pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> {
-    let mut target = db.fetch_user_by_username(&username).await?;
+#[put("/<target>/block")]
+pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
+    let mut target = db.fetch_user(&target).await?;
     user.block_user(db, &mut target).await?;
     Ok(Json(target.with_auto_perspective(db, &user).await))
 }

src/routes/users/remove_friend.rs

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
 use rocket::serde::json::Json;
 use rocket::State;
 
-#[delete("/<username>/friend")]
-pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> {
-    let mut target = db.fetch_user_by_username(&username).await?;
+#[delete("/<target>/friend")]
+pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
+    let mut target = db.fetch_user(&target).await?;
     user.remove_friend(db, &mut target).await?;
     Ok(Json(target.with_auto_perspective(db, &user).await))
 }

src/routes/users/unblock_user.rs

@@ -4,9 +4,9 @@ use revolt_quark::{Database, Result};
 use rocket::serde::json::Json;
 use rocket::State;
 
-#[delete("/<username>/block")]
-pub async fn req(db: &State<Database>, user: User, username: String) -> Result<Json<User>> {
-    let mut target = db.fetch_user_by_username(&username).await?;
+#[delete("/<target>/block")]
+pub async fn req(db: &State<Database>, user: User, target: String) -> Result<Json<User>> {
+    let mut target = db.fetch_user(&target).await?;
     user.unblock_user(db, &mut target).await?;
     Ok(Json(target.with_auto_perspective(db, &user).await))
 }