crates/api/src/post/lock.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   post::{LockPost, PostResponse},
   5   utils::{
   6     blocking,
   7     check_community_ban,
   8     check_community_deleted_or_removed,
   9     get_local_user_view_from_jwt,
  10     is_mod_or_admin,
  11   },
  12 };
  13 use lemmy_apub::{
  14   objects::post::ApubPost,
  15   protocol::activities::{create_or_update::post::CreateOrUpdatePost, CreateOrUpdateType},
  16 };
  17 use lemmy_db_schema::{
  18   source::{
  19     moderator::{ModLockPost, ModLockPostForm},
  20     post::{Post, PostUpdateForm},
  21   },
  22   traits::Crud,
  23 };
  24 use lemmy_utils::{error::LemmyError, ConnectionId};
  25 use lemmy_websocket::{send::send_post_ws_message, LemmyContext, UserOperation};
  26
  27 #[async_trait::async_trait(?Send)]
  28 impl Perform for LockPost {
  29   type Response = PostResponse;
  30
  31   #[tracing::instrument(skip(context, websocket_id))]
  32   async fn perform(
  33     &self,
  34     context: &Data<LemmyContext>,
  35     websocket_id: Option<ConnectionId>,
  36   ) -> Result<PostResponse, LemmyError> {
  37     let data: &LockPost = self;
  38     let local_user_view =
  39       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  40
  41     let post_id = data.post_id;
  42     let orig_post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
  43
  44     check_community_ban(
  45       local_user_view.person.id,
  46       orig_post.community_id,
  47       context.pool(),
  48     )
  49     .await?;
  50     check_community_deleted_or_removed(orig_post.community_id, context.pool()).await?;
  51
  52     // Verify that only the mods can lock
  53     is_mod_or_admin(
  54       context.pool(),
  55       local_user_view.person.id,
  56       orig_post.community_id,
  57     )
  58     .await?;
  59
  60     // Update the post
  61     let post_id = data.post_id;
  62     let locked = data.locked;
  63     let updated_post: ApubPost = blocking(context.pool(), move |conn| {
  64       Post::update(
  65         conn,
  66         post_id,
  67         &PostUpdateForm::builder().locked(Some(locked)).build(),
  68       )
  69     })
  70     .await??
  71     .into();
  72
  73     // Mod tables
  74     let form = ModLockPostForm {
  75       mod_person_id: local_user_view.person.id,
  76       post_id: data.post_id,
  77       locked: Some(locked),
  78     };
  79     blocking(context.pool(), move |conn| ModLockPost::create(conn, &form)).await??;
  80
  81     // apub updates
  82     CreateOrUpdatePost::send(
  83       updated_post,
  84       &local_user_view.person.clone().into(),
  85       CreateOrUpdateType::Update,
  86       context,
  87     )
  88     .await?;
  89
  90     send_post_ws_message(
  91       data.post_id,
  92       UserOperation::LockPost,
  93       websocket_id,
  94       Some(local_user_view.person.id),
  95       context,
  96     )
  97     .await
  98   }
  99 }