crates/apub/src/api/resolve_object.rs

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