use crate::{
- activities::{verify_activity, verify_person},
- fetcher::{community::get_or_fetch_and_upsert_community, person::get_or_fetch_and_upsert_person},
- CommunityType,
+ activities::{
+ generate_activity_id,
+ send_lemmy_activity,
+ verify_person,
+ verify_person_in_community,
+ },
+ fetcher::user_or_community::UserOrCommunity,
+ insert_activity,
+ objects::{community::ApubCommunity, person::ApubPerson},
+ protocol::activities::following::{
+ accept::AcceptFollow,
+ follow::Follow,
+ undo_follow::UndoFollow,
+ },
+ SendActivity,
};
-use activitystreams::{
- activity::{kind::FollowType, Follow},
- base::{AnyBase, ExtendsExt},
+use activitypub_federation::{
+ config::Data,
+ kinds::activity::FollowType,
+ protocol::verification::verify_urls_match,
+ traits::{ActivityHandler, Actor},
};
-use anyhow::Context;
-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::{CommunityFollower, CommunityFollowerForm};
-use lemmy_utils::{location_info, LemmyError};
-use lemmy_websocket::LemmyContext;
+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::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")]
- kind: FollowType,
- #[serde(flatten)]
- pub(in crate::activities::following) common: ActivityCommonFields,
-}
-
-#[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(())
+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(),
+ )?,
+ })
}
- async fn receive(
- &self,
- context: &LemmyContext,
- request_counter: &mut i32,
+ #[tracing::instrument(skip_all)]
+ pub async fn send(
+ actor: &ApubPerson,
+ community: &ApubCommunity,
+ context: &Data<LemmyContext>,
) -> 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,
+ pending: true,
};
+ CommunityFollower::follow(&mut context.pool(), &community_follower_form)
+ .await
+ .ok();
- // 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?;
+ 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]
+impl ActivityHandler for Follow {
+ type DataType = LemmyContext;
+ type Error = LemmyError;
- // TODO: avoid the conversion and pass our own follow struct directly
- let anybase = AnyBase::from_arbitrary_json(serde_json::to_string(self)?)?;
- let anybase = Follow::from_any_base(anybase)?.context(location_info!())?;
- community.send_accept_follow(anybase, context).await
+ fn id(&self) -> &Url {
+ &self.id
}
- fn common(&self) -> &ActivityCommonFields {
- &self.common
+ fn actor(&self) -> &Url {
+ self.actor.inner()
+ }
+
+ #[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?;
+ }
+ }
+
+ AcceptFollow::send(self, context).await
+ }
+}
+
+#[async_trait::async_trait]
+impl SendActivity for BlockCommunity {
+ type Response = BlockCommunityResponse;
+
+ 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
}
}