feat(admin): log retention policies (#2053)

Author: jonaro00

admin/src/args.rs

@@ -119,4 +119,5 @@ pub enum Command {
     },
 
     DowngradeProTrials,
+    FixRetentionPolicies,
 }

admin/src/client.rs

@@ -166,7 +166,6 @@ impl Client {
     pub async fn get_expired_protrials(&self) -> Result<Vec<String>> {
         self.inner.get_json("/admin/users/protrial-downgrade").await
     }
-
     pub async fn downgrade_protrial(&self, user_id: &str) -> Result<String> {
         self.inner
             .put_json(
@@ -175,4 +174,16 @@ impl Client {
             )
             .await
     }
+
+    pub async fn get_projects_for_retention_policy(&self) -> Result<Vec<String>> {
+        self.inner.get_json("/admin/log-retention").await
+    }
+    pub async fn fix_retention_policy(&self, project_id: &str) -> Result<String> {
+        self.inner
+            .put_json(
+                format!("/admin/log-retention/{project_id}"),
+                Option::<()>::None,
+            )
+            .await
+    }
 }

admin/src/lib.rs

@@ -164,6 +164,16 @@ pub async fn run(args: Args) {
                 tokio::time::sleep(tokio::time::Duration::from_millis(1000)).await;
             }
         }
+        Command::FixRetentionPolicies => {
+            let projects = client.get_projects_for_retention_policy().await.unwrap();
+            eprintln!("Starting fix of {} log retention policies", projects.len());
+            for pid in projects {
+                println!("{pid}");
+                println!("  {:?}", client.fix_retention_policy(&pid).await);
+                // prevent api rate limiting
+                tokio::time::sleep(tokio::time::Duration::from_millis(1000)).await;
+            }
+        }
     };
 }