2020-05-16 14:04:08 +00:00
|
|
|
use crate::{
|
2020-07-22 18:20:08 +00:00
|
|
|
api::{claims::Claims, is_mod_or_admin, APIError, Oper, Perform},
|
2020-05-16 14:04:08 +00:00
|
|
|
apub::{ApubLikeableType, ApubObjectType},
|
2020-07-01 12:54:29 +00:00
|
|
|
blocking,
|
2020-05-16 14:04:08 +00:00
|
|
|
websocket::{
|
|
|
|
server::{JoinCommunityRoom, SendComment},
|
|
|
|
UserOperation,
|
|
|
|
WebsocketInfo,
|
|
|
|
},
|
2020-07-01 12:54:29 +00:00
|
|
|
DbPool,
|
|
|
|
LemmyError,
|
2020-07-10 18:15:41 +00:00
|
|
|
};
|
|
|
|
use lemmy_db::{
|
|
|
|
comment::*,
|
|
|
|
comment_view::*,
|
|
|
|
community_view::*,
|
|
|
|
moderator::*,
|
|
|
|
post::*,
|
|
|
|
site_view::*,
|
|
|
|
user::*,
|
|
|
|
user_mention::*,
|
|
|
|
Crud,
|
|
|
|
Likeable,
|
|
|
|
ListingType,
|
|
|
|
Saveable,
|
|
|
|
SortType,
|
|
|
|
};
|
|
|
|
use lemmy_utils::{
|
|
|
|
make_apub_endpoint,
|
|
|
|
remove_slurs,
|
|
|
|
scrape_text_for_mentions,
|
|
|
|
send_email,
|
|
|
|
settings::Settings,
|
|
|
|
EndpointType,
|
2020-05-17 01:09:26 +00:00
|
|
|
MentionData,
|
2020-05-16 14:04:08 +00:00
|
|
|
};
|
|
|
|
use log::error;
|
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use std::str::FromStr;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct CreateComment {
|
|
|
|
content: String,
|
|
|
|
parent_id: Option<i32>,
|
|
|
|
pub post_id: i32,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: Option<String>,
|
2019-09-07 15:35:05 +00:00
|
|
|
auth: String,
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct EditComment {
|
|
|
|
content: String,
|
|
|
|
edit_id: i32,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: Option<String>,
|
2020-07-21 01:37:44 +00:00
|
|
|
auth: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct DeleteComment {
|
|
|
|
edit_id: i32,
|
|
|
|
deleted: bool,
|
|
|
|
auth: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct RemoveComment {
|
|
|
|
edit_id: i32,
|
|
|
|
removed: bool,
|
2019-05-05 05:20:38 +00:00
|
|
|
reason: Option<String>,
|
2020-07-21 01:37:44 +00:00
|
|
|
auth: String,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct MarkCommentAsRead {
|
|
|
|
edit_id: i32,
|
|
|
|
read: bool,
|
2019-09-07 15:35:05 +00:00
|
|
|
auth: String,
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct SaveComment {
|
|
|
|
comment_id: i32,
|
|
|
|
save: bool,
|
2019-09-07 15:35:05 +00:00
|
|
|
auth: String,
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Clone)]
|
|
|
|
pub struct CommentResponse {
|
2019-09-07 15:35:05 +00:00
|
|
|
pub comment: CommentView,
|
2020-02-01 01:02:20 +00:00
|
|
|
pub recipient_ids: Vec<i32>,
|
2020-07-21 14:56:41 +00:00
|
|
|
pub form_id: Option<String>,
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct CreateCommentLike {
|
|
|
|
comment_id: i32,
|
|
|
|
score: i16,
|
2019-09-07 15:35:05 +00:00
|
|
|
auth: String,
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
2020-02-08 04:05:15 +00:00
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct GetComments {
|
|
|
|
type_: String,
|
|
|
|
sort: String,
|
|
|
|
page: Option<i64>,
|
|
|
|
limit: Option<i64>,
|
|
|
|
pub community_id: Option<i32>,
|
|
|
|
auth: Option<String>,
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
|
|
pub struct GetCommentsResponse {
|
|
|
|
comments: Vec<CommentView>,
|
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
2020-04-20 03:59:07 +00:00
|
|
|
impl Perform for Oper<CreateComment> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
async fn perform(
|
2020-04-19 22:08:25 +00:00
|
|
|
&self,
|
2020-07-01 12:54:29 +00:00
|
|
|
pool: &DbPool,
|
2020-04-19 22:08:25 +00:00
|
|
|
websocket_info: Option<WebsocketInfo>,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
2019-05-05 16:20:30 +00:00
|
|
|
let data: &CreateComment = &self.data;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
2020-01-16 14:39:08 +00:00
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
|
|
|
let content_slurs_removed = remove_slurs(&data.content.to_owned());
|
|
|
|
|
|
|
|
let comment_form = CommentForm {
|
|
|
|
content: content_slurs_removed,
|
|
|
|
parent_id: data.parent_id.to_owned(),
|
|
|
|
post_id: data.post_id,
|
|
|
|
creator_id: user_id,
|
|
|
|
removed: None,
|
|
|
|
deleted: None,
|
|
|
|
read: None,
|
2020-04-27 16:57:00 +00:00
|
|
|
published: None,
|
2019-09-07 15:35:05 +00:00
|
|
|
updated: None,
|
2020-06-27 01:12:41 +00:00
|
|
|
ap_id: "http://fake.com".into(),
|
2020-04-04 00:04:57 +00:00
|
|
|
local: true,
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
// Check for a community ban
|
|
|
|
let post_id = data.post_id;
|
|
|
|
let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
|
|
|
|
|
|
|
|
let community_id = post.community_id;
|
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
|
|
|
return Err(APIError::err("community_ban").into());
|
|
|
|
}
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
// Check for a site ban
|
|
|
|
let user = blocking(pool, move |conn| User_::read(&conn, user_id)).await??;
|
|
|
|
if user.banned {
|
|
|
|
return Err(APIError::err("site_ban").into());
|
|
|
|
}
|
|
|
|
|
2020-07-21 03:46:36 +00:00
|
|
|
// Check if post is locked, no new comments
|
|
|
|
if post.locked {
|
|
|
|
return Err(APIError::err("locked").into());
|
|
|
|
}
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Create the comment
|
2020-07-01 12:54:29 +00:00
|
|
|
let comment_form2 = comment_form.clone();
|
|
|
|
let inserted_comment =
|
|
|
|
match blocking(pool, move |conn| Comment::create(&conn, &comment_form2)).await? {
|
|
|
|
Ok(comment) => comment,
|
|
|
|
Err(_e) => return Err(APIError::err("couldnt_create_comment").into()),
|
|
|
|
};
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Necessary to update the ap_id
|
2020-07-01 12:54:29 +00:00
|
|
|
let inserted_comment_id = inserted_comment.id;
|
|
|
|
let updated_comment: Comment = match blocking(pool, move |conn| {
|
2020-07-10 18:15:41 +00:00
|
|
|
let apub_id =
|
|
|
|
make_apub_endpoint(EndpointType::Comment, &inserted_comment_id.to_string()).to_string();
|
|
|
|
Comment::update_ap_id(&conn, inserted_comment_id, apub_id)
|
2020-07-01 12:54:29 +00:00
|
|
|
})
|
|
|
|
.await?
|
|
|
|
{
|
2020-04-04 00:04:57 +00:00
|
|
|
Ok(comment) => comment,
|
|
|
|
Err(_e) => return Err(APIError::err("couldnt_create_comment").into()),
|
|
|
|
};
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
updated_comment
|
|
|
|
.send_create(&user, &self.client, pool)
|
|
|
|
.await?;
|
2020-04-27 16:57:00 +00:00
|
|
|
|
2019-10-20 00:46:29 +00:00
|
|
|
// Scan the comment for user mentions, add those rows
|
2020-05-15 16:36:11 +00:00
|
|
|
let mentions = scrape_text_for_mentions(&comment_form.content);
|
2020-07-01 12:54:29 +00:00
|
|
|
let recipient_ids =
|
2020-07-28 16:08:28 +00:00
|
|
|
send_local_notifs(mentions, updated_comment.clone(), &user, post, pool, true).await?;
|
2020-01-02 21:55:54 +00:00
|
|
|
|
2019-05-05 05:20:38 +00:00
|
|
|
// You like your own comment by default
|
|
|
|
let like_form = CommentLikeForm {
|
|
|
|
comment_id: inserted_comment.id,
|
|
|
|
post_id: data.post_id,
|
2019-12-09 19:08:19 +00:00
|
|
|
user_id,
|
2019-09-07 15:35:05 +00:00
|
|
|
score: 1,
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let like = move |conn: &'_ _| CommentLike::like(&conn, &like_form);
|
|
|
|
if blocking(pool, like).await?.is_err() {
|
|
|
|
return Err(APIError::err("couldnt_like_comment").into());
|
|
|
|
}
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
updated_comment.send_like(&user, &self.client, pool).await?;
|
2020-04-28 04:16:02 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(&conn, inserted_comment.id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-04-19 22:08:25 +00:00
|
|
|
let mut res = CommentResponse {
|
2019-09-07 15:35:05 +00:00
|
|
|
comment: comment_view,
|
2020-02-01 01:02:20 +00:00
|
|
|
recipient_ids,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: data.form_id.to_owned(),
|
2020-04-19 22:08:25 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
ws.chatserver.do_send(SendComment {
|
|
|
|
op: UserOperation::CreateComment,
|
|
|
|
comment: res.clone(),
|
|
|
|
my_id: ws.id,
|
|
|
|
});
|
|
|
|
|
|
|
|
// strip out the recipient_ids, so that
|
|
|
|
// users don't get double notifs
|
|
|
|
res.recipient_ids = Vec::new();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(res)
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
2020-04-20 03:59:07 +00:00
|
|
|
impl Perform for Oper<EditComment> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
async fn perform(
|
2020-04-19 22:08:25 +00:00
|
|
|
&self,
|
2020-07-01 12:54:29 +00:00
|
|
|
pool: &DbPool,
|
2020-04-19 22:08:25 +00:00
|
|
|
websocket_info: Option<WebsocketInfo>,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
2019-05-05 16:20:30 +00:00
|
|
|
let data: &EditComment = &self.data;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
2020-01-16 14:39:08 +00:00
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let orig_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, edit_id, None)).await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Check for a site ban
|
|
|
|
let user = blocking(pool, move |conn| User_::read(conn, user_id)).await??;
|
|
|
|
if user.banned {
|
|
|
|
return Err(APIError::err("site_ban").into());
|
|
|
|
}
|
2020-07-14 13:17:25 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Check for a community ban
|
2020-07-14 13:17:25 +00:00
|
|
|
let community_id = orig_comment.community_id;
|
2020-07-21 01:37:44 +00:00
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
|
|
|
return Err(APIError::err("community_ban").into());
|
|
|
|
}
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Verify that only the creator can edit
|
|
|
|
if user_id != orig_comment.creator_id {
|
|
|
|
return Err(APIError::err("no_comment_edit_allowed").into());
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Do the update
|
2019-05-05 05:20:38 +00:00
|
|
|
let content_slurs_removed = remove_slurs(&data.content.to_owned());
|
2020-07-21 01:37:44 +00:00
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let updated_comment = match blocking(pool, move |conn| {
|
|
|
|
Comment::update_content(conn, edit_id, &content_slurs_removed)
|
|
|
|
})
|
|
|
|
.await?
|
|
|
|
{
|
|
|
|
Ok(comment) => comment,
|
|
|
|
Err(_e) => return Err(APIError::err("couldnt_update_comment").into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
// Send the apub update
|
|
|
|
updated_comment
|
|
|
|
.send_update(&user, &self.client, pool)
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
// Do the mentions / recipients
|
|
|
|
let post_id = orig_comment.post_id;
|
|
|
|
let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
|
|
|
|
|
|
|
|
let updated_comment_content = updated_comment.content.to_owned();
|
|
|
|
let mentions = scrape_text_for_mentions(&updated_comment_content);
|
|
|
|
let recipient_ids =
|
2020-07-28 16:08:28 +00:00
|
|
|
send_local_notifs(mentions, updated_comment, &user, post, pool, false).await?;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let edit_id = data.edit_id;
|
2020-07-21 01:37:44 +00:00
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, edit_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
let mut res = CommentResponse {
|
|
|
|
comment: comment_view,
|
|
|
|
recipient_ids,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: data.form_id.to_owned(),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
ws.chatserver.do_send(SendComment {
|
|
|
|
op: UserOperation::EditComment,
|
|
|
|
comment: res.clone(),
|
|
|
|
my_id: ws.id,
|
|
|
|
});
|
|
|
|
|
|
|
|
// strip out the recipient_ids, so that
|
|
|
|
// users don't get double notifs
|
|
|
|
res.recipient_ids = Vec::new();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
|
|
impl Perform for Oper<DeleteComment> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
|
|
|
async fn perform(
|
|
|
|
&self,
|
|
|
|
pool: &DbPool,
|
|
|
|
websocket_info: Option<WebsocketInfo>,
|
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
|
|
|
let data: &DeleteComment = &self.data;
|
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
let user_id = claims.id;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let edit_id = data.edit_id;
|
2020-07-21 01:37:44 +00:00
|
|
|
let orig_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, edit_id, None)).await??;
|
|
|
|
|
|
|
|
// Check for a site ban
|
|
|
|
let user = blocking(pool, move |conn| User_::read(conn, user_id)).await??;
|
|
|
|
if user.banned {
|
|
|
|
return Err(APIError::err("site_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for a community ban
|
|
|
|
let community_id = orig_comment.community_id;
|
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
|
|
|
return Err(APIError::err("community_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify that only the creator can delete
|
|
|
|
if user_id != orig_comment.creator_id {
|
|
|
|
return Err(APIError::err("no_comment_edit_allowed").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Do the delete
|
|
|
|
let deleted = data.deleted;
|
2020-07-01 12:54:29 +00:00
|
|
|
let updated_comment = match blocking(pool, move |conn| {
|
2020-07-21 01:37:44 +00:00
|
|
|
Comment::update_deleted(conn, edit_id, deleted)
|
2020-07-01 12:54:29 +00:00
|
|
|
})
|
|
|
|
.await?
|
|
|
|
{
|
2019-05-05 05:20:38 +00:00
|
|
|
Ok(comment) => comment,
|
2020-01-16 14:39:08 +00:00
|
|
|
Err(_e) => return Err(APIError::err("couldnt_update_comment").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Send the apub message
|
|
|
|
if deleted {
|
|
|
|
updated_comment
|
|
|
|
.send_delete(&user, &self.client, pool)
|
|
|
|
.await?;
|
|
|
|
} else {
|
|
|
|
updated_comment
|
|
|
|
.send_undo_delete(&user, &self.client, pool)
|
|
|
|
.await?;
|
2020-04-29 14:51:25 +00:00
|
|
|
}
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Refetch it
|
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, edit_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
// Build the recipients
|
|
|
|
let post_id = comment_view.post_id;
|
2020-07-01 12:54:29 +00:00
|
|
|
let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
|
2020-07-21 01:37:44 +00:00
|
|
|
let mentions = vec![];
|
|
|
|
let recipient_ids =
|
2020-07-28 16:08:28 +00:00
|
|
|
send_local_notifs(mentions, updated_comment, &user, post, pool, false).await?;
|
2019-10-20 00:46:29 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
let mut res = CommentResponse {
|
|
|
|
comment: comment_view,
|
|
|
|
recipient_ids,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: None,
|
2020-07-21 01:37:44 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
ws.chatserver.do_send(SendComment {
|
|
|
|
op: UserOperation::DeleteComment,
|
|
|
|
comment: res.clone(),
|
|
|
|
my_id: ws.id,
|
|
|
|
});
|
|
|
|
|
|
|
|
// strip out the recipient_ids, so that
|
|
|
|
// users don't get double notifs
|
|
|
|
res.recipient_ids = Vec::new();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[async_trait::async_trait(?Send)]
|
|
|
|
impl Perform for Oper<RemoveComment> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
|
|
|
async fn perform(
|
|
|
|
&self,
|
|
|
|
pool: &DbPool,
|
|
|
|
websocket_info: Option<WebsocketInfo>,
|
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
|
|
|
let data: &RemoveComment = &self.data;
|
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let orig_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, edit_id, None)).await??;
|
|
|
|
|
|
|
|
// Check for a site ban
|
|
|
|
let user = blocking(pool, move |conn| User_::read(conn, user_id)).await??;
|
|
|
|
if user.banned {
|
|
|
|
return Err(APIError::err("site_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for a community ban
|
|
|
|
let community_id = orig_comment.community_id;
|
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
|
|
|
return Err(APIError::err("community_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify that only a mod or admin can remove
|
2020-07-22 18:20:08 +00:00
|
|
|
is_mod_or_admin(pool, user_id, community_id).await?;
|
2020-07-21 01:37:44 +00:00
|
|
|
|
|
|
|
// Do the remove
|
|
|
|
let removed = data.removed;
|
|
|
|
let updated_comment = match blocking(pool, move |conn| {
|
|
|
|
Comment::update_removed(conn, edit_id, removed)
|
|
|
|
})
|
|
|
|
.await?
|
|
|
|
{
|
|
|
|
Ok(comment) => comment,
|
|
|
|
Err(_e) => return Err(APIError::err("couldnt_update_comment").into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
// Mod tables
|
|
|
|
let form = ModRemoveCommentForm {
|
|
|
|
mod_user_id: user_id,
|
|
|
|
comment_id: data.edit_id,
|
|
|
|
removed: Some(removed),
|
|
|
|
reason: data.reason.to_owned(),
|
|
|
|
};
|
|
|
|
blocking(pool, move |conn| ModRemoveComment::create(conn, &form)).await??;
|
|
|
|
|
|
|
|
// Send the apub message
|
|
|
|
if removed {
|
|
|
|
updated_comment
|
|
|
|
.send_remove(&user, &self.client, pool)
|
|
|
|
.await?;
|
|
|
|
} else {
|
|
|
|
updated_comment
|
|
|
|
.send_undo_remove(&user, &self.client, pool)
|
|
|
|
.await?;
|
|
|
|
}
|
2020-02-01 01:02:20 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Refetch it
|
2020-07-01 12:54:29 +00:00
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, edit_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
// Build the recipients
|
|
|
|
let post_id = comment_view.post_id;
|
|
|
|
let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
|
|
|
|
let mentions = vec![];
|
|
|
|
let recipient_ids =
|
2020-07-28 16:08:28 +00:00
|
|
|
send_local_notifs(mentions, updated_comment, &user, post, pool, false).await?;
|
2020-07-21 01:37:44 +00:00
|
|
|
|
2020-04-19 22:08:25 +00:00
|
|
|
let mut res = CommentResponse {
|
2019-09-07 15:35:05 +00:00
|
|
|
comment: comment_view,
|
2020-02-01 01:02:20 +00:00
|
|
|
recipient_ids,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: None,
|
2020-04-19 22:08:25 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
ws.chatserver.do_send(SendComment {
|
2020-07-21 01:37:44 +00:00
|
|
|
op: UserOperation::RemoveComment,
|
2020-04-19 22:08:25 +00:00
|
|
|
comment: res.clone(),
|
|
|
|
my_id: ws.id,
|
|
|
|
});
|
|
|
|
|
|
|
|
// strip out the recipient_ids, so that
|
|
|
|
// users don't get double notifs
|
|
|
|
res.recipient_ids = Vec::new();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(res)
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
|
|
|
impl Perform for Oper<MarkCommentAsRead> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
|
|
|
async fn perform(
|
|
|
|
&self,
|
|
|
|
pool: &DbPool,
|
|
|
|
_websocket_info: Option<WebsocketInfo>,
|
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
|
|
|
let data: &MarkCommentAsRead = &self.data;
|
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let orig_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, edit_id, None)).await??;
|
|
|
|
|
|
|
|
// Check for a site ban
|
|
|
|
let user = blocking(pool, move |conn| User_::read(conn, user_id)).await??;
|
|
|
|
if user.banned {
|
|
|
|
return Err(APIError::err("site_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Check for a community ban
|
|
|
|
let community_id = orig_comment.community_id;
|
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
|
|
|
return Err(APIError::err("community_ban").into());
|
|
|
|
}
|
|
|
|
|
|
|
|
// Verify that only the recipient can mark as read
|
|
|
|
// Needs to fetch the parent comment / post to get the recipient
|
|
|
|
let parent_id = orig_comment.parent_id;
|
|
|
|
match parent_id {
|
|
|
|
Some(pid) => {
|
|
|
|
let parent_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, pid, None)).await??;
|
|
|
|
if user_id != parent_comment.creator_id {
|
|
|
|
return Err(APIError::err("no_comment_edit_allowed").into());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
None => {
|
|
|
|
let parent_post_id = orig_comment.post_id;
|
|
|
|
let parent_post = blocking(pool, move |conn| Post::read(conn, parent_post_id)).await??;
|
|
|
|
if user_id != parent_post.creator_id {
|
|
|
|
return Err(APIError::err("no_comment_edit_allowed").into());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Do the mark as read
|
|
|
|
let read = data.read;
|
|
|
|
match blocking(pool, move |conn| Comment::update_read(conn, edit_id, read)).await? {
|
|
|
|
Ok(comment) => comment,
|
|
|
|
Err(_e) => return Err(APIError::err("couldnt_update_comment").into()),
|
|
|
|
};
|
|
|
|
|
|
|
|
// Refetch it
|
|
|
|
let edit_id = data.edit_id;
|
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, edit_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
|
|
|
|
|
|
|
let res = CommentResponse {
|
|
|
|
comment: comment_view,
|
|
|
|
recipient_ids: Vec::new(),
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: None,
|
2020-07-21 01:37:44 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
2020-04-20 03:59:07 +00:00
|
|
|
impl Perform for Oper<SaveComment> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
async fn perform(
|
2020-04-19 22:08:25 +00:00
|
|
|
&self,
|
2020-07-01 12:54:29 +00:00
|
|
|
pool: &DbPool,
|
2020-04-19 22:08:25 +00:00
|
|
|
_websocket_info: Option<WebsocketInfo>,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
2019-05-05 16:20:30 +00:00
|
|
|
let data: &SaveComment = &self.data;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
2020-01-16 14:39:08 +00:00
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
|
|
|
let comment_saved_form = CommentSavedForm {
|
|
|
|
comment_id: data.comment_id,
|
2019-12-09 19:08:19 +00:00
|
|
|
user_id,
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if data.save {
|
2020-07-01 12:54:29 +00:00
|
|
|
let save_comment = move |conn: &'_ _| CommentSaved::save(conn, &comment_saved_form);
|
|
|
|
if blocking(pool, save_comment).await?.is_err() {
|
|
|
|
return Err(APIError::err("couldnt_save_comment").into());
|
|
|
|
}
|
2019-05-05 05:20:38 +00:00
|
|
|
} else {
|
2020-07-01 12:54:29 +00:00
|
|
|
let unsave_comment = move |conn: &'_ _| CommentSaved::unsave(conn, &comment_saved_form);
|
|
|
|
if blocking(pool, unsave_comment).await?.is_err() {
|
|
|
|
return Err(APIError::err("couldnt_save_comment").into());
|
|
|
|
}
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let comment_id = data.comment_id;
|
|
|
|
let comment_view = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, comment_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2019-09-07 15:35:05 +00:00
|
|
|
Ok(CommentResponse {
|
|
|
|
comment: comment_view,
|
2020-02-01 01:02:20 +00:00
|
|
|
recipient_ids: Vec::new(),
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: None,
|
2019-09-07 15:35:05 +00:00
|
|
|
})
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
2020-04-20 03:59:07 +00:00
|
|
|
impl Perform for Oper<CreateCommentLike> {
|
|
|
|
type Response = CommentResponse;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
async fn perform(
|
2020-04-19 22:08:25 +00:00
|
|
|
&self,
|
2020-07-01 12:54:29 +00:00
|
|
|
pool: &DbPool,
|
2020-04-19 22:08:25 +00:00
|
|
|
websocket_info: Option<WebsocketInfo>,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<CommentResponse, LemmyError> {
|
2019-05-05 16:20:30 +00:00
|
|
|
let data: &CreateCommentLike = &self.data;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
let claims = match Claims::decode(&data.auth) {
|
|
|
|
Ok(claims) => claims.claims,
|
2020-01-16 14:39:08 +00:00
|
|
|
Err(_e) => return Err(APIError::err("not_logged_in").into()),
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = claims.id;
|
|
|
|
|
2020-02-01 01:02:20 +00:00
|
|
|
let mut recipient_ids = Vec::new();
|
|
|
|
|
2019-12-11 20:21:47 +00:00
|
|
|
// Don't do a downvote if site has downvotes disabled
|
|
|
|
if data.score == -1 {
|
2020-07-01 12:54:29 +00:00
|
|
|
let site = blocking(pool, move |conn| SiteView::read(conn)).await??;
|
2020-01-02 11:30:00 +00:00
|
|
|
if !site.enable_downvotes {
|
2020-01-16 14:39:08 +00:00
|
|
|
return Err(APIError::err("downvotes_disabled").into());
|
2019-12-11 20:21:47 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
let comment_id = data.comment_id;
|
|
|
|
let orig_comment =
|
|
|
|
blocking(pool, move |conn| CommentView::read(&conn, comment_id, None)).await??;
|
|
|
|
|
2019-05-05 05:20:38 +00:00
|
|
|
// Check for a community ban
|
2020-07-21 01:37:44 +00:00
|
|
|
let post_id = orig_comment.post_id;
|
2020-07-01 12:54:29 +00:00
|
|
|
let post = blocking(pool, move |conn| Post::read(conn, post_id)).await??;
|
|
|
|
let community_id = post.community_id;
|
|
|
|
let is_banned =
|
|
|
|
move |conn: &'_ _| CommunityUserBanView::get(conn, user_id, community_id).is_ok();
|
|
|
|
if blocking(pool, is_banned).await? {
|
2020-01-16 14:39:08 +00:00
|
|
|
return Err(APIError::err("community_ban").into());
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Check for a site ban
|
2020-07-01 12:54:29 +00:00
|
|
|
let user = blocking(pool, move |conn| User_::read(conn, user_id)).await??;
|
2020-04-28 04:16:02 +00:00
|
|
|
if user.banned {
|
2020-01-16 14:39:08 +00:00
|
|
|
return Err(APIError::err("site_ban").into());
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let comment_id = data.comment_id;
|
|
|
|
let comment = blocking(pool, move |conn| Comment::read(conn, comment_id)).await??;
|
2020-02-01 01:02:20 +00:00
|
|
|
|
|
|
|
// Add to recipient ids
|
|
|
|
match comment.parent_id {
|
|
|
|
Some(parent_id) => {
|
2020-07-01 12:54:29 +00:00
|
|
|
let parent_comment = blocking(pool, move |conn| Comment::read(conn, parent_id)).await??;
|
2020-02-01 01:02:20 +00:00
|
|
|
if parent_comment.creator_id != user_id {
|
2020-07-01 12:54:29 +00:00
|
|
|
let parent_user = blocking(pool, move |conn| {
|
|
|
|
User_::read(conn, parent_comment.creator_id)
|
|
|
|
})
|
|
|
|
.await??;
|
2020-02-01 01:02:20 +00:00
|
|
|
recipient_ids.push(parent_user.id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
None => {
|
|
|
|
recipient_ids.push(post.creator_id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-05-05 05:20:38 +00:00
|
|
|
let like_form = CommentLikeForm {
|
|
|
|
comment_id: data.comment_id,
|
2020-07-21 01:37:44 +00:00
|
|
|
post_id,
|
2019-12-09 19:08:19 +00:00
|
|
|
user_id,
|
2019-09-07 15:35:05 +00:00
|
|
|
score: data.score,
|
2019-05-05 05:20:38 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
// Remove any likes first
|
2020-07-01 12:54:29 +00:00
|
|
|
let like_form2 = like_form.clone();
|
|
|
|
blocking(pool, move |conn| CommentLike::remove(conn, &like_form2)).await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
|
|
|
// Only add the like if the score isnt 0
|
2019-12-19 21:59:13 +00:00
|
|
|
let do_add = like_form.score != 0 && (like_form.score == 1 || like_form.score == -1);
|
2019-05-15 16:46:39 +00:00
|
|
|
if do_add {
|
2020-07-01 12:54:29 +00:00
|
|
|
let like_form2 = like_form.clone();
|
|
|
|
let like = move |conn: &'_ _| CommentLike::like(conn, &like_form2);
|
|
|
|
if blocking(pool, like).await?.is_err() {
|
|
|
|
return Err(APIError::err("couldnt_like_comment").into());
|
|
|
|
}
|
2020-04-28 04:16:02 +00:00
|
|
|
|
|
|
|
if like_form.score == 1 {
|
2020-07-01 12:54:29 +00:00
|
|
|
comment.send_like(&user, &self.client, pool).await?;
|
2020-04-28 04:16:02 +00:00
|
|
|
} else if like_form.score == -1 {
|
2020-07-01 12:54:29 +00:00
|
|
|
comment.send_dislike(&user, &self.client, pool).await?;
|
2020-04-28 04:16:02 +00:00
|
|
|
}
|
|
|
|
} else {
|
2020-07-01 12:54:29 +00:00
|
|
|
comment.send_undo_like(&user, &self.client, pool).await?;
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Have to refetch the comment to get the current state
|
2020-07-01 12:54:29 +00:00
|
|
|
let comment_id = data.comment_id;
|
|
|
|
let liked_comment = blocking(pool, move |conn| {
|
|
|
|
CommentView::read(conn, comment_id, Some(user_id))
|
|
|
|
})
|
|
|
|
.await??;
|
2019-05-05 05:20:38 +00:00
|
|
|
|
2020-04-19 22:08:25 +00:00
|
|
|
let mut res = CommentResponse {
|
2019-09-07 15:35:05 +00:00
|
|
|
comment: liked_comment,
|
2020-02-01 01:02:20 +00:00
|
|
|
recipient_ids,
|
2020-07-21 14:56:41 +00:00
|
|
|
form_id: None,
|
2020-04-19 22:08:25 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
ws.chatserver.do_send(SendComment {
|
|
|
|
op: UserOperation::CreateCommentLike,
|
|
|
|
comment: res.clone(),
|
|
|
|
my_id: ws.id,
|
|
|
|
});
|
|
|
|
|
|
|
|
// strip out the recipient_ids, so that
|
|
|
|
// users don't get double notifs
|
|
|
|
res.recipient_ids = Vec::new();
|
|
|
|
}
|
|
|
|
|
|
|
|
Ok(res)
|
2019-05-05 05:20:38 +00:00
|
|
|
}
|
|
|
|
}
|
2020-02-08 04:05:15 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
#[async_trait::async_trait(?Send)]
|
2020-04-20 03:59:07 +00:00
|
|
|
impl Perform for Oper<GetComments> {
|
|
|
|
type Response = GetCommentsResponse;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
async fn perform(
|
2020-04-19 22:08:25 +00:00
|
|
|
&self,
|
2020-07-01 12:54:29 +00:00
|
|
|
pool: &DbPool,
|
2020-04-19 22:08:25 +00:00
|
|
|
websocket_info: Option<WebsocketInfo>,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<GetCommentsResponse, LemmyError> {
|
2020-02-08 04:05:15 +00:00
|
|
|
let data: &GetComments = &self.data;
|
|
|
|
|
|
|
|
let user_claims: Option<Claims> = match &data.auth {
|
|
|
|
Some(auth) => match Claims::decode(&auth) {
|
|
|
|
Ok(claims) => Some(claims.claims),
|
|
|
|
Err(_e) => None,
|
|
|
|
},
|
|
|
|
None => None,
|
|
|
|
};
|
|
|
|
|
|
|
|
let user_id = match &user_claims {
|
|
|
|
Some(claims) => Some(claims.id),
|
|
|
|
None => None,
|
|
|
|
};
|
|
|
|
|
|
|
|
let type_ = ListingType::from_str(&data.type_)?;
|
|
|
|
let sort = SortType::from_str(&data.sort)?;
|
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
let community_id = data.community_id;
|
|
|
|
let page = data.page;
|
|
|
|
let limit = data.limit;
|
|
|
|
let comments = blocking(pool, move |conn| {
|
|
|
|
CommentQueryBuilder::create(conn)
|
|
|
|
.listing_type(type_)
|
|
|
|
.sort(&sort)
|
|
|
|
.for_community_id(community_id)
|
|
|
|
.my_user_id(user_id)
|
|
|
|
.page(page)
|
|
|
|
.limit(limit)
|
|
|
|
.list()
|
|
|
|
})
|
|
|
|
.await?;
|
|
|
|
let comments = match comments {
|
2020-02-08 04:05:15 +00:00
|
|
|
Ok(comments) => comments,
|
2020-07-01 12:54:29 +00:00
|
|
|
Err(_) => return Err(APIError::err("couldnt_get_comments").into()),
|
2020-02-08 04:05:15 +00:00
|
|
|
};
|
|
|
|
|
2020-04-19 22:08:25 +00:00
|
|
|
if let Some(ws) = websocket_info {
|
|
|
|
// You don't need to join the specific community room, bc this is already handled by
|
|
|
|
// GetCommunity
|
|
|
|
if data.community_id.is_none() {
|
|
|
|
if let Some(id) = ws.id {
|
|
|
|
// 0 is the "all" community
|
|
|
|
ws.chatserver.do_send(JoinCommunityRoom {
|
|
|
|
community_id: 0,
|
|
|
|
id,
|
|
|
|
});
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-02-08 04:05:15 +00:00
|
|
|
Ok(GetCommentsResponse { comments })
|
|
|
|
}
|
|
|
|
}
|
2020-05-15 16:36:11 +00:00
|
|
|
|
2020-07-01 12:54:29 +00:00
|
|
|
pub async fn send_local_notifs(
|
|
|
|
mentions: Vec<MentionData>,
|
|
|
|
comment: Comment,
|
2020-07-23 14:36:45 +00:00
|
|
|
user: &User_,
|
2020-07-01 12:54:29 +00:00
|
|
|
post: Post,
|
|
|
|
pool: &DbPool,
|
2020-07-21 01:37:44 +00:00
|
|
|
do_send_email: bool,
|
2020-07-01 12:54:29 +00:00
|
|
|
) -> Result<Vec<i32>, LemmyError> {
|
2020-07-23 14:36:45 +00:00
|
|
|
let user2 = user.clone();
|
2020-07-01 12:54:29 +00:00
|
|
|
let ids = blocking(pool, move |conn| {
|
2020-07-28 16:08:28 +00:00
|
|
|
do_send_local_notifs(conn, &mentions, &comment, &user2, &post, do_send_email)
|
2020-07-01 12:54:29 +00:00
|
|
|
})
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
Ok(ids)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn do_send_local_notifs(
|
|
|
|
conn: &diesel::PgConnection,
|
2020-05-15 16:36:11 +00:00
|
|
|
mentions: &[MentionData],
|
|
|
|
comment: &Comment,
|
|
|
|
user: &User_,
|
|
|
|
post: &Post,
|
2020-07-21 01:37:44 +00:00
|
|
|
do_send_email: bool,
|
2020-05-15 16:36:11 +00:00
|
|
|
) -> Vec<i32> {
|
|
|
|
let mut recipient_ids = Vec::new();
|
|
|
|
let hostname = &format!("https://{}", Settings::get().hostname);
|
|
|
|
|
|
|
|
// Send the local mentions
|
|
|
|
for mention in mentions
|
|
|
|
.iter()
|
|
|
|
.filter(|m| m.is_local() && m.name.ne(&user.name))
|
|
|
|
.collect::<Vec<&MentionData>>()
|
|
|
|
{
|
|
|
|
if let Ok(mention_user) = User_::read_from_name(&conn, &mention.name) {
|
|
|
|
// TODO
|
|
|
|
// At some point, make it so you can't tag the parent creator either
|
|
|
|
// This can cause two notifications, one for reply and the other for mention
|
|
|
|
recipient_ids.push(mention_user.id);
|
|
|
|
|
|
|
|
let user_mention_form = UserMentionForm {
|
|
|
|
recipient_id: mention_user.id,
|
|
|
|
comment_id: comment.id,
|
|
|
|
read: None,
|
|
|
|
};
|
|
|
|
|
|
|
|
// Allow this to fail softly, since comment edits might re-update or replace it
|
|
|
|
// Let the uniqueness handle this fail
|
|
|
|
match UserMention::create(&conn, &user_mention_form) {
|
|
|
|
Ok(_mention) => (),
|
|
|
|
Err(_e) => error!("{}", &_e),
|
|
|
|
};
|
|
|
|
|
|
|
|
// Send an email to those users that have notifications on
|
2020-07-21 01:37:44 +00:00
|
|
|
if do_send_email && mention_user.send_notifications_to_email {
|
2020-05-15 16:36:11 +00:00
|
|
|
if let Some(mention_email) = mention_user.email {
|
|
|
|
let subject = &format!("{} - Mentioned by {}", Settings::get().hostname, user.name,);
|
|
|
|
let html = &format!(
|
|
|
|
"<h1>User Mention</h1><br><div>{} - {}</div><br><a href={}/inbox>inbox</a>",
|
|
|
|
user.name, comment.content, hostname
|
|
|
|
);
|
|
|
|
match send_email(subject, &mention_email, &mention_user.name, html) {
|
|
|
|
Ok(_o) => _o,
|
|
|
|
Err(e) => error!("{}", e),
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Send notifs to the parent commenter / poster
|
|
|
|
match comment.parent_id {
|
|
|
|
Some(parent_id) => {
|
|
|
|
if let Ok(parent_comment) = Comment::read(&conn, parent_id) {
|
|
|
|
if parent_comment.creator_id != user.id {
|
|
|
|
if let Ok(parent_user) = User_::read(&conn, parent_comment.creator_id) {
|
|
|
|
recipient_ids.push(parent_user.id);
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
if do_send_email && parent_user.send_notifications_to_email {
|
2020-05-15 16:36:11 +00:00
|
|
|
if let Some(comment_reply_email) = parent_user.email {
|
|
|
|
let subject = &format!("{} - Reply from {}", Settings::get().hostname, user.name,);
|
|
|
|
let html = &format!(
|
|
|
|
"<h1>Comment Reply</h1><br><div>{} - {}</div><br><a href={}/inbox>inbox</a>",
|
|
|
|
user.name, comment.content, hostname
|
|
|
|
);
|
|
|
|
match send_email(subject, &comment_reply_email, &parent_user.name, html) {
|
|
|
|
Ok(_o) => _o,
|
|
|
|
Err(e) => error!("{}", e),
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Its a post
|
|
|
|
None => {
|
|
|
|
if post.creator_id != user.id {
|
|
|
|
if let Ok(parent_user) = User_::read(&conn, post.creator_id) {
|
|
|
|
recipient_ids.push(parent_user.id);
|
|
|
|
|
2020-07-21 01:37:44 +00:00
|
|
|
if do_send_email && parent_user.send_notifications_to_email {
|
2020-05-15 16:36:11 +00:00
|
|
|
if let Some(post_reply_email) = parent_user.email {
|
|
|
|
let subject = &format!("{} - Reply from {}", Settings::get().hostname, user.name,);
|
|
|
|
let html = &format!(
|
|
|
|
"<h1>Post Reply</h1><br><div>{} - {}</div><br><a href={}/inbox>inbox</a>",
|
|
|
|
user.name, comment.content, hostname
|
|
|
|
);
|
|
|
|
match send_email(subject, &post_reply_email, &parent_user.name, html) {
|
|
|
|
Ok(_o) => _o,
|
|
|
|
Err(e) => error!("{}", e),
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
recipient_ids
|
|
|
|
}
|