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(
  34         &page,
  35         context,
  36         Some(post_ap_id.to_owned()),
  37         recursion_counter,
  38       )
  39       .await?;
  40
  41       Ok(post)
  42     }
  43     Err(e) => Err(e.into()),
  44   }
  45 }
  46
  47 /// Gets a comment by its apub ID. If it exists locally, it is returned directly. Otherwise it is
  48 /// pulled from its apub ID, inserted and returned.
  49 ///
  50 /// The parent community, post and comment are also pulled if necessary.
  51 pub(crate) async fn get_or_fetch_and_insert_comment(
  52   comment_ap_id: &Url,
  53   context: &LemmyContext,
  54   recursion_counter: &mut i32,
  55 ) -> Result<Comment, LemmyError> {
  56   let comment_ap_id_owned = comment_ap_id.to_owned();
  57   let comment = blocking(context.pool(), move |conn| {
  58     Comment::read_from_apub_id(conn, &comment_ap_id_owned.into())
  59   })
  60   .await?;
  61
  62   match comment {
  63     Ok(p) => Ok(p),
  64     Err(NotFound {}) => {
  65       debug!(
  66         "Fetching and creating remote comment and its parents: {}",
  67         comment_ap_id
  68       );
  69       let comment =
  70         fetch_remote_object::<NoteExt>(context.client(), comment_ap_id, recursion_counter).await?;
  71       let comment = Comment::from_apub(
  72         &comment,
  73         context,
  74         Some(comment_ap_id.to_owned()),
  75         recursion_counter,
  76       )
  77       .await?;
  78
  79       let post_id = comment.post_id;
  80       let post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
  81       if post.locked {
  82         return Err(anyhow!("Post is locked").into());
  83       }
  84
  85       Ok(comment)
  86     }
  87     Err(e) => Err(e.into()),
  88   }
  89 }