feat: allow bulk deleting attribution groups for moderators (#6839)

* feat: allow bulk deleting attribution groups for moderators

* prepr

apps/labrinth/.sqlx/query-14779c6105c15c6a117332caf850c7939af5d70562eb309680047da3c76f74fd.json

@@ -1,14 +1,14 @@
 {
   "db_name": "PostgreSQL",
-  "query": "\n\t\tDELETE FROM project_attribution_files\n\t\tWHERE group_id = $1\n\t\t",
+  "query": "\n\t\tDELETE FROM project_attribution_files\n\t\tWHERE group_id = ANY($1)\n\t\t",
   "describe": {
     "columns": [],
     "parameters": {
       "Left": [
-        "Int8"
+        "Int8Array"
       ]
     },
     "nullable": []
   },
-  "hash": "b69317d234e934aa6bdf8be4d35bc7475308bde8b7f34a4ef3c7f3ef351f46a8"
+  "hash": "14779c6105c15c6a117332caf850c7939af5d70562eb309680047da3c76f74fd"
 }

apps/labrinth/.sqlx/query-41ec69f790366c741cea01aa7e1bfac7912b4ed17b7d19b5ba10221fa93bc3f8.json

@@ -1,14 +1,14 @@
 {
   "db_name": "PostgreSQL",
-  "query": "\n\t\tDELETE FROM project_attribution_groups\n\t\tWHERE id = $1\n\t\t",
+  "query": "\n\t\tDELETE FROM project_attribution_groups\n\t\tWHERE id = ANY($1)\n\t\t",
   "describe": {
     "columns": [],
     "parameters": {
       "Left": [
-        "Int8"
+        "Int8Array"
       ]
     },
     "nullable": []
   },
-  "hash": "827534d191739cd2ea502142a65f62a052715c7441952ee4ee182afde07c44d7"
+  "hash": "41ec69f790366c741cea01aa7e1bfac7912b4ed17b7d19b5ba10221fa93bc3f8"
 }

apps/labrinth/.sqlx/query-78f66d65d665210fd8334d2e1e8a793ce8128f7d6b53b3b408b030bfee4b3841.json

@@ -0,0 +1,22 @@
+{
+  "db_name": "PostgreSQL",
+  "query": "\n\t\tSELECT id AS \"id: DBAttributionGroupId\"\n\t\tFROM project_attribution_groups\n\t\tWHERE project_id = $1\n\t\t",
+  "describe": {
+    "columns": [
+      {
+        "ordinal": 0,
+        "name": "id: DBAttributionGroupId",
+        "type_info": "Int8"
+      }
+    ],
+    "parameters": {
+      "Left": [
+        "Int8"
+      ]
+    },
+    "nullable": [
+      false
+    ]
+  },
+  "hash": "78f66d65d665210fd8334d2e1e8a793ce8128f7d6b53b3b408b030bfee4b3841"
+}

apps/labrinth/.sqlx/query-bfc67441eb246ac6b155db36919a7a62e19e0dd0c87916e761175ec909add15b.json

@@ -1,6 +1,6 @@
 {
   "db_name": "PostgreSQL",
-  "query": "\n\t\tSELECT DISTINCT f.version_id AS \"version_id: DBVersionId\"\n\t\tFROM project_attribution_files paf\n\t\tINNER JOIN project_attribution_groups pag ON pag.id = paf.group_id\n\t\tINNER JOIN override_file_sources ofs ON ofs.sha1 = paf.sha1\n\t\tINNER JOIN files f ON f.id = ofs.file_id\n\t\tINNER JOIN versions v ON v.id = f.version_id\n\t\tWHERE paf.group_id = $1\n\t\t\tAND pag.project_id = v.mod_id\n\t\t",
+  "query": "\n\t\tSELECT DISTINCT f.version_id AS \"version_id: DBVersionId\"\n\t\tFROM project_attribution_files paf\n\t\tINNER JOIN project_attribution_groups pag ON pag.id = paf.group_id\n\t\tINNER JOIN override_file_sources ofs ON ofs.sha1 = paf.sha1\n\t\tINNER JOIN files f ON f.id = ofs.file_id\n\t\tINNER JOIN versions v ON v.id = f.version_id\n\t\tWHERE paf.group_id = ANY($1)\n\t\t\tAND pag.project_id = v.mod_id\n\t\t",
   "describe": {
     "columns": [
       {
@@ -11,12 +11,12 @@
     ],
     "parameters": {
       "Left": [
-        "Int8"
+        "Int8Array"
       ]
     },
     "nullable": [
       false
     ]
   },
-  "hash": "00ef039799c162e21a8fa5ba7aec2a9044a3028f1cb34e8c9c565a670956d938"
+  "hash": "bfc67441eb246ac6b155db36919a7a62e19e0dd0c87916e761175ec909add15b"
 }

apps/labrinth/src/routes/internal/attribution.rs

@@ -13,7 +13,7 @@ use crate::database::models::{
     },
 };
 use crate::file_hosting::FileHost;
-use crate::models::ids::{FileId, ProjectId, VersionId};
+use crate::models::ids::{AttributionGroupId, FileId, ProjectId, VersionId};
 use crate::models::pats::Scopes;
 use crate::models::projects::{
     AttributionModerationStatusKind, AttributionResolution,
@@ -31,7 +31,8 @@ use xredis::RedisPool;
 pub fn config(cfg: &mut actix_web::web::ServiceConfig) {
     cfg.service(list)
         .service(update_group)
-        .service(delete_group)
+        .service(delete_groups)
+        .service(delete_all_groups)
         .service(scan)
         .service(force_scan_file)
         .service(assign)
@@ -706,19 +707,25 @@ pub async fn update_group(
     Ok(())
 }
 
-/// Delete an attribution group and all files inside it.
+#[derive(Deserialize, utoipa::ToSchema)]
+struct DeleteGroupsBody {
+    groups: Vec<AttributionGroupId>,
+}
+
+/// Delete attribution groups and all files inside them.
 #[utoipa::path(
 	context_path = "/attribution",
 	tag = "attribution",
+	request_body = DeleteGroupsBody,
 	responses((status = NO_CONTENT))
 )]
-#[delete("/group/{group_id}")]
-pub async fn delete_group(
+#[delete("/group")]
+pub async fn delete_groups(
     req: HttpRequest,
     pool: web::Data<PgPool>,
     redis: web::Data<RedisPool>,
     session_queue: web::Data<AuthQueue>,
-    path: web::Path<i64>,
+    web::Json(body): web::Json<DeleteGroupsBody>,
 ) -> Result<(), ApiError> {
     check_is_moderator_from_headers(
         &req,
@@ -729,11 +736,79 @@ pub async fn delete_group(
     )
     .await?;
 
+    let group_ids = body
+        .groups
+        .into_iter()
+        .map(|id| DBAttributionGroupId::from(id).0)
+        .collect::<Vec<_>>();
+
+    delete_attribution_groups(pool.as_ref(), redis.as_ref(), group_ids).await
+}
+
+#[derive(Deserialize, utoipa::ToSchema)]
+struct DeleteAllGroupsBody {
+    project_id: ProjectId,
+}
+
+/// Delete all attribution groups and files for a project.
+#[utoipa::path(
+	context_path = "/attribution",
+	tag = "attribution",
+	request_body = DeleteAllGroupsBody,
+	responses((status = NO_CONTENT))
+)]
+#[delete("/all-groups")]
+pub async fn delete_all_groups(
+    req: HttpRequest,
+    pool: web::Data<PgPool>,
+    redis: web::Data<RedisPool>,
+    session_queue: web::Data<AuthQueue>,
+    web::Json(body): web::Json<DeleteAllGroupsBody>,
+) -> Result<(), ApiError> {
+    check_is_moderator_from_headers(
+        &req,
+        &**pool,
+        &redis,
+        &session_queue,
+        Scopes::PROJECT_READ,
+    )
+    .await?;
+
+    let project_id = DBProjectId::from(body.project_id).0;
+    let group_ids = sqlx::query_scalar!(
+        r#"
+		SELECT id AS "id: DBAttributionGroupId"
+		FROM project_attribution_groups
+		WHERE project_id = $1
+		"#,
+        project_id,
+    )
+    .fetch_all(pool.as_ref())
+    .await
+    .wrap_internal_err("failed to fetch project attribution groups")?
+    .into_iter()
+    .map(|id| id.0)
+    .collect::<Vec<_>>();
+
+    delete_attribution_groups(pool.as_ref(), redis.as_ref(), group_ids).await
+}
+
+async fn delete_attribution_groups(
+    pool: &PgPool,
+    redis: &RedisPool,
+    mut group_ids: Vec<i64>,
+) -> Result<(), ApiError> {
+    group_ids.sort_unstable();
+    group_ids.dedup();
+
+    if group_ids.is_empty() {
+        return Ok(());
+    }
+
     let mut txn = pool.begin().await.wrap_internal_err(
         "failed to begin attribution group deletion transaction",
     )?;
 
-    let group_id = path.into_inner();
     let version_ids = sqlx::query_scalar!(
         r#"
 		SELECT DISTINCT f.version_id AS "version_id: DBVersionId"
@@ -742,10 +817,10 @@ pub async fn delete_group(
 		INNER JOIN override_file_sources ofs ON ofs.sha1 = paf.sha1
 		INNER JOIN files f ON f.id = ofs.file_id
 		INNER JOIN versions v ON v.id = f.version_id
-		WHERE paf.group_id = $1
+		WHERE paf.group_id = ANY($1)
 			AND pag.project_id = v.mod_id
 		"#,
-        group_id,
+        &group_ids,
     )
     .fetch_all(&mut txn)
     .await
@@ -754,9 +829,9 @@ pub async fn delete_group(
     sqlx::query!(
         "
 		DELETE FROM project_attribution_files
-		WHERE group_id = $1
+		WHERE group_id = ANY($1)
 		",
-        group_id,
+        &group_ids,
     )
     .execute(&mut txn)
     .await
@@ -765,15 +840,15 @@ pub async fn delete_group(
     let result = sqlx::query!(
         "
 		DELETE FROM project_attribution_groups
-		WHERE id = $1
+		WHERE id = ANY($1)
 		",
-        group_id,
+        &group_ids,
     )
     .execute(&mut txn)
     .await
-    .wrap_internal_err("failed to delete attribution group")?;
+    .wrap_internal_err("failed to delete attribution groups")?;
 
-    if result.rows_affected() == 0 {
+    if result.rows_affected() != group_ids.len() as u64 {
         return Err(ApiError::NotFound);
     }
 
@@ -781,7 +856,7 @@ pub async fn delete_group(
         "failed to commit attribution group deletion transaction",
     )?;
 
-    DBVersion::clear_cache_ids(&version_ids, redis.as_ref())
+    DBVersion::clear_cache_ids(&version_ids, redis)
         .await
         .wrap_internal_err("failed to clear version attribution cache")?;
 

apps/labrinth/src/routes/internal/mod.rs

@@ -142,7 +142,8 @@ pub fn config(cfg: &mut web::ServiceConfig) {
 		attribution::scan,
 		attribution::list,
 		attribution::update_group,
-		attribution::delete_group,
+		attribution::delete_groups,
+		attribution::delete_all_groups,
 		attribution::assign,
 		attribution::split,
 		billing::products,