objectiveai_sdk/cli/command/plugins/install/
mod.rs1pub mod filesystem;
2pub mod github;
3
4#[derive(clap::Subcommand)]
5pub enum Command {
6 Filesystem(filesystem::Command),
7 Github(github::Command),
8}
9
10#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
11#[serde(untagged)]
12#[schemars(rename = "cli.command.plugins.install.Request")]
13pub enum Request {
14 #[schemars(title = "Filesystem")]
15 Filesystem(filesystem::Request),
16 #[schemars(title = "FilesystemRequestSchema")]
17 FilesystemRequestSchema(filesystem::request_schema::Request),
18 #[schemars(title = "FilesystemResponseSchema")]
19 FilesystemResponseSchema(filesystem::response_schema::Request),
20 #[schemars(title = "Github")]
21 Github(github::Request),
22 #[schemars(title = "GithubRequestSchema")]
23 GithubRequestSchema(github::request_schema::Request),
24 #[schemars(title = "GithubResponseSchema")]
25 GithubResponseSchema(github::response_schema::Request),
26}
27
28#[objectiveai_sdk_macros::json_schema_ignore]
31#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize, schemars::JsonSchema)]
32#[schemars(rename = "cli.command.plugins.install.Response")]
33#[serde(untagged)]
34pub enum Response {
35 #[schemars(title = "Filesystem")]
36 Filesystem(filesystem::Response),
37 #[schemars(title = "FilesystemRequestSchema")]
38 FilesystemRequestSchema(filesystem::request_schema::Response),
39 #[schemars(title = "FilesystemResponseSchema")]
40 FilesystemResponseSchema(filesystem::response_schema::Response),
41 #[schemars(title = "Github")]
42 Github(github::Response),
43 #[schemars(title = "GithubRequestSchema")]
44 GithubRequestSchema(github::request_schema::Response),
45 #[schemars(title = "GithubResponseSchema")]
46 GithubResponseSchema(github::response_schema::Response),
47}
48
49#[cfg(feature = "mcp")]
50impl crate::cli::command::CommandResponse for Response {
51 fn into_mcp(self) -> crate::cli::command::McpResponseItem {
52 match self {
53 Response::Filesystem(v) => v.into_mcp(),
54 Response::FilesystemRequestSchema(v) => v.into_mcp(),
55 Response::FilesystemResponseSchema(v) => v.into_mcp(),
56 Response::Github(v) => v.into_mcp(),
57 Response::GithubRequestSchema(v) => v.into_mcp(),
58 Response::GithubResponseSchema(v) => v.into_mcp(),
59 }
60 }
61}
62
63impl TryFrom<Command> for Request {
64 type Error = crate::cli::command::FromArgsError;
65 fn try_from(command: Command) -> Result<Self, Self::Error> {
66 match command {
67 Command::Filesystem(cmd) => match cmd.schema {
68 None => Ok(Request::Filesystem(filesystem::Request::try_from(cmd.args)?)),
69 Some(filesystem::Schema::RequestSchema(args)) =>
70 Ok(Request::FilesystemRequestSchema(filesystem::request_schema::Request::try_from(args)?)),
71 Some(filesystem::Schema::ResponseSchema(args)) =>
72 Ok(Request::FilesystemResponseSchema(filesystem::response_schema::Request::try_from(args)?)),
73 },
74 Command::Github(cmd) => match cmd.schema {
75 None => Ok(Request::Github(github::Request::try_from(cmd.args)?)),
76 Some(github::Schema::RequestSchema(args)) =>
77 Ok(Request::GithubRequestSchema(github::request_schema::Request::try_from(args)?)),
78 Some(github::Schema::ResponseSchema(args)) =>
79 Ok(Request::GithubResponseSchema(github::response_schema::Request::try_from(args)?)),
80 },
81 }
82 }
83}
84
85impl crate::cli::command::CommandRequest for Request {
86 fn into_command(&self) -> Vec<String> {
87 match self {
88 Request::Filesystem(inner) => inner.into_command(),
89 Request::FilesystemRequestSchema(inner) => inner.into_command(),
90 Request::FilesystemResponseSchema(inner) => inner.into_command(),
91 Request::Github(inner) => inner.into_command(),
92 Request::GithubRequestSchema(inner) => inner.into_command(),
93 Request::GithubResponseSchema(inner) => inner.into_command(),
94 }
95 }
96}
97
98#[cfg(feature = "cli-executor")]
99pub async fn execute<E: crate::cli::command::CommandExecutor>(
100 executor: &E,
101 request: Request,
102
103 agent_arguments: Option<&crate::cli::command::AgentArguments>,
104 ) -> Result<
105 std::pin::Pin<Box<dyn futures::Stream<Item = Result<Response, E::Error>> + Send>>,
106 E::Error,
107> {
108 use futures::StreamExt;
109 let stream: std::pin::Pin<Box<dyn futures::Stream<Item = Result<Response, E::Error>> + Send>> =
110 match request {
111 Request::Filesystem(req) => {
112 let value = filesystem::execute(executor, req, agent_arguments).await?;
113 Box::pin(crate::cli::command::StreamOnce::new(Ok(
114 Response::Filesystem(value),
115 )))
116 }
117 Request::FilesystemRequestSchema(req) => {
118 let value = filesystem::request_schema::execute(executor, req, agent_arguments).await?;
119 Box::pin(crate::cli::command::StreamOnce::new(Ok(
120 Response::FilesystemRequestSchema(value),
121 )))
122 }
123 Request::FilesystemResponseSchema(req) => {
124 let value = filesystem::response_schema::execute(executor, req, agent_arguments).await?;
125 Box::pin(crate::cli::command::StreamOnce::new(Ok(
126 Response::FilesystemResponseSchema(value),
127 )))
128 }
129 Request::Github(req) => {
130 let value = github::execute(executor, req, agent_arguments).await?;
131 Box::pin(crate::cli::command::StreamOnce::new(Ok(
132 Response::Github(value),
133 )))
134 }
135 Request::GithubRequestSchema(req) => {
136 let value = github::request_schema::execute(executor, req, agent_arguments).await?;
137 Box::pin(crate::cli::command::StreamOnce::new(Ok(
138 Response::GithubRequestSchema(value),
139 )))
140 }
141 Request::GithubResponseSchema(req) => {
142 let value = github::response_schema::execute(executor, req, agent_arguments).await?;
143 Box::pin(crate::cli::command::StreamOnce::new(Ok(
144 Response::GithubResponseSchema(value),
145 )))
146 }
147 };
148 Ok(stream)
149}
150
151#[cfg(feature = "cli-executor")]
152pub async fn execute_jq<E: crate::cli::command::CommandExecutor>(
153 executor: &E,
154 request: Request,
155 jq: String,
156
157 agent_arguments: Option<&crate::cli::command::AgentArguments>,
158 ) -> Result<
159 std::pin::Pin<Box<dyn futures::Stream<Item = Result<serde_json::Value, E::Error>> + Send>>,
160 E::Error,
161> {
162 let stream: std::pin::Pin<Box<dyn futures::Stream<Item = Result<serde_json::Value, E::Error>> + Send>> =
163 match request {
164 Request::Filesystem(req) => {
165 let value = filesystem::execute_jq(executor, req, jq, agent_arguments).await?;
166 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
167 }
168 Request::FilesystemRequestSchema(req) => {
169 let value = filesystem::request_schema::execute_jq(executor, req, jq, agent_arguments).await?;
170 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
171 }
172 Request::FilesystemResponseSchema(req) => {
173 let value = filesystem::response_schema::execute_jq(executor, req, jq, agent_arguments).await?;
174 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
175 }
176 Request::Github(req) => {
177 let value = github::execute_jq(executor, req, jq, agent_arguments).await?;
178 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
179 }
180 Request::GithubRequestSchema(req) => {
181 let value = github::request_schema::execute_jq(executor, req, jq, agent_arguments).await?;
182 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
183 }
184 Request::GithubResponseSchema(req) => {
185 let value = github::response_schema::execute_jq(executor, req, jq, agent_arguments).await?;
186 Box::pin(crate::cli::command::StreamOnce::new(Ok(value)))
187 }
188 };
189 Ok(stream)
190}