mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-06-01 00:28:05 +00:00
99 lines
3.1 KiB
Rust
99 lines
3.1 KiB
Rust
use crate::{
|
|
activities::{generate_activity_id, verify_activity, verify_person, CreateOrUpdateType},
|
|
context::lemmy_context,
|
|
fetcher::object_id::ObjectId,
|
|
objects::{private_message::Note, FromApub, ToApub},
|
|
send_lemmy_activity,
|
|
};
|
|
use activitystreams::{base::AnyBase, primitives::OneOrMany, unparsed::Unparsed};
|
|
use lemmy_api_common::blocking;
|
|
use lemmy_apub_lib::{
|
|
data::Data,
|
|
traits::{ActivityFields, ActivityHandler, ActorType},
|
|
verify::verify_domains_match,
|
|
};
|
|
use lemmy_db_schema::{
|
|
source::{person::Person, private_message::PrivateMessage},
|
|
traits::Crud,
|
|
};
|
|
use lemmy_utils::LemmyError;
|
|
use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
|
|
use serde::{Deserialize, Serialize};
|
|
use url::Url;
|
|
|
|
#[derive(Clone, Debug, Deserialize, Serialize, ActivityFields)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct CreateOrUpdatePrivateMessage {
|
|
#[serde(rename = "@context")]
|
|
pub context: OneOrMany<AnyBase>,
|
|
id: Url,
|
|
actor: ObjectId<Person>,
|
|
to: ObjectId<Person>,
|
|
object: Note,
|
|
#[serde(rename = "type")]
|
|
kind: CreateOrUpdateType,
|
|
#[serde(flatten)]
|
|
pub unparsed: Unparsed,
|
|
}
|
|
|
|
impl CreateOrUpdatePrivateMessage {
|
|
pub async fn send(
|
|
private_message: &PrivateMessage,
|
|
actor: &Person,
|
|
kind: CreateOrUpdateType,
|
|
context: &LemmyContext,
|
|
) -> Result<(), LemmyError> {
|
|
let recipient_id = private_message.recipient_id;
|
|
let recipient =
|
|
blocking(context.pool(), move |conn| Person::read(conn, recipient_id)).await??;
|
|
|
|
let id = generate_activity_id(
|
|
kind.clone(),
|
|
&context.settings().get_protocol_and_hostname(),
|
|
)?;
|
|
let create_or_update = CreateOrUpdatePrivateMessage {
|
|
context: lemmy_context(),
|
|
id: id.clone(),
|
|
actor: ObjectId::new(actor.actor_id()),
|
|
to: ObjectId::new(recipient.actor_id()),
|
|
object: private_message.to_apub(context.pool()).await?,
|
|
kind,
|
|
unparsed: Default::default(),
|
|
};
|
|
let inbox = vec![recipient.shared_inbox_or_inbox_url()];
|
|
send_lemmy_activity(context, &create_or_update, &id, actor, inbox, true).await
|
|
}
|
|
}
|
|
#[async_trait::async_trait(?Send)]
|
|
impl ActivityHandler for CreateOrUpdatePrivateMessage {
|
|
type DataType = LemmyContext;
|
|
async fn verify(
|
|
&self,
|
|
context: &Data<LemmyContext>,
|
|
request_counter: &mut i32,
|
|
) -> Result<(), LemmyError> {
|
|
verify_activity(self, &context.settings())?;
|
|
verify_person(&self.actor, context, request_counter).await?;
|
|
verify_domains_match(self.actor.inner(), self.object.id_unchecked())?;
|
|
self.object.verify(context, request_counter).await?;
|
|
Ok(())
|
|
}
|
|
|
|
async fn receive(
|
|
self,
|
|
context: &Data<LemmyContext>,
|
|
request_counter: &mut i32,
|
|
) -> Result<(), LemmyError> {
|
|
let private_message =
|
|
PrivateMessage::from_apub(&self.object, context, self.actor.inner(), request_counter).await?;
|
|
|
|
let notif_type = match self.kind {
|
|
CreateOrUpdateType::Create => UserOperationCrud::CreatePrivateMessage,
|
|
CreateOrUpdateType::Update => UserOperationCrud::EditPrivateMessage,
|
|
};
|
|
send_pm_ws_message(private_message.id, notif_type, None, context).await?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|