6a65fa7c98
* Make it work with Lemmy * working but needs cleanup * almost everything working * debug * stack overflow fix
74 lines
2 KiB
Rust
74 lines
2 KiB
Rust
use crate::{
|
|
activities::accept::Accept,
|
|
generate_object_id,
|
|
instance::DatabaseHandle,
|
|
objects::person::DbUser,
|
|
};
|
|
use activitypub_federation::{
|
|
config::Data,
|
|
fetch::object_id::ObjectId,
|
|
kinds::activity::FollowType,
|
|
traits::{ActivityHandler, Actor},
|
|
};
|
|
use serde::{Deserialize, Serialize};
|
|
use url::Url;
|
|
|
|
#[derive(Deserialize, Serialize, Clone, Debug)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct Follow {
|
|
pub(crate) actor: ObjectId<DbUser>,
|
|
pub(crate) object: ObjectId<DbUser>,
|
|
#[serde(rename = "type")]
|
|
kind: FollowType,
|
|
id: Url,
|
|
}
|
|
|
|
impl Follow {
|
|
pub fn new(actor: ObjectId<DbUser>, object: ObjectId<DbUser>, id: Url) -> Follow {
|
|
Follow {
|
|
actor,
|
|
object,
|
|
kind: Default::default(),
|
|
id,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[async_trait::async_trait]
|
|
impl ActivityHandler for Follow {
|
|
type DataType = DatabaseHandle;
|
|
type Error = crate::error::Error;
|
|
|
|
fn id(&self) -> &Url {
|
|
&self.id
|
|
}
|
|
|
|
fn actor(&self) -> &Url {
|
|
self.actor.inner()
|
|
}
|
|
|
|
async fn verify(&self, _data: &Data<Self::DataType>) -> Result<(), Self::Error> {
|
|
Ok(())
|
|
}
|
|
|
|
// Ignore clippy false positive: https://github.com/rust-lang/rust-clippy/issues/6446
|
|
#[allow(clippy::await_holding_lock)]
|
|
async fn receive(self, data: &Data<Self::DataType>) -> Result<(), Self::Error> {
|
|
// add to followers
|
|
let local_user = {
|
|
let mut users = data.users.lock().unwrap();
|
|
let local_user = users.first_mut().unwrap();
|
|
local_user.followers.push(self.actor.inner().clone());
|
|
local_user.clone()
|
|
};
|
|
|
|
// send back an accept
|
|
let follower = self.actor.dereference(data).await?;
|
|
let id = generate_object_id(data.domain())?;
|
|
let accept = Accept::new(local_user.ap_id.clone(), self, id.clone());
|
|
local_user
|
|
.send(accept, vec![follower.shared_inbox_or_inbox()], data)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
}
|