crates/api/src/site/resolve_object.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use diesel::NotFound;
   4 use lemmy_api_common::{
   5   site::{ResolveObject, ResolveObjectResponse},
   6   utils::{blocking, check_private_instance, get_local_user_view_from_jwt_opt},
   7 };
   8 use lemmy_apub::fetcher::search::{search_query_to_object_id, SearchableObjects};
   9 use lemmy_db_schema::{newtypes::PersonId, source::local_site::LocalSite, utils::DbPool};
  10 use lemmy_db_views::structs::{CommentView, PostView};
  11 use lemmy_db_views_actor::structs::{CommunityView, PersonViewSafe};
  12 use lemmy_utils::{error::LemmyError, ConnectionId};
  13 use lemmy_websocket::LemmyContext;
  14
  15 #[async_trait::async_trait(?Send)]
  16 impl Perform for ResolveObject {
  17   type Response = ResolveObjectResponse;
  18
  19   #[tracing::instrument(skip(context, _websocket_id))]
  20   async fn perform(
  21     &self,
  22     context: &Data<LemmyContext>,
  23     _websocket_id: Option<ConnectionId>,
  24   ) -> Result<ResolveObjectResponse, LemmyError> {
  25     let local_user_view =
  26       get_local_user_view_from_jwt_opt(self.auth.as_ref(), context.pool(), context.secret())
  27         .await?;
  28     let local_site = blocking(context.pool(), LocalSite::read).await??;
  29     check_private_instance(&local_user_view, &local_site)?;
  30
  31     let res = search_query_to_object_id(&self.q, local_user_view.is_none(), context)
  32       .await
  33       .map_err(|e| e.with_message("couldnt_find_object"))?;
  34     convert_response(res, local_user_view.map(|l| l.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: Option<PersonId>,
  43   pool: &DbPool,
  44 ) -> Result<ResolveObjectResponse, LemmyError> {
  45   let removed_or_deleted;
  46   let mut res = ResolveObjectResponse {
  47     comment: None,
  48     post: None,
  49     community: None,
  50     person: None,
  51   };
  52   use SearchableObjects::*;
  53   match object {
  54     Person(p) => {
  55       removed_or_deleted = p.deleted;
  56       res.person = Some(blocking(pool, move |conn| PersonViewSafe::read(conn, p.id)).await??)
  57     }
  58     Community(c) => {
  59       removed_or_deleted = c.deleted || c.removed;
  60       res.community =
  61         Some(blocking(pool, move |conn| CommunityView::read(conn, c.id, user_id)).await??)
  62     }
  63     Post(p) => {
  64       removed_or_deleted = p.deleted || p.removed;
  65       res.post = Some(blocking(pool, move |conn| PostView::read(conn, p.id, user_id)).await??)
  66     }
  67     Comment(c) => {
  68       removed_or_deleted = c.deleted || c.removed;
  69       res.comment = Some(blocking(pool, move |conn| CommentView::read(conn, c.id, user_id)).await??)
  70     }
  71   };
  72   // if the object was deleted from database, dont return it
  73   if removed_or_deleted {
  74     return Err(NotFound {}.into());
  75   }
  76   Ok(res)
  77 }