objectiveai-cli 2.2.9

//! `agents laboratories attach` — acquire the target's lock(s), insert
//! the `(target, laboratory_id)` row, release. Errors if the laboratory
//! is already attached to the target.

use objectiveai_sdk::cli::command::agents::laboratories::attach::{Request, Response};

use crate::context::Context;
use crate::error::Error;

pub async fn execute(ctx: &Context, request: Request) -> Result<Response, Error> {
    let (target, claims) = super::lock_target(ctx, &request.selector).await?;
    let pool = ctx.db_client().await?.clone();
    let result =
        crate::db::laboratory_attachments::attach(&pool, &target, &request.laboratory_id).await;
    super::release_all(claims);
    let inserted = result?;
    if !inserted {
        return Err(Error::LaboratoryAlreadyAttached {
            laboratory_id: request.laboratory_id,
        });
    }
    Ok(Response {
        laboratory_id: request.laboratory_id,
    })
}

pub mod request_schema {
    use objectiveai_sdk::cli::command::agents::laboratories::attach as sdk;
    use objectiveai_sdk::cli::command::agents::laboratories::attach::request_schema::{
        Request, Response,
    };

    use crate::context::Context;
    use crate::error::Error;

    pub async fn execute(_ctx: &Context, _request: Request) -> Result<Response, Error> {
        Ok(objectiveai_sdk::cli::command::ResponseSchema(
            schemars::schema_for!(sdk::Request),
        ))
    }
}

pub mod response_schema {
    use objectiveai_sdk::cli::command::agents::laboratories::attach as sdk;
    use objectiveai_sdk::cli::command::agents::laboratories::attach::response_schema::{
        Request, Response,
    };

    use crate::context::Context;
    use crate::error::Error;

    pub async fn execute(_ctx: &Context, _request: Request) -> Result<Response, Error> {
        Ok(objectiveai_sdk::cli::command::ResponseSchema(
            schemars::schema_for!(sdk::Response),
        ))
    }
}