Multi-tenancy: recalculateQuota and resetTenantQuotas (MT-21)

crates/services/Cargo.toml

@@ -16,6 +16,7 @@ types = { path = "../types" }
 jmap_proto = { path = "../jmap-proto" }
 jmap-tools = { version = "0.1" }
 registry = { path =  "../registry" }
+inbuxa-features = { path = "../features" }
 smtp-proto = { version = "0.2", features = ["rkyv", "serde"] }
 tokio = { version = "1.53", features = ["rt"] }
 mail-parser = { version = "0.11", features = ["full_encoding", "rkyv"] }

crates/services/src/task_manager/maintenance.rs

@@ -227,7 +227,6 @@ async fn store_maintenance(
                 .await
                 .caused_by(trc::location!())?;
 
-
             trc::event!(
                 Store(StoreEvent::DataStorePurged),
                 Elapsed = started.elapsed()
@@ -328,7 +327,12 @@ async fn store_maintenance(
                     .await?;
             }
         }
+        // inbuxa: MT-21: recompute every tenant's usage
         TaskStoreMaintenanceType::ResetTenantQuotas => {
+            for tenant_id in inbuxa_features::tenancy::quota::all_tenants(server.registry()).await?
+            {
+                recalculate_tenant_quota(server, tenant_id).await?;
+            }
         }
     }
 
@@ -431,10 +435,15 @@ async fn recalculate_quota(server: &Server, account_id: u32) -> trc::Result<()>
         .map(|_| ())
 }
 
-
-#[cfg(not(feature = "enterprise"))]
-async fn recalculate_tenant_quota(_server: &Server, _tenant_id: u32) -> trc::Result<()> {
-    Ok(())
+// inbuxa: MT-21: recompute a tenant's usage from its members'
+async fn recalculate_tenant_quota(server: &Server, tenant_id: u32) -> trc::Result<()> {
+    inbuxa_features::tenancy::quota::recalculate(
+        &server.core.storage.data,
+        server.registry(),
+        tenant_id,
+    )
+    .await
+    .map(|_| ())
 }
 
 async fn reset_imap_uids(server: &Server, account_id: u32) -> trc::Result<(u32, u32)> {