use crate::{
activities::{
- following::accept::AcceptFollowCommunity,
generate_activity_id,
- verify_activity,
+ send_lemmy_activity,
verify_person,
+ verify_person_in_community,
},
- activity_queue::send_activity_new,
- extensions::context::lemmy_context,
- fetcher::{community::get_or_fetch_and_upsert_community, person::get_or_fetch_and_upsert_person},
- ActorType,
+ fetcher::user_or_community::UserOrCommunity,
+ insert_activity,
+ objects::{community::ApubCommunity, person::ApubPerson},
+ protocol::activities::following::{
+ accept::AcceptFollow,
+ follow::Follow,
+ undo_follow::UndoFollow,
+ },
+ SendActivity,
+};
+use activitypub_federation::{
+ config::Data,
+ kinds::activity::FollowType,
+ protocol::verification::verify_urls_match,
+ traits::{ActivityHandler, Actor},
};
-use activitystreams::activity::kind::FollowType;
-use lemmy_api_common::blocking;
-use lemmy_apub_lib::{verify_urls_match, ActivityCommonFields, ActivityHandler};
-use lemmy_db_queries::Followable;
-use lemmy_db_schema::source::{
- community::{Community, CommunityFollower, CommunityFollowerForm},
- person::Person,
+use lemmy_api_common::{
+ community::{BlockCommunity, BlockCommunityResponse},
+ context::LemmyContext,
+ utils::local_user_view_from_jwt,
+};
+use lemmy_db_schema::{
+ source::{
+ community::{Community, CommunityFollower, CommunityFollowerForm},
+ person::{PersonFollower, PersonFollowerForm},
+ },
+ traits::{Crud, Followable},
};
-use lemmy_utils::LemmyError;
-use lemmy_websocket::LemmyContext;
+use lemmy_utils::error::LemmyError;
use url::Url;
-#[derive(Clone, Debug, serde::Deserialize, serde::Serialize)]
-#[serde(rename_all = "camelCase")]
-pub struct FollowCommunity {
- pub(in crate::activities::following) to: Url,
- pub(in crate::activities::following) object: Url,
- #[serde(rename = "type")]
- pub(in crate::activities::following) kind: FollowType,
- #[serde(flatten)]
- pub(in crate::activities::following) common: ActivityCommonFields,
-}
+impl Follow {
+ pub(in crate::activities::following) fn new(
+ actor: &ApubPerson,
+ community: &ApubCommunity,
+ context: &Data<LemmyContext>,
+ ) -> Result<Follow, LemmyError> {
+ Ok(Follow {
+ actor: actor.id().into(),
+ object: community.id().into(),
+ to: Some([community.id().into()]),
+ kind: FollowType::Follow,
+ id: generate_activity_id(
+ FollowType::Follow,
+ &context.settings().get_protocol_and_hostname(),
+ )?,
+ })
+ }
-impl FollowCommunity {
+ #[tracing::instrument(skip_all)]
pub async fn send(
- actor: &Person,
- community: &Community,
- context: &LemmyContext,
+ actor: &ApubPerson,
+ community: &ApubCommunity,
+ context: &Data<LemmyContext>,
) -> Result<(), LemmyError> {
let community_follower_form = CommunityFollowerForm {
community_id: community.id,
person_id: actor.id,
pending: true,
};
- blocking(context.pool(), move |conn| {
- CommunityFollower::follow(conn, &community_follower_form).ok()
- })
- .await?;
+ CommunityFollower::follow(&mut context.pool(), &community_follower_form)
+ .await
+ .ok();
- let id = generate_activity_id(FollowType::Follow)?;
- let follow = FollowCommunity {
- to: community.actor_id(),
- object: community.actor_id(),
- kind: FollowType::Follow,
- common: ActivityCommonFields {
- context: lemmy_context(),
- id: id.clone(),
- actor: actor.actor_id(),
- unparsed: Default::default(),
- },
- };
- let inbox = vec![community.inbox_url.clone().into()];
- send_activity_new(context, &follow, &id, actor, inbox, true).await
+ let follow = Follow::new(actor, community, context)?;
+ let inbox = vec![community.shared_inbox_or_inbox()];
+ send_lemmy_activity(context, follow, actor, inbox, true).await
}
}
-#[async_trait::async_trait(?Send)]
-impl ActivityHandler for FollowCommunity {
- async fn verify(
- &self,
- context: &LemmyContext,
- request_counter: &mut i32,
- ) -> Result<(), LemmyError> {
- verify_activity(self.common())?;
- verify_urls_match(&self.to, &self.object)?;
- verify_person(&self.common.actor, context, request_counter).await?;
- Ok(())
+#[async_trait::async_trait]
+impl ActivityHandler for Follow {
+ type DataType = LemmyContext;
+ type Error = LemmyError;
+
+ fn id(&self) -> &Url {
+ &self.id
}
- async fn receive(
- self,
- context: &LemmyContext,
- request_counter: &mut i32,
- ) -> Result<(), LemmyError> {
- let actor =
- get_or_fetch_and_upsert_person(&self.common.actor, context, request_counter).await?;
- let community =
- get_or_fetch_and_upsert_community(&self.object, context, request_counter).await?;
- let community_follower_form = CommunityFollowerForm {
- community_id: community.id,
- person_id: actor.id,
- pending: false,
- };
+ fn actor(&self) -> &Url {
+ self.actor.inner()
+ }
- // This will fail if they're already a follower, but ignore the error.
- blocking(context.pool(), move |conn| {
- CommunityFollower::follow(conn, &community_follower_form).ok()
- })
- .await?;
+ #[tracing::instrument(skip_all)]
+ async fn verify(&self, context: &Data<LemmyContext>) -> Result<(), LemmyError> {
+ verify_person(&self.actor, context).await?;
+ let object = self.object.dereference(context).await?;
+ if let UserOrCommunity::Community(c) = object {
+ verify_person_in_community(&self.actor, &c, context).await?;
+ }
+ if let Some(to) = &self.to {
+ verify_urls_match(to[0].inner(), self.object.inner())?;
+ }
+ Ok(())
+ }
+
+ #[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 {
+ UserOrCommunity::User(u) => {
+ let form = PersonFollowerForm {
+ person_id: u.id,
+ follower_id: actor.id,
+ pending: false,
+ };
+ PersonFollower::follow(&mut context.pool(), &form).await?;
+ }
+ UserOrCommunity::Community(c) => {
+ let form = CommunityFollowerForm {
+ community_id: c.id,
+ person_id: actor.id,
+ pending: false,
+ };
+ CommunityFollower::follow(&mut context.pool(), &form).await?;
+ }
+ }
- AcceptFollowCommunity::send(self, context).await
+ AcceptFollow::send(self, context).await
}
+}
+
+#[async_trait::async_trait]
+impl SendActivity for BlockCommunity {
+ type Response = BlockCommunityResponse;
- fn common(&self) -> &ActivityCommonFields {
- &self.common
+ async fn send_activity(
+ request: &Self,
+ _response: &Self::Response,
+ context: &Data<LemmyContext>,
+ ) -> Result<(), LemmyError> {
+ let local_user_view = local_user_view_from_jwt(&request.auth, context).await?;
+ let community = Community::read(&mut context.pool(), request.community_id).await?;
+ UndoFollow::send(&local_user_view.person.into(), &community.into(), context).await
}
}