mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-02 11:32:22 +00:00
128 lines
3.8 KiB
Rust
128 lines
3.8 KiB
Rust
use crate::{
|
|
activities::{
|
|
following::accept::AcceptFollowCommunity,
|
|
generate_activity_id,
|
|
verify_activity,
|
|
verify_person,
|
|
verify_person_in_community,
|
|
},
|
|
context::lemmy_context,
|
|
fetcher::object_id::ObjectId,
|
|
objects::{community::ApubCommunity, person::ApubPerson},
|
|
send_lemmy_activity,
|
|
};
|
|
use activitystreams::{
|
|
activity::kind::FollowType,
|
|
base::AnyBase,
|
|
primitives::OneOrMany,
|
|
unparsed::Unparsed,
|
|
};
|
|
use lemmy_api_common::blocking;
|
|
use lemmy_apub_lib::{
|
|
data::Data,
|
|
traits::{ActivityFields, ActivityHandler, ActorType},
|
|
verify::verify_urls_match,
|
|
};
|
|
use lemmy_db_schema::{
|
|
source::community::{CommunityFollower, CommunityFollowerForm},
|
|
traits::Followable,
|
|
};
|
|
use lemmy_utils::LemmyError;
|
|
use lemmy_websocket::LemmyContext;
|
|
use serde::{Deserialize, Serialize};
|
|
use url::Url;
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct FollowCommunity {
|
|
pub(in crate::activities::following) actor: ObjectId<ApubPerson>,
|
|
pub(in crate::activities::following) to: [ObjectId<ApubCommunity>; 1],
|
|
pub(in crate::activities::following) object: ObjectId<ApubCommunity>,
|
|
#[serde(rename = "type")]
|
|
kind: FollowType,
|
|
id: Url,
|
|
#[serde(rename = "@context")]
|
|
context: OneOrMany<AnyBase>,
|
|
#[serde(flatten)]
|
|
unparsed: Unparsed,
|
|
}
|
|
|
|
impl FollowCommunity {
|
|
pub(in crate::activities::following) fn new(
|
|
actor: &ApubPerson,
|
|
community: &ApubCommunity,
|
|
context: &LemmyContext,
|
|
) -> Result<FollowCommunity, LemmyError> {
|
|
Ok(FollowCommunity {
|
|
actor: ObjectId::new(actor.actor_id()),
|
|
to: [ObjectId::new(community.actor_id())],
|
|
object: ObjectId::new(community.actor_id()),
|
|
kind: FollowType::Follow,
|
|
id: generate_activity_id(
|
|
FollowType::Follow,
|
|
&context.settings().get_protocol_and_hostname(),
|
|
)?,
|
|
context: lemmy_context(),
|
|
unparsed: Default::default(),
|
|
})
|
|
}
|
|
pub async fn send(
|
|
actor: &ApubPerson,
|
|
community: &ApubCommunity,
|
|
context: &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?;
|
|
|
|
let follow = FollowCommunity::new(actor, community, context)?;
|
|
let inbox = vec![community.inbox_url.clone().into()];
|
|
send_lemmy_activity(context, &follow, &follow.id, actor, inbox, true).await
|
|
}
|
|
}
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
impl ActivityHandler for FollowCommunity {
|
|
type DataType = LemmyContext;
|
|
async fn verify(
|
|
&self,
|
|
context: &Data<LemmyContext>,
|
|
request_counter: &mut i32,
|
|
) -> Result<(), LemmyError> {
|
|
verify_activity(self, &context.settings())?;
|
|
verify_urls_match(self.to[0].inner(), self.object.inner())?;
|
|
verify_person(&self.actor, context, request_counter).await?;
|
|
let community = self.to[0].dereference(context, request_counter).await?;
|
|
verify_person_in_community(&self.actor, &community, context, request_counter).await?;
|
|
Ok(())
|
|
}
|
|
|
|
async fn receive(
|
|
self,
|
|
context: &Data<LemmyContext>,
|
|
request_counter: &mut i32,
|
|
) -> Result<(), LemmyError> {
|
|
let actor = self.actor.dereference(context, request_counter).await?;
|
|
let community = self.to[0].dereference(context, request_counter).await?;
|
|
let community_follower_form = CommunityFollowerForm {
|
|
community_id: community.id,
|
|
person_id: actor.id,
|
|
pending: false,
|
|
};
|
|
|
|
// 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?;
|
|
|
|
AcceptFollowCommunity::send(self, context, request_counter).await
|
|
}
|
|
}
|