feat: admin command to update scaling config (#2019)

Author: oddgrd

admin/src/args.rs

@@ -1,5 +1,8 @@
 use clap::{Parser, Subcommand};
-use shuttle_common::{constants::SHUTTLE_API_URL, models::user::AccountTier};
+use shuttle_common::{
+    constants::SHUTTLE_API_URL,
+    models::{project::ComputeTier, user::AccountTier},
+};
 
 #[derive(Parser, Debug)]
 pub struct Args {
@@ -46,14 +49,17 @@ pub enum Command {
         project_id: String,
     },
 
-    /// Update replica count for a project with a dedicated load balancer.
-    UpdateLbProjectReplicas {
+    /// Update compute tier for a given project.
+    UpdateProjectScale {
         /// Project to update
         #[arg(long, visible_alias = "id")]
         project_id: String,
-        /// Replicas to set for the given project
+        /// Compute tier to set for the given project
+        #[arg(long, visible_alias = "tier")]
+        compute_tier: Option<ComputeTier>,
+        /// Replica count to set for the given project
         #[arg(long)]
-        replicas: u8,
+        replicas: Option<u8>,
     },
 
     /// Renew all old custom domain certificates

admin/src/client.rs

@@ -58,15 +58,15 @@ impl Client {
             .await
     }
 
-    pub async fn update_lb_project_replicas(
+    pub async fn update_project_scale(
         &self,
         project_id: &str,
-        replicas: u8,
+        update_config: &Value,
     ) -> Result<Value> {
         self.inner
             .put_json(
-                format!("/admin/projects/{project_id}/{replicas}"),
-                Option::<()>::None,
+                format!("/admin/projects/{project_id}/scale"),
+                Some(update_config),
             )
             .await
     }

admin/src/lib.rs

@@ -52,12 +52,15 @@ pub async fn run(args: Args) {
             let res = client.upgrade_project_to_lb(&project_id).await.unwrap();
             println!("{res:#?}");
         }
-        Command::UpdateLbProjectReplicas {
+        Command::UpdateProjectScale {
             project_id,
+            compute_tier,
             replicas,
         } => {
+            let update_config =
+                serde_json::json!({"compute_tier": compute_tier, "replicas": replicas});
             let res = client
-                .update_lb_project_replicas(&project_id, replicas)
+                .update_project_scale(&project_id, &update_config)
                 .await
                 .unwrap();
             println!("{res:#?}");