Skip to main content

objectiveai_cli/command/db/config/
mod.rs

1//! `config db` sub-tier.
2
3use std::pin::Pin;
4
5use futures::{Stream, StreamExt};
6use objectiveai_sdk::cli::command::db::config::{Request, Response};
7
8use crate::context::Context;
9use crate::error::Error;
10
11pub mod address;
12pub mod database;
13pub mod get;
14pub mod password;
15pub mod user;
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::Get(req) => {
28            let value = get::execute(ctx, req).await?;
29            once(Ok(Response::Get(value)))
30        }
31        Request::GetRequestSchema(req) => {
32            let value = get::request_schema::execute(ctx, req).await?;
33            once(Ok(Response::GetRequestSchema(value)))
34        }
35        Request::GetResponseSchema(req) => {
36            let value = get::response_schema::execute(ctx, req).await?;
37            once(Ok(Response::GetResponseSchema(value)))
38        }
39        Request::Address(req) => {
40            let inner = address::execute(ctx, req).await?;
41            Box::pin(inner.map(|r| r.map(Response::Address)))
42        }
43        Request::User(req) => {
44            let inner = user::execute(ctx, req).await?;
45            Box::pin(inner.map(|r| r.map(Response::User)))
46        }
47        Request::Password(req) => {
48            let inner = password::execute(ctx, req).await?;
49            Box::pin(inner.map(|r| r.map(Response::Password)))
50        }
51        Request::Database(req) => {
52            let inner = database::execute(ctx, req).await?;
53            Box::pin(inner.map(|r| r.map(Response::Database)))
54        }
55    };
56    Ok(stream)
57}