objectiveai_cli/command/viewer/
mod.rs

1//! `viewer` tier dispatch. Mirrors
2//! `objectiveai-sdk-rs/src/cli/command/viewer/mod.rs`. All viewer leaves
3//! are unary, so every arm wraps a single value in a one-shot stream.
4
5use std::pin::Pin;
6
7use futures::Stream;
8use objectiveai_sdk::cli::command::viewer::{Request, Response};
9
10use crate::context::Context;
11use crate::error::Error;
12
13pub mod generate_secret_signature_pair;
14pub mod kill;
15pub mod spawn;
16
17type ItemStream = Pin<Box<dyn Stream<Item = Result<Response, Error>> + Send>>;
18
19fn once<T: Send + 'static>(
20    item: Result<T, Error>,
21) -> Pin<Box<dyn Stream<Item = Result<T, Error>> + Send>> {
22    Box::pin(futures::stream::once(async move { item }))
23}
24
25pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
26    let stream: ItemStream = match request {
27        Request::GenerateSecretSignaturePair(req) => {
28            let value = generate_secret_signature_pair::execute(ctx, req).await?;
29            once(Ok(Response::GenerateSecretSignaturePair(value)))
30        }
31        Request::GenerateSecretSignaturePairRequestSchema(req) => {
32            let value = generate_secret_signature_pair::request_schema::execute(ctx, req).await?;
33            once(Ok(Response::GenerateSecretSignaturePairRequestSchema(value)))
34        }
35        Request::GenerateSecretSignaturePairResponseSchema(req) => {
36            let value = generate_secret_signature_pair::response_schema::execute(ctx, req).await?;
37            once(Ok(Response::GenerateSecretSignaturePairResponseSchema(value)))
38        }
39        Request::Kill(req) => {
40            let value = kill::execute(ctx, req).await?;
41            once(Ok(Response::Kill(value)))
42        }
43        Request::KillRequestSchema(req) => {
44            let value = kill::request_schema::execute(ctx, req).await?;
45            once(Ok(Response::KillRequestSchema(value)))
46        }
47        Request::KillResponseSchema(req) => {
48            let value = kill::response_schema::execute(ctx, req).await?;
49            once(Ok(Response::KillResponseSchema(value)))
50        }
51        Request::Spawn(req) => {
52            let value = spawn::execute(ctx, req).await?;
53            once(Ok(Response::Spawn(value)))
54        }
55        Request::SpawnRequestSchema(req) => {
56            let value = spawn::request_schema::execute(ctx, req).await?;
57            once(Ok(Response::SpawnRequestSchema(value)))
58        }
59        Request::SpawnResponseSchema(req) => {
60            let value = spawn::response_schema::execute(ctx, req).await?;
61            once(Ok(Response::SpawnResponseSchema(value)))
62        }
63    };
64    Ok(stream)
65}
objectiveai_cli/command/viewer/mod.rs

objectiveai_cli/command/viewer/
mod.rs
