crates/api_crud/src/private_message/read.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   private_message::{GetPrivateMessages, PrivateMessagesResponse},
   5   utils::{blocking, get_local_user_view_from_jwt},
   6 };
   7 use lemmy_db_schema::traits::DeleteableOrRemoveable;
   8 use lemmy_db_views::private_message_view::PrivateMessageQuery;
   9 use lemmy_utils::{error::LemmyError, ConnectionId};
  10 use lemmy_websocket::LemmyContext;
  11
  12 #[async_trait::async_trait(?Send)]
  13 impl PerformCrud for GetPrivateMessages {
  14   type Response = PrivateMessagesResponse;
  15
  16   #[tracing::instrument(skip(self, context, _websocket_id))]
  17   async fn perform(
  18     &self,
  19     context: &Data<LemmyContext>,
  20     _websocket_id: Option<ConnectionId>,
  21   ) -> Result<PrivateMessagesResponse, LemmyError> {
  22     let data: &GetPrivateMessages = self;
  23     let local_user_view =
  24       get_local_user_view_from_jwt(data.auth.as_ref(), context.pool(), context.secret()).await?;
  25     let person_id = local_user_view.person.id;
  26
  27     let page = data.page;
  28     let limit = data.limit;
  29     let unread_only = data.unread_only;
  30     let mut messages = blocking(context.pool(), move |conn| {
  31       PrivateMessageQuery::builder()
  32         .conn(conn)
  33         .recipient_id(person_id)
  34         .page(page)
  35         .limit(limit)
  36         .unread_only(unread_only)
  37         .build()
  38         .list()
  39     })
  40     .await??;
  41
  42     // Messages sent by ourselves should be marked as read. The `read` column in database is only
  43     // for the recipient, and shouldnt be exposed to sender.
  44     messages.iter_mut().for_each(|pmv| {
  45       if pmv.creator.id == person_id {
  46         pmv.private_message.read = true
  47       }
  48     });
  49
  50     // Blank out deleted or removed info
  51     for pmv in messages
  52       .iter_mut()
  53       .filter(|pmv| pmv.private_message.deleted)
  54     {
  55       pmv.private_message = pmv
  56         .to_owned()
  57         .private_message
  58         .blank_out_deleted_or_removed_info();
  59     }
  60
  61     Ok(PrivateMessagesResponse {
  62       private_messages: messages,
  63     })
  64   }
  65 }