-use diesel::{result::Error, *};
-use lemmy_db_queries::{limit_and_offset, MaybeOptional, ToSafe, ViewToVec};
+use crate::structs::CommentReportView;
+use diesel::{
+ dsl::now,
+ result::Error,
+ BoolExpressionMethods,
+ ExpressionMethods,
+ JoinOnDsl,
+ NullableExpressionMethods,
+ QueryDsl,
+};
+use diesel_async::RunQueryDsl;
use lemmy_db_schema::{
- schema::{comment, comment_report, community, post, user_, user_alias_1, user_alias_2},
+ aggregates::structs::CommentAggregates,
+ newtypes::{CommentReportId, CommunityId, PersonId},
+ schema::{
+ comment,
+ comment_aggregates,
+ comment_like,
+ comment_report,
+ community,
+ community_moderator,
+ community_person_ban,
+ person,
+ post,
+ },
source::{
comment::Comment,
comment_report::CommentReport,
- community::{Community, CommunitySafe},
+ community::{Community, CommunityPersonBan},
+ person::Person,
post::Post,
- user::{UserAlias1, UserAlias2, UserSafe, UserSafeAlias1, UserSafeAlias2, User_},
},
+ traits::JoinView,
+ utils::{get_conn, limit_and_offset, DbPool},
};
-use serde::Serialize;
-
-#[derive(Debug, PartialEq, Serialize, Clone)]
-pub struct CommentReportView {
- pub comment_report: CommentReport,
- pub comment: Comment,
- pub post: Post,
- pub community: CommunitySafe,
- pub creator: UserSafe,
- pub comment_creator: UserSafeAlias1,
- pub resolver: Option<UserSafeAlias2>,
-}
-
-type CommentReportViewTuple = (
- CommentReport,
- Comment,
- Post,
- CommunitySafe,
- UserSafe,
- UserSafeAlias1,
- Option<UserSafeAlias2>,
-);
impl CommentReportView {
/// returns the CommentReportView for the provided report_id
///
/// * `report_id` - the report id to obtain
- pub fn read(conn: &PgConnection, report_id: i32) -> Result<Self, Error> {
- let (comment_report, comment, post, community, creator, comment_creator, resolver) =
- comment_report::table
- .find(report_id)
- .inner_join(comment::table)
- .inner_join(post::table.on(comment::post_id.eq(post::id)))
- .inner_join(community::table.on(post::community_id.eq(community::id)))
- .inner_join(user_::table.on(comment_report::creator_id.eq(user_::id)))
- .inner_join(user_alias_1::table.on(post::creator_id.eq(user_alias_1::id)))
- .left_join(
- user_alias_2::table.on(comment_report::resolver_id.eq(user_alias_2::id.nullable())),
- )
- .select((
- comment_report::all_columns,
- comment::all_columns,
- post::all_columns,
- Community::safe_columns_tuple(),
- User_::safe_columns_tuple(),
- UserAlias1::safe_columns_tuple(),
- UserAlias2::safe_columns_tuple().nullable(),
- ))
- .first::<CommentReportViewTuple>(conn)?;
-
- Ok(Self {
- comment_report,
- comment,
- post,
- community,
- creator,
- comment_creator,
- resolver,
- })
- }
+ pub async fn read(
+ pool: &mut DbPool<'_>,
+ report_id: CommentReportId,
+ my_person_id: PersonId,
+ ) -> Result<Self, Error> {
+ let conn = &mut get_conn(pool).await?;
- /// returns the current unresolved post report count for the supplied community ids
- ///
- /// * `community_ids` - a Vec<i32> of community_ids to get a count for
- /// TODO this eq_any is a bad way to do this, would be better to join to communitymoderator
- /// for a user id
- pub fn get_report_count(conn: &PgConnection, community_ids: &[i32]) -> Result<i64, Error> {
- use diesel::dsl::*;
- comment_report::table
+ let (person_alias_1, person_alias_2) = diesel::alias!(person as person1, person as person2);
+
+ let res = comment_report::table
+ .find(report_id)
.inner_join(comment::table)
.inner_join(post::table.on(comment::post_id.eq(post::id)))
- .filter(
- comment_report::resolved
- .eq(false)
- .and(post::community_id.eq_any(community_ids)),
+ .inner_join(community::table.on(post::community_id.eq(community::id)))
+ .inner_join(person::table.on(comment_report::creator_id.eq(person::id)))
+ .inner_join(person_alias_1.on(comment::creator_id.eq(person_alias_1.field(person::id))))
+ .inner_join(
+ comment_aggregates::table.on(comment_report::comment_id.eq(comment_aggregates::comment_id)),
+ )
+ .left_join(
+ community_person_ban::table.on(
+ community::id
+ .eq(community_person_ban::community_id)
+ .and(community_person_ban::person_id.eq(comment::creator_id)),
+ ),
)
- .select(count(comment_report::id))
- .first::<i64>(conn)
+ .left_join(
+ comment_like::table.on(
+ comment::id
+ .eq(comment_like::comment_id)
+ .and(comment_like::person_id.eq(my_person_id)),
+ ),
+ )
+ .left_join(
+ person_alias_2
+ .on(comment_report::resolver_id.eq(person_alias_2.field(person::id).nullable())),
+ )
+ .select((
+ comment_report::all_columns,
+ comment::all_columns,
+ post::all_columns,
+ community::all_columns,
+ person::all_columns,
+ person_alias_1.fields(person::all_columns),
+ comment_aggregates::all_columns,
+ community_person_ban::all_columns.nullable(),
+ comment_like::score.nullable(),
+ person_alias_2.fields(person::all_columns).nullable(),
+ ))
+ .first::<<CommentReportView as JoinView>::JoinTuple>(conn)
+ .await?;
+
+ Ok(Self::from_tuple(res))
}
-}
-pub struct CommentReportQueryBuilder<'a> {
- conn: &'a PgConnection,
- community_ids: Option<Vec<i32>>, // TODO bad way to do this
- page: Option<i64>,
- limit: Option<i64>,
- resolved: Option<bool>,
-}
+ /// Returns the current unresolved post report count for the communities you mod
+ pub async fn get_report_count(
+ pool: &mut DbPool<'_>,
+ my_person_id: PersonId,
+ admin: bool,
+ community_id: Option<CommunityId>,
+ ) -> Result<i64, Error> {
+ use diesel::dsl::count;
+
+ let conn = &mut get_conn(pool).await?;
-impl<'a> CommentReportQueryBuilder<'a> {
- pub fn create(conn: &'a PgConnection) -> Self {
- CommentReportQueryBuilder {
- conn,
- community_ids: None,
- page: None,
- limit: None,
- resolved: Some(false),
+ let mut query = comment_report::table
+ .inner_join(comment::table)
+ .inner_join(post::table.on(comment::post_id.eq(post::id)))
+ .filter(comment_report::resolved.eq(false))
+ .into_boxed();
+
+ if let Some(community_id) = community_id {
+ query = query.filter(post::community_id.eq(community_id))
}
- }
- pub fn community_ids<T: MaybeOptional<Vec<i32>>>(mut self, community_ids: T) -> Self {
- self.community_ids = community_ids.get_optional();
- self
+ // If its not an admin, get only the ones you mod
+ if !admin {
+ query
+ .inner_join(
+ community_moderator::table.on(
+ community_moderator::community_id
+ .eq(post::community_id)
+ .and(community_moderator::person_id.eq(my_person_id)),
+ ),
+ )
+ .select(count(comment_report::id))
+ .first::<i64>(conn)
+ .await
+ } else {
+ query
+ .select(count(comment_report::id))
+ .first::<i64>(conn)
+ .await
+ }
}
+}
- pub fn page<T: MaybeOptional<i64>>(mut self, page: T) -> Self {
- self.page = page.get_optional();
- self
- }
+#[derive(Default)]
+pub struct CommentReportQuery {
+ pub community_id: Option<CommunityId>,
+ pub page: Option<i64>,
+ pub limit: Option<i64>,
+ pub unresolved_only: Option<bool>,
+}
- pub fn limit<T: MaybeOptional<i64>>(mut self, limit: T) -> Self {
- self.limit = limit.get_optional();
- self
- }
+impl CommentReportQuery {
+ pub async fn list(
+ self,
+ pool: &mut DbPool<'_>,
+ my_person: &Person,
+ ) -> Result<Vec<CommentReportView>, Error> {
+ let conn = &mut get_conn(pool).await?;
- pub fn resolved<T: MaybeOptional<bool>>(mut self, resolved: T) -> Self {
- self.resolved = resolved.get_optional();
- self
- }
+ let (person_alias_1, person_alias_2) = diesel::alias!(person as person1, person as person2);
- pub fn list(self) -> Result<Vec<CommentReportView>, Error> {
let mut query = comment_report::table
.inner_join(comment::table)
.inner_join(post::table.on(comment::post_id.eq(post::id)))
.inner_join(community::table.on(post::community_id.eq(community::id)))
- .inner_join(user_::table.on(comment_report::creator_id.eq(user_::id)))
- .inner_join(user_alias_1::table.on(post::creator_id.eq(user_alias_1::id)))
+ .inner_join(person::table.on(comment_report::creator_id.eq(person::id)))
+ .inner_join(person_alias_1.on(comment::creator_id.eq(person_alias_1.field(person::id))))
+ .inner_join(
+ comment_aggregates::table.on(comment_report::comment_id.eq(comment_aggregates::comment_id)),
+ )
+ .left_join(
+ community_person_ban::table.on(
+ community::id
+ .eq(community_person_ban::community_id)
+ .and(community_person_ban::person_id.eq(comment::creator_id))
+ .and(
+ community_person_ban::expires
+ .is_null()
+ .or(community_person_ban::expires.gt(now)),
+ ),
+ ),
+ )
+ .left_join(
+ comment_like::table.on(
+ comment::id
+ .eq(comment_like::comment_id)
+ .and(comment_like::person_id.eq(my_person.id)),
+ ),
+ )
.left_join(
- user_alias_2::table.on(comment_report::resolver_id.eq(user_alias_2::id.nullable())),
+ person_alias_2
+ .on(comment_report::resolver_id.eq(person_alias_2.field(person::id).nullable())),
)
.select((
comment_report::all_columns,
comment::all_columns,
post::all_columns,
- Community::safe_columns_tuple(),
- User_::safe_columns_tuple(),
- UserAlias1::safe_columns_tuple(),
- UserAlias2::safe_columns_tuple().nullable(),
+ community::all_columns,
+ person::all_columns,
+ person_alias_1.fields(person::all_columns),
+ comment_aggregates::all_columns,
+ community_person_ban::all_columns.nullable(),
+ comment_like::score.nullable(),
+ person_alias_2.fields(person::all_columns).nullable(),
))
.into_boxed();
- if let Some(comm_ids) = self.community_ids {
- query = query.filter(post::community_id.eq_any(comm_ids));
+ if let Some(community_id) = self.community_id {
+ query = query.filter(post::community_id.eq(community_id));
}
- if let Some(resolved_flag) = self.resolved {
- query = query.filter(comment_report::resolved.eq(resolved_flag));
+ if self.unresolved_only.unwrap_or(false) {
+ query = query.filter(comment_report::resolved.eq(false));
}
- let (limit, offset) = limit_and_offset(self.page, self.limit);
+ let (limit, offset) = limit_and_offset(self.page, self.limit)?;
- let res = query
- .order_by(comment_report::published.asc())
+ query = query
+ .order_by(comment_report::published.desc())
.limit(limit)
- .offset(offset)
- .load::<CommentReportViewTuple>(self.conn)?;
+ .offset(offset);
- Ok(CommentReportView::from_tuple_to_vec(res))
+ // If its not an admin, get only the ones you mod
+ let res = if !my_person.admin {
+ query
+ .inner_join(
+ community_moderator::table.on(
+ community_moderator::community_id
+ .eq(post::community_id)
+ .and(community_moderator::person_id.eq(my_person.id)),
+ ),
+ )
+ .load::<<CommentReportView as JoinView>::JoinTuple>(conn)
+ .await?
+ } else {
+ query
+ .load::<<CommentReportView as JoinView>::JoinTuple>(conn)
+ .await?
+ };
+
+ Ok(res.into_iter().map(CommentReportView::from_tuple).collect())
}
}
-impl ViewToVec for CommentReportView {
- type DbTuple = CommentReportViewTuple;
- fn from_tuple_to_vec(items: Vec<Self::DbTuple>) -> Vec<Self> {
- items
- .iter()
- .map(|a| Self {
- comment_report: a.0.to_owned(),
- comment: a.1.to_owned(),
- post: a.2.to_owned(),
- community: a.3.to_owned(),
- creator: a.4.to_owned(),
- comment_creator: a.5.to_owned(),
- resolver: a.6.to_owned(),
- })
- .collect::<Vec<Self>>()
+impl JoinView for CommentReportView {
+ type JoinTuple = (
+ CommentReport,
+ Comment,
+ Post,
+ Community,
+ Person,
+ Person,
+ CommentAggregates,
+ Option<CommunityPersonBan>,
+ Option<i16>,
+ Option<Person>,
+ );
+
+ fn from_tuple(a: Self::JoinTuple) -> Self {
+ Self {
+ comment_report: a.0,
+ comment: a.1,
+ post: a.2,
+ community: a.3,
+ creator: a.4,
+ comment_creator: a.5,
+ counts: a.6,
+ creator_banned_from_community: a.7.is_some(),
+ my_vote: a.8,
+ resolver: a.9,
+ }
+ }
+}
+
+#[cfg(test)]
+mod tests {
+ #![allow(clippy::unwrap_used)]
+ #![allow(clippy::indexing_slicing)]
+
+ use crate::comment_report_view::{CommentReportQuery, CommentReportView};
+ use lemmy_db_schema::{
+ aggregates::structs::CommentAggregates,
+ source::{
+ comment::{Comment, CommentInsertForm},
+ comment_report::{CommentReport, CommentReportForm},
+ community::{Community, CommunityInsertForm, CommunityModerator, CommunityModeratorForm},
+ instance::Instance,
+ person::{Person, PersonInsertForm},
+ post::{Post, PostInsertForm},
+ },
+ traits::{Crud, Joinable, Reportable},
+ utils::build_db_pool_for_tests,
+ };
+ use serial_test::serial;
+
+ #[tokio::test]
+ #[serial]
+ async fn test_crud() {
+ let pool = &build_db_pool_for_tests().await;
+ let pool = &mut pool.into();
+
+ let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string())
+ .await
+ .unwrap();
+
+ let new_person = PersonInsertForm::builder()
+ .name("timmy_crv".into())
+ .public_key("pubkey".to_string())
+ .instance_id(inserted_instance.id)
+ .build();
+
+ let inserted_timmy = Person::create(pool, &new_person).await.unwrap();
+
+ let new_person_2 = PersonInsertForm::builder()
+ .name("sara_crv".into())
+ .public_key("pubkey".to_string())
+ .instance_id(inserted_instance.id)
+ .build();
+
+ let inserted_sara = Person::create(pool, &new_person_2).await.unwrap();
+
+ // Add a third person, since new ppl can only report something once.
+ let new_person_3 = PersonInsertForm::builder()
+ .name("jessica_crv".into())
+ .public_key("pubkey".to_string())
+ .instance_id(inserted_instance.id)
+ .build();
+
+ let inserted_jessica = Person::create(pool, &new_person_3).await.unwrap();
+
+ let new_community = CommunityInsertForm::builder()
+ .name("test community crv".to_string())
+ .title("nada".to_owned())
+ .public_key("pubkey".to_string())
+ .instance_id(inserted_instance.id)
+ .build();
+
+ let inserted_community = Community::create(pool, &new_community).await.unwrap();
+
+ // Make timmy a mod
+ let timmy_moderator_form = CommunityModeratorForm {
+ community_id: inserted_community.id,
+ person_id: inserted_timmy.id,
+ };
+
+ let _inserted_moderator = CommunityModerator::join(pool, &timmy_moderator_form)
+ .await
+ .unwrap();
+
+ let new_post = PostInsertForm::builder()
+ .name("A test post crv".into())
+ .creator_id(inserted_timmy.id)
+ .community_id(inserted_community.id)
+ .build();
+
+ let inserted_post = Post::create(pool, &new_post).await.unwrap();
+
+ let comment_form = CommentInsertForm::builder()
+ .content("A test comment 32".into())
+ .creator_id(inserted_timmy.id)
+ .post_id(inserted_post.id)
+ .build();
+
+ let inserted_comment = Comment::create(pool, &comment_form, None).await.unwrap();
+
+ // sara reports
+ let sara_report_form = CommentReportForm {
+ creator_id: inserted_sara.id,
+ comment_id: inserted_comment.id,
+ original_comment_text: "this was it at time of creation".into(),
+ reason: "from sara".into(),
+ };
+
+ let inserted_sara_report = CommentReport::report(pool, &sara_report_form)
+ .await
+ .unwrap();
+
+ // jessica reports
+ let jessica_report_form = CommentReportForm {
+ creator_id: inserted_jessica.id,
+ comment_id: inserted_comment.id,
+ original_comment_text: "this was it at time of creation".into(),
+ reason: "from jessica".into(),
+ };
+
+ let inserted_jessica_report = CommentReport::report(pool, &jessica_report_form)
+ .await
+ .unwrap();
+
+ let agg = CommentAggregates::read(pool, inserted_comment.id)
+ .await
+ .unwrap();
+
+ let read_jessica_report_view =
+ CommentReportView::read(pool, inserted_jessica_report.id, inserted_timmy.id)
+ .await
+ .unwrap();
+ let expected_jessica_report_view = CommentReportView {
+ comment_report: inserted_jessica_report.clone(),
+ comment: inserted_comment.clone(),
+ post: inserted_post,
+ community: Community {
+ id: inserted_community.id,
+ name: inserted_community.name,
+ icon: None,
+ removed: false,
+ deleted: false,
+ nsfw: false,
+ actor_id: inserted_community.actor_id.clone(),
+ local: true,
+ title: inserted_community.title,
+ description: None,
+ updated: None,
+ banner: None,
+ hidden: false,
+ posting_restricted_to_mods: false,
+ published: inserted_community.published,
+ private_key: inserted_community.private_key,
+ public_key: inserted_community.public_key,
+ last_refreshed_at: inserted_community.last_refreshed_at,
+ followers_url: inserted_community.followers_url,
+ inbox_url: inserted_community.inbox_url,
+ shared_inbox_url: inserted_community.shared_inbox_url,
+ moderators_url: inserted_community.moderators_url,
+ featured_url: inserted_community.featured_url,
+ instance_id: inserted_instance.id,
+ },
+ creator: Person {
+ id: inserted_jessica.id,
+ name: inserted_jessica.name,
+ display_name: None,
+ published: inserted_jessica.published,
+ avatar: None,
+ actor_id: inserted_jessica.actor_id.clone(),
+ local: true,
+ banned: false,
+ deleted: false,
+ admin: false,
+ bot_account: false,
+ bio: None,
+ banner: None,
+ updated: None,
+ inbox_url: inserted_jessica.inbox_url.clone(),
+ shared_inbox_url: None,
+ matrix_user_id: None,
+ ban_expires: None,
+ instance_id: inserted_instance.id,
+ private_key: inserted_jessica.private_key,
+ public_key: inserted_jessica.public_key,
+ last_refreshed_at: inserted_jessica.last_refreshed_at,
+ },
+ comment_creator: Person {
+ id: inserted_timmy.id,
+ name: inserted_timmy.name.clone(),
+ display_name: None,
+ published: inserted_timmy.published,
+ avatar: None,
+ actor_id: inserted_timmy.actor_id.clone(),
+ local: true,
+ banned: false,
+ deleted: false,
+ admin: false,
+ bot_account: false,
+ bio: None,
+ banner: None,
+ updated: None,
+ inbox_url: inserted_timmy.inbox_url.clone(),
+ shared_inbox_url: None,
+ matrix_user_id: None,
+ ban_expires: None,
+ instance_id: inserted_instance.id,
+ private_key: inserted_timmy.private_key.clone(),
+ public_key: inserted_timmy.public_key.clone(),
+ last_refreshed_at: inserted_timmy.last_refreshed_at,
+ },
+ creator_banned_from_community: false,
+ counts: CommentAggregates {
+ id: agg.id,
+ comment_id: inserted_comment.id,
+ score: 0,
+ upvotes: 0,
+ downvotes: 0,
+ published: agg.published,
+ child_count: 0,
+ hot_rank: 1728,
+ },
+ my_vote: None,
+ resolver: None,
+ };
+
+ assert_eq!(read_jessica_report_view, expected_jessica_report_view);
+
+ let mut expected_sara_report_view = expected_jessica_report_view.clone();
+ expected_sara_report_view.comment_report = inserted_sara_report;
+ expected_sara_report_view.creator = Person {
+ id: inserted_sara.id,
+ name: inserted_sara.name,
+ display_name: None,
+ published: inserted_sara.published,
+ avatar: None,
+ actor_id: inserted_sara.actor_id.clone(),
+ local: true,
+ banned: false,
+ deleted: false,
+ admin: false,
+ bot_account: false,
+ bio: None,
+ banner: None,
+ updated: None,
+ inbox_url: inserted_sara.inbox_url.clone(),
+ shared_inbox_url: None,
+ matrix_user_id: None,
+ ban_expires: None,
+ instance_id: inserted_instance.id,
+ private_key: inserted_sara.private_key,
+ public_key: inserted_sara.public_key,
+ last_refreshed_at: inserted_sara.last_refreshed_at,
+ };
+
+ // Do a batch read of timmys reports
+ let reports = CommentReportQuery::default()
+ .list(pool, &inserted_timmy)
+ .await
+ .unwrap();
+
+ assert_eq!(
+ reports,
+ [
+ expected_jessica_report_view.clone(),
+ expected_sara_report_view.clone()
+ ]
+ );
+
+ // Make sure the counts are correct
+ let report_count = CommentReportView::get_report_count(pool, inserted_timmy.id, false, None)
+ .await
+ .unwrap();
+ assert_eq!(2, report_count);
+
+ // Try to resolve the report
+ CommentReport::resolve(pool, inserted_jessica_report.id, inserted_timmy.id)
+ .await
+ .unwrap();
+ let read_jessica_report_view_after_resolve =
+ CommentReportView::read(pool, inserted_jessica_report.id, inserted_timmy.id)
+ .await
+ .unwrap();
+
+ let mut expected_jessica_report_view_after_resolve = expected_jessica_report_view;
+ expected_jessica_report_view_after_resolve
+ .comment_report
+ .resolved = true;
+ expected_jessica_report_view_after_resolve
+ .comment_report
+ .resolver_id = Some(inserted_timmy.id);
+ expected_jessica_report_view_after_resolve
+ .comment_report
+ .updated = read_jessica_report_view_after_resolve
+ .comment_report
+ .updated;
+ expected_jessica_report_view_after_resolve.resolver = Some(Person {
+ id: inserted_timmy.id,
+ name: inserted_timmy.name.clone(),
+ display_name: None,
+ published: inserted_timmy.published,
+ avatar: None,
+ actor_id: inserted_timmy.actor_id.clone(),
+ local: true,
+ banned: false,
+ deleted: false,
+ admin: false,
+ bot_account: false,
+ bio: None,
+ banner: None,
+ updated: None,
+ inbox_url: inserted_timmy.inbox_url.clone(),
+ private_key: inserted_timmy.private_key.clone(),
+ public_key: inserted_timmy.public_key.clone(),
+ last_refreshed_at: inserted_timmy.last_refreshed_at,
+ shared_inbox_url: None,
+ matrix_user_id: None,
+ ban_expires: None,
+ instance_id: inserted_instance.id,
+ });
+
+ assert_eq!(
+ read_jessica_report_view_after_resolve,
+ expected_jessica_report_view_after_resolve
+ );
+
+ // Do a batch read of timmys reports
+ // It should only show saras, which is unresolved
+ let reports_after_resolve = CommentReportQuery {
+ unresolved_only: (Some(true)),
+ ..Default::default()
+ }
+ .list(pool, &inserted_timmy)
+ .await
+ .unwrap();
+ assert_eq!(reports_after_resolve[0], expected_sara_report_view);
+ assert_eq!(reports_after_resolve.len(), 1);
+
+ // Make sure the counts are correct
+ let report_count_after_resolved =
+ CommentReportView::get_report_count(pool, inserted_timmy.id, false, None)
+ .await
+ .unwrap();
+ assert_eq!(1, report_count_after_resolved);
+
+ Person::delete(pool, inserted_timmy.id).await.unwrap();
+ Person::delete(pool, inserted_sara.id).await.unwrap();
+ Person::delete(pool, inserted_jessica.id).await.unwrap();
+ Community::delete(pool, inserted_community.id)
+ .await
+ .unwrap();
+ Instance::delete(pool, inserted_instance.id).await.unwrap();
}
}
projects / lemmy.git / blobdiff