use activitypub::activity; use chrono::NaiveDateTime; use diesel::{self, ExpressionMethods, QueryDsl, RunQueryDsl}; use notifications::*; use plume_common::activity_pub::{ inbox::{AsObject, FromId}, Id, IntoId, PUBLIC_VISIBILITY, }; use posts::Post; use schema::likes; use users::User; use {Connection, Error, PlumeRocket, Result}; #[derive(Clone, Queryable, Identifiable)] pub struct Like { pub id: i32, pub user_id: i32, pub post_id: i32, pub creation_date: NaiveDateTime, pub ap_url: String, } #[derive(Default, Insertable)] #[table_name = "likes"] pub struct NewLike { pub user_id: i32, pub post_id: i32, pub ap_url: String, } impl Like { insert!(likes, NewLike); get!(likes); find_by!(likes, find_by_ap_url, ap_url as &str); find_by!(likes, find_by_user_on_post, user_id as i32, post_id as i32); pub fn to_activity(&self, conn: &Connection) -> Result { let mut act = activity::Like::default(); act.like_props .set_actor_link(User::get(conn, self.user_id)?.into_id())?; act.like_props .set_object_link(Post::get(conn, self.post_id)?.into_id())?; act.object_props .set_to_link_vec(vec![Id::new(PUBLIC_VISIBILITY.to_string())])?; act.object_props.set_cc_link_vec(vec![Id::new( User::get(conn, self.user_id)?.followers_endpoint, )])?; act.object_props.set_id_string(self.ap_url.clone())?; Ok(act) } pub fn notify(&self, conn: &Connection) -> Result<()> { let post = Post::get(conn, self.post_id)?; for author in post.get_authors(conn)? { Notification::insert( conn, NewNotification { kind: notification_kind::LIKE.to_string(), object_id: self.id, user_id: author.id, }, )?; } Ok(()) } pub fn build_undo(&self, conn: &Connection) -> Result { let mut act = activity::Undo::default(); act.undo_props .set_actor_link(User::get(conn, self.user_id)?.into_id())?; act.undo_props.set_object_object(self.to_activity(conn)?)?; act.object_props .set_id_string(format!("{}#delete", self.ap_url))?; act.object_props .set_to_link_vec(vec![Id::new(PUBLIC_VISIBILITY.to_string())])?; act.object_props.set_cc_link_vec(vec![Id::new( User::get(conn, self.user_id)?.followers_endpoint, )])?; Ok(act) } } impl AsObject for Post { type Error = Error; type Output = Like; fn activity(self, c: &PlumeRocket, actor: User, id: &str) -> Result { let res = Like::insert( &c.conn, NewLike { post_id: self.id, user_id: actor.id, ap_url: id.to_string(), }, )?; res.notify(&c.conn)?; Ok(res) } } impl FromId for Like { type Error = Error; type Object = activity::Like; fn from_db(c: &PlumeRocket, id: &str) -> Result { Like::find_by_ap_url(&c.conn, id) } fn from_activity(c: &PlumeRocket, act: activity::Like) -> Result { let res = Like::insert( &c.conn, NewLike { post_id: Post::from_id( c, &{ let res: String = act.like_props.object_link::()?.into(); res }, None, ) .map_err(|(_, e)| e)? .id, user_id: User::from_id( c, &{ let res: String = act.like_props.actor_link::()?.into(); res }, None, ) .map_err(|(_, e)| e)? .id, ap_url: act.object_props.id_string()?, }, )?; res.notify(&c.conn)?; Ok(res) } } impl AsObject for Like { type Error = Error; type Output = (); fn activity(self, c: &PlumeRocket, actor: User, _id: &str) -> Result<()> { let conn = &*c.conn; if actor.id == self.user_id { diesel::delete(&self).execute(conn)?; // delete associated notification if any if let Ok(notif) = Notification::find(conn, notification_kind::LIKE, self.id) { diesel::delete(¬if).execute(conn)?; } Ok(()) } else { Err(Error::Unauthorized) } } } impl NewLike { pub fn new(p: &Post, u: &User) -> Self { // TODO: this URL is not valid let ap_url = format!("{}/like/{}", u.ap_url, p.ap_url); NewLike { post_id: p.id, user_id: u.id, ap_url, } } }