use activitypub::{Actor, activity::{Accept, Follow as FollowAct}}; use diesel::{self, PgConnection, ExpressionMethods, QueryDsl, RunQueryDsl}; use activity_pub::{broadcast, Id, IntoId, inbox::{FromActivity, Notify, WithInbox}, sign::Signer}; use models::{ blogs::Blog, notifications::*, 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); pub fn accept_follow( conn: &PgConnection, from: &A, target: &B, 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(); accept.accept_props.set_actor_link::(from.clone().into_id()).unwrap(); accept.accept_props.set_object_object(follow).unwrap(); broadcast(&*from, accept, vec![target.clone()]); res } } impl FromActivity for Follow { fn from_activity(conn: &PgConnection, follow: FollowAct, _actor: Id) -> Follow { let from = User::from_url(conn, follow.follow_props.actor.as_str().unwrap().to_string()).unwrap(); match User::from_url(conn, follow.follow_props.object.as_str().unwrap().to_string()) { Some(u) => Follow::accept_follow(conn, &from, &u, follow, from.id, u.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) { let follower = User::get(conn, self.follower_id).unwrap(); Notification::insert(conn, NewNotification { title: "{{ data }} started following you".to_string(), data: Some(follower.display_name.clone()), content: None, link: Some(follower.ap_url), user_id: self.following_id }); } }