use crate::PerformCrud; use actix_web::web::Data; use lemmy_api_common::{ private_message::{GetPrivateMessages, PrivateMessagesResponse}, utils::{blocking, get_local_user_view_from_jwt}, }; use lemmy_db_schema::traits::DeleteableOrRemoveable; use lemmy_db_views::private_message_view::PrivateMessageQuery; use lemmy_utils::{error::LemmyError, ConnectionId}; use lemmy_websocket::LemmyContext; #[async_trait::async_trait(?Send)] impl PerformCrud for GetPrivateMessages { type Response = PrivateMessagesResponse; #[tracing::instrument(skip(self, context, _websocket_id))] async fn perform( &self, context: &Data, _websocket_id: Option, ) -> Result { let data: &GetPrivateMessages = self; let local_user_view = get_local_user_view_from_jwt(data.auth.as_ref(), context.pool(), context.secret()).await?; let person_id = local_user_view.person.id; let page = data.page; let limit = data.limit; let unread_only = data.unread_only; let mut messages = blocking(context.pool(), move |conn| { PrivateMessageQuery::builder() .conn(conn) .recipient_id(person_id) .page(page) .limit(limit) .unread_only(unread_only) .build() .list() }) .await??; // Messages sent by ourselves should be marked as read. The `read` column in database is only // for the recipient, and shouldnt be exposed to sender. messages.iter_mut().for_each(|pmv| { if pmv.creator.id == person_id { pmv.private_message.read = true } }); // Blank out deleted or removed info for pmv in messages .iter_mut() .filter(|pmv| pmv.private_message.deleted) { pmv.private_message = pmv .to_owned() .private_message .blank_out_deleted_or_removed_info(); } Ok(PrivateMessagesResponse { private_messages: messages, }) } }