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 actix_web::web::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_opt},
  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, PersonViewSafe};
  15 use lemmy_utils::{error::LemmyError, ConnectionId};
  16
  17 #[async_trait::async_trait(?Send)]
  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_opt(self.auth.as_ref(), context.pool(), context.secret())
  29         .await?;
  30     let local_site = LocalSite::read(context.pool()).await?;
  31     check_private_instance(&local_user_view, &local_site)?;
  32
  33     // In release builds only allow for authenticated users to fetch remote objects
  34     let local_only = local_user_view.is_none() && cfg!(not(debug_assertions));
  35     let res = search_query_to_object_id(&self.q, local_only, context)
  36       .await
  37       .map_err(|e| e.with_message("couldnt_find_object"))?;
  38     convert_response(res, local_user_view.map(|l| l.person.id), context.pool())
  39       .await
  40       .map_err(|e| e.with_message("couldnt_find_object"))
  41   }
  42 }
  43
  44 async fn convert_response(
  45   object: SearchableObjects,
  46   user_id: Option<PersonId>,
  47   pool: &DbPool,
  48 ) -> Result<ResolveObjectResponse, LemmyError> {
  49   use SearchableObjects::*;
  50   let removed_or_deleted;
  51   let mut res = ResolveObjectResponse::default();
  52   match object {
  53     Person(p) => {
  54       removed_or_deleted = p.deleted;
  55       res.person = Some(PersonViewSafe::read(pool, p.id).await?)
  56     }
  57     Community(c) => {
  58       removed_or_deleted = c.deleted || c.removed;
  59       res.community = Some(CommunityView::read(pool, c.id, user_id, None).await?)
  60     }
  61     Post(p) => {
  62       removed_or_deleted = p.deleted || p.removed;
  63       res.post = Some(PostView::read(pool, p.id, user_id, None).await?)
  64     }
  65     Comment(c) => {
  66       removed_or_deleted = c.deleted || c.removed;
  67       res.comment = Some(CommentView::read(pool, c.id, user_id).await?)
  68     }
  69   };
  70   // if the object was deleted from database, dont return it
  71   if removed_or_deleted {
  72     return Err(NotFound {}.into());
  73   }
  74   Ok(res)
  75 }