crates/api_crud/src/comment/list.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   comment::{GetComments, GetCommentsResponse},
   5   utils::{
   6     blocking,
   7     check_private_instance,
   8     get_local_user_view_from_jwt_opt,
   9     listing_type_with_site_default,
  10   },
  11 };
  12 use lemmy_apub::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
  13 use lemmy_db_schema::{
  14   source::{comment::Comment, community::Community, local_site::LocalSite},
  15   traits::{Crud, DeleteableOrRemoveable},
  16 };
  17 use lemmy_db_views::comment_view::CommentQuery;
  18 use lemmy_utils::{error::LemmyError, ConnectionId};
  19 use lemmy_websocket::LemmyContext;
  20
  21 #[async_trait::async_trait(?Send)]
  22 impl PerformCrud for GetComments {
  23   type Response = GetCommentsResponse;
  24
  25   #[tracing::instrument(skip(context, _websocket_id))]
  26   async fn perform(
  27     &self,
  28     context: &Data<LemmyContext>,
  29     _websocket_id: Option<ConnectionId>,
  30   ) -> Result<GetCommentsResponse, LemmyError> {
  31     let data: &GetComments = self;
  32     let local_user_view =
  33       get_local_user_view_from_jwt_opt(data.auth.as_ref(), context.pool(), context.secret())
  34         .await?;
  35     let local_site = blocking(context.pool(), LocalSite::read).await??;
  36     check_private_instance(&local_user_view, &local_site)?;
  37
  38     let community_id = data.community_id;
  39     let listing_type = listing_type_with_site_default(data.type_, &local_site)?;
  40
  41     let community_actor_id = if let Some(name) = &data.community_name {
  42       resolve_actor_identifier::<ApubCommunity, Community>(name, context, true)
  43         .await
  44         .ok()
  45         .map(|c| c.actor_id)
  46     } else {
  47       None
  48     };
  49     let sort = data.sort;
  50     let max_depth = data.max_depth;
  51     let saved_only = data.saved_only;
  52     let page = data.page;
  53     let limit = data.limit;
  54     let parent_id = data.parent_id;
  55
  56     // If a parent_id is given, fetch the comment to get the path
  57     let parent_path = if let Some(parent_id) = parent_id {
  58       Some(
  59         blocking(context.pool(), move |conn| Comment::read(conn, parent_id))
  60           .await??
  61           .path,
  62       )
  63     } else {
  64       None
  65     };
  66
  67     let parent_path_cloned = parent_path.to_owned();
  68     let post_id = data.post_id;
  69     let local_user = local_user_view.map(|l| l.local_user);
  70     let mut comments = blocking(context.pool(), move |conn| {
  71       CommentQuery::builder()
  72         .conn(conn)
  73         .listing_type(Some(listing_type))
  74         .sort(sort)
  75         .max_depth(max_depth)
  76         .saved_only(saved_only)
  77         .community_id(community_id)
  78         .community_actor_id(community_actor_id)
  79         .parent_path(parent_path_cloned)
  80         .post_id(post_id)
  81         .local_user(local_user.as_ref())
  82         .page(page)
  83         .limit(limit)
  84         .build()
  85         .list()
  86     })
  87     .await?
  88     .map_err(|e| LemmyError::from_error_message(e, "couldnt_get_comments"))?;
  89
  90     // Blank out deleted or removed info
  91     for cv in comments
  92       .iter_mut()
  93       .filter(|cv| cv.comment.deleted || cv.comment.removed)
  94     {
  95       cv.comment = cv.to_owned().comment.blank_out_deleted_or_removed_info();
  96     }
  97
  98     Ok(GetCommentsResponse { comments })
  99   }
 100 }