]> Untitled Git - lemmy.git/blob - crates/api_crud/src/private_message/delete.rs
Make functions work with both connection and pool (#3420)
[lemmy.git] / crates / api_crud / src / private_message / delete.rs
1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   context::LemmyContext,
5   private_message::{DeletePrivateMessage, PrivateMessageResponse},
6   utils::local_user_view_from_jwt,
7 };
8 use lemmy_db_schema::{
9   source::private_message::{PrivateMessage, PrivateMessageUpdateForm},
10   traits::Crud,
11 };
12 use lemmy_db_views::structs::PrivateMessageView;
13 use lemmy_utils::error::{LemmyError, LemmyErrorExt, LemmyErrorType};
14
15 #[async_trait::async_trait(?Send)]
16 impl PerformCrud for DeletePrivateMessage {
17   type Response = PrivateMessageResponse;
18
19   #[tracing::instrument(skip(self, context))]
20   async fn perform(
21     &self,
22     context: &Data<LemmyContext>,
23   ) -> Result<PrivateMessageResponse, LemmyError> {
24     let data: &DeletePrivateMessage = self;
25     let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
26
27     // Checking permissions
28     let private_message_id = data.private_message_id;
29     let orig_private_message =
30       PrivateMessage::read(&mut context.pool(), private_message_id).await?;
31     if local_user_view.person.id != orig_private_message.creator_id {
32       return Err(LemmyErrorType::EditPrivateMessageNotAllowed)?;
33     }
34
35     // Doing the update
36     let private_message_id = data.private_message_id;
37     let deleted = data.deleted;
38     PrivateMessage::update(
39       &mut context.pool(),
40       private_message_id,
41       &PrivateMessageUpdateForm::builder()
42         .deleted(Some(deleted))
43         .build(),
44     )
45     .await
46     .with_lemmy_type(LemmyErrorType::CouldntUpdatePrivateMessage)?;
47
48     let view = PrivateMessageView::read(&mut context.pool(), private_message_id).await?;
49     Ok(PrivateMessageResponse {
50       private_message_view: view,
51     })
52   }
53 }