use crate::PerformCrud; use actix_web::web::Data; use lemmy_api_common::{ blocking, check_community_ban, get_local_user_view_from_jwt, is_mod_or_admin, post::*, }; use lemmy_apub::ApubObjectType; use lemmy_db_queries::{source::post::Post_, Crud}; use lemmy_db_schema::source::{moderator::*, post::*}; use lemmy_db_views::post_view::PostView; use lemmy_utils::{ApiError, ConnectionId, LemmyError}; use lemmy_websocket::{messages::SendPost, LemmyContext, UserOperation}; #[async_trait::async_trait(?Send)] impl PerformCrud for DeletePost { type Response = PostResponse; async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &DeletePost = &self; let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?; let post_id = data.post_id; let orig_post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??; check_community_ban( local_user_view.person.id, orig_post.community_id, context.pool(), ) .await?; // Verify that only the creator can delete if !Post::is_post_creator(local_user_view.person.id, orig_post.creator_id) { return Err(ApiError::err("no_post_edit_allowed").into()); } // Update the post let post_id = data.post_id; let deleted = data.deleted; let updated_post = blocking(context.pool(), move |conn| { Post::update_deleted(conn, post_id, deleted) }) .await??; // apub updates if deleted { updated_post .send_delete(&local_user_view.person, context) .await?; } else { updated_post .send_undo_delete(&local_user_view.person, context) .await?; } // Refetch the post let post_id = data.post_id; let post_view = blocking(context.pool(), move |conn| { PostView::read(conn, post_id, Some(local_user_view.person.id)) }) .await??; let res = PostResponse { post_view }; context.chat_server().do_send(SendPost { op: UserOperation::DeletePost, post: res.clone(), websocket_id, }); Ok(res) } } #[async_trait::async_trait(?Send)] impl PerformCrud for RemovePost { type Response = PostResponse; async fn perform( &self, context: &Data, websocket_id: Option, ) -> Result { let data: &RemovePost = &self; let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?; let post_id = data.post_id; let orig_post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??; check_community_ban( local_user_view.person.id, orig_post.community_id, context.pool(), ) .await?; // Verify that only the mods can remove is_mod_or_admin( context.pool(), local_user_view.person.id, orig_post.community_id, ) .await?; // Update the post let post_id = data.post_id; let removed = data.removed; let updated_post = blocking(context.pool(), move |conn| { Post::update_removed(conn, post_id, removed) }) .await??; // Mod tables let form = ModRemovePostForm { mod_person_id: local_user_view.person.id, post_id: data.post_id, removed: Some(removed), reason: data.reason.to_owned(), }; blocking(context.pool(), move |conn| { ModRemovePost::create(conn, &form) }) .await??; // apub updates if removed { updated_post .send_remove(&local_user_view.person, context) .await?; } else { updated_post .send_undo_remove(&local_user_view.person, context) .await?; } // Refetch the post let post_id = data.post_id; let person_id = local_user_view.person.id; let post_view = blocking(context.pool(), move |conn| { PostView::read(conn, post_id, Some(person_id)) }) .await??; let res = PostResponse { post_view }; context.chat_server().do_send(SendPost { op: UserOperation::RemovePost, post: res.clone(), websocket_id, }); Ok(res) } }