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