crates/apub/src/fetcher/objects.rs

   1 use crate::{fetcher::fetch::fetch_remote_object, objects::FromApub, NoteExt, PageExt};
   2 use anyhow::anyhow;
   3 use diesel::result::Error::NotFound;
   4 use lemmy_api_structs::blocking;
   5 use lemmy_db_queries::{ApubObject, Crud};
   6 use lemmy_db_schema::source::{comment::Comment, post::Post};
   7 use lemmy_utils::LemmyError;
   8 use lemmy_websocket::LemmyContext;
   9 use log::debug;
  10 use url::Url;
  11
  12 /// Gets a post by its apub ID. If it exists locally, it is returned directly. Otherwise it is
  13 /// pulled from its apub ID, inserted and returned.
  14 ///
  15 /// The parent community is also pulled if necessary. Comments are not pulled.
  16 pub(crate) async fn get_or_fetch_and_insert_post(
  17   post_ap_id: &Url,
  18   context: &LemmyContext,
  19   recursion_counter: &mut i32,
  20 ) -> Result<Post, LemmyError> {
  21   let post_ap_id_owned = post_ap_id.to_owned();
  22   let post = blocking(context.pool(), move |conn| {
  23     Post::read_from_apub_id(conn, &post_ap_id_owned.into())
  24   })
  25   .await?;
  26
  27   match post {
  28     Ok(p) => Ok(p),
  29     Err(NotFound {}) => {
  30       debug!("Fetching and creating remote post: {}", post_ap_id);
  31       let page =
  32         fetch_remote_object::<PageExt>(context.client(), post_ap_id, recursion_counter).await?;
  33       let post = Post::from_apub(&page, context, post_ap_id.to_owned(), recursion_counter).await?;
  34
  35       Ok(post)
  36     }
  37     Err(e) => Err(e.into()),
  38   }
  39 }
  40
  41 /// Gets a comment by its apub ID. If it exists locally, it is returned directly. Otherwise it is
  42 /// pulled from its apub ID, inserted and returned.
  43 ///
  44 /// The parent community, post and comment are also pulled if necessary.
  45 pub(crate) async fn get_or_fetch_and_insert_comment(
  46   comment_ap_id: &Url,
  47   context: &LemmyContext,
  48   recursion_counter: &mut i32,
  49 ) -> Result<Comment, LemmyError> {
  50   let comment_ap_id_owned = comment_ap_id.to_owned();
  51   let comment = blocking(context.pool(), move |conn| {
  52     Comment::read_from_apub_id(conn, &comment_ap_id_owned.into())
  53   })
  54   .await?;
  55
  56   match comment {
  57     Ok(p) => Ok(p),
  58     Err(NotFound {}) => {
  59       debug!(
  60         "Fetching and creating remote comment and its parents: {}",
  61         comment_ap_id
  62       );
  63       let comment =
  64         fetch_remote_object::<NoteExt>(context.client(), comment_ap_id, recursion_counter).await?;
  65       let comment = Comment::from_apub(
  66         &comment,
  67         context,
  68         comment_ap_id.to_owned(),
  69         recursion_counter,
  70       )
  71       .await?;
  72
  73       let post_id = comment.post_id;
  74       let post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
  75       if post.locked {
  76         return Err(anyhow!("Post is locked").into());
  77       }
  78
  79       Ok(comment)
  80     }
  81     Err(e) => Err(e.into()),
  82   }
  83 }