crates/apub/src/objects/post.rs

   1 use crate::{
   2   extensions::{context::lemmy_context, page_extension::PageExtension},
   3   fetcher::user::get_or_fetch_and_upsert_user,
   4   objects::{
   5     check_object_domain,
   6     check_object_for_community_or_site_ban,
   7     create_tombstone,
   8     get_object_from_apub,
   9     get_source_markdown_value,
  10     get_to_community,
  11     set_content_and_source,
  12     FromApub,
  13     FromApubToForm,
  14     ToApub,
  15   },
  16   PageExt,
  17 };
  18 use activitystreams::{
  19   object::{kind::PageType, ApObject, Image, Page, Tombstone},
  20   prelude::*,
  21   public,
  22 };
  23 use activitystreams_ext::Ext1;
  24 use anyhow::Context;
  25 use lemmy_db_queries::{Crud, DbPool};
  26 use lemmy_db_schema::source::{
  27   community::Community,
  28   post::{Post, PostForm},
  29   user::User_,
  30 };
  31 use lemmy_structs::blocking;
  32 use lemmy_utils::{
  33   location_info,
  34   request::fetch_iframely_and_pictrs_data,
  35   utils::{check_slurs, convert_datetime, remove_slurs},
  36   LemmyError,
  37 };
  38 use lemmy_websocket::LemmyContext;
  39 use url::Url;
  40
  41 #[async_trait::async_trait(?Send)]
  42 impl ToApub for Post {
  43   type ApubType = PageExt;
  44
  45   // Turn a Lemmy post into an ActivityPub page that can be sent out over the network.
  46   async fn to_apub(&self, pool: &DbPool) -> Result<PageExt, LemmyError> {
  47     let mut page = ApObject::new(Page::new());
  48
  49     let creator_id = self.creator_id;
  50     let creator = blocking(pool, move |conn| User_::read(conn, creator_id)).await??;
  51
  52     let community_id = self.community_id;
  53     let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??;
  54
  55     page
  56       // Not needed when the Post is embedded in a collection (like for community outbox)
  57       // TODO: need to set proper context defining sensitive/commentsEnabled fields
  58       // https://git.asonix.dog/Aardwolf/activitystreams/issues/5
  59       .set_many_contexts(lemmy_context()?)
  60       .set_id(self.ap_id.to_owned().into_inner())
  61       .set_name(self.name.to_owned())
  62       // `summary` field for compatibility with lemmy v0.9.9 and older,
  63       // TODO: remove this after some time
  64       .set_summary(self.name.to_owned())
  65       .set_published(convert_datetime(self.published))
  66       .set_many_tos(vec![community.actor_id.into_inner(), public()])
  67       .set_attributed_to(creator.actor_id.into_inner());
  68
  69     if let Some(body) = &self.body {
  70       set_content_and_source(&mut page, &body)?;
  71     }
  72
  73     // TODO: hacky code because we get self.url == Some("")
  74     // https://github.com/LemmyNet/lemmy/issues/602
  75     let url = self.url.as_ref().filter(|u| !u.is_empty());
  76     if let Some(u) = url {
  77       page.set_url(Url::parse(u)?);
  78     }
  79
  80     if let Some(thumbnail_url) = &self.thumbnail_url {
  81       let mut image = Image::new();
  82       image.set_url(Url::parse(thumbnail_url)?);
  83       page.set_image(image.into_any_base()?);
  84     }
  85
  86     if let Some(u) = self.updated {
  87       page.set_updated(convert_datetime(u));
  88     }
  89
  90     let ext = PageExtension {
  91       comments_enabled: Some(!self.locked),
  92       sensitive: Some(self.nsfw),
  93       stickied: Some(self.stickied),
  94     };
  95     Ok(Ext1::new(page, ext))
  96   }
  97
  98   fn to_tombstone(&self) -> Result<Tombstone, LemmyError> {
  99     create_tombstone(
 100       self.deleted,
 101       self.ap_id.to_owned().into(),
 102       self.updated,
 103       PageType::Page,
 104     )
 105   }
 106 }
 107
 108 #[async_trait::async_trait(?Send)]
 109 impl FromApub for Post {
 110   type ApubType = PageExt;
 111
 112   /// Converts a `PageExt` to `PostForm`.
 113   ///
 114   /// If the post's community or creator are not known locally, these are also fetched.
 115   async fn from_apub(
 116     page: &PageExt,
 117     context: &LemmyContext,
 118     expected_domain: Url,
 119     request_counter: &mut i32,
 120   ) -> Result<Post, LemmyError> {
 121     let post: Post = get_object_from_apub(page, context, expected_domain, request_counter).await?;
 122     check_object_for_community_or_site_ban(page, post.community_id, context, request_counter)
 123       .await?;
 124     Ok(post)
 125   }
 126 }
 127
 128 #[async_trait::async_trait(?Send)]
 129 impl FromApubToForm<PageExt> for PostForm {
 130   async fn from_apub(
 131     page: &PageExt,
 132     context: &LemmyContext,
 133     expected_domain: Url,
 134     request_counter: &mut i32,
 135   ) -> Result<PostForm, LemmyError> {
 136     let ext = &page.ext_one;
 137     let creator_actor_id = page
 138       .inner
 139       .attributed_to()
 140       .as_ref()
 141       .context(location_info!())?
 142       .as_single_xsd_any_uri()
 143       .context(location_info!())?;
 144
 145     let creator = get_or_fetch_and_upsert_user(creator_actor_id, context, request_counter).await?;
 146
 147     let community = get_to_community(page, context, request_counter).await?;
 148
 149     let thumbnail_url = match &page.inner.image() {
 150       Some(any_image) => Image::from_any_base(
 151         any_image
 152           .to_owned()
 153           .as_one()
 154           .context(location_info!())?
 155           .to_owned(),
 156       )?
 157       .context(location_info!())?
 158       .url()
 159       .context(location_info!())?
 160       .as_single_xsd_any_uri()
 161       .map(|u| u.to_string()),
 162       None => None,
 163     };
 164     let url = page
 165       .inner
 166       .url()
 167       .map(|u| u.as_single_xsd_any_uri())
 168       .flatten()
 169       .map(|s| s.to_string());
 170
 171     let (iframely_title, iframely_description, iframely_html, pictrs_thumbnail) =
 172       if let Some(url) = &url {
 173         fetch_iframely_and_pictrs_data(context.client(), Some(url.to_owned())).await
 174       } else {
 175         (None, None, None, thumbnail_url)
 176       };
 177
 178     let name = page
 179       .inner
 180       .name()
 181       .map(|s| s.map(|s2| s2.to_owned()))
 182       // The following is for compatibility with lemmy v0.9.9 and older
 183       // TODO: remove it after some time (along with the map above)
 184       .or_else(|| page.inner.summary().map(|s| s.to_owned()))
 185       .context(location_info!())?
 186       .as_single_xsd_string()
 187       .context(location_info!())?
 188       .to_string();
 189     let body = get_source_markdown_value(page)?;
 190
 191     check_slurs(&name)?;
 192     let body_slurs_removed = body.map(|b| remove_slurs(&b));
 193     Ok(PostForm {
 194       name,
 195       url,
 196       body: body_slurs_removed,
 197       creator_id: creator.id,
 198       community_id: community.id,
 199       removed: None,
 200       locked: ext.comments_enabled.map(|e| !e),
 201       published: page
 202         .inner
 203         .published()
 204         .as_ref()
 205         .map(|u| u.to_owned().naive_local()),
 206       updated: page
 207         .inner
 208         .updated()
 209         .as_ref()
 210         .map(|u| u.to_owned().naive_local()),
 211       deleted: None,
 212       nsfw: ext.sensitive.unwrap_or(false),
 213       stickied: ext.stickied.or(Some(false)),
 214       embed_title: iframely_title,
 215       embed_description: iframely_description,
 216       embed_html: iframely_html,
 217       thumbnail_url: pictrs_thumbnail,
 218       ap_id: Some(check_object_domain(page, expected_domain)?),
 219       local: false,
 220     })
 221   }
 222 }