Plume/src/models/follows.rs

use activitypub::{Actor, activity::{Accept, Follow as FollowAct}};
use diesel::{self, PgConnection, ExpressionMethods, QueryDsl, RunQueryDsl};

use activity_pub::{broadcast, Id, IntoId, actor::Actor as ApActor, 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 {
    pub fn insert(conn: &PgConnection, new: NewFollow) -> Follow {
        diesel::insert_into(follows::table)
            .values(new)
            .get_result(conn)
            .expect("Unable to insert new follow")
    }

    pub fn get(conn: &PgConnection, id: i32) -> Option<Follow> {
        follows::table.filter(follows::id.eq(id))
            .limit(1)
            .load::<Follow>(conn)
            .expect("Unable to load follow by id")
            .into_iter().nth(0)
    }

    pub fn accept_follow<A: Signer + IntoId + Clone, B: Clone + WithInbox + Actor>(
        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::<Id>(from.clone().into_id()).unwrap();
        accept.accept_props.set_object_object(follow).unwrap();
        broadcast(conn, &*from, accept, vec![target.clone()]);
        res
    }
}

impl FromActivity<FollowAct> 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<FollowAct> for Follow {
    fn notify(conn: &PgConnection, follow: FollowAct, actor: Id) {
        let follower = User::from_url(conn, actor.into()).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: User::from_url(conn, follow.follow_props.object_link::<Id>().unwrap().into()).unwrap().id
        });
    }
}