use crate::{ api::check_slurs, apub::{ activities::{generate_activity_id, send_activity_to_community}, check_actor_domain, create_apub_response, create_apub_tombstone_response, create_tombstone, fetch_webfinger_url, fetcher::{ get_or_fetch_and_insert_comment, get_or_fetch_and_insert_post, get_or_fetch_and_upsert_user, }, ActorType, ApubLikeableType, ApubObjectType, FromApub, ToApub, }, blocking, routes::DbPoolParam, DbPool, LemmyError, }; use activitystreams::{ activity::{ kind::{CreateType, DeleteType, DislikeType, LikeType, RemoveType, UndoType, UpdateType}, Create, Delete, Dislike, Like, Remove, Undo, Update, }, base::AnyBase, context, link::Mention, object::{kind::NoteType, Note, Tombstone}, prelude::*, public, }; use actix_web::{body::Body, client::Client, web::Path, HttpResponse}; use itertools::Itertools; use lemmy_db::{ comment::{Comment, CommentForm}, community::Community, post::Post, user::User_, Crud, }; use lemmy_utils::{convert_datetime, scrape_text_for_mentions, MentionData}; use log::debug; use serde::Deserialize; use serde_json::Error; use url::Url; #[derive(Deserialize)] pub struct CommentQuery { comment_id: String, } /// Return the post json over HTTP. pub async fn get_apub_comment( info: Path, db: DbPoolParam, ) -> Result, LemmyError> { let id = info.comment_id.parse::()?; let comment = blocking(&db, move |conn| Comment::read(conn, id)).await??; if !comment.deleted { Ok(create_apub_response(&comment.to_apub(&db).await?)) } else { Ok(create_apub_tombstone_response(&comment.to_tombstone()?)) } } #[async_trait::async_trait(?Send)] impl ToApub for Comment { type Response = Note; async fn to_apub(&self, pool: &DbPool) -> Result { let mut comment = Note::new(); let creator_id = self.creator_id; let creator = blocking(pool, move |conn| User_::read(conn, creator_id)).await??; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; // Add a vector containing some important info to the "in_reply_to" field // [post_ap_id, Option(parent_comment_ap_id)] let mut in_reply_to_vec = vec![post.ap_id]; if let Some(parent_id) = self.parent_id { let parent_comment = blocking(pool, move |conn| Comment::read(conn, parent_id)).await??; in_reply_to_vec.push(parent_comment.ap_id); } comment // Not needed when the Post is embedded in a collection (like for community outbox) .set_context(context()) .set_id(Url::parse(&self.ap_id)?) .set_published(convert_datetime(self.published)) .set_to(community.actor_id) .set_many_in_reply_tos(in_reply_to_vec) .set_content(self.content.to_owned()) .set_attributed_to(creator.actor_id); if let Some(u) = self.updated { comment.set_updated(convert_datetime(u)); } Ok(comment) } fn to_tombstone(&self) -> Result { create_tombstone(self.deleted, &self.ap_id, self.updated, NoteType::Note) } } #[async_trait::async_trait(?Send)] impl FromApub for CommentForm { type ApubType = Note; /// Parse an ActivityPub note received from another instance into a Lemmy comment async fn from_apub( note: &Note, client: &Client, pool: &DbPool, expected_domain: Option, ) -> Result { let creator_actor_id = ¬e .attributed_to() .unwrap() .as_single_xsd_any_uri() .unwrap(); let creator = get_or_fetch_and_upsert_user(creator_actor_id, client, pool).await?; let mut in_reply_tos = note .in_reply_to() .as_ref() .unwrap() .as_many() .unwrap() .iter() .map(|i| i.as_xsd_any_uri().unwrap()); let post_ap_id = in_reply_tos.next().unwrap(); // This post, or the parent comment might not yet exist on this server yet, fetch them. let post = get_or_fetch_and_insert_post(&post_ap_id, client, pool).await?; // The 2nd item, if it exists, is the parent comment apub_id // For deeply nested comments, FromApub automatically gets called recursively let parent_id: Option = match in_reply_tos.next() { Some(parent_comment_uri) => { let parent_comment_ap_id = &parent_comment_uri; let parent_comment = get_or_fetch_and_insert_comment(&parent_comment_ap_id, client, pool).await?; Some(parent_comment.id) } None => None, }; let content = note .content() .unwrap() .as_single_xsd_string() .unwrap() .to_string(); check_slurs(&content)?; Ok(CommentForm { creator_id: creator.id, post_id: post.id, parent_id, content, removed: None, read: None, published: note.published().map(|u| u.to_owned().naive_local()), updated: note.updated().map(|u| u.to_owned().naive_local()), deleted: None, ap_id: check_actor_domain(note, expected_domain)?, local: false, }) } } #[async_trait::async_trait(?Send)] impl ApubObjectType for Comment { /// Send out information about a newly created comment, to the followers of the community. async fn send_create( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let maa = collect_non_local_mentions_and_addresses(&self.content, &community, client, pool).await?; let mut create = Create::new(creator.actor_id.to_owned(), note.into_any_base()?); create .set_context(context()) .set_id(generate_activity_id(CreateType::Create)?) .set_to(public()) .set_many_ccs(maa.addressed_ccs.to_owned()) // Set the mention tags .set_many_tags(maa.get_tags()?); send_activity_to_community( &creator, &community, maa.inboxes, create.into_any_base()?, client, pool, ) .await?; Ok(()) } /// Send out information about an edited post, to the followers of the community. async fn send_update( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let maa = collect_non_local_mentions_and_addresses(&self.content, &community, client, pool).await?; let mut update = Update::new(creator.actor_id.to_owned(), note.into_any_base()?); update .set_context(context()) .set_id(generate_activity_id(UpdateType::Update)?) .set_to(public()) .set_many_ccs(maa.addressed_ccs.to_owned()) // Set the mention tags .set_many_tags(maa.get_tags()?); send_activity_to_community( &creator, &community, maa.inboxes, update.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let mut delete = Delete::new(creator.actor_id.to_owned(), note.into_any_base()?); delete .set_context(context()) .set_id(generate_activity_id(DeleteType::Delete)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &creator, &community, vec![community.get_shared_inbox_url()], delete.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_undo_delete( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; // Generate a fake delete activity, with the correct object let mut delete = Delete::new(creator.actor_id.to_owned(), note.into_any_base()?); delete .set_context(context()) .set_id(generate_activity_id(DeleteType::Delete)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); // Undo that fake activity let mut undo = Undo::new(creator.actor_id.to_owned(), delete.into_any_base()?); undo .set_context(context()) .set_id(generate_activity_id(UndoType::Undo)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &creator, &community, vec![community.get_shared_inbox_url()], undo.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let mut remove = Remove::new(mod_.actor_id.to_owned(), note.into_any_base()?); remove .set_context(context()) .set_id(generate_activity_id(RemoveType::Remove)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &mod_, &community, vec![community.get_shared_inbox_url()], remove.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_undo_remove( &self, mod_: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; // Generate a fake delete activity, with the correct object let mut remove = Remove::new(mod_.actor_id.to_owned(), note.into_any_base()?); remove .set_context(context()) .set_id(generate_activity_id(RemoveType::Remove)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); // Undo that fake activity let mut undo = Undo::new(mod_.actor_id.to_owned(), remove.into_any_base()?); undo .set_context(context()) .set_id(generate_activity_id(UndoType::Undo)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &mod_, &community, vec![community.get_shared_inbox_url()], undo.into_any_base()?, client, pool, ) .await?; Ok(()) } } #[async_trait::async_trait(?Send)] impl ApubLikeableType for Comment { async fn send_like( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let mut like = Like::new(creator.actor_id.to_owned(), note.into_any_base()?); like .set_context(context()) .set_id(generate_activity_id(LikeType::Like)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &creator, &community, vec![community.get_shared_inbox_url()], like.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_dislike( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let mut dislike = Dislike::new(creator.actor_id.to_owned(), note.into_any_base()?); dislike .set_context(context()) .set_id(generate_activity_id(DislikeType::Dislike)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &creator, &community, vec![community.get_shared_inbox_url()], dislike.into_any_base()?, client, pool, ) .await?; Ok(()) } async fn send_undo_like( &self, creator: &User_, client: &Client, pool: &DbPool, ) -> Result<(), LemmyError> { let note = self.to_apub(pool).await?; let post_id = self.post_id; let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??; let community_id = post.community_id; let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??; let mut like = Like::new(creator.actor_id.to_owned(), note.into_any_base()?); like .set_context(context()) .set_id(generate_activity_id(DislikeType::Dislike)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); // Undo that fake activity let mut undo = Undo::new(creator.actor_id.to_owned(), like.into_any_base()?); undo .set_context(context()) .set_id(generate_activity_id(UndoType::Undo)?) .set_to(public()) .set_many_ccs(vec![community.get_followers_url()]); send_activity_to_community( &creator, &community, vec![community.get_shared_inbox_url()], undo.into_any_base()?, client, pool, ) .await?; Ok(()) } } struct MentionsAndAddresses { addressed_ccs: Vec, inboxes: Vec, tags: Vec, } impl MentionsAndAddresses { fn get_tags(&self) -> Result, Error> { self .tags .iter() .map(|t| t.to_owned().into_any_base()) .collect::, Error>>() } } /// This takes a comment, and builds a list of to_addresses, inboxes, /// and mention tags, so they know where to be sent to. /// Addresses are the users / addresses that go in the cc field. async fn collect_non_local_mentions_and_addresses( content: &str, community: &Community, client: &Client, pool: &DbPool, ) -> Result { let mut addressed_ccs = vec![community.get_followers_url()]; // Add the mention tag let mut tags = Vec::new(); // Get the inboxes for any mentions let mentions = scrape_text_for_mentions(&content) .into_iter() // Filter only the non-local ones .filter(|m| !m.is_local()) .collect::>(); let mut mention_inboxes = Vec::new(); for mention in &mentions { // TODO should it be fetching it every time? if let Ok(actor_id) = fetch_webfinger_url(mention, client).await { debug!("mention actor_id: {}", actor_id); addressed_ccs.push(actor_id.to_owned().to_string()); let mention_user = get_or_fetch_and_upsert_user(&actor_id, client, pool).await?; let shared_inbox = mention_user.get_shared_inbox_url(); mention_inboxes.push(shared_inbox); let mut mention_tag = Mention::new(); mention_tag.set_href(actor_id).set_name(mention.full_name()); tags.push(mention_tag); } } let mut inboxes = vec![community.get_shared_inbox_url()]; inboxes.extend(mention_inboxes); inboxes = inboxes.into_iter().unique().collect(); Ok(MentionsAndAddresses { addressed_ccs, inboxes, tags, }) }