mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-09-25 04:50:02 +00:00
179 lines
5.1 KiB
Rust
179 lines
5.1 KiB
Rust
use crate::{
|
|
activities::community::send_activity_in_community,
|
|
activity_lists::AnnouncableActivities,
|
|
fetcher::post_or_comment::PostOrComment,
|
|
objects::{comment::ApubComment, person::ApubPerson, post::ApubPost},
|
|
protocol::activities::voting::{
|
|
undo_vote::UndoVote,
|
|
vote::{Vote, VoteType},
|
|
},
|
|
SendActivity,
|
|
};
|
|
use activitypub_federation::core::object_id::ObjectId;
|
|
use lemmy_api_common::{
|
|
comment::{CommentResponse, CreateCommentLike},
|
|
context::LemmyContext,
|
|
post::{CreatePostLike, PostResponse},
|
|
sensitive::Sensitive,
|
|
utils::get_local_user_view_from_jwt,
|
|
websocket::{
|
|
send::{send_comment_ws_message_simple, send_post_ws_message},
|
|
UserOperation,
|
|
},
|
|
};
|
|
use lemmy_db_schema::{
|
|
newtypes::CommunityId,
|
|
source::{
|
|
comment::{CommentLike, CommentLikeForm},
|
|
community::Community,
|
|
person::Person,
|
|
post::{PostLike, PostLikeForm},
|
|
},
|
|
traits::{Crud, Likeable},
|
|
};
|
|
use lemmy_utils::error::LemmyError;
|
|
|
|
pub mod undo_vote;
|
|
pub mod vote;
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl SendActivity for CreatePostLike {
|
|
type Response = PostResponse;
|
|
|
|
async fn send_activity(
|
|
request: &Self,
|
|
response: &Self::Response,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let object_id = ObjectId::new(response.post_view.post.ap_id.clone());
|
|
let community_id = response.post_view.community.id;
|
|
send_activity(
|
|
object_id,
|
|
community_id,
|
|
request.score,
|
|
&request.auth,
|
|
context,
|
|
)
|
|
.await
|
|
}
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl SendActivity for CreateCommentLike {
|
|
type Response = CommentResponse;
|
|
|
|
async fn send_activity(
|
|
request: &Self,
|
|
response: &Self::Response,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let object_id = ObjectId::new(response.comment_view.comment.ap_id.clone());
|
|
let community_id = response.comment_view.community.id;
|
|
send_activity(
|
|
object_id,
|
|
community_id,
|
|
request.score,
|
|
&request.auth,
|
|
context,
|
|
)
|
|
.await
|
|
}
|
|
}
|
|
|
|
async fn send_activity(
|
|
object_id: ObjectId<PostOrComment>,
|
|
community_id: CommunityId,
|
|
score: i16,
|
|
jwt: &Sensitive<String>,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let community = Community::read(context.pool(), community_id).await?.into();
|
|
let local_user_view = get_local_user_view_from_jwt(jwt, context.pool(), context.secret()).await?;
|
|
let actor = Person::read(context.pool(), local_user_view.person.id)
|
|
.await?
|
|
.into();
|
|
|
|
// score of 1 means upvote, -1 downvote, 0 undo a previous vote
|
|
if score != 0 {
|
|
let vote = Vote::new(object_id, &actor, &community, score.try_into()?, context)?;
|
|
let activity = AnnouncableActivities::Vote(vote);
|
|
send_activity_in_community(activity, &actor, &community, vec![], false, context).await
|
|
} else {
|
|
// Lemmy API doesnt distinguish between Undo/Like and Undo/Dislike, so we hardcode it here.
|
|
let vote = Vote::new(object_id, &actor, &community, VoteType::Like, context)?;
|
|
let undo_vote = UndoVote::new(vote, &actor, &community, context)?;
|
|
let activity = AnnouncableActivities::UndoVote(undo_vote);
|
|
send_activity_in_community(activity, &actor, &community, vec![], false, context).await
|
|
}
|
|
}
|
|
|
|
#[tracing::instrument(skip_all)]
|
|
async fn vote_comment(
|
|
vote_type: &VoteType,
|
|
actor: ApubPerson,
|
|
comment: &ApubComment,
|
|
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;
|
|
CommentLike::remove(context.pool(), person_id, comment_id).await?;
|
|
CommentLike::like(context.pool(), &like_form).await?;
|
|
|
|
send_comment_ws_message_simple(comment_id, UserOperation::CreateCommentLike, context).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(skip_all)]
|
|
async fn vote_post(
|
|
vote_type: &VoteType,
|
|
actor: ApubPerson,
|
|
post: &ApubPost,
|
|
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;
|
|
PostLike::remove(context.pool(), person_id, post_id).await?;
|
|
PostLike::like(context.pool(), &like_form).await?;
|
|
|
|
send_post_ws_message(post.id, UserOperation::CreatePostLike, None, None, context).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(skip_all)]
|
|
async fn undo_vote_comment(
|
|
actor: ApubPerson,
|
|
comment: &ApubComment,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let comment_id = comment.id;
|
|
let person_id = actor.id;
|
|
CommentLike::remove(context.pool(), person_id, comment_id).await?;
|
|
|
|
send_comment_ws_message_simple(comment_id, UserOperation::CreateCommentLike, context).await?;
|
|
Ok(())
|
|
}
|
|
|
|
#[tracing::instrument(skip_all)]
|
|
async fn undo_vote_post(
|
|
actor: ApubPerson,
|
|
post: &ApubPost,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let post_id = post.id;
|
|
let person_id = actor.id;
|
|
PostLike::remove(context.pool(), person_id, post_id).await?;
|
|
|
|
send_post_ws_message(post_id, UserOperation::CreatePostLike, None, None, context).await?;
|
|
Ok(())
|
|
}
|