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