shrike 0.1.1

AT Protocol library for Rust
Documentation
// Code generated by lexgen. DO NOT EDIT.

#[derive(Debug, Clone, Default, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct ModerationGetMessageContextParams {
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub after: Option<i64>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub before: Option<i64>,
    /// Conversation that the message is from. NOTE: this field will eventually be required.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub convo_id: Option<String>,
    pub message_id: String,
}

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

/// ModerationGetMessageContextOutputMessagesUnion is a union type.
#[derive(Debug, Clone)]
pub enum ModerationGetMessageContextOutputMessagesUnion {
    ConvoDefsMessageView(Box<crate::api::chat::bsky::ConvoDefsMessageView>),
    ConvoDefsDeletedMessageView(Box<crate::api::chat::bsky::ConvoDefsDeletedMessageView>),
    Unknown(crate::api::UnknownUnionVariant),
}

impl serde::Serialize for ModerationGetMessageContextOutputMessagesUnion {
    fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
        match self {
            ModerationGetMessageContextOutputMessagesUnion::ConvoDefsMessageView(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("chat.bsky.convo.defs#messageView".to_string()),
                    );
                }
                map.serialize(serializer)
            }
            ModerationGetMessageContextOutputMessagesUnion::ConvoDefsDeletedMessageView(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(
                            "chat.bsky.convo.defs#deletedMessageView".to_string(),
                        ),
                    );
                }
                map.serialize(serializer)
            }
            ModerationGetMessageContextOutputMessagesUnion::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 ModerationGetMessageContextOutputMessagesUnion {
    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 {
            "chat.bsky.convo.defs#messageView" => {
                let inner: crate::api::chat::bsky::ConvoDefsMessageView =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(
                    ModerationGetMessageContextOutputMessagesUnion::ConvoDefsMessageView(Box::new(
                        inner,
                    )),
                )
            }
            "chat.bsky.convo.defs#deletedMessageView" => {
                let inner: crate::api::chat::bsky::ConvoDefsDeletedMessageView =
                    serde_json::from_value(value).map_err(serde::de::Error::custom)?;
                Ok(
                    ModerationGetMessageContextOutputMessagesUnion::ConvoDefsDeletedMessageView(
                        Box::new(inner),
                    ),
                )
            }
            _ => Ok(ModerationGetMessageContextOutputMessagesUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: Some(value),
                    cbor: None,
                },
            )),
        }
    }
}

impl ModerationGetMessageContextOutputMessagesUnion {
    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 {
            ModerationGetMessageContextOutputMessagesUnion::ConvoDefsMessageView(inner) => {
                inner.encode_cbor(buf)
            }
            ModerationGetMessageContextOutputMessagesUnion::ConvoDefsDeletedMessageView(inner) => {
                inner.encode_cbor(buf)
            }
            ModerationGetMessageContextOutputMessagesUnion::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 {
            "chat.bsky.convo.defs#messageView" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner = crate::api::chat::bsky::ConvoDefsMessageView::decode_cbor(&mut dec)?;
                Ok(
                    ModerationGetMessageContextOutputMessagesUnion::ConvoDefsMessageView(Box::new(
                        inner,
                    )),
                )
            }
            "chat.bsky.convo.defs#deletedMessageView" => {
                let mut dec = crate::cbor::Decoder::new(raw);
                let inner =
                    crate::api::chat::bsky::ConvoDefsDeletedMessageView::decode_cbor(&mut dec)?;
                Ok(
                    ModerationGetMessageContextOutputMessagesUnion::ConvoDefsDeletedMessageView(
                        Box::new(inner),
                    ),
                )
            }
            _ => Ok(ModerationGetMessageContextOutputMessagesUnion::Unknown(
                crate::api::UnknownUnionVariant {
                    r#type: type_str.to_string(),
                    json: None,
                    cbor: Some(raw.to_vec()),
                },
            )),
        }
    }
}

/// ModerationGetMessageContext XRPC query.
pub async fn moderation_get_message_context(
    client: &crate::xrpc::Client,
    params: &ModerationGetMessageContextParams,
) -> Result<ModerationGetMessageContextOutput, crate::xrpc::Error> {
    client
        .query("chat.bsky.moderation.getMessageContext", params)
        .await
}