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,317 @@
+use chrono::{DateTime, Utc};
+use clap::{Args, Subcommand, ValueEnum};
+
+use crate::json_filter::JsonOutput;
+
+/// Task-related subcommands.
+#[derive(Debug, Clone, Subcommand)]
+pub enum TaskCommand {
+    /// List ambient agent tasks.
+    List(ListTasksArgs),
+    /// Get status of a specific ambient agent task.
+    Get(TaskGetArgs),
+    /// Retrieve the conversation for a specific run or conversation.
+    #[command(subcommand)]
+    Conversation(ConversationCommand),
+    /// Messages sent to and from runs.
+    #[command(subcommand)]
+    Message(MessageCommand),
+}
+
+/// Conversation-related subcommands.
+#[derive(Debug, Clone, Subcommand)]
+pub enum ConversationCommand {
+    /// Get a conversation by conversation ID.
+    Get(ConversationGetArgs),
+}
+
+/// Message-related subcommands.
+#[derive(Debug, Clone, Subcommand)]
+pub enum MessageCommand {
+    /// Watch for new messages delivered to a run.
+    Watch(MessageWatchArgs),
+    /// Send a message from one run to one or more recipient runs.
+    Send(MessageSendArgs),
+    /// List inbox message headers for a run.
+    List(MessageListArgs),
+    /// Read a full message body.
+    Read(MessageReadArgs),
+    /// Mark a message as delivered.
+    #[command(alias = "delivered")]
+    MarkDelivered(MessageDeliveredArgs),
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct ConversationGetArgs {
+    /// The conversation ID to retrieve.
+    pub conversation_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct MessageSendArgs {
+    /// Recipient run ID. Repeat the flag to send to multiple recipients.
+    #[arg(long = "to", required = true, num_args = 1.., value_delimiter = ',')]
+    pub to: Vec<String>,
+
+    /// Message subject.
+    #[arg(long = "subject")]
+    pub subject: String,
+
+    /// Message body.
+    #[arg(long = "body")]
+    pub body: String,
+
+    /// Sender run ID.
+    #[arg(long = "sender-run-id")]
+    pub sender_run_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct MessageListArgs {
+    /// The run ID whose inbox should be listed.
+    pub run_id: String,
+
+    /// Only return unread messages.
+    #[arg(long = "unread")]
+    pub unread: bool,
+
+    /// Only return messages sent at or after this RFC3339 timestamp.
+    #[arg(long = "since")]
+    pub since: Option<String>,
+
+    /// Maximum number of messages to return (default: 50).
+    #[arg(
+        short = 'L',
+        long = "limit",
+        default_value = "50",
+        value_parser = clap::value_parser!(i32).range(1..)
+    )]
+    pub limit: i32,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct MessageWatchArgs {
+    /// The run ID whose inbox should be watched.
+    pub run_id: String,
+
+    /// Resume after this event sequence (inclusive cursor for reconnects).
+    #[arg(
+        long = "since-sequence",
+        default_value = "0",
+        value_parser = clap::value_parser!(i64).range(0..)
+    )]
+    pub since_sequence: i64,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct MessageReadArgs {
+    /// The message ID to read.
+    pub message_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct MessageDeliveredArgs {
+    /// The message ID to mark as delivered.
+    pub message_id: String,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct ListTasksArgs {
+    /// Maximum number of tasks to return (default: 10).
+    #[arg(short = 'L', long = "limit", default_value = "10")]
+    pub limit: i32,
+
+    /// Filter by run state. Repeat the flag to match any of multiple states.
+    #[arg(long = "state", value_enum, value_name = "STATE")]
+    pub state: Vec<RunStateArg>,
+
+    /// Filter by run source.
+    #[arg(long = "source", value_enum, value_name = "SOURCE")]
+    pub source: Option<RunSourceArg>,
+
+    /// Filter by where the run executed.
+    #[arg(long = "execution-location", value_enum, value_name = "LOC")]
+    pub execution_location: Option<ExecutionLocationArg>,
+
+    /// Filter by creator ID.
+    #[arg(long = "creator", value_name = "UID")]
+    pub creator: Option<String>,
+
+    /// Filter by environment ID.
+    #[arg(long = "environment", value_name = "ENV_ID")]
+    pub environment: Option<String>,
+
+    /// Filter by skill specification (e.g. `owner/repo:path/to/SKILL.md`).
+    #[arg(long = "skill", value_name = "SPEC")]
+    pub skill: Option<String>,
+
+    /// Filter to runs created by a specific scheduled agent.
+    #[arg(long = "schedule", value_name = "SCHEDULE_ID")]
+    pub schedule: Option<String>,
+
+    /// Filter to descendants of a specific run.
+    #[arg(long = "ancestor-run", value_name = "RUN_ID")]
+    pub ancestor_run: Option<String>,
+
+    /// Filter by agent config name.
+    #[arg(long = "name", value_name = "NAME")]
+    pub name: Option<String>,
+
+    /// Filter by model ID.
+    #[arg(long = "model", value_name = "MODEL_ID")]
+    pub model: Option<String>,
+
+    /// Filter by produced artifact type.
+    #[arg(long = "artifact-type", value_enum, value_name = "TYPE")]
+    pub artifact_type: Option<ArtifactTypeArg>,
+
+    /// Only include runs created after the given timestamp.
+    #[arg(long = "created-after", value_name = "RFC3339", value_parser = parse_rfc3339)]
+    pub created_after: Option<DateTime<Utc>>,
+
+    /// Only include runs created before the given timestamp.
+    #[arg(long = "created-before", value_name = "RFC3339", value_parser = parse_rfc3339)]
+    pub created_before: Option<DateTime<Utc>>,
+
+    /// Only include runs updated after the given timestamp.
+    #[arg(long = "updated-after", value_name = "RFC3339", value_parser = parse_rfc3339)]
+    pub updated_after: Option<DateTime<Utc>>,
+
+    /// Fuzzy search across run title, prompt, and skill spec.
+    #[arg(short = 'q', long = "query", value_name = "TEXT")]
+    pub query: Option<String>,
+
+    /// Sort field.
+    #[arg(long = "sort-by", value_enum, value_name = "FIELD")]
+    pub sort_by: Option<RunSortByArg>,
+
+    /// Sort direction.
+    #[arg(long = "sort-order", value_enum, value_name = "DIR")]
+    pub sort_order: Option<RunSortOrderArg>,
+
+    /// Opaque pagination cursor from a previous list response.
+    ///
+    /// When using `--cursor`, `--sort-by` and `--sort-order` must match the
+    /// values used to obtain the cursor.
+    #[arg(long = "cursor", value_name = "CURSOR")]
+    pub cursor: Option<String>,
+
+    /// JSON formatting configuration.
+    #[command(flatten)]
+    pub json_output: JsonOutput,
+}
+
+/// Parse an RFC 3339 timestamp into a UTC `DateTime`.
+fn parse_rfc3339(s: &str) -> Result<DateTime<Utc>, String> {
+    DateTime::parse_from_rfc3339(s)
+        .map(|dt| dt.with_timezone(&Utc))
+        .map_err(|e| format!("invalid RFC 3339 timestamp '{s}': {e}"))
+}
+
+/// Run state values accepted by `--state`. Repeatable; multiple values match any of them.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum RunStateArg {
+    #[value(name = "queued")]
+    Queued,
+    #[value(name = "pending")]
+    Pending,
+    #[value(name = "claimed")]
+    Claimed,
+    #[value(name = "in-progress")]
+    InProgress,
+    #[value(name = "succeeded")]
+    Succeeded,
+    #[value(name = "failed")]
+    Failed,
+    #[value(name = "error")]
+    Error,
+    #[value(name = "blocked")]
+    Blocked,
+    #[value(name = "cancelled")]
+    Cancelled,
+}
+
+/// Run source values accepted by `--source`.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum RunSourceArg {
+    #[value(name = "api")]
+    Api,
+    #[value(name = "cli")]
+    Cli,
+    #[value(name = "slack")]
+    Slack,
+    #[value(name = "linear")]
+    Linear,
+    #[value(name = "scheduled-agent")]
+    ScheduledAgent,
+    #[value(name = "web-app")]
+    WebApp,
+    #[value(name = "cloud-mode")]
+    CloudMode,
+    #[value(name = "github-action")]
+    GitHubAction,
+    #[value(name = "interactive")]
+    Interactive,
+}
+
+/// Execution-location values accepted by `--execution-location`.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum ExecutionLocationArg {
+    #[value(name = "local")]
+    Local,
+    #[value(name = "remote")]
+    Remote,
+}
+
+/// Artifact-type values accepted by `--artifact-type`.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum ArtifactTypeArg {
+    #[value(name = "plan")]
+    Plan,
+    #[value(name = "pull-request")]
+    PullRequest,
+    #[value(name = "screenshot")]
+    Screenshot,
+    #[value(name = "file")]
+    File,
+}
+
+/// Sort-by values accepted by `--sort-by`.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum RunSortByArg {
+    #[value(name = "updated-at")]
+    UpdatedAt,
+    #[value(name = "created-at")]
+    CreatedAt,
+    #[value(name = "title")]
+    Title,
+    #[value(name = "agent")]
+    Agent,
+}
+
+/// Sort-order values accepted by `--sort-order`.
+#[derive(Debug, Clone, Copy, PartialEq, Eq, ValueEnum)]
+pub enum RunSortOrderArg {
+    #[value(name = "asc")]
+    Asc,
+    #[value(name = "desc")]
+    Desc,
+}
+
+#[derive(Debug, Clone, Args)]
+pub struct TaskGetArgs {
+    /// The task ID to get status for.
+    pub task_id: String,
+
+    /// Retrieve the conversation for this run instead of the run status.
+    #[arg(long = "conversation")]
+    pub conversation: bool,
+
+    /// JSON formatting configuration.
+    #[command(flatten)]
+    pub json_output: JsonOutput,
+}
+
+#[cfg(test)]
+#[path = "task_tests.rs"]
+mod tests;