lemmy/server/src/apub/private_message.rs

use crate::{
  apub::{
    activities::send_activity, create_tombstone, fetcher::get_or_fetch_and_upsert_remote_user,
    insert_activity, ApubObjectType, FromApub, ToApub,
  },
  blocking, DbPool, LemmyError,
};
use activitystreams_new::{
  activity::{Create, Delete, Undo, Update},
  context,
  object::{kind::NoteType, Note, Tombstone},
  prelude::*,
};
use actix_web::client::Client;
use lemmy_db::{
  private_message::{PrivateMessage, PrivateMessageForm},
  user::User_,
  Crud,
};
use lemmy_utils::convert_datetime;
use url::Url;

#[async_trait::async_trait(?Send)]
impl ToApub for PrivateMessage {
  type Response = Note;

  async fn to_apub(&self, pool: &DbPool) -> Result<Note, LemmyError> {
    let mut private_message = Note::new();

    let creator_id = self.creator_id;
    let creator = blocking(pool, move |conn| User_::read(conn, creator_id)).await??;

    let recipient_id = self.recipient_id;
    let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;

    private_message
      .set_context(context())
      .set_id(Url::parse(&self.ap_id.to_owned())?)
      .set_published(convert_datetime(self.published))
      .set_content(self.content.to_owned())
      .set_to(recipient.actor_id)
      .set_attributed_to(creator.actor_id);

    if let Some(u) = self.updated {
      private_message.set_updated(convert_datetime(u));
    }

    Ok(private_message)
  }

  fn to_tombstone(&self) -> Result<Tombstone, LemmyError> {
    create_tombstone(
      self.deleted,
      &self.ap_id,
      self.updated,
      NoteType::Note.to_string(),
    )
  }
}

#[async_trait::async_trait(?Send)]
impl FromApub for PrivateMessageForm {
  type ApubType = Note;

  /// Parse an ActivityPub note received from another instance into a Lemmy Private message
  async fn from_apub(
    note: &Note,
    client: &Client,
    pool: &DbPool,
    actor_id: &Url,
  ) -> Result<PrivateMessageForm, LemmyError> {
    let creator_actor_id = note
      .attributed_to()
      .unwrap()
      .clone()
      .single_xsd_any_uri()
      .unwrap();

    let creator = get_or_fetch_and_upsert_remote_user(&creator_actor_id, client, pool).await?;

    let recipient_actor_id = note.to().unwrap().clone().single_xsd_any_uri().unwrap();

    let recipient = get_or_fetch_and_upsert_remote_user(&recipient_actor_id, client, pool).await?;

    Ok(PrivateMessageForm {
      creator_id: creator.id,
      recipient_id: recipient.id,
      content: note
        .content()
        .unwrap()
        .as_single_xsd_string()
        .unwrap()
        .to_string(),
      published: note.published().map(|u| u.to_owned().naive_local()),
      updated: note.updated().map(|u| u.to_owned().naive_local()),
      deleted: None,
      read: None,
      ap_id: note.id(actor_id.domain().unwrap())?.unwrap().to_string(),
      local: false,
    })
  }
}

#[async_trait::async_trait(?Send)]
impl ApubObjectType for PrivateMessage {
  /// Send out information about a newly created private message
  async fn send_create(
    &self,
    creator: &User_,
    client: &Client,
    pool: &DbPool,
  ) -> Result<(), LemmyError> {
    let note = self.to_apub(pool).await?;
    let id = format!("{}/create/{}", self.ap_id, uuid::Uuid::new_v4());

    let recipient_id = self.recipient_id;
    let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;

    let mut create = Create::new(creator.actor_id.to_owned(), note.into_any_base()?);
    let to = format!("{}/inbox", recipient.actor_id);
    create
      .set_context(context())
      .set_id(Url::parse(&id)?)
      .set_to(to.clone());

    insert_activity(creator.id, create.clone(), true, pool).await?;

    send_activity(client, &create.into_any_base()?, creator, vec![to]).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 id = format!("{}/update/{}", self.ap_id, uuid::Uuid::new_v4());

    let recipient_id = self.recipient_id;
    let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;

    let mut update = Update::new(creator.actor_id.to_owned(), note.into_any_base()?);
    let to = format!("{}/inbox", recipient.actor_id);
    update
      .set_context(context())
      .set_id(Url::parse(&id)?)
      .set_to(to.clone());

    insert_activity(creator.id, update.clone(), true, pool).await?;

    send_activity(client, &update.into_any_base()?, creator, vec![to]).await?;
    Ok(())
  }

  async fn send_delete(
    &self,
    creator: &User_,
    client: &Client,
    pool: &DbPool,
  ) -> Result<(), LemmyError> {
    let note = self.to_apub(pool).await?;
    let id = format!("{}/delete/{}", self.ap_id, uuid::Uuid::new_v4());

    let recipient_id = self.recipient_id;
    let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;

    let mut delete = Delete::new(creator.actor_id.to_owned(), note.into_any_base()?);
    let to = format!("{}/inbox", recipient.actor_id);
    delete
      .set_context(context())
      .set_id(Url::parse(&id)?)
      .set_to(to.clone());

    insert_activity(creator.id, delete.clone(), true, pool).await?;

    send_activity(client, &delete.into_any_base()?, creator, vec![to]).await?;
    Ok(())
  }

  async fn send_undo_delete(
    &self,
    creator: &User_,
    client: &Client,
    pool: &DbPool,
  ) -> Result<(), LemmyError> {
    let note = self.to_apub(pool).await?;
    let id = format!("{}/delete/{}", self.ap_id, uuid::Uuid::new_v4());

    let recipient_id = self.recipient_id;
    let recipient = blocking(pool, move |conn| User_::read(conn, recipient_id)).await??;

    let mut delete = Delete::new(creator.actor_id.to_owned(), note.into_any_base()?);
    let to = format!("{}/inbox", recipient.actor_id);
    delete
      .set_context(context())
      .set_id(Url::parse(&id)?)
      .set_to(to.clone());

    // TODO
    // Undo that fake activity
    let undo_id = format!("{}/undo/delete/{}", self.ap_id, uuid::Uuid::new_v4());
    let mut undo = Undo::new(creator.actor_id.to_owned(), delete.into_any_base()?);
    undo
      .set_context(context())
      .set_id(Url::parse(&undo_id)?)
      .set_to(to.clone());

    insert_activity(creator.id, undo.clone(), true, pool).await?;

    send_activity(client, &undo.into_any_base()?, creator, vec![to]).await?;
    Ok(())
  }

  async fn send_remove(
    &self,
    _mod_: &User_,
    _client: &Client,
    _pool: &DbPool,
  ) -> Result<(), LemmyError> {
    unimplemented!()
  }

  async fn send_undo_remove(
    &self,
    _mod_: &User_,
    _client: &Client,
    _pool: &DbPool,
  ) -> Result<(), LemmyError> {
    unimplemented!()
  }
}