lemmy/crates/apub/src/extensions/group_extensions.rs

use activitystreams::{
  collection::{CollectionExt, OrderedCollection},
  unparsed::UnparsedMutExt,
};
use activitystreams_ext::UnparsedExtension;
use lemmy_utils::LemmyError;
use serde::{Deserialize, Serialize};
use url::Url;

/// Activitystreams extension to allow (de)serializing additional Community field
/// `sensitive` (called 'nsfw' in Lemmy).
#[derive(Clone, Debug, Default, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct GroupExtension {
  pub sensitive: Option<bool>,
  pub moderators: Option<OrderedCollection>,
}

impl GroupExtension {
  pub fn new(sensitive: bool, moderators: Vec<Url>) -> Result<GroupExtension, LemmyError> {
    let mut mods = OrderedCollection::new();
    mods.set_total_items(moderators.len() as u64);
    mods.set_many_items(moderators);
    Ok(GroupExtension {
      sensitive: Some(sensitive),
      moderators: Some(mods),
    })
  }
}

impl<U> UnparsedExtension<U> for GroupExtension
where
  U: UnparsedMutExt,
{
  type Error = serde_json::Error;

  fn try_from_unparsed(unparsed_mut: &mut U) -> Result<Self, Self::Error> {
    Ok(GroupExtension {
      sensitive: unparsed_mut.remove("sensitive")?,
      moderators: unparsed_mut.remove("moderators")?,
    })
  }

  fn try_into_unparsed(self, unparsed_mut: &mut U) -> Result<(), Self::Error> {
    unparsed_mut.insert("sensitive", self.sensitive)?;
    unparsed_mut.insert("moderators", self.moderators)?;
    Ok(())
  }
}