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, get_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 =
  28       get_local_user_view_from_jwt(&self.auth, context.pool(), context.secret()).await?;
  29     let local_site = LocalSite::read(context.pool()).await?;
  30     let person_id = local_user_view.person.id;
  31     check_private_instance(&Some(local_user_view), &local_site)?;
  32
  33     let res = search_query_to_object_id(&self.q, context)
  34       .await
  35       .map_err(|e| e.with_message("couldnt_find_object"))?;
  36     convert_response(res, person_id, context.pool())
  37       .await
  38       .map_err(|e| e.with_message("couldnt_find_object"))
  39   }
  40 }
  41
  42 async fn convert_response(
  43   object: SearchableObjects,
  44   user_id: PersonId,
  45   pool: &DbPool,
  46 ) -> Result<ResolveObjectResponse, LemmyError> {
  47   use SearchableObjects::*;
  48   let removed_or_deleted;
  49   let mut res = ResolveObjectResponse::default();
  50   match object {
  51     Person(p) => {
  52       removed_or_deleted = p.deleted;
  53       res.person = Some(PersonView::read(pool, p.id).await?)
  54     }
  55     Community(c) => {
  56       removed_or_deleted = c.deleted || c.removed;
  57       res.community = Some(CommunityView::read(pool, c.id, Some(user_id), None).await?)
  58     }
  59     Post(p) => {
  60       removed_or_deleted = p.deleted || p.removed;
  61       res.post = Some(PostView::read(pool, p.id, Some(user_id), None).await?)
  62     }
  63     Comment(c) => {
  64       removed_or_deleted = c.deleted || c.removed;
  65       res.comment = Some(CommentView::read(pool, c.id, Some(user_id)).await?)
  66     }
  67   };
  68   // if the object was deleted from database, dont return it
  69   if removed_or_deleted {
  70     return Err(NotFound {}.into());
  71   }
  72   Ok(res)
  73 }