mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-11-05 20:15:01 +00:00
3da47352be
Remove remaining usages of unwrap() from activitypub code Remove most usage of Option::unwrap() from activitypub code Co-authored-by: Felix Ableitner <me@nutomic.com> Reviewed-on: https://yerbamate.dev/LemmyNet/lemmy/pulls/80
245 lines
6.8 KiB
Rust
245 lines
6.8 KiB
Rust
use crate::{
|
|
apub::{
|
|
activities::{generate_activity_id, send_activity},
|
|
check_actor_domain,
|
|
check_is_apub_id_valid,
|
|
create_tombstone,
|
|
fetcher::get_or_fetch_and_upsert_user,
|
|
insert_activity,
|
|
ActorType,
|
|
ApubObjectType,
|
|
FromApub,
|
|
ToApub,
|
|
},
|
|
blocking,
|
|
DbPool,
|
|
LemmyError,
|
|
};
|
|
use activitystreams::{
|
|
activity::{
|
|
kind::{CreateType, DeleteType, UndoType, UpdateType},
|
|
Create,
|
|
Delete,
|
|
Undo,
|
|
Update,
|
|
},
|
|
context,
|
|
object::{kind::NoteType, Note, Tombstone},
|
|
prelude::*,
|
|
};
|
|
use actix_web::client::Client;
|
|
use anyhow::Context;
|
|
use lemmy_db::{
|
|
private_message::{PrivateMessage, PrivateMessageForm},
|
|
user::User_,
|
|
Crud,
|
|
};
|
|
use lemmy_utils::{convert_datetime, location_info};
|
|
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)
|
|
}
|
|
}
|
|
|
|
#[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,
|
|
expected_domain: Option<Url>,
|
|
) -> Result<PrivateMessageForm, LemmyError> {
|
|
let creator_actor_id = note
|
|
.attributed_to()
|
|
.context(location_info!())?
|
|
.clone()
|
|
.single_xsd_any_uri()
|
|
.context(location_info!())?;
|
|
|
|
let creator = get_or_fetch_and_upsert_user(&creator_actor_id, client, pool).await?;
|
|
let recipient_actor_id = note
|
|
.to()
|
|
.context(location_info!())?
|
|
.clone()
|
|
.single_xsd_any_uri()
|
|
.context(location_info!())?;
|
|
let recipient = get_or_fetch_and_upsert_user(&recipient_actor_id, client, pool).await?;
|
|
let ap_id = note.id_unchecked().context(location_info!())?.to_string();
|
|
check_is_apub_id_valid(&Url::parse(&ap_id)?)?;
|
|
|
|
Ok(PrivateMessageForm {
|
|
creator_id: creator.id,
|
|
recipient_id: recipient.id,
|
|
content: note
|
|
.content()
|
|
.context(location_info!())?
|
|
.as_single_xsd_string()
|
|
.context(location_info!())?
|
|
.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: check_actor_domain(note, expected_domain)?,
|
|
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 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 = recipient.get_inbox_url()?;
|
|
create
|
|
.set_context(context())
|
|
.set_id(generate_activity_id(CreateType::Create)?)
|
|
.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 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 = recipient.get_inbox_url()?;
|
|
update
|
|
.set_context(context())
|
|
.set_id(generate_activity_id(UpdateType::Update)?)
|
|
.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 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 = recipient.get_inbox_url()?;
|
|
delete
|
|
.set_context(context())
|
|
.set_id(generate_activity_id(DeleteType::Delete)?)
|
|
.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 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 = recipient.get_inbox_url()?;
|
|
delete
|
|
.set_context(context())
|
|
.set_id(generate_activity_id(DeleteType::Delete)?)
|
|
.set_to(to.clone());
|
|
|
|
// 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(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!()
|
|
}
|
|
}
|