crates/api_crud/src/post/create.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   context::LemmyContext,
   5   post::{CreatePost, PostResponse},
   6   request::fetch_site_data,
   7   utils::{
   8     check_community_ban,
   9     check_community_deleted_or_removed,
  10     generate_local_apub_endpoint,
  11     get_local_user_view_from_jwt,
  12     honeypot_check,
  13     local_site_to_slur_regex,
  14     mark_post_as_read,
  15     EndpointType,
  16   },
  17   websocket::UserOperationCrud,
  18 };
  19 use lemmy_db_schema::{
  20   impls::actor_language::default_post_language,
  21   source::{
  22     actor_language::CommunityLanguage,
  23     community::Community,
  24     local_site::LocalSite,
  25     post::{Post, PostInsertForm, PostLike, PostLikeForm, PostUpdateForm},
  26   },
  27   traits::{Crud, Likeable},
  28 };
  29 use lemmy_db_views_actor::structs::CommunityView;
  30 use lemmy_utils::{
  31   error::LemmyError,
  32   utils::{
  33     slurs::{check_slurs, check_slurs_opt},
  34     validation::{clean_url_params, is_valid_body_field, is_valid_post_title},
  35   },
  36   ConnectionId,
  37 };
  38 use tracing::{warn, Instrument};
  39 use url::Url;
  40 use webmention::{Webmention, WebmentionError};
  41
  42 #[async_trait::async_trait(?Send)]
  43 impl PerformCrud for CreatePost {
  44   type Response = PostResponse;
  45
  46   #[tracing::instrument(skip(context, websocket_id))]
  47   async fn perform(
  48     &self,
  49     context: &Data<LemmyContext>,
  50     websocket_id: Option<ConnectionId>,
  51   ) -> Result<PostResponse, LemmyError> {
  52     let data: &CreatePost = self;
  53     let local_user_view =
  54       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  55     let local_site = LocalSite::read(context.pool()).await?;
  56
  57     let slur_regex = local_site_to_slur_regex(&local_site);
  58     check_slurs(&data.name, &slur_regex)?;
  59     check_slurs_opt(&data.body, &slur_regex)?;
  60     honeypot_check(&data.honeypot)?;
  61
  62     let data_url = data.url.as_ref();
  63     let url = data_url.map(clean_url_params).map(Into::into); // TODO no good way to handle a "clear"
  64
  65     is_valid_post_title(&data.name)?;
  66     is_valid_body_field(&data.body)?;
  67
  68     check_community_ban(local_user_view.person.id, data.community_id, context.pool()).await?;
  69     check_community_deleted_or_removed(data.community_id, context.pool()).await?;
  70
  71     let community_id = data.community_id;
  72     let community = Community::read(context.pool(), community_id).await?;
  73     if community.posting_restricted_to_mods {
  74       let community_id = data.community_id;
  75       let is_mod = CommunityView::is_mod_or_admin(
  76         context.pool(),
  77         local_user_view.local_user.person_id,
  78         community_id,
  79       )
  80       .await?;
  81       if !is_mod {
  82         return Err(LemmyError::from_message("only_mods_can_post_in_community"));
  83       }
  84     }
  85
  86     // Fetch post links and pictrs cached image
  87     let (metadata_res, thumbnail_url) =
  88       fetch_site_data(context.client(), context.settings(), data_url).await;
  89     let (embed_title, embed_description, embed_video_url) = metadata_res
  90       .map(|u| (u.title, u.description, u.embed_video_url))
  91       .unwrap_or_default();
  92
  93     let language_id = match data.language_id {
  94       Some(lid) => Some(lid),
  95       None => {
  96         default_post_language(context.pool(), community_id, local_user_view.local_user.id).await?
  97       }
  98     };
  99     CommunityLanguage::is_allowed_community_language(context.pool(), language_id, community_id)
 100       .await?;
 101
 102     let post_form = PostInsertForm::builder()
 103       .name(data.name.trim().to_owned())
 104       .url(url)
 105       .body(data.body.clone())
 106       .community_id(data.community_id)
 107       .creator_id(local_user_view.person.id)
 108       .nsfw(data.nsfw)
 109       .embed_title(embed_title)
 110       .embed_description(embed_description)
 111       .embed_video_url(embed_video_url)
 112       .language_id(language_id)
 113       .thumbnail_url(thumbnail_url)
 114       .build();
 115
 116     let inserted_post = match Post::create(context.pool(), &post_form).await {
 117       Ok(post) => post,
 118       Err(e) => {
 119         let err_type = if e.to_string() == "value too long for type character varying(200)" {
 120           "post_title_too_long"
 121         } else {
 122           "couldnt_create_post"
 123         };
 124
 125         return Err(LemmyError::from_error_message(e, err_type));
 126       }
 127     };
 128
 129     let inserted_post_id = inserted_post.id;
 130     let protocol_and_hostname = context.settings().get_protocol_and_hostname();
 131     let apub_id = generate_local_apub_endpoint(
 132       EndpointType::Post,
 133       &inserted_post_id.to_string(),
 134       &protocol_and_hostname,
 135     )?;
 136     let updated_post = Post::update(
 137       context.pool(),
 138       inserted_post_id,
 139       &PostUpdateForm::builder().ap_id(Some(apub_id)).build(),
 140     )
 141     .await
 142     .map_err(|e| LemmyError::from_error_message(e, "couldnt_create_post"))?;
 143
 144     // They like their own post by default
 145     let person_id = local_user_view.person.id;
 146     let post_id = inserted_post.id;
 147     let like_form = PostLikeForm {
 148       post_id,
 149       person_id,
 150       score: 1,
 151     };
 152
 153     PostLike::like(context.pool(), &like_form)
 154       .await
 155       .map_err(|e| LemmyError::from_error_message(e, "couldnt_like_post"))?;
 156
 157     // Mark the post as read
 158     mark_post_as_read(person_id, post_id, context.pool()).await?;
 159
 160     if let Some(url) = &updated_post.url {
 161       let mut webmention =
 162         Webmention::new::<Url>(updated_post.ap_id.clone().into(), url.clone().into())?;
 163       webmention.set_checked(true);
 164       match webmention
 165         .send()
 166         .instrument(tracing::info_span!("Sending webmention"))
 167         .await
 168       {
 169         Ok(_) => {}
 170         Err(WebmentionError::NoEndpointDiscovered(_)) => {}
 171         Err(e) => warn!("Failed to send webmention: {}", e),
 172       }
 173     }
 174
 175     context
 176       .send_post_ws_message(
 177         &UserOperationCrud::CreatePost,
 178         inserted_post.id,
 179         websocket_id,
 180         Some(local_user_view.person.id),
 181       )
 182       .await
 183   }
 184 }