proto-blue-api 0.3.0

// Generated by atproto-codegen. Do not edit.
//! Lexicon: com.atproto.repo.applyWrites

use serde::{Deserialize, Serialize};

/// Operation which creates a new record.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Create {
    pub collection: proto_blue_syntax::Nsid,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub rkey: Option<proto_blue_syntax::RecordKey>,
    pub value: serde_json::Value,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct CreateResult {
    pub cid: String,
    pub uri: proto_blue_syntax::AtUri,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub validation_status: Option<String>,
}

/// Operation which deletes an existing record.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Delete {
    pub collection: proto_blue_syntax::Nsid,
    pub rkey: proto_blue_syntax::RecordKey,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct DeleteResult {}

/// Apply a batch transaction of repository creates, updates, and deletes. Requires auth, implemented by PDS.
/// XRPC Procedure: com.atproto.repo.applyWrites
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "$type")]
pub enum InputWritesItemRefs {
    #[serde(rename = "com.atproto.repo.applyWrites#create")]
    AtprotoRepoApplyWritesCreate(Box<Create>),
    #[serde(rename = "com.atproto.repo.applyWrites#update")]
    AtprotoRepoApplyWritesUpdate(Box<Update>),
    #[serde(rename = "com.atproto.repo.applyWrites#delete")]
    AtprotoRepoApplyWritesDelete(Box<Delete>),
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Input {
    pub repo: proto_blue_syntax::AtIdentifier,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub swap_commit: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub validate: Option<bool>,
    pub writes: Vec<InputWritesItemRefs>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "$type")]
pub enum OutputResultsItemRefs {
    #[serde(rename = "com.atproto.repo.applyWrites#createResult")]
    AtprotoRepoApplyWritesCreateResult(Box<CreateResult>),
    #[serde(rename = "com.atproto.repo.applyWrites#updateResult")]
    AtprotoRepoApplyWritesUpdateResult(Box<UpdateResult>),
    #[serde(rename = "com.atproto.repo.applyWrites#deleteResult")]
    AtprotoRepoApplyWritesDeleteResult(Box<DeleteResult>),
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Output {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub commit: Option<crate::com::atproto::repo::defs::CommitMeta>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub results: Option<Vec<OutputResultsItemRefs>>,
}

/// Errors a `call()` on this method can return.
#[derive(Debug, thiserror::Error)]
pub enum CallError {
    /// Indicates that the 'swapCommit' parameter did not match current commit.
    #[error("InvalidSwap")]
    InvalidSwap,
    #[error("{0}")]
    Xrpc(proto_blue_xrpc::XrpcError),
    #[error(transparent)]
    Transport(#[from] proto_blue_xrpc::Error),
    #[error(transparent)]
    Json(#[from] serde_json::Error),
}

fn map_xrpc_error(err: proto_blue_xrpc::XrpcError) -> CallError {
    match err.error.as_deref() {
        Some("InvalidSwap") => CallError::InvalidSwap,
        _ => CallError::Xrpc(err),
    }
}

/// Execute the procedure.
pub async fn call(
    client: &proto_blue_xrpc::XrpcClient,
    input: &Input,
    opts: Option<&proto_blue_xrpc::CallOptions>,
) -> Result<Output, CallError> {
    let qp_ref: Option<&proto_blue_xrpc::QueryParams> = None;
    let body = proto_blue_xrpc::XrpcBody::Json(serde_json::to_value(input)?);
    let response = match client
        .procedure("com.atproto.repo.applyWrites", qp_ref, Some(body), opts)
        .await
    {
        Ok(r) => r,
        Err(proto_blue_xrpc::Error::Xrpc(x)) => return Err(map_xrpc_error(x)),
        Err(e) => return Err(CallError::Transport(e)),
    };
    Ok(serde_json::from_value(response.data)?)
}

/// Register a typed handler for this procedure on an [`proto_blue_xrpc::XrpcServer`].
#[cfg(feature = "server")]
pub fn register<F, Fut>(
    server: proto_blue_xrpc::XrpcServer,
    handler: F,
) -> proto_blue_xrpc::XrpcServer
where
    F: Fn(proto_blue_xrpc::HandlerContext, Option<Input>) -> Fut + Send + Sync + 'static,
    Fut: std::future::Future<Output = Result<Output, proto_blue_xrpc::XrpcServerError>>
        + Send
        + 'static,
{
    let handler = std::sync::Arc::new(handler);
    server.procedure("com.atproto.repo.applyWrites", move |ctx| {
        let handler = handler.clone();
        async move {
            let input = match ctx.json_body()? {
                Some(v) => Some(serde_json::from_value::<Input>(v).map_err(|e| {
                    proto_blue_xrpc::XrpcServerError::new(
                        proto_blue_xrpc::ResponseType::InvalidRequest,
                        format!("input deserialize: {e}"),
                    )
                })?),
                None => None,
            };
            let out = handler(ctx, input).await?;
            let value = serde_json::to_value(&out).map_err(|e| {
                proto_blue_xrpc::XrpcServerError::new(
                    proto_blue_xrpc::ResponseType::InternalServerError,
                    format!("output serialize: {e}"),
                )
            })?;
            Ok::<_, proto_blue_xrpc::XrpcServerError>(value)
        }
    })
}

/// Operation which updates an existing record.
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Update {
    pub collection: proto_blue_syntax::Nsid,
    pub rkey: proto_blue_syntax::RecordKey,
    pub value: serde_json::Value,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct UpdateResult {
    pub cid: String,
    pub uri: proto_blue_syntax::AtUri,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub validation_status: Option<String>,
}