use crate::cli::command::CommandRequest;
pub use crate::cli::command::agents::logs::read::all::Target;
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.tasks.list.Request")]
pub struct Request {
pub path_type: Path,
pub targets: Vec<Target>,
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
pub oneshot: bool,
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
pub interval: bool,
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
pub pending: bool,
#[serde(default, skip_serializing_if = "std::ops::Not::not")]
pub exhausted: bool,
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub after_id: Option<i64>,
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub count: Option<u64>,
#[serde(flatten)]
pub base: crate::cli::command::RequestBase,
}
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.tasks.list.Path")]
pub enum Path {
#[serde(rename = "tasks/list")]
AgentsTasksList,
}
impl CommandRequest for Request {
fn into_command(&self) -> Vec<String> {
let mut argv = vec![
"tasks".to_string(),
"list".to_string(),
];
for target in &self.targets {
argv.push("--target".to_string());
argv.push(target.into_arg_string());
}
if self.oneshot {
argv.push("--oneshot".to_string());
}
if self.interval {
argv.push("--interval".to_string());
}
if self.pending {
argv.push("--pending".to_string());
}
if self.exhausted {
argv.push("--exhausted".to_string());
}
if let Some(after_id) = self.after_id {
argv.push("--after-id".to_string());
argv.push(after_id.to_string());
}
if let Some(c) = self.count {
argv.push("--count".to_string());
argv.push(c.to_string());
}
self.base.push_flags(&mut argv);
argv
}
fn request_base(&self) -> &crate::cli::command::RequestBase {
&self.base
}
fn request_base_mut(&mut self) -> Option<&mut crate::cli::command::RequestBase> {
Some(&mut self.base)
}
}
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.tasks.list.Plugin")]
pub struct Plugin {
pub owner: String,
pub repository: String,
pub version: String,
}
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.tasks.list.ResponseItem")]
pub struct ResponseItem {
pub id: i64,
pub name: String,
pub agent_instance_hierarchy: String,
pub command: Vec<String>,
pub description: String,
pub created_at: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub last_ran_at: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub interval: Option<String>,
pub version: u64,
#[serde(default, skip_serializing_if = "Option::is_none")]
#[schemars(extend("omitempty" = true))]
pub plugin: Option<Plugin>,
}
#[derive(clap::Args)]
#[command(group(
clap::ArgGroup::new("kind")
.multiple(false)
.args(["oneshot", "interval"])
))]
#[command(group(
clap::ArgGroup::new("readiness")
.multiple(false)
.args(["pending", "exhausted"])
))]
pub struct Args {
#[arg(long = "target", required = true)]
pub targets: Vec<String>,
#[arg(long)]
pub oneshot: bool,
#[arg(long)]
pub interval: bool,
#[arg(long)]
pub pending: bool,
#[arg(long)]
pub exhausted: bool,
#[arg(long)]
pub after_id: Option<i64>,
#[arg(long)]
pub count: Option<u64>,
#[command(flatten)]
pub base: crate::cli::command::RequestBaseArgs,
}
#[derive(clap::Args)]
#[command(args_conflicts_with_subcommands = true)]
pub struct Command {
#[command(flatten)]
pub args: Args,
#[command(subcommand)]
pub schema: Option<Schema>,
}
#[derive(clap::Subcommand)]
pub enum Schema {
RequestSchema(request_schema::Args),
ResponseSchema(response_schema::Args),
}
impl TryFrom<Args> for Request {
type Error = crate::cli::command::FromArgsError;
fn try_from(args: Args) -> Result<Self, Self::Error> {
let targets = args
.targets
.iter()
.map(|s| {
s.parse::<Target>().map_err(|msg| {
crate::cli::command::FromArgsError::path_parse("target", msg)
})
})
.collect::<Result<Vec<_>, _>>()?;
Ok(Self {
path_type: Path::AgentsTasksList,
targets,
oneshot: args.oneshot,
interval: args.interval,
pending: args.pending,
exhausted: args.exhausted,
after_id: args.after_id,
count: args.count,
base: args.base.into(),
})
}
}
#[cfg(feature = "cli-executor")]
pub async fn execute<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<E::Stream<ResponseItem>, E::Error> {
request.base.clear_transform();
executor.execute(request, agent_arguments).await
}
#[cfg(feature = "cli-executor")]
pub async fn execute_transform<E: crate::cli::command::CommandExecutor>(
executor: &E,
mut request: Request,
transform: crate::cli::command::Transform,
agent_arguments: Option<&crate::cli::command::AgentArguments>,
) -> Result<E::Stream<serde_json::Value>, E::Error> {
request.base.set_transform(transform);
executor.execute(request, agent_arguments).await
}
#[cfg(feature = "mcp")]
impl crate::cli::command::CommandResponse for ResponseItem {
fn into_mcp(self) -> crate::cli::command::McpResponseItem {
crate::cli::command::McpResponseItem::JSONL(serde_json::to_value(self).unwrap())
}
}
pub mod request_schema;
pub mod response_schema;