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::{generate_apub_endpoint, ApubLikeableType, ApubObjectType, EndpointType};
  11 use lemmy_db_queries::{source::post::Post_, Crud, Likeable};
  12 use lemmy_db_schema::source::post::*;
  13 use lemmy_db_views::post_view::PostView;
  14 use lemmy_utils::{
  15   request::fetch_iframely_and_pictrs_data,
  16   utils::{check_slurs, check_slurs_opt, clean_url_params, is_valid_post_title},
  17   ApiError,
  18   ConnectionId,
  19   LemmyError,
  20 };
  21 use lemmy_websocket::{messages::SendPost, LemmyContext, UserOperationCrud};
  22
  23 #[async_trait::async_trait(?Send)]
  24 impl PerformCrud for CreatePost {
  25   type Response = PostResponse;
  26
  27   async fn perform(
  28     &self,
  29     context: &Data<LemmyContext>,
  30     websocket_id: Option<ConnectionId>,
  31   ) -> Result<PostResponse, LemmyError> {
  32     let data: &CreatePost = &self;
  33     let local_user_view = get_local_user_view_from_jwt(&data.auth, context.pool()).await?;
  34
  35     check_slurs(&data.name)?;
  36     check_slurs_opt(&data.body)?;
  37
  38     if !is_valid_post_title(&data.name) {
  39       return Err(ApiError::err("invalid_post_title").into());
  40     }
  41
  42     check_community_ban(local_user_view.person.id, data.community_id, context.pool()).await?;
  43
  44     // Fetch Iframely and pictrs cached image
  45     let data_url = data.url.as_ref();
  46     let (iframely_title, iframely_description, iframely_html, pictrs_thumbnail) =
  47       fetch_iframely_and_pictrs_data(context.client(), data_url).await;
  48
  49     let post_form = PostForm {
  50       name: data.name.trim().to_owned(),
  51       url: data_url.map(|u| clean_url_params(u.to_owned()).into()),
  52       body: data.body.to_owned(),
  53       community_id: data.community_id,
  54       creator_id: local_user_view.person.id,
  55       nsfw: data.nsfw,
  56       embed_title: iframely_title,
  57       embed_description: iframely_description,
  58       embed_html: iframely_html,
  59       thumbnail_url: pictrs_thumbnail.map(|u| u.into()),
  60       ..PostForm::default()
  61     };
  62
  63     let inserted_post =
  64       match blocking(context.pool(), move |conn| Post::create(conn, &post_form)).await? {
  65         Ok(post) => post,
  66         Err(e) => {
  67           let err_type = if e.to_string() == "value too long for type character varying(200)" {
  68             "post_title_too_long"
  69           } else {
  70             "couldnt_create_post"
  71           };
  72
  73           return Err(ApiError::err(err_type).into());
  74         }
  75       };
  76
  77     let inserted_post_id = inserted_post.id;
  78     let updated_post = blocking(context.pool(), move |conn| -> Result<Post, LemmyError> {
  79       let apub_id = generate_apub_endpoint(EndpointType::Post, &inserted_post_id.to_string())?;
  80       Ok(Post::update_ap_id(conn, inserted_post_id, apub_id)?)
  81     })
  82     .await?
  83     .map_err(|_| ApiError::err("couldnt_create_post"))?;
  84
  85     updated_post
  86       .send_create(&local_user_view.person, context)
  87       .await?;
  88
  89     // They like their own post by default
  90     let person_id = local_user_view.person.id;
  91     let post_id = inserted_post.id;
  92     let like_form = PostLikeForm {
  93       post_id,
  94       person_id,
  95       score: 1,
  96     };
  97
  98     let like = move |conn: &'_ _| PostLike::like(conn, &like_form);
  99     if blocking(context.pool(), like).await?.is_err() {
 100       return Err(ApiError::err("couldnt_like_post").into());
 101     }
 102
 103     // Mark the post as read
 104     mark_post_as_read(person_id, post_id, context.pool()).await?;
 105
 106     updated_post
 107       .send_like(&local_user_view.person, context)
 108       .await?;
 109
 110     // Refetch the view
 111     let inserted_post_id = inserted_post.id;
 112     let post_view = blocking(context.pool(), move |conn| {
 113       PostView::read(conn, inserted_post_id, Some(local_user_view.person.id))
 114     })
 115     .await?
 116     .map_err(|_| ApiError::err("couldnt_find_post"))?;
 117
 118     let res = PostResponse { post_view };
 119
 120     context.chat_server().do_send(SendPost {
 121       op: UserOperationCrud::CreatePost,
 122       post: res.clone(),
 123       websocket_id,
 124     });
 125
 126     Ok(res)
 127   }
 128 }