]> Untitled Git - lemmy.git/blobdiff - crates/apub/src/activities/create_or_update/comment.rs
Split activity table into sent and received parts (fixes #3103) (#3583)
[lemmy.git] / crates / apub / src / activities / create_or_update / comment.rs
index 451cb867030ba5edfde006956d6e7a680a6a51d8..51b87ed27fc2271230f434973f008c327937d721 100644 (file)
 use crate::{
   activities::{
     check_community_deleted_or_removed,
-    community::{announce::GetCommunity, send_activity_in_community},
-    create_or_update::get_comment_notif_recipients,
+    community::send_activity_in_community,
     generate_activity_id,
-    verify_activity,
     verify_is_public,
     verify_person_in_community,
   },
   activity_lists::AnnouncableActivities,
+  insert_received_activity,
+  mentions::MentionOrValue,
   objects::{comment::ApubComment, community::ApubCommunity, person::ApubPerson},
-  protocol::activities::{create_or_update::comment::CreateOrUpdateComment, CreateOrUpdateType},
+  protocol::{
+    activities::{create_or_update::note::CreateOrUpdateNote, CreateOrUpdateType},
+    InCommunity,
+  },
+  SendActivity,
+};
+use activitypub_federation::{
+  config::Data,
+  fetch::object_id::ObjectId,
+  kinds::public,
+  protocol::verification::verify_domains_match,
+  traits::{ActivityHandler, Actor, Object},
 };
-use activitystreams_kinds::public;
-use lemmy_api_common::{blocking, check_post_deleted_or_removed};
-use lemmy_apub_lib::{
-  data::Data,
-  object_id::ObjectId,
-  traits::{ActivityHandler, ActorType, ApubObject},
-  verify::verify_domains_match,
+use lemmy_api_common::{
+  build_response::send_local_notifs,
+  comment::{CommentResponse, CreateComment, EditComment},
+  context::LemmyContext,
+  utils::{check_post_deleted_or_removed, is_mod_or_admin},
 };
 use lemmy_db_schema::{
-  source::{community::Community, post::Post},
-  traits::Crud,
+  aggregates::structs::CommentAggregates,
+  newtypes::PersonId,
+  source::{
+    comment::{Comment, CommentLike, CommentLikeForm},
+    community::Community,
+    person::Person,
+    post::Post,
+  },
+  traits::{Crud, Likeable},
 };
-use lemmy_utils::LemmyError;
-use lemmy_websocket::{send::send_comment_ws_message, LemmyContext, UserOperationCrud};
-use tracing::info;
-
-impl CreateOrUpdateComment {
-  #[tracing::instrument(skip(comment, actor, kind, context))]
-  pub async fn send(
-    comment: ApubComment,
-    actor: &ApubPerson,
+use lemmy_utils::{error::LemmyError, utils::mention::scrape_text_for_mentions};
+use url::Url;
+
+#[async_trait::async_trait]
+impl SendActivity for CreateComment {
+  type Response = CommentResponse;
+
+  async fn send_activity(
+    _request: &Self,
+    response: &Self::Response,
+    context: &Data<LemmyContext>,
+  ) -> Result<(), LemmyError> {
+    CreateOrUpdateNote::send(
+      &response.comment_view.comment,
+      response.comment_view.creator.id,
+      CreateOrUpdateType::Create,
+      context,
+    )
+    .await
+  }
+}
+
+#[async_trait::async_trait]
+impl SendActivity for EditComment {
+  type Response = CommentResponse;
+
+  async fn send_activity(
+    _request: &Self,
+    response: &Self::Response,
+    context: &Data<LemmyContext>,
+  ) -> Result<(), LemmyError> {
+    CreateOrUpdateNote::send(
+      &response.comment_view.comment,
+      response.comment_view.creator.id,
+      CreateOrUpdateType::Update,
+      context,
+    )
+    .await
+  }
+}
+
+impl CreateOrUpdateNote {
+  #[tracing::instrument(skip(comment, person_id, kind, context))]
+  async fn send(
+    comment: &Comment,
+    person_id: PersonId,
     kind: CreateOrUpdateType,
-    context: &LemmyContext,
-    request_counter: &mut i32,
+    context: &Data<LemmyContext>,
   ) -> Result<(), LemmyError> {
     // TODO: might be helpful to add a comment method to retrieve community directly
     let post_id = comment.post_id;
-    let post = blocking(context.pool(), move |conn| Post::read(conn, post_id)).await??;
+    let post = Post::read(&mut context.pool(), post_id).await?;
     let community_id = post.community_id;
-    let community: ApubCommunity = blocking(context.pool(), move |conn| {
-      Community::read(conn, community_id)
-    })
-    .await??
-    .into();
+    let person: ApubPerson = Person::read(&mut context.pool(), person_id).await?.into();
+    let community: ApubCommunity = Community::read(&mut context.pool(), community_id)
+      .await?
+      .into();
 
     let id = generate_activity_id(
       kind.clone(),
       &context.settings().get_protocol_and_hostname(),
     )?;
-    info!("Sending Create/Comment for {} as {}", comment.ap_id, id);
-    let note = comment.into_apub(context).await?;
+    let note = ApubComment(comment.clone()).into_json(context).await?;
 
-    let create_or_update = CreateOrUpdateComment {
-      actor: ObjectId::new(actor.actor_id()),
+    let create_or_update = CreateOrUpdateNote {
+      actor: person.id().into(),
       to: vec![public()],
       cc: note.cc.clone(),
       tag: note.tag.clone(),
       object: note,
       kind,
       id: id.clone(),
-      unparsed: Default::default(),
+      audience: Some(community.id().into()),
     };
 
     let tagged_users: Vec<ObjectId<ApubPerson>> = create_or_update
       .tag
       .iter()
+      .filter_map(|t| {
+        if let MentionOrValue::Mention(t) = t {
+          Some(t)
+        } else {
+          None
+        }
+      })
       .map(|t| t.href.clone())
-      .map(ObjectId::new)
+      .map(ObjectId::from)
       .collect();
     let mut inboxes = vec![];
     for t in tagged_users {
-      let person = t
-        .dereference(context, context.client(), request_counter)
-        .await?;
-      inboxes.push(person.shared_inbox_or_inbox_url());
+      let person = t.dereference(context).await?;
+      inboxes.push(person.shared_inbox_or_inbox());
     }
 
     let activity = AnnouncableActivities::CreateOrUpdateComment(create_or_update);
-    send_activity_in_community(activity, &id, actor, &community, inboxes, context).await
+    send_activity_in_community(activity, &person, &community, inboxes, false, context).await
   }
 }
 
-#[async_trait::async_trait(?Send)]
-impl ActivityHandler for CreateOrUpdateComment {
+#[async_trait::async_trait]
+impl ActivityHandler for CreateOrUpdateNote {
   type DataType = LemmyContext;
+  type Error = LemmyError;
+
+  fn id(&self) -> &Url {
+    &self.id
+  }
+
+  fn actor(&self) -> &Url {
+    self.actor.inner()
+  }
 
   #[tracing::instrument(skip_all)]
-  async fn verify(
-    &self,
-    context: &Data<LemmyContext>,
-    request_counter: &mut i32,
-  ) -> Result<(), LemmyError> {
+  async fn verify(&self, context: &Data<Self::DataType>) -> Result<(), LemmyError> {
+    insert_received_activity(&self.id, context).await?;
     verify_is_public(&self.to, &self.cc)?;
-    let post = self.object.get_parents(context, request_counter).await?.0;
-    let community = self.get_community(context, request_counter).await?;
+    let post = self.object.get_parents(context).await?.0;
+    let community = self.community(context).await?;
 
-    verify_activity(&self.id, self.actor.inner(), &context.settings())?;
-    verify_person_in_community(&self.actor, &community, context, request_counter).await?;
+    verify_person_in_community(&self.actor, &community, context).await?;
     verify_domains_match(self.actor.inner(), self.object.id.inner())?;
     check_community_deleted_or_removed(&community)?;
     check_post_deleted_or_removed(&post)?;
 
-    ApubComment::verify(&self.object, self.actor.inner(), context, request_counter).await?;
+    ApubComment::verify(&self.object, self.actor.inner(), context).await?;
     Ok(())
   }
 
   #[tracing::instrument(skip_all)]
-  async fn receive(
-    self,
-    context: &Data<LemmyContext>,
-    request_counter: &mut i32,
-  ) -> Result<(), LemmyError> {
-    let comment = ApubComment::from_apub(self.object, context, request_counter).await?;
-    let do_send_email = self.kind == CreateOrUpdateType::Create;
-    let recipients = get_comment_notif_recipients(
-      &self.actor,
-      &comment,
-      do_send_email,
-      context,
-      request_counter,
-    )
-    .await?;
-    let notif_type = match self.kind {
-      CreateOrUpdateType::Create => UserOperationCrud::CreateComment,
-      CreateOrUpdateType::Update => UserOperationCrud::EditComment,
+  async fn receive(self, context: &Data<Self::DataType>) -> Result<(), LemmyError> {
+    // Need to do this check here instead of Note::from_json because we need the person who
+    // send the activity, not the comment author.
+    let existing_comment = self.object.id.dereference_local(context).await.ok();
+    if let (Some(distinguished), Some(existing_comment)) =
+      (self.object.distinguished, existing_comment)
+    {
+      if distinguished != existing_comment.distinguished {
+        let creator = self.actor.dereference(context).await?;
+        let (post, _) = self.object.get_parents(context).await?;
+        is_mod_or_admin(&mut context.pool(), creator.id, post.community_id).await?;
+      }
+    }
+
+    let comment = ApubComment::from_json(self.object, context).await?;
+
+    // author likes their own comment by default
+    let like_form = CommentLikeForm {
+      comment_id: comment.id,
+      post_id: comment.post_id,
+      person_id: comment.creator_id,
+      score: 1,
     };
-    send_comment_ws_message(
-      comment.id, notif_type, None, None, None, recipients, context,
-    )
-    .await?;
-    Ok(())
-  }
-}
+    CommentLike::like(&mut context.pool(), &like_form).await?;
 
-#[async_trait::async_trait(?Send)]
-impl GetCommunity for CreateOrUpdateComment {
-  #[tracing::instrument(skip_all)]
-  async fn get_community(
-    &self,
-    context: &LemmyContext,
-    request_counter: &mut i32,
-  ) -> Result<ApubCommunity, LemmyError> {
-    let post = self.object.get_parents(context, request_counter).await?.0;
-    let community = blocking(context.pool(), move |conn| {
-      Community::read(conn, post.community_id)
-    })
-    .await??;
-    Ok(community.into())
+    // Calculate initial hot_rank
+    CommentAggregates::update_hot_rank(&mut context.pool(), comment.id).await?;
+
+    let do_send_email = self.kind == CreateOrUpdateType::Create;
+    let post_id = comment.post_id;
+    let post = Post::read(&mut context.pool(), post_id).await?;
+    let actor = self.actor.dereference(context).await?;
+
+    // Note:
+    // Although mentions could be gotten from the post tags (they are included there), or the ccs,
+    // Its much easier to scrape them from the comment body, since the API has to do that
+    // anyway.
+    // TODO: for compatibility with other projects, it would be much better to read this from cc or tags
+    let mentions = scrape_text_for_mentions(&comment.content);
+    send_local_notifs(mentions, &comment.0, &actor, &post, do_send_email, context).await?;
+    Ok(())
   }
 }