pub mod comment; pub mod community; pub mod post; pub mod site; pub mod user; pub mod websocket; use diesel::PgConnection; use lemmy_db::{ comment::Comment, post::Post, user::User_, user_mention::{UserMention, UserMentionForm}, Crud, DbPool, }; use lemmy_utils::{email::send_email, settings::Settings, utils::MentionData, LemmyError}; use log::error; use serde::{Deserialize, Serialize}; #[derive(Serialize, Deserialize, Debug)] pub struct WebFingerLink { pub rel: Option, #[serde(rename(serialize = "type", deserialize = "type"))] pub type_: Option, pub href: Option, #[serde(skip_serializing_if = "Option::is_none")] pub template: Option, } #[derive(Serialize, Deserialize, Debug)] pub struct WebFingerResponse { pub subject: String, pub aliases: Vec, pub links: Vec, } pub async fn blocking(pool: &DbPool, f: F) -> Result where F: FnOnce(&diesel::PgConnection) -> T + Send + 'static, T: Send + 'static, { let pool = pool.clone(); let res = actix_web::web::block(move || { let conn = pool.get()?; let res = (f)(&conn); Ok(res) as Result<_, LemmyError> }) .await?; Ok(res) } pub async fn send_local_notifs( mentions: Vec, comment: Comment, user: &User_, post: Post, pool: &DbPool, do_send_email: bool, ) -> Result, LemmyError> { let user2 = user.clone(); let ids = blocking(pool, move |conn| { do_send_local_notifs(conn, &mentions, &comment, &user2, &post, do_send_email) }) .await?; Ok(ids) } fn do_send_local_notifs( conn: &PgConnection, mentions: &[MentionData], comment: &Comment, user: &User_, post: &Post, do_send_email: bool, ) -> Vec { let mut recipient_ids = Vec::new(); // Send the local mentions for mention in mentions .iter() .filter(|m| m.is_local() && m.name.ne(&user.name)) .collect::>() { 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 if do_send_email && mention_user.send_notifications_to_email { send_email_to_user( mention_user, "Mentioned by", "User Mention", &comment.content, ) } } } // 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); if do_send_email && parent_user.send_notifications_to_email { send_email_to_user(parent_user, "Reply from", "Comment Reply", &comment.content) } } } } } // 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); if do_send_email && parent_user.send_notifications_to_email { send_email_to_user(parent_user, "Reply from", "Post Reply", &comment.content) } } } } }; recipient_ids } pub fn send_email_to_user(user: User_, subject_text: &str, body_text: &str, comment_content: &str) { if user.banned { return; } if let Some(user_email) = user.email { let subject = &format!( "{} - {} {}", subject_text, Settings::get().hostname, user.name, ); let html = &format!( "

{}


{} - {}

inbox", body_text, user.name, comment_content, Settings::get().get_protocol_and_hostname() ); match send_email(subject, &user_email, &user.name, html) { Ok(_o) => _o, Err(e) => error!("{}", e), }; } }