objectiveai_cli/command/daemon/
mod.rs

1//! `daemon` tier dispatch. Mirrors
2//! `objectiveai-sdk-rs/src/cli/command/daemon/mod.rs`. `spawn` is
3//! streaming (the resident daemon yields readiness then serves);
4//! `kill` is unary.
5
6use std::pin::Pin;
7
8use futures::{Stream, StreamExt};
9use objectiveai_sdk::cli::command::daemon::{Request, ResponseItem};
10
11use crate::context::Context;
12use crate::error::Error;
13
14pub mod kill;
15pub mod spawn;
16
17/// Per-state daemon singleton lock key (under `state_dir/locks`).
18pub const DAEMON_LOCK_KEY: &str = "plugins-daemon";
19
20/// Init gate key — serializes daemon startup so the singleton lock is
21/// acquired without racing (mirrors `objectiveai-db`'s `db-init` gate).
22pub const DAEMON_INIT_LOCK_KEY: &str = "plugins-daemon-init";
23
24type ItemStream = Pin<Box<dyn Stream<Item = Result<ResponseItem, Error>> + Send>>;
25
26fn once<T: Send + 'static>(
27    item: Result<T, Error>,
28) -> Pin<Box<dyn Stream<Item = Result<T, Error>> + Send>> {
29    Box::pin(futures::stream::once(async move { item }))
30}
31
32pub async fn execute(ctx: &Context, request: Request) -> Result<ItemStream, Error> {
33    let stream: ItemStream = match request {
34        Request::Spawn(req) => {
35            let inner = spawn::execute(ctx, req).await?;
36            Box::pin(inner.map(|r| r.map(ResponseItem::Spawn)))
37        }
38        Request::SpawnRequestSchema(req) => {
39            let value = spawn::request_schema::execute(ctx, req).await?;
40            once(Ok(ResponseItem::SpawnRequestSchema(value)))
41        }
42        Request::SpawnResponseSchema(req) => {
43            let value = spawn::response_schema::execute(ctx, req).await?;
44            once(Ok(ResponseItem::SpawnResponseSchema(value)))
45        }
46        Request::Kill(req) => {
47            let value = kill::execute(ctx, req).await?;
48            once(Ok(ResponseItem::Kill(value)))
49        }
50        Request::KillRequestSchema(req) => {
51            let value = kill::request_schema::execute(ctx, req).await?;
52            once(Ok(ResponseItem::KillRequestSchema(value)))
53        }
54        Request::KillResponseSchema(req) => {
55            let value = kill::response_schema::execute(ctx, req).await?;
56            once(Ok(ResponseItem::KillResponseSchema(value)))
57        }
58    };
59    Ok(stream)
60}
objectiveai_cli/command/daemon/mod.rs

objectiveai_cli/command/daemon/
mod.rs
