feat: admin commands to upgrade proj to lb, replica count (#2015)

Author: oddgrd

admin/src/args.rs

@@ -33,6 +33,29 @@ pub enum Command {
         json: String,
     },
 
+    GetProjectConfig {
+        /// Project to get config for
+        #[arg(long, visible_alias = "id")]
+        project_id: String,
+    },
+
+    /// Upgrade project to use a dedicated load balancer.
+    UpgradeProjectToLb {
+        /// Project to upgrade to ALB
+        #[arg(long, visible_alias = "id")]
+        project_id: String,
+    },
+
+    /// Update replica count for a project with a dedicated load balancer.
+    UpdateLbProjectReplicas {
+        /// Project to update
+        #[arg(long, visible_alias = "id")]
+        project_id: String,
+        /// Replicas to set for the given project
+        #[arg(long)]
+        replicas: u8,
+    },
+
     /// Renew all old custom domain certificates
     RenewCerts,
 

admin/src/client.rs

@@ -46,6 +46,31 @@ impl Client {
             .await
     }
 
+    pub async fn get_project_config(&self, project_id: &str) -> Result<Value> {
+        self.inner
+            .get_json(format!("/admin/projects/{project_id}"))
+            .await
+    }
+
+    pub async fn upgrade_project_to_lb(&self, project_id: &str) -> Result<Value> {
+        self.inner
+            .put_json(format!("/admin/projects/{project_id}"), Option::<()>::None)
+            .await
+    }
+
+    pub async fn update_lb_project_replicas(
+        &self,
+        project_id: &str,
+        replicas: u8,
+    ) -> Result<Value> {
+        self.inner
+            .put_json(
+                format!("/admin/projects/{project_id}/{replicas}"),
+                Option::<()>::None,
+            )
+            .await
+    }
+
     pub async fn update_project_owner(
         &self,
         project_id: &str,

admin/src/lib.rs

@@ -45,6 +45,24 @@ pub async fn run(args: Args) {
                 .unwrap();
             println!("{res:?}");
         }
+        Command::UpgradeProjectToLb { project_id } => {
+            let res = client.upgrade_project_to_lb(&project_id).await.unwrap();
+            println!("{res:#?}");
+        }
+        Command::UpdateLbProjectReplicas {
+            project_id,
+            replicas,
+        } => {
+            let res = client
+                .update_lb_project_replicas(&project_id, replicas)
+                .await
+                .unwrap();
+            println!("{res:#?}");
+        }
+        Command::GetProjectConfig { project_id } => {
+            let res = client.get_project_config(&project_id).await.unwrap();
+            println!("{res:#?}");
+        }
         Command::AddFeatureFlag { entity, flag } => {
             client.feature_flag(&entity, &flag, true).await.unwrap();
             println!("Added flag {flag} for {entity}");