2 extensions::context::lemmy_context,
3 fetcher::objects::{get_or_fetch_and_insert_comment, get_or_fetch_and_insert_post},
4 get_community_from_to_or_cc,
7 check_object_for_community_or_site_ban,
10 get_or_fetch_and_upsert_person,
11 get_source_markdown_value,
12 set_content_and_source,
19 use activitystreams::{
20 object::{kind::NoteType, ApObject, Note, Tombstone},
24 use anyhow::{anyhow, Context};
25 use lemmy_api_common::blocking;
26 use lemmy_db_queries::{Crud, DbPool};
27 use lemmy_db_schema::{
29 comment::{Comment, CommentForm},
38 utils::{convert_datetime, remove_slurs},
41 use lemmy_websocket::LemmyContext;
44 #[async_trait::async_trait(?Send)]
45 impl ToApub for Comment {
46 type ApubType = NoteExt;
48 async fn to_apub(&self, pool: &DbPool) -> Result<NoteExt, LemmyError> {
49 let mut comment = ApObject::new(Note::new());
51 let creator_id = self.creator_id;
52 let creator = blocking(pool, move |conn| Person::read(conn, creator_id)).await??;
54 let post_id = self.post_id;
55 let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
57 let community_id = post.community_id;
58 let community = blocking(pool, move |conn| Community::read(conn, community_id)).await??;
60 // Add a vector containing some important info to the "in_reply_to" field
61 // [post_ap_id, Option(parent_comment_ap_id)]
62 let mut in_reply_to_vec = vec![post.ap_id.into_inner()];
64 if let Some(parent_id) = self.parent_id {
65 let parent_comment = blocking(pool, move |conn| Comment::read(conn, parent_id)).await??;
67 in_reply_to_vec.push(parent_comment.ap_id.into_inner());
71 // Not needed when the Post is embedded in a collection (like for community outbox)
72 .set_many_contexts(lemmy_context()?)
73 .set_id(self.ap_id.to_owned().into_inner())
74 .set_published(convert_datetime(self.published))
75 // NOTE: included community id for compatibility with lemmy v0.9.9
76 .set_many_tos(vec![community.actor_id.into_inner(), public()])
77 .set_many_in_reply_tos(in_reply_to_vec)
78 .set_attributed_to(creator.actor_id.into_inner());
80 set_content_and_source(&mut comment, &self.content)?;
82 if let Some(u) = self.updated {
83 comment.set_updated(convert_datetime(u));
89 fn to_tombstone(&self) -> Result<Tombstone, LemmyError> {
92 self.ap_id.to_owned().into(),
99 #[async_trait::async_trait(?Send)]
100 impl FromApub for Comment {
101 type ApubType = NoteExt;
103 /// Converts a `Note` to `Comment`.
105 /// If the parent community, post and comment(s) are not known locally, these are also fetched.
108 context: &LemmyContext,
109 expected_domain: Url,
110 request_counter: &mut i32,
111 mod_action_allowed: bool,
112 ) -> Result<Comment, LemmyError> {
113 let comment: Comment = get_object_from_apub(
122 let post_id = comment.post_id;
123 let post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
124 check_object_for_community_or_site_ban(note, post.community_id, context, request_counter)
127 // This is not very efficient because a comment gets inserted just to be deleted right
128 // afterwards, but it seems to be the easiest way to implement it.
129 blocking(context.pool(), move |conn| {
130 Comment::delete(conn, comment.id)
133 Err(anyhow!("Post is locked").into())
140 #[async_trait::async_trait(?Send)]
141 impl FromApubToForm<NoteExt> for CommentForm {
144 context: &LemmyContext,
145 expected_domain: Url,
146 request_counter: &mut i32,
147 _mod_action_allowed: bool,
148 ) -> Result<CommentForm, LemmyError> {
149 let community = get_community_from_to_or_cc(note, context, request_counter).await?;
150 let ap_id = Some(check_object_domain(note, expected_domain, community.local)?);
151 let creator_actor_id = ¬e
153 .context(location_info!())?
154 .as_single_xsd_any_uri()
155 .context(location_info!())?;
158 get_or_fetch_and_upsert_person(creator_actor_id, context, request_counter).await?;
160 let mut in_reply_tos = note
163 .context(location_info!())?
165 .context(location_info!())?
167 .map(|i| i.as_xsd_any_uri().context(""));
168 let post_ap_id = in_reply_tos.next().context(location_info!())??;
170 // This post, or the parent comment might not yet exist on this server yet, fetch them.
171 let post = Box::pin(get_or_fetch_and_insert_post(
178 // The 2nd item, if it exists, is the parent comment apub_id
179 // For deeply nested comments, FromApub automatically gets called recursively
180 let parent_id: Option<CommentId> = match in_reply_tos.next() {
181 Some(parent_comment_uri) => {
182 let parent_comment_ap_id = &parent_comment_uri?;
183 let parent_comment = Box::pin(get_or_fetch_and_insert_comment(
184 &parent_comment_ap_id,
190 Some(parent_comment.id)
195 let content = get_source_markdown_value(note)?.context(location_info!())?;
196 let content_slurs_removed = remove_slurs(&content);
199 creator_id: creator.id,
202 content: content_slurs_removed,
205 published: note.published().map(|u| u.to_owned().naive_local()),
206 updated: note.updated().map(|u| u.to_owned().naive_local()),