proto-blue-api 0.3.0

AT Protocol high-level API: agent, rich text, moderation, generated types
Documentation
// Generated by atproto-codegen. Do not edit.
//! Lexicon: tools.ozone.moderation.queryEvents

use serde::{Deserialize, Serialize};

/// List moderation events related to a subject.
/// XRPC Query: tools.ozone.moderation.queryEvents
#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Params {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub added_labels: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub added_tags: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub age_assurance_state: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub batch_id: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub collections: Option<Vec<proto_blue_syntax::Nsid>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub comment: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub created_after: Option<proto_blue_syntax::Datetime>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub created_before: Option<proto_blue_syntax::Datetime>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub created_by: Option<proto_blue_syntax::Did>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub cursor: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub has_comment: Option<bool>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub include_all_user_records: Option<bool>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub limit: Option<i64>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub mod_tool: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub policies: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub removed_labels: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub removed_tags: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub report_types: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub sort_direction: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub subject: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub subject_type: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub types: Option<Vec<String>>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub with_strike: Option<bool>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct Output {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub cursor: Option<String>,
    pub events: Vec<crate::tools::ozone::moderation::defs::ModEventView>,
}

/// Errors a `call()` on this method can return.
#[derive(Debug, thiserror::Error)]
pub enum CallError {
    #[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 {
    CallError::Xrpc(err)
}

fn to_query_params(p: &Params) -> proto_blue_xrpc::QueryParams {
    let mut qp = proto_blue_xrpc::QueryParams::new();
    if let Some(v) = &p.added_labels {
        qp.insert(
            "addedLabels".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.added_tags {
        qp.insert(
            "addedTags".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.age_assurance_state {
        qp.insert(
            "ageAssuranceState".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.batch_id {
        qp.insert(
            "batchId".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.collections {
        qp.insert(
            "collections".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.comment {
        qp.insert(
            "comment".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.created_after {
        qp.insert(
            "createdAfter".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.created_before {
        qp.insert(
            "createdBefore".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.created_by {
        qp.insert(
            "createdBy".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.cursor {
        qp.insert(
            "cursor".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.has_comment {
        qp.insert(
            "hasComment".to_string(),
            proto_blue_xrpc::QueryValue::Boolean(*v),
        );
    }
    if let Some(v) = &p.include_all_user_records {
        qp.insert(
            "includeAllUserRecords".to_string(),
            proto_blue_xrpc::QueryValue::Boolean(*v),
        );
    }
    if let Some(v) = &p.limit {
        qp.insert(
            "limit".to_string(),
            proto_blue_xrpc::QueryValue::Integer(*v),
        );
    }
    if let Some(v) = &p.mod_tool {
        qp.insert(
            "modTool".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.policies {
        qp.insert(
            "policies".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.removed_labels {
        qp.insert(
            "removedLabels".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.removed_tags {
        qp.insert(
            "removedTags".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.report_types {
        qp.insert(
            "reportTypes".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.sort_direction {
        qp.insert(
            "sortDirection".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.subject {
        qp.insert(
            "subject".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.subject_type {
        qp.insert(
            "subjectType".to_string(),
            proto_blue_xrpc::QueryValue::String(v.to_string()),
        );
    }
    if let Some(v) = &p.types {
        qp.insert(
            "types".to_string(),
            proto_blue_xrpc::QueryValue::Array(
                v.iter()
                    .map(|x| proto_blue_xrpc::QueryValue::String(x.to_string()))
                    .collect(),
            ),
        );
    }
    if let Some(v) = &p.with_strike {
        qp.insert(
            "withStrike".to_string(),
            proto_blue_xrpc::QueryValue::Boolean(*v),
        );
    }
    qp
}

/// Execute the query.
pub async fn call(
    client: &proto_blue_xrpc::XrpcClient,
    params: Option<&Params>,
    opts: Option<&proto_blue_xrpc::CallOptions>,
) -> Result<Output, CallError> {
    let qp = params.map(to_query_params);
    let response = match client
        .query("tools.ozone.moderation.queryEvents", qp.as_ref(), 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 method 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<Params>) -> 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.query("tools.ozone.moderation.queryEvents", move |ctx| {
        let handler = handler.clone();
        async move {
            let params = params_from_ctx(&ctx);
            let out = handler(ctx, params).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)
        }
    })
}

#[cfg(feature = "server")]
fn params_from_ctx(ctx: &proto_blue_xrpc::HandlerContext) -> Option<Params> {
    // Always construct a `Params` — required fields are
    // validated upstream by the lexicon validator when enabled;
    // missing values surface as runtime errors from the handler.
    Some(Params {
        added_labels: Some(
            ctx.params
                .get("addedLabels")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        added_tags: Some(
            ctx.params
                .get("addedTags")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        age_assurance_state: ctx.params.get("ageAssuranceState").cloned(),
        batch_id: ctx.params.get("batchId").cloned(),
        collections: ctx.params.get("collections").and_then(|v| {
            v.split(',')
                .map(proto_blue_syntax::Nsid::new)
                .collect::<Result<Vec<_>, _>>()
                .ok()
        }),
        comment: ctx.params.get("comment").cloned(),
        created_after: ctx
            .params
            .get("createdAfter")
            .and_then(|v| proto_blue_syntax::Datetime::new(v).ok()),
        created_before: ctx
            .params
            .get("createdBefore")
            .and_then(|v| proto_blue_syntax::Datetime::new(v).ok()),
        created_by: ctx
            .params
            .get("createdBy")
            .and_then(|v| proto_blue_syntax::Did::new(v).ok()),
        cursor: ctx.params.get("cursor").cloned(),
        has_comment: ctx
            .params
            .get("hasComment")
            .and_then(|v| v.parse::<bool>().ok()),
        include_all_user_records: ctx
            .params
            .get("includeAllUserRecords")
            .and_then(|v| v.parse::<bool>().ok()),
        limit: ctx.params.get("limit").and_then(|v| v.parse::<i64>().ok()),
        mod_tool: Some(
            ctx.params
                .get("modTool")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        policies: Some(
            ctx.params
                .get("policies")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        removed_labels: Some(
            ctx.params
                .get("removedLabels")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        removed_tags: Some(
            ctx.params
                .get("removedTags")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        report_types: Some(
            ctx.params
                .get("reportTypes")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        sort_direction: ctx.params.get("sortDirection").cloned(),
        subject: ctx.params.get("subject").cloned(),
        subject_type: ctx.params.get("subjectType").cloned(),
        types: Some(
            ctx.params
                .get("types")
                .map(|v| v.split(',').map(String::from).collect::<Vec<_>>())
                .unwrap_or_default(),
        ),
        with_strike: ctx
            .params
            .get("withStrike")
            .and_then(|v| v.parse::<bool>().ok()),
    })
}