crates/api_crud/src/post/create.rs

   1 use crate::PerformCrud;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   blocking,
   5   check_community_ban,
   6   get_local_user_view_from_jwt,
   7   mark_post_as_read,
   8   post::*,
   9 };
  10 use lemmy_apub::{
  11   activities::{
  12     post::create_or_update::CreateOrUpdatePost,
  13     voting::vote::{Vote, VoteType},
  14     CreateOrUpdateType,
  15   },
  16   fetcher::post_or_comment::PostOrComment,
  17   generate_apub_endpoint,
  18   EndpointType,
  19 };
  20 use lemmy_db_queries::{source::post::Post_, Crud, Likeable};
  21 use lemmy_db_schema::source::post::*;
  22 use lemmy_utils::{
  23   request::fetch_site_data,
  24   utils::{check_slurs, check_slurs_opt, clean_url_params, is_valid_post_title},
  25   ApiError,
  26   ConnectionId,
  27   LemmyError,
  28 };
  29 use lemmy_websocket::{send::send_post_ws_message, LemmyContext, UserOperationCrud};
  30
  31 #[async_trait::async_trait(?Send)]
  32 impl PerformCrud for CreatePost {
  33   type Response = PostResponse;
  34
  35   async fn perform(
  36     &self,
  37     context: &Data<LemmyContext>,
  38     websocket_id: Option<ConnectionId>,
  39   ) -> Result<PostResponse, LemmyError> {
  40     let data: &CreatePost = self;
  41     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?;
  42
  43     check_slurs(&data.name)?;
  44     check_slurs_opt(&data.body)?;
  45
  46     if !is_valid_post_title(&data.name) {
  47       return Err(ApiError::err("invalid_post_title").into());
  48     }
  49
  50     check_community_ban(local_user_view.person.id, data.community_id, context.pool()).await?;
  51
  52     // Fetch post links and pictrs cached image
  53     let data_url = data.url.as_ref();
  54     let (metadata_res, pictrs_thumbnail) = fetch_site_data(context.client(), data_url).await;
  55     let (embed_title, embed_description, embed_html) = metadata_res
  56       .map(|u| (u.title, u.description, u.html))
  57       .unwrap_or((None, None, None));
  58
  59     let post_form = PostForm {
  60       name: data.name.trim().to_owned(),
  61       url: data_url.map(|u| clean_url_params(u.to_owned()).into()),
  62       body: data.body.to_owned(),
  63       community_id: data.community_id,
  64       creator_id: local_user_view.person.id,
  65       nsfw: data.nsfw,
  66       embed_title,
  67       embed_description,
  68       embed_html,
  69       thumbnail_url: pictrs_thumbnail.map(|u| u.into()),
  70       ..PostForm::default()
  71     };
  72
  73     let inserted_post =
  74       match blocking(context.pool(), move |conn| Post::create(conn, &post_form)).await? {
  75         Ok(post) => post,
  76         Err(e) => {
  77           let err_type = if e.to_string() == "value too long for type character varying(200)" {
  78             "post_title_too_long"
  79           } else {
  80             "couldnt_create_post"
  81           };
  82
  83           return Err(ApiError::err(err_type).into());
  84         }
  85       };
  86
  87     let inserted_post_id = inserted_post.id;
  88     let updated_post = blocking(context.pool(), move |conn| -> Result<Post, LemmyError> {
  89       let apub_id = generate_apub_endpoint(EndpointType::Post, &inserted_post_id.to_string())?;
  90       Ok(Post::update_ap_id(conn, inserted_post_id, apub_id)?)
  91     })
  92     .await?
  93     .map_err(|_| ApiError::err("couldnt_create_post"))?;
  94
  95     CreateOrUpdatePost::send(
  96       &updated_post,
  97       &local_user_view.person,
  98       CreateOrUpdateType::Create,
  99       context,
 100     )
 101     .await?;
 102
 103     // They like their own post by default
 104     let person_id = local_user_view.person.id;
 105     let post_id = inserted_post.id;
 106     let like_form = PostLikeForm {
 107       post_id,
 108       person_id,
 109       score: 1,
 110     };
 111
 112     let like = move |conn: &'_ _| PostLike::like(conn, &like_form);
 113     if blocking(context.pool(), like).await?.is_err() {
 114       return Err(ApiError::err("couldnt_like_post").into());
 115     }
 116
 117     // Mark the post as read
 118     mark_post_as_read(person_id, post_id, context.pool()).await?;
 119
 120     let object = PostOrComment::Post(Box::new(updated_post));
 121     Vote::send(
 122       &object,
 123       &local_user_view.person,
 124       inserted_post.community_id,
 125       VoteType::Like,
 126       context,
 127     )
 128     .await?;
 129
 130     send_post_ws_message(
 131       inserted_post.id,
 132       UserOperationCrud::CreatePost,
 133       websocket_id,
 134       Some(local_user_view.person.id),
 135       context,
 136     )
 137     .await
 138   }
 139 }