crates/api_crud/src/comment/list.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   blocking,
   5   check_private_instance,
   6   comment::*,
   7   get_local_user_view_from_jwt_opt,
   8 };
   9 use lemmy_apub::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
  10 use lemmy_db_schema::{
  11   from_opt_str_to_opt_enum,
  12   source::community::Community,
  13   traits::DeleteableOrRemoveable,
  14   ListingType,
  15   SortType,
  16 };
  17 use lemmy_db_views::comment_view::CommentQueryBuilder;
  18 use lemmy_utils::{ConnectionId, LemmyError};
  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
  36     check_private_instance(&local_user_view, context.pool()).await?;
  37
  38     let show_bot_accounts = local_user_view
  39       .as_ref()
  40       .map(|t| t.local_user.show_bot_accounts);
  41     let person_id = local_user_view.map(|u| u.person.id);
  42
  43     let sort: Option<SortType> = from_opt_str_to_opt_enum(&data.sort);
  44     let listing_type: Option<ListingType> = from_opt_str_to_opt_enum(&data.type_);
  45
  46     let community_id = data.community_id;
  47     let community_actor_id = if let Some(name) = &data.community_name {
  48       resolve_actor_identifier::<ApubCommunity, Community>(name, context)
  49         .await
  50         .ok()
  51         .map(|c| c.actor_id)
  52     } else {
  53       None
  54     };
  55     let saved_only = data.saved_only;
  56     let page = data.page;
  57     let limit = data.limit;
  58     let mut comments = blocking(context.pool(), move |conn| {
  59       CommentQueryBuilder::create(conn)
  60         .listing_type(listing_type)
  61         .sort(sort)
  62         .saved_only(saved_only)
  63         .community_id(community_id)
  64         .community_actor_id(community_actor_id)
  65         .my_person_id(person_id)
  66         .show_bot_accounts(show_bot_accounts)
  67         .page(page)
  68         .limit(limit)
  69         .list()
  70     })
  71     .await?
  72     .map_err(|e| LemmyError::from_error_message(e, "couldnt_get_comments"))?;
  73
  74     // Blank out deleted or removed info
  75     for cv in comments
  76       .iter_mut()
  77       .filter(|cv| cv.comment.deleted || cv.comment.removed)
  78     {
  79       cv.comment = cv.to_owned().comment.blank_out_deleted_or_removed_info();
  80     }
  81
  82     Ok(GetCommentsResponse { comments })
  83   }
  84 }