use activitypub::{Actor, activity::{Accept, Follow as FollowAct}, actor::Person}; use diesel::{self, PgConnection, ExpressionMethods, QueryDsl, RunQueryDsl}; use plume_common::activity_pub::{broadcast, Id, IntoId, inbox::{FromActivity, Notify, WithInbox}, sign::Signer}; use blogs::Blog; use notifications::*; use users::User; use schema::follows; #[derive(Queryable, Identifiable, Associations)] #[belongs_to(User, foreign_key = "following_id")] pub struct Follow { pub id: i32, pub follower_id: i32, pub following_id: i32 } #[derive(Insertable)] #[table_name = "follows"] pub struct NewFollow { pub follower_id: i32, pub following_id: i32 } impl Follow { insert!(follows, NewFollow); get!(follows); /// from -> The one sending the follow request /// target -> The target of the request, responding with Accept pub fn accept_follow( conn: &PgConnection, from: &B, target: &A, follow: FollowAct, from_id: i32, target_id: i32 ) -> Follow { let res = Follow::insert(conn, NewFollow { follower_id: from_id, following_id: target_id }); let mut accept = Accept::default(); let accept_id = format!("{}#accept", follow.object_props.id_string().unwrap_or(String::new())); accept.object_props.set_id_string(accept_id).expect("accept_follow: id error"); accept.object_props.set_to_link(from.clone().into_id()).expect("accept_follow: to error"); accept.object_props.set_cc_link_vec::(vec![]).expect("accept_follow: cc error"); accept.accept_props.set_actor_link::(target.clone().into_id()).unwrap(); accept.accept_props.set_object_object(follow).unwrap(); broadcast(&*target, accept, vec![from.clone()]); res } } impl FromActivity for Follow { fn from_activity(conn: &PgConnection, follow: FollowAct, _actor: Id) -> Follow { let from_id = follow.follow_props.actor_link::().map(|l| l.into()) .unwrap_or_else(|_| follow.follow_props.actor_object::().expect("No actor object (nor ID) on Follow").object_props.id_string().expect("No ID on actor on Follow")); let from = User::from_url(conn, from_id).unwrap(); match User::from_url(conn, follow.follow_props.object.as_str().unwrap().to_string()) { Some(user) => Follow::accept_follow(conn, &from, &user, follow, from.id, user.id), None => { let blog = Blog::from_url(conn, follow.follow_props.object.as_str().unwrap().to_string()).unwrap(); Follow::accept_follow(conn, &from, &blog, follow, from.id, blog.id) } } } } impl Notify for Follow { fn notify(&self, conn: &PgConnection) { Notification::insert(conn, NewNotification { kind: notification_kind::FOLLOW.to_string(), object_id: self.id, user_id: self.following_id }); } }