Rebrand to Galaxy, major improvements to Bedrock support, still needs some TLC though

2026-05-07 11:29:34 -05:00
parent f4e2475c60
commit a41cbd8cc7
2433 changed files with 14208 additions and 9409 deletions
@@ -0,0 +1,217 @@
+use clap::{Args, Subcommand};
+
+use crate::{
+    config_file::ConfigFileArgs,
+    environment::{EnvironmentCreateArgs, EnvironmentUpdateArgs},
+    mcp::MCPSpec,
+    model::ModelArgs,
+    scope::ObjectScope,
+    skill::SkillSpec,
+};
+
+/// `ScheduleCommand` has a slightly unusual definition because we allow `oz schedule` as
+// a shorthand for `oz schedule create`.
+#[derive(Debug, Clone, Args)]
+#[clap(args_conflicts_with_subcommands = true)]
+pub struct ScheduleCommand {
+    #[clap(subcommand)]
+    subcommand: Option<ScheduleSubcommand>,
+
+    #[clap(flatten)]
+    create: Option<CreateScheduleArgs>,
+}
+
+impl ScheduleCommand {
+    /// Get the specific scheduling subcommand. Returns `None` if using the `oz schedule` creation shorthand.
+    pub fn subcommand(&self) -> Option<&ScheduleSubcommand> {
+        self.subcommand.as_ref()
+    }
+
+    /// Convert into the specific scheduling subcommand to run.
+    pub fn into_subcommand(self) -> ScheduleSubcommand {
+        if let Some(create) = self.create {
+            ScheduleSubcommand::Create(create)
+        } else if let Some(cmd) = self.subcommand {
+            cmd
+        } else {
+            panic!("Either subcommand or create args are required");
+        }
+    }
+}
+
+/// Schedule-related subcommands.
+#[derive(Debug, Clone, Subcommand)]
+pub enum ScheduleSubcommand {
+    /// Create a scheduled Oz agent.
+    Create(CreateScheduleArgs),
+    /// List scheduled Oz agents.
+    List,
+    /// Get a scheduled Oz agent's configuration.
+    Get(GetScheduleArgs),
+    /// Update a scheduled Oz agent.
+    Update(UpdateScheduleArgs),
+    /// Pause a scheduled Oz agent.
+    ///
+    /// A paused agent still exists, but will not run according to its schedule.
+    Pause(PauseScheduleArgs),
+    /// Unpause a scheduled Oz agent.
+    ///
+    /// The agent will resume executing on its previously-configured schedule.
+    #[command(alias = "resume")]
+    Unpause(UnpauseScheduleArgs),
+    /// Delete a scheduled Oz agent.
+    Delete(DeleteScheduleArgs),
+}
+
+#[derive(Debug, Clone, Args)]
+#[command(
+    group(
+        clap::ArgGroup::new("prompt_group")
+            .required(true)
+            .multiple(true)
+            .args(["prompt", "skill"])
+    )
+)]
+pub struct CreateScheduleArgs {
+    /// Name of the scheduled agent.
+    #[arg(long = "name")]
+    pub name: String,
+
+    /// Cron schedule expression (e.g., "0 9 * * 1" for 9 AM every Monday).
+    #[arg(long = "cron")]
+    pub cron: String,
+
+    #[command(flatten)]
+    pub model: ModelArgs,
+
+    #[command(flatten)]
+    pub environment: EnvironmentCreateArgs,
+
+    #[command(flatten)]
+    pub config_file: ConfigFileArgs,
+
+    #[command(flatten)]
+    pub scope: ObjectScope,
+
+    /// MCP servers to configure for this schedule.
+    ///
+    /// Can be specified as:
+    /// - A path to a JSON file containing MCP configuration
+    /// - Inline JSON with MCP server configuration
+    ///
+    /// Can be specified multiple times to include multiple servers.
+    #[arg(long = "mcp", value_name = "SPEC")]
+    pub mcp_specs: Vec<MCPSpec>,
+
+    /// Prompt for what the scheduled agent should do.
+    #[arg(long = "prompt", short = 'p')]
+    pub prompt: Option<String>,
+
+    /// Automate a skill to run on a schedule.
+    ///
+    /// Format: `repo:skill_name` or `org/repo:skill_name`
+    ///
+    /// Skills are searched in `.agents/skills/`, `.warp-core/skills/`, `.claude/skills/`, and `.codex/skills/` directories.
+    /// The skill is resolved at runtime in the agent's cloud environment.
+    ///
+    /// When used with --prompt, the skill provides the base context and the prompt is the user task.
+    /// This is useful for running recurring workflows like code reviews, dependency updates, or reports.
+    #[arg(long = "skill", value_name = "SPEC")]
+    pub skill: Option<SkillSpec>,
+
+    /// Where this job should be hosted.
+    ///
+    /// Setting "warp" (or omitting this flag) runs it on Warp's infrastructure.
+    /// Any other value is treated as a self-hosted job and the value will be matched
+    /// with the self-hosted worker's name.
+    #[arg(long = "host", value_name = "WORKER_ID")]
+    pub worker_host: Option<String>,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct PauseScheduleArgs {
+    /// ID of the schedule to pause.
+    pub schedule_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct UnpauseScheduleArgs {
+    /// ID of the schedule to unpause.
+    pub schedule_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct UpdateScheduleArgs {
+    /// ID of the schedule to update.
+    pub schedule_id: String,
+
+    /// Update the scheduled agent name.
+    #[arg(long = "name")]
+    pub name: Option<String>,
+
+    /// Update the cron schedule on which the agent is executed.
+    #[arg(long = "cron")]
+    pub cron: Option<String>,
+
+    #[command(flatten)]
+    pub model: ModelArgs,
+
+    #[command(flatten)]
+    pub environment: EnvironmentUpdateArgs,
+
+    #[command(flatten)]
+    pub config_file: ConfigFileArgs,
+
+    /// MCP servers to configure for this schedule.
+    ///
+    /// Can be specified as:
+    /// - A path to a JSON file containing MCP configuration
+    /// - Inline JSON with MCP server configuration
+    ///
+    /// Can be specified multiple times to include multiple servers.
+    #[arg(long = "mcp", value_name = "SPEC")]
+    pub mcp_specs: Vec<MCPSpec>,
+
+    /// Remove MCP servers from this schedule by server name.
+    ///
+    /// This removes the server entry whose key matches `SERVER_NAME`.
+    #[arg(long = "remove-mcp", value_name = "SERVER_NAME")]
+    pub remove_mcp: Vec<String>,
+
+    /// Update the scheduled agent's prompt.
+    #[arg(long = "prompt", short = 'p')]
+    pub prompt: Option<String>,
+
+    /// Update the skill used as the base prompt for the scheduled agent.
+    ///
+    /// Format: `skill_name`, `repo:skill_name`, or `org/repo:skill_name`
+    ///
+    /// Skills are searched in `.agents/skills/`, `.warp-core/skills/`, `.claude/skills/`, and `.codex/skills/` directories.
+    /// The skill is resolved at runtime in the agent's cloud environment.
+    #[arg(long = "skill", value_name = "SPEC", conflicts_with = "remove_skill")]
+    pub skill: Option<SkillSpec>,
+
+    /// Remove the skill from this scheduled agent.
+    #[arg(long = "remove-skill", conflicts_with = "skill")]
+    pub remove_skill: bool,
+
+    /// Where this job should be hosted.
+    ///
+    /// Setting "warp" runs it on Warp's infrastructure.
+    /// Any other value is treated as a self-hosted job and the value will be matched
+    /// with the self-hosted worker's name.
+    #[arg(long = "host", value_name = "WORKER_ID")]
+    pub worker_host: Option<String>,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct DeleteScheduleArgs {
+    /// ID of the schedule to delete.
+    pub schedule_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct GetScheduleArgs {
+    /// ID of the schedule to get.
+    pub schedule_id: String,
+}