pub const NSID_LABELER_SERVICE: &str = "app.bsky.labeler.service";
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct LabelerService {
pub created_at: crate::syntax::Datetime,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub labels: Option<LabelerServiceLabelsUnion>,
pub policies: crate::api::app::bsky::LabelerDefsLabelerPolicies,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub reason_types: Vec<crate::api::com::atproto::ModerationDefsReasonType>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub subject_collections: Vec<crate::syntax::Nsid>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub subject_types: Vec<crate::api::com::atproto::ModerationDefsSubjectType>,
#[serde(flatten)]
pub extra: std::collections::HashMap<String, serde_json::Value>,
#[serde(skip)]
pub extra_cbor: Vec<(String, Vec<u8>)>,
}
#[derive(Debug, Clone)]
pub enum LabelerServiceLabelsUnion {
LabelDefsSelfLabels(Box<crate::api::com::atproto::LabelDefsSelfLabels>),
Unknown(crate::api::UnknownUnionVariant),
}
impl serde::Serialize for LabelerServiceLabelsUnion {
fn serialize<S: serde::Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
match self {
LabelerServiceLabelsUnion::LabelDefsSelfLabels(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.label.defs#selfLabels".to_string()),
);
}
map.serialize(serializer)
}
LabelerServiceLabelsUnion::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 LabelerServiceLabelsUnion {
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.label.defs#selfLabels" => {
let inner: crate::api::com::atproto::LabelDefsSelfLabels =
serde_json::from_value(value).map_err(serde::de::Error::custom)?;
Ok(LabelerServiceLabelsUnion::LabelDefsSelfLabels(Box::new(
inner,
)))
}
_ => Ok(LabelerServiceLabelsUnion::Unknown(
crate::api::UnknownUnionVariant {
r#type: type_str.to_string(),
json: Some(value),
cbor: None,
},
)),
}
}
}
impl LabelerServiceLabelsUnion {
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 {
LabelerServiceLabelsUnion::LabelDefsSelfLabels(inner) => inner.encode_cbor(buf),
LabelerServiceLabelsUnion::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> {
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.label.defs#selfLabels" => {
let mut dec = crate::cbor::Decoder::new(raw);
let inner = crate::api::com::atproto::LabelDefsSelfLabels::decode_cbor(&mut dec)?;
Ok(LabelerServiceLabelsUnion::LabelDefsSelfLabels(Box::new(
inner,
)))
}
_ => Ok(LabelerServiceLabelsUnion::Unknown(
crate::api::UnknownUnionVariant {
r#type: type_str.to_string(),
json: None,
cbor: Some(raw.to_vec()),
},
)),
}
}
}
impl LabelerService {
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> {
if self.extra_cbor.is_empty() {
let mut count = 2u64;
if self.labels.is_some() {
count += 1;
}
if !self.reason_types.is_empty() {
count += 1;
}
if !self.subject_types.is_empty() {
count += 1;
}
if !self.subject_collections.is_empty() {
count += 1;
}
crate::cbor::Encoder::new(&mut *buf).encode_map_header(count)?;
if self.labels.is_some() {
crate::cbor::Encoder::new(&mut *buf).encode_text("labels")?;
if let Some(ref val) = self.labels {
val.encode_cbor(buf)?;
}
}
crate::cbor::Encoder::new(&mut *buf).encode_text("policies")?;
self.policies.encode_cbor(buf)?;
crate::cbor::Encoder::new(&mut *buf).encode_text("createdAt")?;
crate::cbor::Encoder::new(&mut *buf).encode_text(self.created_at.as_str())?;
if !self.reason_types.is_empty() {
crate::cbor::Encoder::new(&mut *buf).encode_text("reasonTypes")?;
crate::cbor::Encoder::new(&mut *buf)
.encode_array_header(self.reason_types.len() as u64)?;
for item in &self.reason_types {
crate::cbor::Encoder::new(&mut *buf).encode_text(item)?;
}
}
if !self.subject_types.is_empty() {
crate::cbor::Encoder::new(&mut *buf).encode_text("subjectTypes")?;
crate::cbor::Encoder::new(&mut *buf)
.encode_array_header(self.subject_types.len() as u64)?;
for item in &self.subject_types {
crate::cbor::Encoder::new(&mut *buf).encode_text(item)?;
}
}
if !self.subject_collections.is_empty() {
crate::cbor::Encoder::new(&mut *buf).encode_text("subjectCollections")?;
crate::cbor::Encoder::new(&mut *buf)
.encode_array_header(self.subject_collections.len() as u64)?;
for item in &self.subject_collections {
crate::cbor::Encoder::new(&mut *buf).encode_text(item.as_str())?;
}
}
} else {
let mut pairs: Vec<(&str, Vec<u8>)> = Vec::new();
if self.labels.is_some() {
let mut vbuf = Vec::new();
if let Some(ref val) = self.labels {
val.encode_cbor(&mut vbuf)?;
}
pairs.push(("labels", vbuf));
}
{
let mut vbuf = Vec::new();
self.policies.encode_cbor(&mut vbuf)?;
pairs.push(("policies", vbuf));
}
{
let mut vbuf = Vec::new();
crate::cbor::Encoder::new(&mut vbuf).encode_text(self.created_at.as_str())?;
pairs.push(("createdAt", vbuf));
}
if !self.reason_types.is_empty() {
let mut vbuf = Vec::new();
crate::cbor::Encoder::new(&mut vbuf)
.encode_array_header(self.reason_types.len() as u64)?;
for item in &self.reason_types {
crate::cbor::Encoder::new(&mut vbuf).encode_text(item)?;
}
pairs.push(("reasonTypes", vbuf));
}
if !self.subject_types.is_empty() {
let mut vbuf = Vec::new();
crate::cbor::Encoder::new(&mut vbuf)
.encode_array_header(self.subject_types.len() as u64)?;
for item in &self.subject_types {
crate::cbor::Encoder::new(&mut vbuf).encode_text(item)?;
}
pairs.push(("subjectTypes", vbuf));
}
if !self.subject_collections.is_empty() {
let mut vbuf = Vec::new();
crate::cbor::Encoder::new(&mut vbuf)
.encode_array_header(self.subject_collections.len() as u64)?;
for item in &self.subject_collections {
crate::cbor::Encoder::new(&mut vbuf).encode_text(item.as_str())?;
}
pairs.push(("subjectCollections", vbuf));
}
for (k, v) in &self.extra_cbor {
pairs.push((k.as_str(), v.clone()));
}
pairs.sort_by(|a, b| crate::cbor::cbor_key_cmp(a.0, b.0));
crate::cbor::Encoder::new(&mut *buf).encode_map_header(pairs.len() as u64)?;
for (k, v) in &pairs {
crate::cbor::Encoder::new(&mut *buf).encode_text(k)?;
buf.extend_from_slice(v);
}
}
Ok(())
}
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> {
let val = decoder.decode()?;
let entries = match val {
crate::cbor::Value::Map(entries) => entries,
_ => return Err(crate::cbor::CborError::InvalidCbor("expected map".into())),
};
let mut field_labels: Option<LabelerServiceLabelsUnion> = None;
let mut field_policies: Option<crate::api::app::bsky::LabelerDefsLabelerPolicies> = None;
let mut field_created_at: Option<crate::syntax::Datetime> = None;
let mut field_reason_types: Vec<crate::api::com::atproto::ModerationDefsReasonType> =
Vec::new();
let mut field_subject_types: Vec<crate::api::com::atproto::ModerationDefsSubjectType> =
Vec::new();
let mut field_subject_collections: Vec<crate::syntax::Nsid> = Vec::new();
let mut extra_cbor: Vec<(String, Vec<u8>)> = Vec::new();
for (key, value) in entries {
match key {
"labels" => {
let raw = crate::cbor::encode_value(&value)?;
let mut dec = crate::cbor::Decoder::new(&raw);
field_labels = Some(LabelerServiceLabelsUnion::decode_cbor(&mut dec)?);
}
"policies" => {
let raw = crate::cbor::encode_value(&value)?;
let mut dec = crate::cbor::Decoder::new(&raw);
field_policies = Some(
crate::api::app::bsky::LabelerDefsLabelerPolicies::decode_cbor(&mut dec)?,
);
}
"createdAt" => {
if let crate::cbor::Value::Text(s) = value {
field_created_at = Some(
crate::syntax::Datetime::try_from(s)
.map_err(|e| crate::cbor::CborError::InvalidCbor(e.to_string()))?,
);
} else {
return Err(crate::cbor::CborError::InvalidCbor("expected text".into()));
}
}
"reasonTypes" => {
if let crate::cbor::Value::Array(items) = value {
for item in items {
if let crate::cbor::Value::Text(s) = item {
field_reason_types.push(s.to_string());
} else {
return Err(crate::cbor::CborError::InvalidCbor(
"expected text in array".into(),
));
}
}
} else {
return Err(crate::cbor::CborError::InvalidCbor("expected array".into()));
}
}
"subjectTypes" => {
if let crate::cbor::Value::Array(items) = value {
for item in items {
if let crate::cbor::Value::Text(s) = item {
field_subject_types.push(s.to_string());
} else {
return Err(crate::cbor::CborError::InvalidCbor(
"expected text in array".into(),
));
}
}
} else {
return Err(crate::cbor::CborError::InvalidCbor("expected array".into()));
}
}
"subjectCollections" => {
if let crate::cbor::Value::Array(items) = value {
for item in items {
if let crate::cbor::Value::Text(s) = item {
field_subject_collections.push(
crate::syntax::Nsid::try_from(s).map_err(|e| {
crate::cbor::CborError::InvalidCbor(e.to_string())
})?,
);
} else {
return Err(crate::cbor::CborError::InvalidCbor(
"expected text in array".into(),
));
}
}
} else {
return Err(crate::cbor::CborError::InvalidCbor("expected array".into()));
}
}
_ => {
let raw = crate::cbor::encode_value(&value)?;
extra_cbor.push((key.to_string(), raw));
}
}
}
Ok(LabelerService {
labels: field_labels,
policies: field_policies.ok_or_else(|| {
crate::cbor::CborError::InvalidCbor("missing required field 'policies'".into())
})?,
created_at: field_created_at.ok_or_else(|| {
crate::cbor::CborError::InvalidCbor("missing required field 'createdAt'".into())
})?,
reason_types: field_reason_types,
subject_types: field_subject_types,
subject_collections: field_subject_collections,
extra: std::collections::HashMap::new(),
extra_cbor,
})
}
}