objectiveai_cli/command/agents/queue/
mod.rs1use std::pin::Pin;
7
8use futures::{Stream, StreamExt};
9use objectiveai_sdk::cli::command::agents::queue::{Request, ResponseItem};
10
11use crate::context::Context;
12use crate::error::Error;
13
14pub mod delete;
15pub mod deliver;
16pub mod list;
17pub mod open;
18
19type ItemStream = Pin<Box<dyn Stream<Item = Result<ResponseItem, Error>> + Send>>;
20
21fn once<T: Send + 'static>(
22 item: Result<T, Error>,
23) -> Pin<Box<dyn Stream<Item = Result<T, Error>> + Send>> {
24 Box::pin(futures::stream::once(async move { item }))
25}
26
27pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
28 let stream: ItemStream = match request {
29 Request::Delete(req) => {
30 let value = delete::execute(ctx, req).await?;
31 once(Ok(ResponseItem::Delete(value)))
32 }
33 Request::DeleteRequestSchema(req) => {
34 let value = delete::request_schema::execute(ctx, req).await?;
35 once(Ok(ResponseItem::DeleteRequestSchema(value)))
36 }
37 Request::DeleteResponseSchema(req) => {
38 let value = delete::response_schema::execute(ctx, req).await?;
39 once(Ok(ResponseItem::DeleteResponseSchema(value)))
40 }
41 Request::Deliver(req) => {
42 let inner = deliver::execute(ctx, req).await?;
43 Box::pin(inner.map(|r| r.map(ResponseItem::Deliver)))
44 }
45 Request::DeliverRequestSchema(req) => {
46 let value = deliver::request_schema::execute(ctx, req).await?;
47 once(Ok(ResponseItem::DeliverRequestSchema(value)))
48 }
49 Request::DeliverResponseSchema(req) => {
50 let value = deliver::response_schema::execute(ctx, req).await?;
51 once(Ok(ResponseItem::DeliverResponseSchema(value)))
52 }
53 Request::List(req) => {
54 let inner = list::execute(ctx, req).await?;
55 Box::pin(inner.map(|r| r.map(ResponseItem::List)))
56 }
57 Request::ListRequestSchema(req) => {
58 let value = list::request_schema::execute(ctx, req).await?;
59 once(Ok(ResponseItem::ListRequestSchema(value)))
60 }
61 Request::ListResponseSchema(req) => {
62 let value = list::response_schema::execute(ctx, req).await?;
63 once(Ok(ResponseItem::ListResponseSchema(value)))
64 }
65 Request::Open(req) => {
66 let value = open::execute(ctx, req).await?;
67 once(Ok(ResponseItem::Open(value)))
68 }
69 Request::OpenRequestSchema(req) => {
70 let value = open::request_schema::execute(ctx, req).await?;
71 once(Ok(ResponseItem::OpenRequestSchema(value)))
72 }
73 Request::OpenResponseSchema(req) => {
74 let value = open::response_schema::execute(ctx, req).await?;
75 once(Ok(ResponseItem::OpenResponseSchema(value)))
76 }
77 };
78 Ok(stream)
79}