use crate::cli::command::CommandRequest;
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.plugins.run.Request")]
pub struct Request {
pub path_type: Path,
pub owner: String,
pub name: String,
pub version: String,
pub args: Vec<String>,
#[serde(flatten)]
pub base: crate::cli::command::RequestBase,
}
#[derive(Debug, Clone, PartialEq, Eq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.plugins.run.Path")]
pub enum Path {
#[serde(rename = "plugins/run")]
PluginsRun,
}
impl CommandRequest for Request {
fn into_command(&self) -> Vec<String> {
let mut argv = vec!["plugins".to_string(), "run".to_string()];
argv.push("--owner".to_string());
argv.push(self.owner.clone());
argv.push("--name".to_string());
argv.push(self.name.clone());
argv.push("--version".to_string());
argv.push(self.version.clone());
self.base.push_flags(&mut argv);
if !self.args.is_empty() {
argv.push("--args".to_string());
argv.push(serde_json::to_string(&self.args).expect("Vec<String> serializes"));
}
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, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[serde(untagged)]
#[schemars(rename = "cli.command.plugins.run.ResponseItem")]
pub enum ResponseItem {
#[schemars(title = "Mcp")]
Mcp(Mcp),
#[schemars(title = "Error")]
Error(crate::cli::Error),
#[schemars(title = "Notification")]
Notification(serde_json::Value),
}
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
#[schemars(rename = "cli.command.plugins.run.Mcp")]
pub struct Mcp {
pub r#type: McpType,
pub url: String,
}
#[derive(
Debug,
Clone,
Copy,
PartialEq,
Eq,
serde::Serialize,
serde::Deserialize,
schemars::JsonSchema,
)]
#[serde(rename_all = "snake_case")]
#[schemars(rename = "cli.command.plugins.run.McpType")]
pub enum McpType {
Mcp,
}
#[derive(clap::Args)]
pub struct Args {
#[arg(long)]
pub owner: String,
#[arg(long)]
pub name: String,
#[arg(long)]
pub version: String,
#[arg(long)]
pub args: Option<String>,
#[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 parsed_args: Vec<String> = match args.args {
Some(s) => {
let mut de = serde_json::Deserializer::from_str(&s);
serde_path_to_error::deserialize(&mut de).map_err(|source| {
crate::cli::command::FromArgsError {
field: "args",
source: source.into(),
}
})?
}
None => Vec::new(),
};
Ok(Self {
path_type: Path::PluginsRun,
owner: args.owner,
name: args.name,
version: args.version,
args: parsed_args,
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 {
use crate::agent::completions::message::RichContentPart;
use crate::cli::command::McpResponseItem;
match self {
ResponseItem::Mcp(m) => {
McpResponseItem::JSONL(serde_json::to_value(m).unwrap())
}
ResponseItem::Error(e) => e.into_mcp(),
ResponseItem::Notification(value) => {
if let serde_json::Value::String(s) = &value
&& let Some((mime, payload)) = crate::data_url::parse_data_url(s)
{
let part = RichContentPart::from_blob(
mime,
payload.to_string(),
None,
);
return McpResponseItem::Media(part.into());
}
McpResponseItem::JSONL(value)
}
}
}
}
pub mod request_schema;
pub mod response_schema;