crates/api/src/post/like.rs

   1 use crate::Perform;
   2 use actix_web::web::Data;
   3 use lemmy_api_common::{
   4   post::{CreatePostLike, PostResponse},
   5   utils::{
   6     blocking,
   7     check_community_ban,
   8     check_community_deleted_or_removed,
   9     check_downvotes_enabled,
  10     get_local_user_view_from_jwt,
  11     mark_post_as_read,
  12   },
  13 };
  14 use lemmy_apub::{
  15   fetcher::post_or_comment::PostOrComment,
  16   objects::post::ApubPost,
  17   protocol::activities::voting::{
  18     undo_vote::UndoVote,
  19     vote::{Vote, VoteType},
  20   },
  21 };
  22 use lemmy_db_schema::{
  23   source::{
  24     local_site::LocalSite,
  25     post::{Post, PostLike, PostLikeForm},
  26   },
  27   traits::{Crud, Likeable},
  28 };
  29 use lemmy_utils::{error::LemmyError, ConnectionId};
  30 use lemmy_websocket::{send::send_post_ws_message, LemmyContext, UserOperation};
  31
  32 #[async_trait::async_trait(?Send)]
  33 impl Perform for CreatePostLike {
  34   type Response = PostResponse;
  35
  36   #[tracing::instrument(skip(context, websocket_id))]
  37   async fn perform(
  38     &self,
  39     context: &Data<LemmyContext>,
  40     websocket_id: Option<ConnectionId>,
  41   ) -> Result<PostResponse, LemmyError> {
  42     let data: &CreatePostLike = self;
  43     let local_user_view =
  44       get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
  45     let local_site = blocking(context.pool(), LocalSite::read).await??;
  46
  47     // Don't do a downvote if site has downvotes disabled
  48     check_downvotes_enabled(data.score, &local_site)?;
  49
  50     // Check for a community ban
  51     let post_id = data.post_id;
  52     let post: ApubPost = blocking(context.pool(), move |conn| Post::read(conn, post_id))
  53       .await??
  54       .into();
  55
  56     check_community_ban(local_user_view.person.id, post.community_id, context.pool()).await?;
  57     check_community_deleted_or_removed(post.community_id, context.pool()).await?;
  58
  59     let like_form = PostLikeForm {
  60       post_id: data.post_id,
  61       person_id: local_user_view.person.id,
  62       score: data.score,
  63     };
  64
  65     // Remove any likes first
  66     let person_id = local_user_view.person.id;
  67     blocking(context.pool(), move |conn| {
  68       PostLike::remove(conn, person_id, post_id)
  69     })
  70     .await??;
  71
  72     let community_id = post.community_id;
  73     let object = PostOrComment::Post(Box::new(post));
  74
  75     // Only add the like if the score isnt 0
  76     let do_add = like_form.score != 0 && (like_form.score == 1 || like_form.score == -1);
  77     if do_add {
  78       let like_form2 = like_form.clone();
  79       let like = move |conn: &mut _| PostLike::like(conn, &like_form2);
  80       blocking(context.pool(), like)
  81         .await?
  82         .map_err(|e| LemmyError::from_error_message(e, "couldnt_like_post"))?;
  83
  84       Vote::send(
  85         &object,
  86         &local_user_view.person.clone().into(),
  87         community_id,
  88         like_form.score.try_into()?,
  89         context,
  90       )
  91       .await?;
  92     } else {
  93       // API doesn't distinguish between Undo/Like and Undo/Dislike
  94       UndoVote::send(
  95         &object,
  96         &local_user_view.person.clone().into(),
  97         community_id,
  98         VoteType::Like,
  99         context,
 100       )
 101       .await?;
 102     }
 103
 104     // Mark the post as read
 105     mark_post_as_read(person_id, post_id, context.pool()).await?;
 106
 107     send_post_ws_message(
 108       data.post_id,
 109       UserOperation::CreatePostLike,
 110       websocket_id,
 111       Some(local_user_view.person.id),
 112       context,
 113     )
 114     .await
 115   }
 116 }