use crate::{ comments::Comment, notifications::*, posts::Post, schema::mentions, users::User, Connection, Error, PlumeRocket, Result, }; use activitypub::link; use diesel::{self, ExpressionMethods, QueryDsl, RunQueryDsl}; use plume_common::activity_pub::inbox::AsActor; #[derive(Clone, Queryable, Identifiable)] pub struct Mention { pub id: i32, pub mentioned_id: i32, pub post_id: Option, pub comment_id: Option, } #[derive(Insertable)] #[table_name = "mentions"] pub struct NewMention { pub mentioned_id: i32, pub post_id: Option, pub comment_id: Option, } impl Mention { insert!(mentions, NewMention); get!(mentions); list_by!(mentions, list_for_user, mentioned_id as i32); list_by!(mentions, list_for_post, post_id as i32); list_by!(mentions, list_for_comment, comment_id as i32); pub fn get_mentioned(&self, conn: &Connection) -> Result { User::get(conn, self.mentioned_id) } pub fn get_post(&self, conn: &Connection) -> Result { self.post_id .ok_or(Error::NotFound) .and_then(|id| Post::get(conn, id)) } pub fn get_comment(&self, conn: &Connection) -> Result { self.comment_id .ok_or(Error::NotFound) .and_then(|id| Comment::get(conn, id)) } pub fn get_user(&self, conn: &Connection) -> Result { match self.get_post(conn) { Ok(p) => Ok(p.get_authors(conn)?.into_iter().next()?), Err(_) => self.get_comment(conn).and_then(|c| c.get_author(conn)), } } pub fn build_activity(c: &PlumeRocket, ment: &str) -> Result { let user = User::find_by_fqn(c, ment)?; let mut mention = link::Mention::default(); mention.link_props.set_href_string(user.ap_url)?; mention.link_props.set_name_string(format!("@{}", ment))?; Ok(mention) } pub fn to_activity(&self, conn: &Connection) -> Result { let user = self.get_mentioned(conn)?; let mut mention = link::Mention::default(); mention.link_props.set_href_string(user.ap_url.clone())?; mention .link_props .set_name_string(format!("@{}", user.fqn))?; Ok(mention) } pub fn from_activity( conn: &Connection, ment: &link::Mention, inside: i32, in_post: bool, notify: bool, ) -> Result { let ap_url = ment.link_props.href_string().ok()?; let mentioned = User::find_by_ap_url(conn, &ap_url)?; if in_post { Post::get(conn, inside).and_then(|post| { let res = Mention::insert( conn, NewMention { mentioned_id: mentioned.id, post_id: Some(post.id), comment_id: None, }, )?; if notify { res.notify(conn)?; } Ok(res) }) } else { Comment::get(conn, inside).and_then(|comment| { let res = Mention::insert( conn, NewMention { mentioned_id: mentioned.id, post_id: None, comment_id: Some(comment.id), }, )?; if notify { res.notify(conn)?; } Ok(res) }) } } pub fn delete(&self, conn: &Connection) -> Result<()> { //find related notifications and delete them if let Ok(n) = Notification::find(conn, notification_kind::MENTION, self.id) { n.delete(conn)?; } diesel::delete(self) .execute(conn) .map(|_| ()) .map_err(Error::from) } fn notify(&self, conn: &Connection) -> Result<()> { let m = self.get_mentioned(conn)?; if m.is_local() { Notification::insert( conn, NewNotification { kind: notification_kind::MENTION.to_string(), object_id: self.id, user_id: m.id, }, ) .map(|_| ()) } else { Ok(()) } } }