mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-02 04:49:30 +00:00
128 lines
3.6 KiB
Rust
128 lines
3.6 KiB
Rust
use crate::{
|
|
activities::{
|
|
following::accept::AcceptFollowCommunity,
|
|
generate_activity_id,
|
|
verify_activity,
|
|
verify_person,
|
|
},
|
|
context::lemmy_context,
|
|
fetcher::object_id::ObjectId,
|
|
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::{Community, CommunityFollower, CommunityFollowerForm},
|
|
person::Person,
|
|
},
|
|
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<Person>,
|
|
// TODO: is there any reason to put the same community id twice, in to and object?
|
|
pub(in crate::activities::following) to: ObjectId<Community>,
|
|
pub(in crate::activities::following) object: ObjectId<Community>,
|
|
#[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: &Person,
|
|
community: &Community,
|
|
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: &Person,
|
|
community: &Community,
|
|
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.inner(), self.object.inner())?;
|
|
verify_person(&self.actor, 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.object.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
|
|
}
|
|
}
|