refactor(quark): port ban_create, ban_list, ban_remove, emoji_list, invites_fetch

#283

crates/delta/src/routes/servers/ban_create.rs

@@ -1,73 +1,66 @@
-use revolt_quark::{
-    models::{server_member::MemberCompositeKey, ServerBan, User},
-    perms, Db, Error, Permission, Ref, Result,
+use revolt_database::{
+    util::{permissions::DatabasePermissionQuery, reference::Reference},
+    Database, RemovalIntention, ServerBan, User,
 };
+use revolt_models::v0;
 
-use rocket::serde::json::Json;
+use revolt_permissions::{
+    calculate_channel_permissions, calculate_server_permissions, ChannelPermission,
+};
+use revolt_result::{create_error, Result};
+use rocket::{serde::json::Json, State};
 use serde::{Deserialize, Serialize};
 use validator::Validate;
 
-/// # Ban Information
-#[derive(Validate, Serialize, Deserialize, JsonSchema)]
-pub struct DataBanCreate {
-    /// Ban reason
-    #[validate(length(min = 1, max = 1024))]
-    reason: Option<String>,
-}
-
 /// # Ban User
 ///
 /// Ban a user by their id.
 #[openapi(tag = "Server Members")]
 #[put("/<server>/bans/<target>", data = "<data>")]
-pub async fn req(
-    db: &Db,
+pub async fn ban(
+    db: &State<Database>,
     user: User,
-    server: Ref,
-    target: Ref,
-    data: Json<DataBanCreate>,
-) -> Result<Json<ServerBan>> {
+    server: Reference,
+    target: Reference,
+    data: Json<v0::DataBanCreate>,
+) -> Result<Json<v0::ServerBan>> {
     let data = data.into_inner();
-    data.validate()
-        .map_err(|error| Error::FailedValidation { error })?;
+    data.validate().map_err(|error| {
+        create_error!(FailedValidation {
+            error: error.to_string()
+        })
+    })?;
 
     let server = server.as_server(db).await?;
 
     if target.id == user.id {
-        return Err(Error::CannotRemoveYourself);
+        return Err(create_error!(CannotRemoveYourself));
     }
 
     if target.id == server.owner {
-        return Err(Error::InvalidOperation);
+        return Err(create_error!(InvalidOperation));
     }
 
-    let mut permissions = perms(&user).server(&server);
-
-    permissions
-        .throw_permission(db, Permission::BanMembers)
-        .await?;
+    let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
+    calculate_server_permissions(&mut query)
+        .await
+        .throw_if_lacking_channel_permission(ChannelPermission::BanMembers)?;
 
     // If member exists, check privileges against them
     if let Ok(member) = target.as_member(db, &server.id).await {
-        if member.get_ranking(permissions.server.get().unwrap())
-            <= permissions.get_member_rank().unwrap_or(i64::MIN)
+        if member.get_ranking(&query.server_ref().as_ref().unwrap())
+            <= query.get_member_rank().unwrap_or(i64::MIN)
         {
-            return Err(Error::NotElevated);
+            return Err(create_error!(NotElevated));
         }
 
-        server.ban_member(db, member, data.reason).await.map(Json)
-    } else {
-        let server_id = server.id.to_string();
-        server
-            .ban_user(
-                db,
-                MemberCompositeKey {
-                    server: server_id,
-                    user: target.id,
-                },
-                data.reason,
-            )
-            .await
-            .map(Json)
+        member
+            .remove(db, &server, RemovalIntention::Ban, false)
+            .await?;
     }
+
+    ServerBan::create(db, &server, &target.id, data.reason)
+        .await
+        .map(Into::into)
+        .map(Json)
 }

crates/delta/src/routes/servers/ban_list.rs

@@ -1,60 +1,33 @@
-use revolt_quark::models::{File, ServerBan, User};
-use revolt_quark::{perms, Db, Permission, Ref, Result};
+use futures::future::join_all;
+use revolt_database::util::permissions::DatabasePermissionQuery;
+use revolt_database::util::reference::Reference;
+use revolt_database::{Database, User};
+use revolt_models::v0;
 
+use revolt_permissions::{calculate_server_permissions, ChannelPermission};
+use revolt_result::Result;
 use rocket::serde::json::Json;
-use serde::{Deserialize, Serialize};
-
-/// # Banned User
-///
-/// Just enoguh user information to list bans.
-#[derive(Serialize, Deserialize, JsonSchema)]
-struct BannedUser {
-    /// Id of the banned user
-    #[serde(rename = "_id")]
-    pub id: String,
-    /// Username of the banned user
-    pub username: String,
-    /// Discriminator of the banned user
-    pub discriminator: String,
-    /// Avatar of the banned user
-    pub avatar: Option<File>,
-}
-
-/// # Ban List Result
-#[derive(Serialize, Deserialize, JsonSchema)]
-pub struct BanListResult {
-    /// Users objects
-    users: Vec<BannedUser>,
-    /// Ban objects
-    bans: Vec<ServerBan>,
-}
-
-impl From<User> for BannedUser {
-    fn from(user: User) -> Self {
-        BannedUser {
-            id: user.id,
-            username: user.username,
-            discriminator: user.discriminator,
-            avatar: user.avatar,
-        }
-    }
-}
+use rocket::State;
 
 /// # Fetch Bans
 ///
 /// Fetch all bans on a server.
 #[openapi(tag = "Server Members")]
 #[get("/<target>/bans")]
-pub async fn req(db: &Db, user: User, target: Ref) -> Result<Json<BanListResult>> {
+pub async fn list(
+    db: &State<Database>,
+    user: User,
+    target: Reference,
+) -> Result<Json<v0::BanListResult>> {
     let server = target.as_server(db).await?;
-    perms(&user)
-        .server(&server)
-        .throw_permission(db, Permission::BanMembers)
-        .await?;
+    let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
+    calculate_server_permissions(&mut query)
+        .await
+        .throw_if_lacking_channel_permission(ChannelPermission::BanMembers)?;
 
     let bans = db.fetch_bans(&server.id).await?;
-    let users = db
-        .fetch_users(
+    let users = join_all(
+        db.fetch_users(
             &bans
                 .iter()
                 .map(|x| &x.id.user)
@@ -63,8 +36,12 @@ pub async fn req(db: &Db, user: User, target: Ref) -> Result<Json<BanListResult>
         )
         .await?
         .into_iter()
-        .map(|x| x.into())
-        .collect();
+        .map(|u| u.into_self()),
+    )
+    .await;
 
-    Ok(Json(BanListResult { users, bans }))
+    Ok(Json(v0::BanListResult {
+        users: users.into_iter().map(Into::into).collect(),
+        bans: bans.into_iter().map(Into::into).collect(),
+    }))
 }

crates/delta/src/routes/servers/ban_remove.rs

@@ -1,16 +1,28 @@
-use revolt_quark::{models::User, perms, Db, EmptyResponse, Permission, Ref, Result};
+use revolt_database::{
+    util::{permissions::DatabasePermissionQuery, reference::Reference},
+    Database, User,
+};
+use revolt_permissions::{calculate_server_permissions, ChannelPermission};
+use revolt_result::Result;
+use rocket::State;
+use rocket_empty::EmptyResponse;
 
 /// # Unban user
 ///
 /// Remove a user's ban.
 #[openapi(tag = "Server Members")]
 #[delete("/<server>/bans/<target>")]
-pub async fn req(db: &Db, user: User, server: Ref, target: Ref) -> Result<EmptyResponse> {
+pub async fn unban(
+    db: &State<Database>,
+    user: User,
+    server: Reference,
+    target: Reference,
+) -> Result<EmptyResponse> {
     let server = server.as_server(db).await?;
-    perms(&user)
-        .server(&server)
-        .throw_permission(db, Permission::BanMembers)
-        .await?;
+    let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
+    calculate_server_permissions(&mut query)
+        .await
+        .throw_if_lacking_channel_permission(ChannelPermission::BanMembers)?;
 
     let ban = target.as_ban(db, &server.id).await?;
     db.delete_ban(&ban.id).await.map(|_| EmptyResponse)

crates/delta/src/routes/servers/emoji_list.rs

@@ -1,17 +1,31 @@
-use revolt_quark::models::{Emoji, User};
-use revolt_quark::{perms, Db, Ref, Result};
-
-use rocket::serde::json::Json;
+use revolt_database::{
+    util::{permissions::DatabasePermissionQuery, reference::Reference},
+    Database, User,
+};
+use revolt_models::v0;
+use revolt_permissions::PermissionQuery;
+use revolt_result::{create_error, Result};
+use rocket::{serde::json::Json, State};
 
 /// # Fetch Server Emoji
 ///
 /// Fetch all emoji on a server.
 #[openapi(tag = "Server Customisation")]
 #[get("/<target>/emojis")]
-pub async fn list_emoji(db: &Db, user: User, target: Ref) -> Result<Json<Vec<Emoji>>> {
+pub async fn list_emoji(
+    db: &State<Database>,
+    user: User,
+    target: Reference,
+) -> Result<Json<Vec<v0::Emoji>>> {
     let server = target.as_server(db).await?;
-    perms(&user).server(&server).calc(db).await?;
+    let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
+    if !query.are_we_a_member().await {
+        return Err(create_error!(NotFound));
+    }
 
     // Fetch all emoji from server if we can view it
-    db.fetch_emoji_by_parent_id(&server.id).await.map(Json)
+    db.fetch_emoji_by_parent_id(&server.id)
+        .await
+        .map(|v| v.into_iter().map(Into::into).collect())
+        .map(Json)
 }

crates/delta/src/routes/servers/invites_fetch.rs

@@ -1,21 +1,30 @@
-use revolt_quark::{
-    models::{Invite, User},
-    perms, Db, Permission, Ref, Result,
+use revolt_database::{
+    util::{permissions::DatabasePermissionQuery, reference::Reference},
+    Database, User,
 };
-
-use rocket::serde::json::Json;
+use revolt_models::v0;
+use revolt_permissions::{calculate_server_permissions, ChannelPermission};
+use revolt_result::Result;
+use rocket::{serde::json::Json, State};
 
 /// # Fetch Invites
 ///
 /// Fetch all server invites.
 #[openapi(tag = "Server Members")]
 #[get("/<target>/invites")]
-pub async fn req(db: &Db, user: User, target: Ref) -> Result<Json<Vec<Invite>>> {
+pub async fn invites(
+    db: &State<Database>,
+    user: User,
+    target: Reference,
+) -> Result<Json<Vec<v0::Invite>>> {
     let server = target.as_server(db).await?;
-    perms(&user)
-        .server(&server)
-        .throw_permission(db, Permission::ManageServer)
-        .await?;
+    let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
+    calculate_server_permissions(&mut query)
+        .await
+        .throw_if_lacking_channel_permission(ChannelPermission::ManageServer)?;
 
-    db.fetch_invites_for_server(&server.id).await.map(Json)
+    db.fetch_invites_for_server(&server.id)
+        .await
+        .map(|v| v.into_iter().map(Into::into).collect())
+        .map(Json)
 }

crates/delta/src/routes/servers/mod.rs

@@ -37,10 +37,10 @@ pub fn routes() -> (Vec<Route>, OpenApi) {
         member_fetch::req,
         member_edit::req,
         member_experimental_query::member_experimental_query,
-        ban_create::req,
-        ban_remove::req,
-        ban_list::req,
-        invites_fetch::req,
+        ban_create::ban,
+        ban_remove::unban,
+        ban_list::list,
+        invites_fetch::invites,
         roles_create::req,
         roles_edit::req,
         roles_fetch::req,