use crate::{
activities::{
- community::{
- announce::{AnnouncableActivities, GetCommunity},
- send_to_community,
- },
generate_activity_id,
- verify_activity,
- verify_is_public,
verify_person_in_community,
voting::{vote_comment, vote_post},
},
- context::lemmy_context,
- fetcher::object_id::ObjectId,
+ insert_activity,
objects::{community::ApubCommunity, person::ApubPerson},
+ protocol::{
+ activities::voting::vote::{Vote, VoteType},
+ InCommunity,
+ },
PostOrComment,
};
-use activitystreams::{base::AnyBase, primitives::OneOrMany, public, unparsed::Unparsed};
-use anyhow::anyhow;
-use lemmy_api_common::blocking;
-use lemmy_apub_lib::{
- data::Data,
- traits::{ActivityFields, ActivityHandler, ActorType},
-};
-use lemmy_db_schema::{
- newtypes::CommunityId,
- source::{community::Community, post::Post},
- traits::Crud,
+use activitypub_federation::{
+ config::Data,
+ fetch::object_id::ObjectId,
+ traits::{ActivityHandler, Actor},
};
-use lemmy_utils::LemmyError;
-use lemmy_websocket::LemmyContext;
-use serde::{Deserialize, Serialize};
-use std::{convert::TryFrom, ops::Deref};
-use strum_macros::ToString;
+use anyhow::anyhow;
+use lemmy_api_common::context::LemmyContext;
+use lemmy_db_schema::source::local_site::LocalSite;
+use lemmy_utils::error::LemmyError;
use url::Url;
-#[derive(Clone, Debug, ToString, Deserialize, Serialize)]
-pub enum VoteType {
- Like,
- Dislike,
-}
-
-impl TryFrom<i16> for VoteType {
- type Error = LemmyError;
-
- fn try_from(value: i16) -> Result<Self, Self::Error> {
- match value {
- 1 => Ok(VoteType::Like),
- -1 => Ok(VoteType::Dislike),
- _ => Err(anyhow!("invalid vote value").into()),
- }
- }
-}
-
-impl From<&VoteType> for i16 {
- fn from(value: &VoteType) -> i16 {
- match value {
- VoteType::Like => 1,
- VoteType::Dislike => -1,
- }
- }
-}
-
-#[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
-#[serde(rename_all = "camelCase")]
-pub struct Vote {
- actor: ObjectId<ApubPerson>,
- to: Vec<Url>,
- pub(in crate::activities::voting) object: ObjectId<PostOrComment>,
- cc: [ObjectId<ApubCommunity>; 1],
- #[serde(rename = "type")]
- pub(in crate::activities::voting) kind: VoteType,
- id: Url,
- #[serde(rename = "@context")]
- context: OneOrMany<AnyBase>,
- #[serde(flatten)]
- unparsed: Unparsed,
-}
-
impl Vote {
pub(in crate::activities::voting) fn new(
- object: &PostOrComment,
+ object_id: ObjectId<PostOrComment>,
actor: &ApubPerson,
community: &ApubCommunity,
kind: VoteType,
- context: &LemmyContext,
+ context: &Data<LemmyContext>,
) -> Result<Vote, LemmyError> {
Ok(Vote {
- actor: ObjectId::new(actor.actor_id()),
- to: vec![public()],
- object: ObjectId::new(object.ap_id()),
- cc: [ObjectId::new(community.actor_id())],
+ actor: actor.id().into(),
+ object: object_id,
kind: kind.clone(),
id: generate_activity_id(kind, &context.settings().get_protocol_and_hostname())?,
- context: lemmy_context(),
- unparsed: Default::default(),
+ audience: Some(community.id().into()),
})
}
-
- pub async fn send(
- object: &PostOrComment,
- actor: &ApubPerson,
- community_id: CommunityId,
- kind: VoteType,
- context: &LemmyContext,
- ) -> Result<(), LemmyError> {
- let community = blocking(context.pool(), move |conn| {
- Community::read(conn, community_id)
- })
- .await??
- .into();
- let vote = Vote::new(object, actor, &community, kind, context)?;
- let vote_id = vote.id.clone();
-
- let activity = AnnouncableActivities::Vote(vote);
- send_to_community(activity, &vote_id, actor, &community, vec![], context).await
- }
}
-#[async_trait::async_trait(?Send)]
+#[async_trait::async_trait]
impl ActivityHandler for Vote {
type DataType = LemmyContext;
- async fn verify(
- &self,
- context: &Data<LemmyContext>,
- request_counter: &mut i32,
- ) -> Result<(), LemmyError> {
- verify_is_public(&self.to)?;
- verify_activity(self, &context.settings())?;
- let community = self.get_community(context, request_counter).await?;
- verify_person_in_community(&self.actor, &community, context, request_counter).await?;
+ type Error = LemmyError;
+
+ fn id(&self) -> &Url {
+ &self.id
+ }
+
+ fn actor(&self) -> &Url {
+ self.actor.inner()
+ }
+
+ #[tracing::instrument(skip_all)]
+ async fn verify(&self, context: &Data<LemmyContext>) -> Result<(), LemmyError> {
+ let community = self.community(context).await?;
+ verify_person_in_community(&self.actor, &community, context).await?;
+ let enable_downvotes = LocalSite::read(&mut context.pool())
+ .await
+ .map(|l| l.enable_downvotes)
+ .unwrap_or(true);
+ if self.kind == VoteType::Dislike && !enable_downvotes {
+ return Err(anyhow!("Downvotes disabled").into());
+ }
Ok(())
}
- async fn receive(
- self,
- context: &Data<LemmyContext>,
- request_counter: &mut i32,
- ) -> Result<(), LemmyError> {
- let actor = self.actor.dereference(context, request_counter).await?;
- let object = self.object.dereference(context, request_counter).await?;
+ #[tracing::instrument(skip_all)]
+ async fn receive(self, context: &Data<LemmyContext>) -> Result<(), LemmyError> {
+ insert_activity(&self.id, &self, false, true, context).await?;
+ let actor = self.actor.dereference(context).await?;
+ let object = self.object.dereference(context).await?;
match object {
- PostOrComment::Post(p) => vote_post(&self.kind, actor, p.deref(), context).await,
+ PostOrComment::Post(p) => vote_post(&self.kind, actor, &p, context).await,
PostOrComment::Comment(c) => vote_comment(&self.kind, actor, &c, context).await,
}
}
}
-
-#[async_trait::async_trait(?Send)]
-impl GetCommunity for Vote {
- async fn get_community(
- &self,
- context: &LemmyContext,
- request_counter: &mut i32,
- ) -> Result<ApubCommunity, LemmyError> {
- let object = self.object.dereference(context, request_counter).await?;
- let cid = match object {
- PostOrComment::Post(p) => p.community_id,
- PostOrComment::Comment(c) => {
- blocking(context.pool(), move |conn| Post::read(conn, c.post_id))
- .await??
- .community_id
- }
- };
- let community = blocking(context.pool(), move |conn| Community::read(conn, cid)).await??;
- Ok(community.into())
- }
-}