crates/api_crud/src/post/update.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   post::{EditPost, PostResponse},
   5   request::fetch_site_data,
   6   utils::{
   7     check_community_ban,
   8     check_community_deleted_or_removed,
   9     get_local_user_view_from_jwt,
  10     local_site_to_slur_regex,
  11   },
  12   websocket::{send::send_post_ws_message, UserOperationCrud},
  13   LemmyContext,
  14 };
  15 use lemmy_apub::protocol::activities::{
  16   create_or_update::page::CreateOrUpdatePage,
  17   CreateOrUpdateType,
  18 };
  19 use lemmy_db_schema::{
  20   source::{
  21     actor_language::CommunityLanguage,
  22     local_site::LocalSite,
  23     post::{Post, PostUpdateForm},
  24   },
  25   traits::Crud,
  26   utils::diesel_option_overwrite,
  27 };
  28 use lemmy_utils::{
  29   error::LemmyError,
  30   utils::{check_slurs_opt, clean_url_params, is_valid_post_title},
  31   ConnectionId,
  32 };
  33
  34 #[async_trait::async_trait(?Send)]
  35 impl PerformCrud for EditPost {
  36   type Response = PostResponse;
  37
  38   #[tracing::instrument(skip(context, websocket_id))]
  39   async fn perform(
  40     &self,
  41     context: &Data<LemmyContext>,
  42     websocket_id: Option<ConnectionId>,
  43   ) -> Result<PostResponse, LemmyError> {
  44     let data: &EditPost = self;
  45     let local_user_view =
  46       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  47     let local_site = LocalSite::read(context.pool()).await?;
  48
  49     let data_url = data.url.as_ref();
  50
  51     // TODO No good way to handle a clear.
  52     // Issue link: https://github.com/LemmyNet/lemmy/issues/2287
  53     let url = Some(data_url.map(clean_url_params).map(Into::into));
  54     let body = diesel_option_overwrite(&data.body);
  55
  56     let slur_regex = local_site_to_slur_regex(&local_site);
  57     check_slurs_opt(&data.name, &slur_regex)?;
  58     check_slurs_opt(&data.body, &slur_regex)?;
  59
  60     if let Some(name) = &data.name {
  61       if !is_valid_post_title(name) {
  62         return Err(LemmyError::from_message("invalid_post_title"));
  63       }
  64     }
  65
  66     let post_id = data.post_id;
  67     let orig_post = Post::read(context.pool(), post_id).await?;
  68
  69     check_community_ban(
  70       local_user_view.person.id,
  71       orig_post.community_id,
  72       context.pool(),
  73     )
  74     .await?;
  75     check_community_deleted_or_removed(orig_post.community_id, context.pool()).await?;
  76
  77     // Verify that only the creator can edit
  78     if !Post::is_post_creator(local_user_view.person.id, orig_post.creator_id) {
  79       return Err(LemmyError::from_message("no_post_edit_allowed"));
  80     }
  81
  82     // Fetch post links and Pictrs cached image
  83     let data_url = data.url.as_ref();
  84     let (metadata_res, thumbnail_url) =
  85       fetch_site_data(context.client(), context.settings(), data_url).await;
  86     let (embed_title, embed_description, embed_video_url) = metadata_res
  87       .map(|u| (Some(u.title), Some(u.description), Some(u.embed_video_url)))
  88       .unwrap_or_default();
  89
  90     let language_id = self.language_id;
  91     CommunityLanguage::is_allowed_community_language(
  92       context.pool(),
  93       language_id,
  94       orig_post.community_id,
  95     )
  96     .await?;
  97
  98     let post_form = PostUpdateForm::builder()
  99       .name(data.name.clone())
 100       .url(url)
 101       .body(body)
 102       .nsfw(data.nsfw)
 103       .embed_title(embed_title)
 104       .embed_description(embed_description)
 105       .embed_video_url(embed_video_url)
 106       .language_id(data.language_id)
 107       .thumbnail_url(Some(thumbnail_url))
 108       .build();
 109
 110     let post_id = data.post_id;
 111     let res = Post::update(context.pool(), post_id, &post_form).await;
 112     let updated_post: Post = match res {
 113       Ok(post) => post,
 114       Err(e) => {
 115         let err_type = if e.to_string() == "value too long for type character varying(200)" {
 116           "post_title_too_long"
 117         } else {
 118           "couldnt_update_post"
 119         };
 120
 121         return Err(LemmyError::from_error_message(e, err_type));
 122       }
 123     };
 124
 125     // Send apub update
 126     CreateOrUpdatePage::send(
 127       updated_post.into(),
 128       &local_user_view.person.clone().into(),
 129       CreateOrUpdateType::Update,
 130       context,
 131     )
 132     .await?;
 133
 134     send_post_ws_message(
 135       data.post_id,
 136       UserOperationCrud::EditPost,
 137       websocket_id,
 138       Some(local_user_view.person.id),
 139       context,
 140     )
 141     .await
 142   }
 143 }