use crate::PerformCrud;
use actix_web::web::Data;
use lemmy_api_common::{
+ context::LemmyContext,
private_message::{CreatePrivateMessage, PrivateMessageResponse},
utils::{
check_person_block,
+ generate_local_apub_endpoint,
get_interface_language,
- get_local_user_view_from_jwt,
local_site_to_slur_regex,
+ local_user_view_from_jwt,
+ sanitize_html,
send_email_to_user,
+ EndpointType,
},
};
-use lemmy_apub::{
- generate_local_apub_endpoint,
- protocol::activities::{
- create_or_update::private_message::CreateOrUpdatePrivateMessage,
- CreateOrUpdateType,
- },
- EndpointType,
-};
use lemmy_db_schema::{
source::{
local_site::LocalSite,
},
traits::Crud,
};
-use lemmy_db_views::structs::LocalUserView;
-use lemmy_utils::{error::LemmyError, utils::remove_slurs, ConnectionId};
-use lemmy_websocket::{send::send_pm_ws_message, LemmyContext, UserOperationCrud};
+use lemmy_db_views::structs::{LocalUserView, PrivateMessageView};
+use lemmy_utils::{
+ error::{LemmyError, LemmyErrorExt, LemmyErrorType},
+ utils::{slurs::remove_slurs, validation::is_valid_body_field},
+};
#[async_trait::async_trait(?Send)]
impl PerformCrud for CreatePrivateMessage {
type Response = PrivateMessageResponse;
- #[tracing::instrument(skip(self, context, websocket_id))]
+ #[tracing::instrument(skip(self, context))]
async fn perform(
&self,
context: &Data<LemmyContext>,
- websocket_id: Option<ConnectionId>,
) -> Result<PrivateMessageResponse, LemmyError> {
let data: &CreatePrivateMessage = self;
- let local_user_view =
- get_local_user_view_from_jwt(&data.auth, context.pool(), context.secret()).await?;
- let local_site = LocalSite::read(context.pool()).await?;
+ let local_user_view = local_user_view_from_jwt(&data.auth, context).await?;
+ let local_site = LocalSite::read(&mut context.pool()).await?;
- let content_slurs_removed = remove_slurs(
- &data.content.clone(),
- &local_site_to_slur_regex(&local_site),
- );
+ let content = sanitize_html(&data.content);
+ let content = remove_slurs(&content, &local_site_to_slur_regex(&local_site));
+ is_valid_body_field(&Some(content.clone()), false)?;
- check_person_block(local_user_view.person.id, data.recipient_id, context.pool()).await?;
+ check_person_block(
+ local_user_view.person.id,
+ data.recipient_id,
+ &mut context.pool(),
+ )
+ .await?;
let private_message_form = PrivateMessageInsertForm::builder()
- .content(content_slurs_removed.clone())
+ .content(content.clone())
.creator_id(local_user_view.person.id)
.recipient_id(data.recipient_id)
.build();
let inserted_private_message =
- match PrivateMessage::create(context.pool(), &private_message_form).await {
- Ok(private_message) => private_message,
- Err(e) => {
- return Err(LemmyError::from_error_message(
- e,
- "couldnt_create_private_message",
- ));
- }
- };
+ PrivateMessage::create(&mut context.pool(), &private_message_form)
+ .await
+ .with_lemmy_type(LemmyErrorType::CouldntCreatePrivateMessage)?;
let inserted_private_message_id = inserted_private_message.id;
let protocol_and_hostname = context.settings().get_protocol_and_hostname();
&inserted_private_message_id.to_string(),
&protocol_and_hostname,
)?;
- let updated_private_message = PrivateMessage::update(
- context.pool(),
+ PrivateMessage::update(
+ &mut context.pool(),
inserted_private_message.id,
&PrivateMessageUpdateForm::builder()
.ap_id(Some(apub_id))
.build(),
)
.await
- .map_err(|e| LemmyError::from_error_message(e, "couldnt_create_private_message"))?;
+ .with_lemmy_type(LemmyErrorType::CouldntCreatePrivateMessage)?;
- CreateOrUpdatePrivateMessage::send(
- updated_private_message.into(),
- &local_user_view.person.into(),
- CreateOrUpdateType::Create,
- context,
- )
- .await?;
-
- let res = send_pm_ws_message(
- inserted_private_message.id,
- UserOperationCrud::CreatePrivateMessage,
- websocket_id,
- context,
- )
- .await?;
+ let view = PrivateMessageView::read(&mut context.pool(), inserted_private_message.id).await?;
// Send email to the local recipient, if one exists
- if res.private_message_view.recipient.local {
+ if view.recipient.local {
let recipient_id = data.recipient_id;
- let local_recipient = LocalUserView::read_person(context.pool(), recipient_id).await?;
+ let local_recipient = LocalUserView::read_person(&mut context.pool(), recipient_id).await?;
let lang = get_interface_language(&local_recipient);
let inbox_link = format!("{}/inbox", context.settings().get_protocol_and_hostname());
+ let sender_name = &local_user_view.person.name;
send_email_to_user(
&local_recipient,
- &lang.notification_private_message_subject(&local_recipient.person.name),
- &lang.notification_private_message_body(
- &inbox_link,
- &content_slurs_removed,
- &local_recipient.person.name,
- ),
+ &lang.notification_private_message_subject(sender_name),
+ &lang.notification_private_message_body(inbox_link, &content, sender_name),
context.settings(),
- );
+ )
+ .await;
}
- Ok(res)
+ Ok(PrivateMessageResponse {
+ private_message_view: view,
+ })
}
}