mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-16 12:20:32 +00:00
* Merge like/dislike activity handlers into vote * Rewrite vote sending code * Remove old send_create, send_update functions
104 lines
2.3 KiB
Rust
104 lines
2.3 KiB
Rust
use crate::activities::{
|
|
comment::send_websocket_message as send_comment_message,
|
|
post::send_websocket_message as send_post_message,
|
|
voting::vote::VoteType,
|
|
};
|
|
use lemmy_api_common::blocking;
|
|
use lemmy_db_queries::Likeable;
|
|
use lemmy_db_schema::source::{
|
|
comment::{Comment, CommentLike, CommentLikeForm},
|
|
person::Person,
|
|
post::{Post, PostLike, PostLikeForm},
|
|
};
|
|
use lemmy_utils::LemmyError;
|
|
use lemmy_websocket::{LemmyContext, UserOperation};
|
|
|
|
pub mod undo_vote;
|
|
pub mod vote;
|
|
|
|
async fn vote_comment(
|
|
vote_type: &VoteType,
|
|
actor: Person,
|
|
comment: &Comment,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let comment_id = comment.id;
|
|
let like_form = CommentLikeForm {
|
|
comment_id,
|
|
post_id: comment.post_id,
|
|
person_id: actor.id,
|
|
score: vote_type.into(),
|
|
};
|
|
let person_id = actor.id;
|
|
blocking(context.pool(), move |conn| {
|
|
CommentLike::remove(conn, person_id, comment_id)?;
|
|
CommentLike::like(conn, &like_form)
|
|
})
|
|
.await??;
|
|
|
|
send_comment_message(
|
|
comment_id,
|
|
vec![],
|
|
UserOperation::CreateCommentLike,
|
|
context,
|
|
)
|
|
.await
|
|
}
|
|
|
|
async fn vote_post(
|
|
vote_type: &VoteType,
|
|
actor: Person,
|
|
post: &Post,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let post_id = post.id;
|
|
let like_form = PostLikeForm {
|
|
post_id: post.id,
|
|
person_id: actor.id,
|
|
score: vote_type.into(),
|
|
};
|
|
let person_id = actor.id;
|
|
blocking(context.pool(), move |conn| {
|
|
PostLike::remove(conn, person_id, post_id)?;
|
|
PostLike::like(conn, &like_form)
|
|
})
|
|
.await??;
|
|
|
|
send_post_message(post.id, UserOperation::CreatePostLike, context).await
|
|
}
|
|
|
|
async fn undo_vote_comment(
|
|
actor: Person,
|
|
comment: &Comment,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let comment_id = comment.id;
|
|
let person_id = actor.id;
|
|
blocking(context.pool(), move |conn| {
|
|
CommentLike::remove(conn, person_id, comment_id)
|
|
})
|
|
.await??;
|
|
|
|
send_comment_message(
|
|
comment.id,
|
|
vec![],
|
|
UserOperation::CreateCommentLike,
|
|
context,
|
|
)
|
|
.await
|
|
}
|
|
|
|
async fn undo_vote_post(
|
|
actor: Person,
|
|
post: &Post,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let post_id = post.id;
|
|
let person_id = actor.id;
|
|
blocking(context.pool(), move |conn| {
|
|
PostLike::remove(conn, person_id, post_id)
|
|
})
|
|
.await??;
|
|
send_post_message(post.id, UserOperation::CreatePostLike, context).await
|
|
}
|