shrike 0.1.0

// Code generated by lexgen. DO NOT EDIT.

#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ModerationGetRecordsParams {
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub uris: Vec<String>,
}

#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ModerationGetRecordsOutput {
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub records: Vec<ModerationGetRecordsOutputRecordsUnion>,
    /// Extra fields not defined in the schema.
    #[serde(flatten)]
    pub extra: std::collections::HashMap<String, serde_json::Value>,
}

/// ModerationGetRecordsOutputRecordsUnion is a union type.
#[derive(Debug, Clone)]
pub enum ModerationGetRecordsOutputRecordsUnion {
    ModerationDefsRecordViewDetail(Box<crate::api::tools::ozone::ModerationDefsRecordViewDetail>),
    ModerationDefsRecordViewNotFound(
        Box<crate::api::tools::ozone::ModerationDefsRecordViewNotFound>,
    ),
    Unknown(crate::api::UnknownUnionVariant),
}

impl serde::Serialize for ModerationGetRecordsOutputRecordsUnion {
    fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        match self {
            ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewDetail(inner) => {
                let mut map =
                    serde_json::to_value(inner.as_ref()).map_err(serde::ser::Error::custom)?;
                if let serde_json::Value::Object(ref mut m) = map {
                    m.insert(
                        "$type".to_string(),
                        serde_json::Value::String(
                            "tools.ozone.moderation.defs#recordViewDetail".to_string(),
                        ),
                    );
                }
                map.serialize(serializer)
            }
            ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewNotFound(inner) => {
                let mut map =
                    serde_json::to_value(inner.as_ref()).map_err(serde::ser::Error::custom)?;
                if let serde_json::Value::Object(ref mut m) = map {
                    m.insert(
                        "$type".to_string(),
                        serde_json::Value::String(
                            "tools.ozone.moderation.defs#recordViewNotFound".to_string(),
                        ),
                    );
                }
                map.serialize(serializer)
            }
            ModerationGetRecordsOutputRecordsUnion::Unknown(v) => {
                if let Some(ref j) = v.json {
                    j.serialize(serializer)
                } else {
                    Err(serde::ser::Error::custom(
                        "no JSON data for unknown union variant",
                    ))
                }
            }
        }
    }
}

impl<'de> serde::Deserialize<'de> for ModerationGetRecordsOutputRecordsUnion {
    fn deserialize<D: serde::Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error> {
        let value = serde_json::Value::deserialize(deserializer)?;
        let type_str = value
            .get("$type")
            .and_then(|v| v.as_str())
            .unwrap_or_default();
        match type_str {
            "tools.ozone.moderation.defs#recordViewDetail" => {
                let inner: crate::api::tools::ozone::ModerationDefsRecordViewDetail =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(
                    ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewDetail(
                        Box::new(inner),
                    ),
                )
            }
            "tools.ozone.moderation.defs#recordViewNotFound" => {
                let inner: crate::api::tools::ozone::ModerationDefsRecordViewNotFound =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(
                    ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewNotFound(
                        Box::new(inner),
                    ),
                )
            }
            _ => Ok(ModerationGetRecordsOutputRecordsUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: Some(value),
                    cbor: None,
                },
            )),
        }
    }
}

impl ModerationGetRecordsOutputRecordsUnion {
    pub fn to_cbor(&self) -> Result<Vec<u8>, crate::cbor::CborError> {
        let mut buf = Vec::new();
        self.encode_cbor(&mut buf)?;
        Ok(buf)
    }

    pub fn encode_cbor(&self, buf: &mut Vec<u8>) -> Result<(), crate::cbor::CborError> {
        match self {
            ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewDetail(inner) => {
                inner.encode_cbor(buf)
            }
            ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewNotFound(inner) => {
                inner.encode_cbor(buf)
            }
            ModerationGetRecordsOutputRecordsUnion::Unknown(v) => {
                if let Some(ref data) = v.cbor {
                    buf.extend_from_slice(data);
                    Ok(())
                } else {
                    Err(crate::cbor::CborError::InvalidCbor(
                        "no CBOR data for unknown union variant".into(),
                    ))
                }
            }
        }
    }

    pub fn from_cbor(data: &[u8]) -> Result<Self, crate::cbor::CborError> {
        let mut decoder = crate::cbor::Decoder::new(data);
        let result = Self::decode_cbor(&mut decoder)?;
        if !decoder.is_empty() {
            return Err(crate::cbor::CborError::InvalidCbor("trailing data".into()));
        }
        Ok(result)
    }

    pub fn decode_cbor(decoder: &mut crate::cbor::Decoder) -> Result<Self, crate::cbor::CborError> {
        // Save position, decode the value, look for $type key.
        let start = decoder.position();
        let val = decoder.decode()?;
        let end = decoder.position();
        let raw = &decoder.raw_input()[start..end];
        let entries = match val {
            crate::cbor::Value::Map(entries) => entries,
            _ => {
                return Err(crate::cbor::CborError::InvalidCbor(
                    "expected map for union".into(),
                ));
            }
        };
        let type_str = entries
            .iter()
            .find(|(k, _)| *k == "$type")
            .and_then(|(_, v)| match v {
                crate::cbor::Value::Text(s) => Some(*s),
                _ => None,
            })
            .unwrap_or_default();
        match type_str {
            "tools.ozone.moderation.defs#recordViewDetail" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner = crate::api::tools::ozone::ModerationDefsRecordViewDetail::decode_cbor(
                    &mut dec,
                )?;
                Ok(
                    ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewDetail(
                        Box::new(inner),
                    ),
                )
            }
            "tools.ozone.moderation.defs#recordViewNotFound" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner =
                    crate::api::tools::ozone::ModerationDefsRecordViewNotFound::decode_cbor(
                        &mut dec,
                    )?;
                Ok(
                    ModerationGetRecordsOutputRecordsUnion::ModerationDefsRecordViewNotFound(
                        Box::new(inner),
                    ),
                )
            }
            _ => Ok(ModerationGetRecordsOutputRecordsUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: None,
                    cbor: Some(raw.to_vec()),
                },
            )),
        }
    }
}

/// ModerationGetRecords — Get details about some records.
pub async fn moderation_get_records(
    client: &crate::xrpc::Client,
    params: &ModerationGetRecordsParams,
) -> Result<ModerationGetRecordsOutput, crate::xrpc::Error> {
    client
        .query("tools.ozone.moderation.getRecords", params)
        .await
}