use super::*; #[derive(Serialize, Deserialize)] pub struct CreateComment { content: String, parent_id: Option, edit_id: Option, pub post_id: i32, auth: String } #[derive(Serialize, Deserialize)] pub struct EditComment { content: String, parent_id: Option, edit_id: i32, creator_id: i32, pub post_id: i32, removed: Option, deleted: Option, reason: Option, read: Option, auth: String } #[derive(Serialize, Deserialize)] pub struct SaveComment { comment_id: i32, save: bool, auth: String } #[derive(Serialize, Deserialize, Clone)] pub struct CommentResponse { op: String, pub comment: CommentView } #[derive(Serialize, Deserialize)] pub struct CreateCommentLike { comment_id: i32, pub post_id: i32, score: i16, auth: String } impl Perform for Oper { fn perform(&self) -> Result { let data: &CreateComment = &self.data; let conn = establish_connection(); let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => { return Err(APIError::err(&self.op, "Not logged in."))? } }; let user_id = claims.id; // Check for a community ban let post = Post::read(&conn, data.post_id)?; if CommunityUserBanView::get(&conn, user_id, post.community_id).is_ok() { return Err(APIError::err(&self.op, "You have been banned from this community"))? } // Check for a site ban if UserView::read(&conn, user_id)?.banned { return Err(APIError::err(&self.op, "You have been banned from the site"))? } 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, updated: None }; let inserted_comment = match Comment::create(&conn, &comment_form) { Ok(comment) => comment, Err(_e) => { return Err(APIError::err(&self.op, "Couldn't create Comment"))? } }; // You like your own comment by default let like_form = CommentLikeForm { comment_id: inserted_comment.id, post_id: data.post_id, user_id: user_id, score: 1 }; let _inserted_like = match CommentLike::like(&conn, &like_form) { Ok(like) => like, Err(_e) => { return Err(APIError::err(&self.op, "Couldn't like comment."))? } }; let comment_view = CommentView::read(&conn, inserted_comment.id, Some(user_id))?; Ok( CommentResponse { op: self.op.to_string(), comment: comment_view } ) } } impl Perform for Oper { fn perform(&self) -> Result { let data: &EditComment = &self.data; let conn = establish_connection(); let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => { return Err(APIError::err(&self.op, "Not logged in."))? } }; let user_id = claims.id; let orig_comment = CommentView::read(&conn, data.edit_id, None)?; // You are allowed to mark the comment as read even if you're banned. if data.read.is_none() { // Verify its the creator or a mod, or an admin let mut editors: Vec = vec![data.creator_id]; editors.append( &mut CommunityModeratorView::for_community(&conn, orig_comment.community_id) ? .into_iter() .map(|m| m.user_id) .collect() ); editors.append( &mut UserView::admins(&conn) ? .into_iter() .map(|a| a.id) .collect() ); if !editors.contains(&user_id) { return Err(APIError::err(&self.op, "Not allowed to edit comment."))? } // Check for a community ban if CommunityUserBanView::get(&conn, user_id, orig_comment.community_id).is_ok() { return Err(APIError::err(&self.op, "You have been banned from this community"))? } // Check for a site ban if UserView::read(&conn, user_id)?.banned { return Err(APIError::err(&self.op, "You have been banned from the site"))? } } let content_slurs_removed = remove_slurs(&data.content.to_owned()); let comment_form = CommentForm { content: content_slurs_removed, parent_id: data.parent_id, post_id: data.post_id, creator_id: data.creator_id, removed: data.removed.to_owned(), deleted: data.deleted.to_owned(), read: data.read.to_owned(), updated: if data.read.is_some() { orig_comment.updated } else {Some(naive_now())} }; let _updated_comment = match Comment::update(&conn, data.edit_id, &comment_form) { Ok(comment) => comment, Err(_e) => { return Err(APIError::err(&self.op, "Couldn't update Comment"))? } }; // Mod tables if let Some(removed) = data.removed.to_owned() { let form = ModRemoveCommentForm { mod_user_id: user_id, comment_id: data.edit_id, removed: Some(removed), reason: data.reason.to_owned(), }; ModRemoveComment::create(&conn, &form)?; } let comment_view = CommentView::read(&conn, data.edit_id, Some(user_id))?; Ok( CommentResponse { op: self.op.to_string(), comment: comment_view } ) } } impl Perform for Oper { fn perform(&self) -> Result { let data: &SaveComment = &self.data; let conn = establish_connection(); let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => { return Err(APIError::err(&self.op, "Not logged in."))? } }; let user_id = claims.id; let comment_saved_form = CommentSavedForm { comment_id: data.comment_id, user_id: user_id, }; if data.save { match CommentSaved::save(&conn, &comment_saved_form) { Ok(comment) => comment, Err(_e) => { return Err(APIError::err(&self.op, "Couldnt do comment save"))? } }; } else { match CommentSaved::unsave(&conn, &comment_saved_form) { Ok(comment) => comment, Err(_e) => { return Err(APIError::err(&self.op, "Couldnt do comment save"))? } }; } let comment_view = CommentView::read(&conn, data.comment_id, Some(user_id))?; Ok( CommentResponse { op: self.op.to_string(), comment: comment_view } ) } } impl Perform for Oper { fn perform(&self) -> Result { let data: &CreateCommentLike = &self.data; let conn = establish_connection(); let claims = match Claims::decode(&data.auth) { Ok(claims) => claims.claims, Err(_e) => { return Err(APIError::err(&self.op, "Not logged in."))? } }; let user_id = claims.id; // Check for a community ban let post = Post::read(&conn, data.post_id)?; if CommunityUserBanView::get(&conn, user_id, post.community_id).is_ok() { return Err(APIError::err(&self.op, "You have been banned from this community"))? } // Check for a site ban if UserView::read(&conn, user_id)?.banned { return Err(APIError::err(&self.op, "You have been banned from the site"))? } let like_form = CommentLikeForm { comment_id: data.comment_id, post_id: data.post_id, user_id: user_id, score: data.score }; // Remove any likes first CommentLike::remove(&conn, &like_form)?; // Only add the like if the score isnt 0 let do_add = &like_form.score != &0 && (&like_form.score == &1 || &like_form.score == &-1); if do_add { let _inserted_like = match CommentLike::like(&conn, &like_form) { Ok(like) => like, Err(_e) => { return Err(APIError::err(&self.op, "Couldn't like comment."))? } }; } // Have to refetch the comment to get the current state let liked_comment = CommentView::read(&conn, data.comment_id, Some(user_id))?; Ok( CommentResponse { op: self.op.to_string(), comment: liked_comment } ) } }