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 AdminGetSubjectStatusParams {
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub blob: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub did: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub uri: Option<String>,
}

#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct AdminGetSubjectStatusOutput {
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub deactivated: Option<crate::api::com::atproto::AdminDefsStatusAttr>,
    pub subject: AdminGetSubjectStatusOutputSubjectUnion,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub takedown: Option<crate::api::com::atproto::AdminDefsStatusAttr>,
    /// Extra fields not defined in the schema.
    #[serde(flatten)]
    pub extra: std::collections::HashMap<String, serde_json::Value>,
}

/// AdminGetSubjectStatusOutputSubjectUnion is a union type.
#[derive(Debug, Clone)]
pub enum AdminGetSubjectStatusOutputSubjectUnion {
    AdminDefsRepoRef(Box<crate::api::com::atproto::AdminDefsRepoRef>),
    RepoStrongRef(Box<crate::api::com::atproto::RepoStrongRef>),
    AdminDefsRepoBlobRef(Box<crate::api::com::atproto::AdminDefsRepoBlobRef>),
    Unknown(crate::api::UnknownUnionVariant),
}

impl serde::Serialize for AdminGetSubjectStatusOutputSubjectUnion {
    fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        match self {
            AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoRef(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("com.atproto.admin.defs#repoRef".to_string()),
                    );
                }
                map.serialize(serializer)
            }
            AdminGetSubjectStatusOutputSubjectUnion::RepoStrongRef(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("com.atproto.repo.strongRef".to_string()),
                    );
                }
                map.serialize(serializer)
            }
            AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoBlobRef(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("com.atproto.admin.defs#repoBlobRef".to_string()),
                    );
                }
                map.serialize(serializer)
            }
            AdminGetSubjectStatusOutputSubjectUnion::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 AdminGetSubjectStatusOutputSubjectUnion {
    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 {
            "com.atproto.admin.defs#repoRef" => {
                let inner: crate::api::com::atproto::AdminDefsRepoRef =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoRef(
                    Box::new(inner),
                ))
            }
            "com.atproto.repo.strongRef" => {
                let inner: crate::api::com::atproto::RepoStrongRef =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::RepoStrongRef(
                    Box::new(inner),
                ))
            }
            "com.atproto.admin.defs#repoBlobRef" => {
                let inner: crate::api::com::atproto::AdminDefsRepoBlobRef =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoBlobRef(Box::new(inner)))
            }
            _ => Ok(AdminGetSubjectStatusOutputSubjectUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: Some(value),
                    cbor: None,
                },
            )),
        }
    }
}

impl AdminGetSubjectStatusOutputSubjectUnion {
    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 {
            AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoRef(inner) => {
                inner.encode_cbor(buf)
            }
            AdminGetSubjectStatusOutputSubjectUnion::RepoStrongRef(inner) => inner.encode_cbor(buf),
            AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoBlobRef(inner) => {
                inner.encode_cbor(buf)
            }
            AdminGetSubjectStatusOutputSubjectUnion::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 {
            "com.atproto.admin.defs#repoRef" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner = crate::api::com::atproto::AdminDefsRepoRef::decode_cbor(&mut dec)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoRef(
                    Box::new(inner),
                ))
            }
            "com.atproto.repo.strongRef" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner = crate::api::com::atproto::RepoStrongRef::decode_cbor(&mut dec)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::RepoStrongRef(
                    Box::new(inner),
                ))
            }
            "com.atproto.admin.defs#repoBlobRef" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner = crate::api::com::atproto::AdminDefsRepoBlobRef::decode_cbor(&mut dec)?;
                Ok(AdminGetSubjectStatusOutputSubjectUnion::AdminDefsRepoBlobRef(Box::new(inner)))
            }
            _ => Ok(AdminGetSubjectStatusOutputSubjectUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: None,
                    cbor: Some(raw.to_vec()),
                },
            )),
        }
    }
}

/// AdminGetSubjectStatus — Get the service-specific admin status of a subject (account, record, or blob).
pub async fn admin_get_subject_status(
    client: &crate::xrpc::Client,
    params: &AdminGetSubjectStatusParams,
) -> Result<AdminGetSubjectStatusOutput, crate::xrpc::Error> {
    client
        .query("com.atproto.admin.getSubjectStatus", params)
        .await
}