crates/apub/src/api/resolve_object.rs

   1 use crate::{
   2   api::PerformApub,
   3   fetcher::search::{search_query_to_object_id, SearchableObjects},
   4 };
   5 use activitypub_federation::config::Data;
   6 use diesel::NotFound;
   7 use lemmy_api_common::{
   8   context::LemmyContext,
   9   site::{ResolveObject, ResolveObjectResponse},
  10   utils::{check_private_instance, local_user_view_from_jwt},
  11 };
  12 use lemmy_db_schema::{newtypes::PersonId, source::local_site::LocalSite, utils::DbPool};
  13 use lemmy_db_views::structs::{CommentView, PostView};
  14 use lemmy_db_views_actor::structs::{CommunityView, PersonView};
  15 use lemmy_utils::error::LemmyError;
  16
  17 #[async_trait::async_trait]
  18 impl PerformApub for ResolveObject {
  19   type Response = ResolveObjectResponse;
  20
  21   #[tracing::instrument(skip(context))]
  22   async fn perform(
  23     &self,
  24     context: &Data<LemmyContext>,
  25   ) -> Result<ResolveObjectResponse, LemmyError> {
  26     let local_user_view = local_user_view_from_jwt(&self.auth, context).await?;
  27     let local_site = LocalSite::read(context.pool()).await?;
  28     let person_id = local_user_view.person.id;
  29     check_private_instance(&Some(local_user_view), &local_site)?;
  30
  31     let res = search_query_to_object_id(&self.q, context)
  32       .await
  33       .map_err(|e| e.with_message("couldnt_find_object"))?;
  34     convert_response(res, person_id, context.pool())
  35       .await
  36       .map_err(|e| e.with_message("couldnt_find_object"))
  37   }
  38 }
  39
  40 async fn convert_response(
  41   object: SearchableObjects,
  42   user_id: PersonId,
  43   pool: &DbPool,
  44 ) -> Result<ResolveObjectResponse, LemmyError> {
  45   use SearchableObjects::*;
  46   let removed_or_deleted;
  47   let mut res = ResolveObjectResponse::default();
  48   match object {
  49     Person(p) => {
  50       removed_or_deleted = p.deleted;
  51       res.person = Some(PersonView::read(pool, p.id).await?)
  52     }
  53     Community(c) => {
  54       removed_or_deleted = c.deleted || c.removed;
  55       res.community = Some(CommunityView::read(pool, c.id, Some(user_id), None).await?)
  56     }
  57     Post(p) => {
  58       removed_or_deleted = p.deleted || p.removed;
  59       res.post = Some(PostView::read(pool, p.id, Some(user_id), None).await?)
  60     }
  61     Comment(c) => {
  62       removed_or_deleted = c.deleted || c.removed;
  63       res.comment = Some(CommentView::read(pool, c.id, Some(user_id)).await?)
  64     }
  65   };
  66   // if the object was deleted from database, dont return it
  67   if removed_or_deleted {
  68     return Err(NotFound {}.into());
  69   }
  70   Ok(res)
  71 }