crates/api_crud/src/private_message/read.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   context::LemmyContext,
   5   private_message::{GetPrivateMessages, PrivateMessagesResponse},
   6   utils::local_user_view_from_jwt,
   7 };
   8 use lemmy_db_views::private_message_view::PrivateMessageQuery;
   9 use lemmy_utils::error::LemmyError;
  10
  11 #[async_trait::async_trait(?Send)]
  12 impl PerformCrud for GetPrivateMessages {
  13   type Response = PrivateMessagesResponse;
  14
  15   #[tracing::instrument(skip(self, context))]
  16   async fn perform(
  17     &self,
  18     context: &Data<LemmyContext>,
  19   ) -> Result<PrivateMessagesResponse, LemmyError> {
  20     let data: &GetPrivateMessages = self;
  21     let local_user_view = local_user_view_from_jwt(data.auth.as_ref(), context).await?;
  22     let person_id = local_user_view.person.id;
  23
  24     let page = data.page;
  25     let limit = data.limit;
  26     let unread_only = data.unread_only;
  27     let mut messages = PrivateMessageQuery::builder()
  28       .pool(context.pool())
  29       .recipient_id(person_id)
  30       .page(page)
  31       .limit(limit)
  32       .unread_only(unread_only)
  33       .build()
  34       .list()
  35       .await?;
  36
  37     // Messages sent by ourselves should be marked as read. The `read` column in database is only
  38     // for the recipient, and shouldnt be exposed to sender.
  39     messages.iter_mut().for_each(|pmv| {
  40       if pmv.creator.id == person_id {
  41         pmv.private_message.read = true
  42       }
  43     });
  44
  45     Ok(PrivateMessagesResponse {
  46       private_messages: messages,
  47     })
  48   }
  49 }