feat: add ability to get status counts for projects from users and organizations

- Also add ability to get tech review verdict counts for projects from users and organizations
This commit is contained in:
Blodhgarm
2026-08-23 18:31:13 -05:00
parent fe9ae94a5d
commit fba6ee994d
4 changed files with 688 additions and 21 deletions
@@ -206,6 +206,31 @@ impl DBOrganization {
}
}
pub async fn get_projects<'a, E>(
organization_id: DBOrganizationId,
exec: E,
) -> Result<Vec<DBProjectId>, super::DatabaseError>
where
E: database::Executor<'a, Database = sqlx::Postgres>,
{
use futures::TryStreamExt;
let db_projects = sqlx::query!(
"
SELECT m.id FROM organizations o
INNER JOIN mods m ON m.organization_id = o.id
WHERE o.id = $1
",
organization_id as DBOrganizationId,
)
.fetch(exec)
.map_ok(|m| DBProjectId(m.id))
.try_collect::<Vec<_>>()
.await?;
Ok(db_projects)
}
pub async fn remove(
id: DBOrganizationId,
transaction: &mut PgTransaction<'_>,
+1 -1
View File
@@ -444,7 +444,7 @@ impl From<LinkUrl> for Link {
/// Scheduled - Project is scheduled to be released in the future
/// Private - Project is approved, but is not viewable to the public
#[derive(
Serialize, Deserialize, Copy, Clone, Eq, PartialEq, Debug, utoipa::ToSchema,
Serialize, Deserialize, Copy, Clone, Eq, PartialEq, Hash, Debug, utoipa::ToSchema,
)]
#[serde(rename_all = "lowercase")]
pub enum ProjectStatus {
@@ -2,7 +2,7 @@ use super::ApiError;
use crate::auth::get_user_from_headers;
use crate::database;
use crate::database::PgPool;
use crate::database::models::DBModerationLock;
use crate::database::models::{DBModerationLock, DBOrganization, DBOrganizationId, DBProjectId, DBProject};
use crate::database::models::moderation_external_item;
use crate::models::ids::{OrganizationId, ProjectId};
use crate::models::projects::{ProjectStatus, VersionStatus};
@@ -20,7 +20,10 @@ use chrono::{DateTime, Utc};
use eyre::eyre;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use futures_util::future::try_join_all;
use xredis::RedisPool;
use crate::routes::v3::organizations::OrganizationIds;
use crate::routes::v3::users::UserIds;
pub mod external_license;
mod ownership;
@@ -37,6 +40,10 @@ pub fn config(cfg: &mut actix_web::web::ServiceConfig) {
.service(release_lock)
.service(release_lock_beacon)
.service(delete_all_locks)
.service(get_user_project_status_counts)
.service(get_users_project_status_counts)
.service(get_organization_project_status_counts)
.service(get_organizations_project_status_counts)
.service(web::scope("/tech-review").configure(tech_review::config))
.service(
web::scope("/external-license").configure(external_license::config),
@@ -215,7 +222,7 @@ pub struct DeleteAllLocksResponse {
pub deleted_count: u64,
}
/// List projects in the moderation queue.
/// List projects in the moderation queue.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1063,7 +1070,7 @@ fn row_to_ownership(
})
}
/// Get project moderation metadata.
/// Get project moderation metadata.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1226,7 +1233,7 @@ pub enum Judgement {
},
}
/// Update project moderation judgements.
/// Update project moderation judgements.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1328,7 +1335,7 @@ pub async fn set_project_meta(
Ok(())
}
/// Acquire a moderation lock.
/// Acquire a moderation lock.
/// Returns success if acquired, or info about who holds the lock if blocked.
#[utoipa::path(
context_path = "/moderation",
@@ -1393,7 +1400,7 @@ pub async fn acquire_lock(
}
}
/// Override a moderation lock.
/// Override a moderation lock.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1444,7 +1451,7 @@ pub async fn override_lock(
}))
}
/// Get moderation lock status.
/// Get moderation lock status.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1511,7 +1518,7 @@ pub async fn get_lock_status(
}
}
/// Release a moderation lock.
/// Release a moderation lock.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1557,7 +1564,7 @@ pub async fn release_lock(
Ok(web::Json(LockReleaseResponse { success: released }))
}
/// Release a moderation lock by beacon.
/// Release a moderation lock by beacon.
///
/// For use with `navigator.sendBeacon`, which cannot set `Authorization` or send `DELETE`.
/// The body must be `text/plain` containing the same token value as the `Authorization` header
@@ -1633,7 +1640,7 @@ pub async fn release_lock_beacon(
Ok(web::Json(LockReleaseResponse { success: released }))
}
/// Delete all moderation locks.
/// Delete all moderation locks.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
@@ -1672,3 +1679,292 @@ pub async fn delete_all_locks(
Ok(web::Json(DeleteAllLocksResponse { deleted_count }))
}
/// Get counts of each `ProjectStatus` across a user's projects. Only
/// statuses with at least one project are present in the map.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
security(("bearer_auth" = [])),
responses((status = OK, body = HashMap<ProjectStatus, u32>))
)]
#[get("/user/{user_id}/project-status-counts")]
pub async fn get_user_project_status_counts(
req: HttpRequest,
info: web::Path<(String,)>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<HashMap<ProjectStatus, u32>>, ApiError> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let target_user =
database::models::DBUser::get(&info.into_inner().0, &**pool, &redis)
.await
.wrap_internal_err("fetching user from database")?
.wrap_not_found_err("resource not found")?;
let counts =
user_project_status_counts(target_user.id, &**pool, &redis).await?;
Ok(web::Json(counts))
}
#[derive(Debug, Deserialize)]
pub struct ProjectStatusCountsUserIds {
pub ids: String,
}
/// Get counts of each `ProjectStatus` across multiple users' projects.
/// Users that don't exist are silently omitted from the response; users
/// that exist but have no projects are included with an empty map.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
security(("bearer_auth" = [])),
params(("ids" = String, Query)),
responses((status = OK, body = HashMap<UserId, HashMap<ProjectStatus, u32>>))
)]
#[get("/users/project-status-counts")]
pub async fn get_users_project_status_counts(
req: HttpRequest,
ids: web::Query<ProjectStatusCountsUserIds>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<HashMap<UserId, HashMap<ProjectStatus, u32>>>, ApiError>
{
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let user_ids = serde_json::from_str::<Vec<String>>(&ids.ids)
.wrap_request_err("deserializing JSON data")?;
if user_ids.is_empty() {
return Ok(web::Json(HashMap::new()));
}
let target_users =
database::models::DBUser::get_many(&user_ids, &**pool, &redis)
.await
.wrap_internal_err("fetching users from database")?;
let pool_ref = &**pool;
let redis_ref = &*redis;
let counts_by_user = try_join_all(target_users.into_iter().map(
|target_user| async move {
let counts = user_project_status_counts(
target_user.id,
pool_ref,
redis_ref,
)
.await?;
Ok::<_, ApiError>((UserId::from(target_user.id), counts))
},
))
.await?
.into_iter()
.collect::<HashMap<_, _>>();
Ok(web::Json(counts_by_user))
}
/// Get counts of each `ProjectStatus` across an organization's projects.
/// Only statuses with at least one project are present in the map.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
security(("bearer_auth" = [])),
responses((status = OK, body = HashMap<ProjectStatus, u32>))
)]
#[get("/organization/{organization_id}/project-status-counts")]
pub async fn get_organization_project_status_counts(
req: HttpRequest,
info: web::Path<(String,)>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<HashMap<ProjectStatus, u32>>, ApiError> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let target_org = database::models::DBOrganization::get(
&info.into_inner().0,
&**pool,
&redis,
)
.await
.wrap_internal_err("fetching organization from database")?
.wrap_not_found_err("resource not found")?;
let counts = organization_project_status_counts(
target_org.id,
&**pool,
&redis,
)
.await?;
Ok(web::Json(counts))
}
/// Get counts of each `ProjectStatus` across multiple organizations'
/// projects. Organizations that don't exist are silently omitted from the
/// response; organizations that exist but have no projects are included
/// with an empty map.
#[utoipa::path(
context_path = "/moderation",
tag = "moderation",
security(("bearer_auth" = [])),
params(("ids" = String, Query)),
responses((status = OK, body = HashMap<OrganizationId, HashMap<ProjectStatus, u32>>))
)]
#[get("/organizations/project-status-counts")]
pub async fn get_organizations_project_status_counts(
req: HttpRequest,
ids: web::Query<OrganizationIds>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<
web::Json<HashMap<OrganizationId, HashMap<ProjectStatus, u32>>>,
ApiError,
> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let organization_ids = serde_json::from_str::<Vec<String>>(&ids.ids)
.wrap_request_err("deserializing JSON data")?;
if organization_ids.is_empty() {
return Ok(web::Json(HashMap::new()));
}
let target_orgs = database::models::DBOrganization::get_many(
&organization_ids,
&**pool,
&redis,
)
.await
.wrap_internal_err("fetching organizations from database")?;
let pool_ref = &**pool;
let redis_ref = &*redis;
let counts_by_org = try_join_all(target_orgs.into_iter().map(
|target_org| async move {
let counts = organization_project_status_counts(
target_org.id,
pool_ref,
redis_ref,
)
.await?;
Ok::<_, ApiError>((OrganizationId::from(target_org.id), counts))
},
))
.await?
.into_iter()
.collect::<HashMap<_, _>>();
Ok(web::Json(counts_by_org))
}
/// Tallies a single user's projects by `ProjectStatus`. Shared by the
/// single- and multi-user project-status-count endpoints.
async fn user_project_status_counts<'a, E>(
user_id: database::models::DBUserId,
pool: E,
redis: &RedisPool,
) -> Result<HashMap<ProjectStatus, u32>, ApiError>
where
E: database::Executor<'a, Database = sqlx::Postgres>
+ database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
let project_ids =
models::DBUser::get_projects(user_id, pool, redis)
.await
.wrap_internal_err("fetching user's projects from database")?;
project_status_counts_for(&project_ids, pool, redis).await
}
// Tallies a single organization's projects by `ProjectStatus`. Shared by
/// the single- and multi-organization project-status-count endpoints.
async fn organization_project_status_counts<'a, E>(
organization_id: DBOrganizationId,
pool: E,
redis: &RedisPool,
) -> Result<HashMap<ProjectStatus, u32>, ApiError>
where
E: database::Executor<'a, Database = sqlx::Postgres>
+ database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
let project_ids = DBOrganization::get_projects(organization_id, pool)
.await
.wrap_internal_err("fetching project IDs from database")?;
project_status_counts_for(&project_ids, pool, redis).await
}
/// Tallies `ProjectStatus` counts across the given projects. Shared by the
/// user- and organization-scoped project-status-count helpers above.
async fn project_status_counts_for<'a, E>(
project_ids: &[DBProjectId],
pool: E,
redis: &RedisPool,
) -> Result<HashMap<ProjectStatus, u32>, ApiError>
where
E: database::Executor<'a, Database = sqlx::Postgres>
+ database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
if project_ids.is_empty() {
return Ok(HashMap::new());
}
let projects =
DBProject::get_many_ids(project_ids, pool, redis)
.await
.wrap_internal_err("fetching projects from database")?;
let mut counts = HashMap::new();
for project in &projects {
*counts.entry(project.inner.status).or_insert(0u32) += 1;
}
Ok(counts)
}
@@ -14,7 +14,7 @@ use crate::{
database::{
DBProject,
models::{
DBFileId, DBProjectId, DBThread, DBThreadId, DBUser, DBVersion,
DBFileId, DBProjectId, DBThread, DBThreadId, DBUser, DBUserId, DBVersion,
DBVersionId, DelphiReportId, DelphiReportIssueDetailsId,
DelphiReportIssueId,
delphi_report_item::{
@@ -26,7 +26,7 @@ use crate::{
},
},
models::{
ids::{FileId, ProjectId, ThreadId, VersionId},
ids::{FileId, ProjectId, ThreadId, ThreadMessageId, VersionId},
pats::Scopes,
projects::{Project, ProjectStatus},
threads::{MessageBody, Thread},
@@ -43,6 +43,11 @@ use crate::{
util::error::Context,
};
use eyre::eyre;
use futures_util::future::try_join_all;
use ariadne::ids::UserId;
use crate::database::models::{DBOrganization, DBOrganizationId};
use crate::models::ids::OrganizationId;
use crate::routes::v3::users::UserIds;
pub mod global;
@@ -55,7 +60,11 @@ pub fn config(cfg: &mut actix_web::web::ServiceConfig) {
.service(submit_report)
.service(update_issue_details)
.service(update_global_issue_details)
.service(add_report);
.service(add_report)
.service(get_user_flagged_projects)
.service(get_users_flagged_projects)
.service(get_organization_flagged_projects)
.service(get_organizations_flagged_projects);
}
/// Arguments for searching project technical reviews.
@@ -204,7 +213,7 @@ pub enum FlagReason {
Delphi,
}
/// Get a Delphi report issue.
/// Get a Delphi report issue.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
@@ -268,7 +277,7 @@ pub async fn get_issue(
Ok(web::Json(row.data.0))
}
/// Get a project technical report.
/// Get a project technical report.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
@@ -684,7 +693,7 @@ async fn fetch_project_reports(
Ok(project_reports)
}
/// Search projects awaiting technical review.
/// Search projects awaiting technical review.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
@@ -896,7 +905,7 @@ pub async fn search_projects(
}))
}
/// Get a project technical review report.
/// Get a project technical review report.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
@@ -992,7 +1001,7 @@ pub struct SubmitReport {
pub message: Option<String>,
}
/// Submit a technical review verdict.
/// Submit a technical review verdict.
///
/// Before this is called, all issues for this project's reports must have been
/// marked as either safe or unsafe. Otherwise, this will error with
@@ -1210,7 +1219,7 @@ pub struct UpdateGlobalIssue {
pub verdict: DelphiStatus,
}
/// Update technical review issue details.
/// Update technical review issue details.
///
/// This will not automatically reject the project for malware, but just flag
/// this issue with a verdict.
@@ -1497,7 +1506,7 @@ pub struct AddReport {
pub file_id: FileId,
}
/// Add a technical review report.
/// Add a technical review report.
/// does not already exist for it.
#[utoipa::path(
context_path = "/moderation/tech-review",
@@ -1566,3 +1575,340 @@ pub async fn add_report(
Ok(web::Json(report_id))
}
/// A user's project that is stuck in `processing` or `rejected` because the
/// most recent technical review verdict posted to its thread was `unsafe`.
#[derive(Debug, Clone, Serialize, utoipa::ToSchema)]
pub struct FlaggedProject {
pub project_id: ProjectId,
pub thread_id: ThreadId,
pub status: ProjectStatus,
/// The `tech_review` message that carried the `unsafe` verdict.
pub message_id: ThreadMessageId,
/// When that verdict was posted.
pub reviewed: DateTime<Utc>,
}
/// Get all of a user's `processing`/`rejected` projects whose most recent
/// `tech_review` thread message was an `unsafe` verdict.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
security(("bearer_auth" = [])),
responses((status = OK, body = Vec<FlaggedProject>))
)]
#[get("/user/{user_id}/flagged-projects")]
pub async fn get_user_flagged_projects(
req: HttpRequest,
info: web::Path<(String,)>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<Vec<FlaggedProject>>, ApiError> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let target_user = DBUser::get(&info.into_inner().0, &**pool, &redis)
.await
.wrap_internal_err("fetching user from database")?
.wrap_not_found_err("resource not found")?;
let flagged =
user_flagged_projects(target_user.id, &**pool, &redis).await?;
Ok(web::Json(flagged))
}
#[derive(Debug, Deserialize)]
pub struct FlaggedProjectsIds {
pub ids: String,
}
/// Get all of multiple users' `processing`/`rejected` projects whose most
/// recent `tech_review` thread message was an `unsafe` verdict. Users that
/// don't exist are silently omitted from the response; users that exist
/// but have no matching projects are included with an empty list.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
security(("bearer_auth" = [])),
params(("ids" = String, Query)),
responses((status = OK, body = HashMap<UserId, Vec<FlaggedProject>>))
)]
#[get("/users/flagged-projects")]
pub async fn get_users_flagged_projects(
req: HttpRequest,
ids: web::Query<FlaggedProjectsIds>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<HashMap<UserId, Vec<FlaggedProject>>>, ApiError> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let user_ids = serde_json::from_str::<Vec<String>>(&ids.ids)
.wrap_request_err("deserializing JSON data")?;
if user_ids.is_empty() {
return Ok(web::Json(HashMap::new()));
}
let target_users = DBUser::get_many(&user_ids, &**pool, &redis)
.await
.wrap_internal_err("fetching users from database")?;
let pool_ref = &**pool;
let redis_ref = &*redis;
let flagged_by_user = try_join_all(target_users.into_iter().map(
|target_user| async move {
let flagged =
user_flagged_projects(target_user.id, pool_ref, redis_ref)
.await?;
Ok::<_, ApiError>((UserId::from(target_user.id), flagged))
},
))
.await?
.into_iter()
.collect::<HashMap<_, _>>();
Ok(web::Json(flagged_by_user))
}
/// Get all of an organization's `processing`/`rejected` projects whose most
/// recent `tech_review` thread message was an `unsafe` verdict.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
security(("bearer_auth" = [])),
responses((status = OK, body = Vec<FlaggedProject>))
)]
#[get("/organization/{organization_id}/flagged-projects")]
pub async fn get_organization_flagged_projects(
req: HttpRequest,
info: web::Path<(String,)>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<Vec<FlaggedProject>>, ApiError> {
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let target_org =
DBOrganization::get(&info.into_inner().0, &**pool, &redis)
.await
.wrap_internal_err("fetching organization from database")?
.wrap_not_found_err("resource not found")?;
let flagged =
organization_flagged_projects(target_org.id, &**pool, &redis).await?;
Ok(web::Json(flagged))
}
/// Get all of multiple organizations' `processing`/`rejected` projects
/// whose most recent `tech_review` thread message was an `unsafe` verdict.
/// Organizations that don't exist are silently omitted from the response;
/// organizations that exist but have no matching projects are included
/// with an empty list.
#[utoipa::path(
context_path = "/moderation/tech-review",
tag = "moderation",
security(("bearer_auth" = [])),
params(("ids" = String, Query)),
responses((status = OK, body = HashMap<OrganizationId, Vec<FlaggedProject>>))
)]
#[get("/organizations/flagged-projects")]
pub async fn get_organizations_flagged_projects(
req: HttpRequest,
ids: web::Query<FlaggedProjectsIds>,
pool: web::Data<PgPool>,
redis: web::Data<RedisPool>,
session_queue: web::Data<AuthQueue>,
) -> Result<web::Json<HashMap<OrganizationId, Vec<FlaggedProject>>>, ApiError>
{
check_is_moderator_from_headers(
&req,
&**pool,
&redis,
&session_queue,
Scopes::PROJECT_READ,
)
.await
.wrap_auth_err("authenticating API request")?;
let organization_ids = serde_json::from_str::<Vec<String>>(&ids.ids)
.wrap_request_err("deserializing JSON data")?;
if organization_ids.is_empty() {
return Ok(web::Json(HashMap::new()));
}
let target_orgs =
DBOrganization::get_many(&organization_ids, &**pool, &redis)
.await
.wrap_internal_err("fetching organizations from database")?;
let pool_ref = &**pool;
let redis_ref = &*redis;
let flagged_by_org = try_join_all(target_orgs.into_iter().map(
|target_org| async move {
let flagged = organization_flagged_projects(
target_org.id,
pool_ref,
redis_ref,
)
.await?;
Ok::<_, ApiError>((OrganizationId::from(target_org.id), flagged))
},
))
.await?
.into_iter()
.collect::<HashMap<_, _>>();
Ok(web::Json(flagged_by_org))
}
/// Finds a single user's `processing`/`rejected` projects whose most recent
/// `tech_review` thread message was an `unsafe` verdict. Shared by the
/// single- and multi-user flagged-projects endpoints.
async fn user_flagged_projects<'a, E>(
user_id: DBUserId,
pool: E,
redis: &RedisPool,
) -> Result<Vec<FlaggedProject>, ApiError>
where
E: crate::database::Executor<'a, Database = sqlx::Postgres>
+ crate::database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
let project_ids = DBUser::get_projects(user_id, pool, redis)
.await
.wrap_internal_err("fetching user's projects from database")?;
flagged_projects_among(&project_ids, pool, redis).await
}
/// Finds a single organization's `processing`/`rejected` projects whose
/// most recent `tech_review` thread message was an `unsafe` verdict.
/// Shared by the single- and multi-organization flagged-projects
/// endpoints.
async fn organization_flagged_projects<'a, E>(
organization_id: DBOrganizationId,
pool: E,
redis: &RedisPool,
) -> Result<Vec<FlaggedProject>, ApiError>
where
E: crate::database::Executor<'a, Database = sqlx::Postgres>
+ crate::database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
let project_ids = DBOrganization::get_projects(organization_id, pool)
.await
.wrap_internal_err("fetching project IDs from database")?;
flagged_projects_among(&project_ids, pool, redis).await
}
/// Finds `processing`/`rejected` projects (from the given ids) whose most
/// recent `tech_review` thread message was an `unsafe` verdict. Shared by
/// the user- and organization-scoped flagged-projects helpers above.
async fn flagged_projects_among<'a, E>(
project_ids: &[DBProjectId],
pool: E,
redis: &RedisPool,
) -> Result<Vec<FlaggedProject>, ApiError>
where
E: crate::database::Executor<'a, Database = sqlx::Postgres>
+ crate::database::Acquire<'a, Database = sqlx::Postgres>
+ Copy,
{
if project_ids.is_empty() {
return Ok(Vec::new());
}
// Only `processing`/`rejected` projects are relevant, so narrow down
// before pulling any thread data.
let candidate_projects = DBProject::get_many_ids(project_ids, pool, redis)
.await
.wrap_internal_err("fetching projects from database")?
.into_iter()
.filter(|project| {
matches!(
project.inner.status,
ProjectStatus::Processing | ProjectStatus::Rejected
)
})
.collect::<Vec<_>>();
if candidate_projects.is_empty() {
return Ok(Vec::new());
}
let thread_ids = candidate_projects
.iter()
.map(|project| project.thread_id)
.collect::<Vec<_>>();
let threads = DBThread::get_many(&thread_ids, pool)
.await
.wrap_internal_err("fetching threads from database")?
.into_iter()
.map(|thread| (thread.id, thread))
.collect::<HashMap<_, _>>();
Ok(candidate_projects
.into_iter()
.filter_map(|project| {
let thread = threads.get(&project.thread_id)?;
// `DBThread::get_many` returns messages sorted oldest-first, so
// walking backwards finds the most recent `tech_review` entry
// regardless of what (if anything) was posted after it.
let last_review = thread.messages.iter().rev().find(|message| {
matches!(message.body, MessageBody::TechReview { .. })
})?;
let verdict = match &last_review.body {
MessageBody::TechReview { verdict } => *verdict,
_ => return None,
};
if verdict != DelphiVerdict::Unsafe {
return None;
}
Some(FlaggedProject {
project_id: project.inner.id.into(),
thread_id: project.thread_id.into(),
status: project.inner.status,
message_id: last_review.id.into(),
reviewed: last_review.created,
})
})
.collect())
}