Add CLI helpers for managing schedules

crates/tower-cmd/src/api.rs

@@ -8,6 +8,7 @@ use tokio::sync::mpsc;
 use reqwest_eventsource::{Event, EventSource};
 use tower_api::apis::configuration;
 use futures_util::StreamExt;
+use tower_api::models::RunParameter;
 
 /// Helper trait to extract the successful response data from API responses
 pub trait ResponseEntity {
@@ -571,3 +572,132 @@ impl ResponseEntity for tower_api::apis::default_api::DescribeRunSuccess {
         }
     }
 }
+
+pub async fn list_schedules(config: &Config, _app_name: Option<&str>, _environment: Option<&str>) -> Result<tower_api::models::ListSchedulesResponse, Error<tower_api::apis::default_api::ListSchedulesError>> {
+    let api_config = &config.into();
+
+    let params = tower_api::apis::default_api::ListSchedulesParams {
+        page: None,
+        page_size: None,
+    };
+
+    unwrap_api_response(tower_api::apis::default_api::list_schedules(api_config, params)).await
+}
+
+pub async fn create_schedule(
+    config: &Config,
+    app_name: &str,
+    environment: &str,
+    cron: &str,
+    parameters: Option<HashMap<String, String>>,
+) -> Result<tower_api::models::CreateScheduleResponse, Error<tower_api::apis::default_api::CreateScheduleError>> {
+    let api_config = &config.into();
+
+    let run_parameters = parameters.map(|params| {
+        params
+            .into_iter()
+            .map(|(key, value)| RunParameter { name: key, value })
+            .collect()
+    });
+
+    let params = tower_api::apis::default_api::CreateScheduleParams {
+        create_schedule_params: tower_api::models::CreateScheduleParams {
+            schema: None,
+            app_name: app_name.to_string(),
+            cron: cron.to_string(),
+            environment: Some(environment.to_string()),
+            app_version: None,
+            parameters: run_parameters,
+            status: None,
+        },
+    };
+
+    unwrap_api_response(tower_api::apis::default_api::create_schedule(api_config, params)).await
+}
+
+pub async fn update_schedule(
+    config: &Config,
+    schedule_id: &str,
+    cron: Option<&String>,
+    parameters: Option<HashMap<String, String>>,
+) -> Result<tower_api::models::UpdateScheduleResponse, Error<tower_api::apis::default_api::UpdateScheduleError>> {
+    let api_config = &config.into();
+
+    let run_parameters = parameters.map(|params| {
+        params
+            .into_iter()
+            .map(|(key, value)| RunParameter { name: key, value })
+            .collect()
+    });
+
+    let params = tower_api::apis::default_api::UpdateScheduleParams {
+        id: schedule_id.to_string(),
+        update_schedule_params: tower_api::models::UpdateScheduleParams {
+            schema: None,
+            cron: cron.map(|s| s.clone()),
+            environment: None,
+            app_version: None,
+            parameters: run_parameters,
+            status: None,
+        },
+    };
+
+    unwrap_api_response(tower_api::apis::default_api::update_schedule(api_config, params)).await
+}
+
+pub async fn delete_schedule(config: &Config, schedule_id: &str) -> Result<tower_api::models::DeleteScheduleResponse, Error<tower_api::apis::default_api::DeleteScheduleError>> {
+    let api_config = &config.into();
+
+    let params = tower_api::apis::default_api::DeleteScheduleParams {
+        delete_schedule_params: tower_api::models::DeleteScheduleParams {
+            schema: None,
+            ids: vec![schedule_id.to_string()],
+        },
+    };
+
+    unwrap_api_response(tower_api::apis::default_api::delete_schedule(api_config, params)).await
+}
+
+impl ResponseEntity for tower_api::apis::default_api::ListSchedulesSuccess {
+    type Data = tower_api::models::ListSchedulesResponse;
+
+    fn extract_data(self) -> Option<Self::Data> {
+        match self {
+            Self::Status200(data) => Some(data),
+            Self::UnknownValue(_) => None,
+        }
+    }
+}
+
+impl ResponseEntity for tower_api::apis::default_api::CreateScheduleSuccess {
+    type Data = tower_api::models::CreateScheduleResponse;
+
+    fn extract_data(self) -> Option<Self::Data> {
+        match self {
+            Self::Status201(data) => Some(data),
+            Self::UnknownValue(_) => None,
+        }
+    }
+}
+
+impl ResponseEntity for tower_api::apis::default_api::UpdateScheduleSuccess {
+    type Data = tower_api::models::UpdateScheduleResponse;
+
+    fn extract_data(self) -> Option<Self::Data> {
+        match self {
+            Self::Status200(data) => Some(data),
+            Self::UnknownValue(_) => None,
+        }
+    }
+}
+
+impl ResponseEntity for tower_api::apis::default_api::DeleteScheduleSuccess {
+    type Data = tower_api::models::DeleteScheduleResponse;
+
+    fn extract_data(self) -> Option<Self::Data> {
+        match self {
+            Self::Status200(data) => Some(data),
+            Self::UnknownValue(_) => None,
+        }
+    }
+}

crates/tower-cmd/src/lib.rs

@@ -7,6 +7,7 @@ pub mod output;
 pub mod api;
 pub mod error;
 mod run;
+mod schedules;
 mod secrets;
 mod session;
 mod teams;
@@ -118,6 +119,20 @@ impl App {
                     }
                 }
             }
+            Some(("schedules", sub_matches)) => {
+                let schedules_command = sub_matches.subcommand();
+
+                match schedules_command {
+                    Some(("list", args)) => schedules::do_list(sessionized_config, args).await,
+                    Some(("create", args)) => schedules::do_create(sessionized_config, args).await,
+                    Some(("update", args)) => schedules::do_update(sessionized_config, args).await,
+                    Some(("delete", args)) => schedules::do_delete(sessionized_config, args).await,
+                    _ => {
+                        schedules::schedules_cmd().print_help().unwrap();
+                        std::process::exit(2);
+                    }
+                }
+            }
             Some(("deploy", args)) => deploy::do_deploy(sessionized_config, args).await,
             Some(("run", args)) => run::do_run(sessionized_config, args, args.subcommand()).await,
             Some(("teams", sub_matches)) => {
@@ -162,6 +177,7 @@ fn root_cmd() -> Command {
         .arg_required_else_help(false)
         .subcommand(session::login_cmd())
         .subcommand(apps::apps_cmd())
+        .subcommand(schedules::schedules_cmd())
         .subcommand(secrets::secrets_cmd())
         .subcommand(deploy::deploy_cmd())
         .subcommand(run::run_cmd())