reanalyze removal

backend/src/bin/admin.rs

@@ -15,9 +15,12 @@ use uuid::Uuid;
 use backend::{
     config::AppConfig,
     db::{self, PgPool},
+    jobs::{enqueue_job, JOB_ANALYZE_DOCUMENT},
     models::{DocumentAsset, DocumentAssetObject, NewUser, NewUserMembership, Tenant, User},
     s3,
-    schema::{document_asset_objects, document_assets, tenants, user_memberships, users},
+    schema::{
+        document_asset_objects, document_assets, documents, tenants, user_memberships, users,
+    },
     storage::{ObjectStorage, S3Storage, TenantStorage},
     utils::tracing::init_tracing,
 };
@@ -83,6 +86,9 @@ enum Command {
         username: String,
         slug: String,
     },
+    ReanalyzeDocuments {
+        slug: String,
+    },
     ListTenants,
     DeleteAssets(String),
     QuickwitCreate(String),
@@ -100,6 +106,7 @@ impl Command {
             delete-tenant <slug>\n\
             add-user-to-tenant <username> <slug> [role]\n\
             remove-user-from-tenant <username> <slug>\n\
+            reanalyze-documents <slug>\n\
             list-tenants\n\
             delete-assets <slug>\n\
             quickwit-create-index <slug>\n\
@@ -138,6 +145,9 @@ impl Command {
                 username: args.next().ok_or_else(|| anyhow!("username required"))?,
                 slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
             }),
+            Some("reanalyze-documents") => Ok(Self::ReanalyzeDocuments {
+                slug: args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
+            }),
             Some("list-tenants") => Ok(Self::ListTenants),
             Some("delete-assets") => Ok(Self::DeleteAssets(
                 args.next().ok_or_else(|| anyhow!("tenant slug required"))?,
@@ -179,6 +189,7 @@ async fn main() -> Result<()> {
         Command::RemoveUserFromTenant { username, slug } => {
             remove_user_from_tenant(&pool, &username, &slug)?
         }
+        Command::ReanalyzeDocuments { slug } => reanalyze_documents(&pool, &slug)?,
         Command::ListTenants => list_tenants(&pool)?,
         Command::DeleteAssets(slug) => delete_assets_for_tenant(&config, &pool, &slug).await?,
         Command::QuickwitCreate(slug) => {
@@ -436,6 +447,50 @@ fn remove_user_from_tenant(pool: &PgPool, username: &str, slug: &str) -> Result<
     Ok(())
 }
 
+fn reanalyze_documents(pool: &PgPool, slug: &str) -> Result<()> {
+    let mut conn = pool.get().context("failed to get database connection")?;
+
+    let tenant: Tenant = tenants::table
+        .filter(tenants::slug.eq(slug))
+        .first(&mut conn)
+        .optional()?
+        .ok_or_else(|| anyhow!("tenant '{}' not found", slug))?;
+
+    let targets: Vec<(Uuid, Uuid)> = documents::table
+        .filter(documents::tenant_id.eq(tenant.id))
+        .filter(documents::deleted_at.is_null())
+        .select((documents::id, documents::current_version_id))
+        .load(&mut conn)?;
+
+    if targets.is_empty() {
+        println!("tenant '{}' has no active documents", slug);
+        return Ok(());
+    }
+
+    let mut queued = 0usize;
+    for (document_id, version_id) in targets {
+        enqueue_job(
+            &mut conn,
+            tenant.id,
+            JOB_ANALYZE_DOCUMENT,
+            serde_json::json!({
+                "document_id": document_id,
+                "document_version_id": version_id,
+                "force": true,
+            }),
+            None,
+        )
+        .map_err(|err| anyhow!("failed to enqueue analyze job: {}", err))?;
+        queued += 1;
+    }
+
+    println!(
+        "queued {} documents for re-analysis in tenant '{}'",
+        queued, slug
+    );
+    Ok(())
+}
+
 fn list_tenants(pool: &PgPool) -> Result<()> {
     let mut conn = pool.get().context("failed to get database connection")?;
     let tenants: Vec<Tenant> = tenants::table