fedimovies/src/activitypub/handlers/follow.rs
2022-12-03 22:57:09 +00:00

53 lines
1.6 KiB
Rust

use tokio_postgres::GenericClient;
use crate::activitypub::{
activity::Activity,
builders::accept_follow::prepare_accept_follow,
fetcher::helpers::get_or_import_profile_by_actor_id,
identifiers::parse_local_actor_id,
receiver::find_object_id,
vocabulary::PERSON,
};
use crate::config::Config;
use crate::database::DatabaseError;
use crate::models::relationships::queries::follow;
use crate::models::users::queries::get_user_by_name;
use super::{HandlerError, HandlerResult};
pub async fn handle_follow(
config: &Config,
db_client: &mut impl GenericClient,
activity: Activity,
) -> HandlerResult {
let source_profile = get_or_import_profile_by_actor_id(
db_client,
&config.instance(),
&config.media_dir(),
&activity.actor,
).await?;
let source_actor = source_profile.actor_json
.ok_or(HandlerError::LocalObject)?;
let target_actor_id = find_object_id(&activity.object)?;
let target_username = parse_local_actor_id(
&config.instance_url(),
&target_actor_id,
)?;
let target_user = get_user_by_name(db_client, &target_username).await?;
match follow(db_client, &source_profile.id, &target_user.profile.id).await {
Ok(_) => (),
// Proceed even if relationship already exists
Err(DatabaseError::AlreadyExists(_)) => (),
Err(other_error) => return Err(other_error.into()),
};
// Send activity
prepare_accept_follow(
&config.instance(),
&target_user,
&source_actor,
&activity.id,
).spawn_deliver();
Ok(Some(PERSON))
}