]> Untitled Git - lemmy.git/blobdiff - crates/db_views/src/post_report_view.rs
Clean up reporting (#1776)
[lemmy.git] / crates / db_views / src / post_report_view.rs
index a3dc49e288eb56af96066a67bf784795b4845bed..b7becdbc5615589f28e444b932137d23d6852307 100644 (file)
@@ -1,14 +1,33 @@
 use diesel::{result::Error, *};
-use lemmy_db_queries::{limit_and_offset, MaybeOptional, ToSafe, ViewToVec};
+use lemmy_db_queries::{
+  aggregates::post_aggregates::PostAggregates,
+  limit_and_offset,
+  MaybeOptional,
+  ToSafe,
+  ViewToVec,
+};
 use lemmy_db_schema::{
-  schema::{community, person, person_alias_1, person_alias_2, post, post_report},
+  schema::{
+    community,
+    community_moderator,
+    community_person_ban,
+    person,
+    person_alias_1,
+    person_alias_2,
+    post,
+    post_aggregates,
+    post_like,
+    post_report,
+  },
   source::{
-    community::{Community, CommunitySafe},
+    community::{Community, CommunityPersonBan, CommunitySafe},
     person::{Person, PersonAlias1, PersonAlias2, PersonSafe, PersonSafeAlias1, PersonSafeAlias2},
     post::Post,
     post_report::PostReport,
   },
   CommunityId,
+  PersonId,
+  PostReportId,
 };
 use serde::Serialize;
 
@@ -19,6 +38,9 @@ pub struct PostReportView {
   pub community: CommunitySafe,
   pub creator: PersonSafe,
   pub post_creator: PersonSafeAlias1,
+  pub creator_banned_from_community: bool,
+  pub my_vote: Option<i16>,
+  pub counts: PostAggregates,
   pub resolver: Option<PersonSafeAlias2>,
 }
 
@@ -28,6 +50,9 @@ type PostReportViewTuple = (
   CommunitySafe,
   PersonSafe,
   PersonSafeAlias1,
+  Option<CommunityPersonBan>,
+  Option<i16>,
+  PostAggregates,
   Option<PersonSafeAlias2>,
 );
 
@@ -35,13 +60,42 @@ impl PostReportView {
   /// returns the PostReportView for the provided report_id
   ///
   /// * `report_id` - the report id to obtain
-  pub fn read(conn: &PgConnection, report_id: i32) -> Result<Self, Error> {
-    let (post_report, post, community, creator, post_creator, resolver) = post_report::table
+  pub fn read(
+    conn: &PgConnection,
+    report_id: PostReportId,
+    my_person_id: PersonId,
+  ) -> Result<Self, Error> {
+    let (
+      post_report,
+      post,
+      community,
+      creator,
+      post_creator,
+      creator_banned_from_community,
+      post_like,
+      counts,
+      resolver,
+    ) = post_report::table
       .find(report_id)
       .inner_join(post::table)
       .inner_join(community::table.on(post::community_id.eq(community::id)))
       .inner_join(person::table.on(post_report::creator_id.eq(person::id)))
       .inner_join(person_alias_1::table.on(post::creator_id.eq(person_alias_1::id)))
+      .left_join(
+        community_person_ban::table.on(
+          post::community_id
+            .eq(community_person_ban::community_id)
+            .and(community_person_ban::person_id.eq(post::creator_id)),
+        ),
+      )
+      .left_join(
+        post_like::table.on(
+          post::id
+            .eq(post_like::post_id)
+            .and(post_like::person_id.eq(my_person_id)),
+        ),
+      )
+      .inner_join(post_aggregates::table.on(post_report::post_id.eq(post_aggregates::post_id)))
       .left_join(
         person_alias_2::table.on(post_report::resolver_id.eq(person_alias_2::id.nullable())),
       )
@@ -51,16 +105,24 @@ impl PostReportView {
         Community::safe_columns_tuple(),
         Person::safe_columns_tuple(),
         PersonAlias1::safe_columns_tuple(),
+        community_person_ban::all_columns.nullable(),
+        post_like::score.nullable(),
+        post_aggregates::all_columns,
         PersonAlias2::safe_columns_tuple().nullable(),
       ))
       .first::<PostReportViewTuple>(conn)?;
 
+    let my_vote = if post_like.is_none() { None } else { post_like };
+
     Ok(Self {
       post_report,
       post,
       community,
       creator,
       post_creator,
+      creator_banned_from_community: creator_banned_from_community.is_some(),
+      my_vote,
+      counts,
       resolver,
     })
   }
@@ -72,42 +134,54 @@ impl PostReportView {
   /// for a person id
   pub fn get_report_count(
     conn: &PgConnection,
-    community_ids: &[CommunityId],
+    my_person_id: PersonId,
+    community_id: Option<CommunityId>,
   ) -> Result<i64, Error> {
     use diesel::dsl::*;
-    post_report::table
+    let mut query = post_report::table
       .inner_join(post::table)
-      .filter(
-        post_report::resolved
-          .eq(false)
-          .and(post::community_id.eq_any(community_ids)),
+      // Test this join
+      .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(post_report::id))
-      .first::<i64>(conn)
+      .filter(post_report::resolved.eq(false))
+      .into_boxed();
+
+    if let Some(community_id) = community_id {
+      query = query.filter(post::community_id.eq(community_id))
+    }
+
+    query.select(count(post_report::id)).first::<i64>(conn)
   }
 }
 
 pub struct PostReportQueryBuilder<'a> {
   conn: &'a PgConnection,
-  community_ids: Option<Vec<CommunityId>>, // TODO bad way to do this
+  my_person_id: PersonId,
+  community_id: Option<CommunityId>,
   page: Option<i64>,
   limit: Option<i64>,
-  resolved: Option<bool>,
+  unresolved_only: Option<bool>,
 }
 
 impl<'a> PostReportQueryBuilder<'a> {
-  pub fn create(conn: &'a PgConnection) -> Self {
+  pub fn create(conn: &'a PgConnection, my_person_id: PersonId) -> Self {
     PostReportQueryBuilder {
       conn,
-      community_ids: None,
+      my_person_id,
+      community_id: None,
       page: None,
       limit: None,
-      resolved: Some(false),
+      unresolved_only: Some(true),
     }
   }
 
-  pub fn community_ids<T: MaybeOptional<Vec<CommunityId>>>(mut self, community_ids: T) -> Self {
-    self.community_ids = community_ids.get_optional();
+  pub fn community_id<T: MaybeOptional<CommunityId>>(mut self, community_id: T) -> Self {
+    self.community_id = community_id.get_optional();
     self
   }
 
@@ -121,8 +195,8 @@ impl<'a> PostReportQueryBuilder<'a> {
     self
   }
 
-  pub fn resolved<T: MaybeOptional<bool>>(mut self, resolved: T) -> Self {
-    self.resolved = resolved.get_optional();
+  pub fn unresolved_only<T: MaybeOptional<bool>>(mut self, unresolved_only: T) -> Self {
+    self.unresolved_only = unresolved_only.get_optional();
     self
   }
 
@@ -132,6 +206,29 @@ impl<'a> PostReportQueryBuilder<'a> {
       .inner_join(community::table.on(post::community_id.eq(community::id)))
       .inner_join(person::table.on(post_report::creator_id.eq(person::id)))
       .inner_join(person_alias_1::table.on(post::creator_id.eq(person_alias_1::id)))
+      // Test this join
+      .inner_join(
+        community_moderator::table.on(
+          community_moderator::community_id
+            .eq(post::community_id)
+            .and(community_moderator::person_id.eq(self.my_person_id)),
+        ),
+      )
+      .left_join(
+        community_person_ban::table.on(
+          post::community_id
+            .eq(community_person_ban::community_id)
+            .and(community_person_ban::person_id.eq(post::creator_id)),
+        ),
+      )
+      .left_join(
+        post_like::table.on(
+          post::id
+            .eq(post_like::post_id)
+            .and(post_like::person_id.eq(self.my_person_id)),
+        ),
+      )
+      .inner_join(post_aggregates::table.on(post_report::post_id.eq(post_aggregates::post_id)))
       .left_join(
         person_alias_2::table.on(post_report::resolver_id.eq(person_alias_2::id.nullable())),
       )
@@ -141,16 +238,19 @@ impl<'a> PostReportQueryBuilder<'a> {
         Community::safe_columns_tuple(),
         Person::safe_columns_tuple(),
         PersonAlias1::safe_columns_tuple(),
+        community_person_ban::all_columns.nullable(),
+        post_like::score.nullable(),
+        post_aggregates::all_columns,
         PersonAlias2::safe_columns_tuple().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(post_report::resolved.eq(resolved_flag));
+    if self.unresolved_only.unwrap_or(false) {
+      query = query.filter(post_report::resolved.eq(false));
     }
 
     let (limit, offset) = limit_and_offset(self.page, self.limit);
@@ -176,8 +276,283 @@ impl ViewToVec for PostReportView {
         community: a.2.to_owned(),
         creator: a.3.to_owned(),
         post_creator: a.4.to_owned(),
-        resolver: a.5.to_owned(),
+        creator_banned_from_community: a.5.is_some(),
+        my_vote: a.6,
+        counts: a.7.to_owned(),
+        resolver: a.8.to_owned(),
       })
       .collect::<Vec<Self>>()
   }
 }
+
+#[cfg(test)]
+mod tests {
+  use crate::post_report_view::{PostReportQueryBuilder, PostReportView};
+  use lemmy_db_queries::{
+    aggregates::post_aggregates::PostAggregates,
+    establish_unpooled_connection,
+    Crud,
+    Joinable,
+    Reportable,
+  };
+  use lemmy_db_schema::source::{
+    community::*,
+    person::*,
+    post::*,
+    post_report::{PostReport, PostReportForm},
+  };
+  use serial_test::serial;
+
+  #[test]
+  #[serial]
+  fn test_crud() {
+    let conn = establish_unpooled_connection();
+
+    let new_person = PersonForm {
+      name: "timmy_prv".into(),
+      ..PersonForm::default()
+    };
+
+    let inserted_timmy = Person::create(&conn, &new_person).unwrap();
+
+    let new_person_2 = PersonForm {
+      name: "sara_prv".into(),
+      ..PersonForm::default()
+    };
+
+    let inserted_sara = Person::create(&conn, &new_person_2).unwrap();
+
+    // Add a third person, since new ppl can only report something once.
+    let new_person_3 = PersonForm {
+      name: "jessica_prv".into(),
+      ..PersonForm::default()
+    };
+
+    let inserted_jessica = Person::create(&conn, &new_person_3).unwrap();
+
+    let new_community = CommunityForm {
+      name: "test community prv".to_string(),
+      title: "nada".to_owned(),
+      ..CommunityForm::default()
+    };
+
+    let inserted_community = Community::create(&conn, &new_community).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(&conn, &timmy_moderator_form).unwrap();
+
+    let new_post = PostForm {
+      name: "A test post crv".into(),
+      creator_id: inserted_timmy.id,
+      community_id: inserted_community.id,
+      ..PostForm::default()
+    };
+
+    let inserted_post = Post::create(&conn, &new_post).unwrap();
+
+    // sara reports
+    let sara_report_form = PostReportForm {
+      creator_id: inserted_sara.id,
+      post_id: inserted_post.id,
+      original_post_name: "Orig post".into(),
+      original_post_url: None,
+      original_post_body: None,
+      reason: "from sara".into(),
+    };
+
+    let inserted_sara_report = PostReport::report(&conn, &sara_report_form).unwrap();
+
+    // jessica reports
+    let jessica_report_form = PostReportForm {
+      creator_id: inserted_jessica.id,
+      post_id: inserted_post.id,
+      original_post_name: "Orig post".into(),
+      original_post_url: None,
+      original_post_body: None,
+      reason: "from jessica".into(),
+    };
+
+    let inserted_jessica_report = PostReport::report(&conn, &jessica_report_form).unwrap();
+
+    let agg = PostAggregates::read(&conn, inserted_post.id).unwrap();
+
+    let read_jessica_report_view =
+      PostReportView::read(&conn, inserted_jessica_report.id, inserted_timmy.id).unwrap();
+    let expected_jessica_report_view = PostReportView {
+      post_report: inserted_jessica_report.to_owned(),
+      post: inserted_post.to_owned(),
+      community: CommunitySafe {
+        id: inserted_community.id,
+        name: inserted_community.name,
+        icon: None,
+        removed: false,
+        deleted: false,
+        nsfw: false,
+        actor_id: inserted_community.actor_id.to_owned(),
+        local: true,
+        title: inserted_community.title,
+        description: None,
+        updated: None,
+        banner: None,
+        published: inserted_community.published,
+      },
+      creator: PersonSafe {
+        id: inserted_jessica.id,
+        name: inserted_jessica.name,
+        display_name: None,
+        published: inserted_jessica.published,
+        avatar: None,
+        actor_id: inserted_jessica.actor_id.to_owned(),
+        local: true,
+        banned: false,
+        deleted: false,
+        admin: false,
+        bot_account: false,
+        bio: None,
+        banner: None,
+        updated: None,
+        inbox_url: inserted_jessica.inbox_url.to_owned(),
+        shared_inbox_url: None,
+        matrix_user_id: None,
+      },
+      post_creator: PersonSafeAlias1 {
+        id: inserted_timmy.id,
+        name: inserted_timmy.name.to_owned(),
+        display_name: None,
+        published: inserted_timmy.published,
+        avatar: None,
+        actor_id: inserted_timmy.actor_id.to_owned(),
+        local: true,
+        banned: false,
+        deleted: false,
+        admin: false,
+        bot_account: false,
+        bio: None,
+        banner: None,
+        updated: None,
+        inbox_url: inserted_timmy.inbox_url.to_owned(),
+        shared_inbox_url: None,
+        matrix_user_id: None,
+      },
+      creator_banned_from_community: false,
+      my_vote: None,
+      counts: PostAggregates {
+        id: agg.id,
+        post_id: inserted_post.id,
+        comments: 0,
+        score: 0,
+        upvotes: 0,
+        downvotes: 0,
+        stickied: false,
+        published: agg.published,
+        newest_comment_time_necro: inserted_post.published,
+        newest_comment_time: inserted_post.published,
+      },
+      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.post_report = inserted_sara_report;
+    expected_sara_report_view.my_vote = None;
+    expected_sara_report_view.creator = PersonSafe {
+      id: inserted_sara.id,
+      name: inserted_sara.name,
+      display_name: None,
+      published: inserted_sara.published,
+      avatar: None,
+      actor_id: inserted_sara.actor_id.to_owned(),
+      local: true,
+      banned: false,
+      deleted: false,
+      admin: false,
+      bot_account: false,
+      bio: None,
+      banner: None,
+      updated: None,
+      inbox_url: inserted_sara.inbox_url.to_owned(),
+      shared_inbox_url: None,
+      matrix_user_id: None,
+    };
+
+    // Do a batch read of timmys reports
+    let reports = PostReportQueryBuilder::create(&conn, inserted_timmy.id)
+      .list()
+      .unwrap();
+
+    assert_eq!(
+      reports,
+      [
+        expected_sara_report_view.to_owned(),
+        expected_jessica_report_view.to_owned()
+      ]
+    );
+
+    // Make sure the counts are correct
+    let report_count = PostReportView::get_report_count(&conn, inserted_timmy.id, None).unwrap();
+    assert_eq!(2, report_count);
+
+    // Try to resolve the report
+    PostReport::resolve(&conn, inserted_jessica_report.id, inserted_timmy.id).unwrap();
+    let read_jessica_report_view_after_resolve =
+      PostReportView::read(&conn, inserted_jessica_report.id, inserted_timmy.id).unwrap();
+
+    let mut expected_jessica_report_view_after_resolve = expected_jessica_report_view;
+    expected_jessica_report_view_after_resolve
+      .post_report
+      .resolved = true;
+    expected_jessica_report_view_after_resolve
+      .post_report
+      .resolver_id = Some(inserted_timmy.id);
+    expected_jessica_report_view_after_resolve
+      .post_report
+      .updated = read_jessica_report_view_after_resolve.post_report.updated;
+    expected_jessica_report_view_after_resolve.resolver = Some(PersonSafeAlias2 {
+      id: inserted_timmy.id,
+      name: inserted_timmy.name.to_owned(),
+      display_name: None,
+      published: inserted_timmy.published,
+      avatar: None,
+      actor_id: inserted_timmy.actor_id.to_owned(),
+      local: true,
+      banned: false,
+      deleted: false,
+      admin: false,
+      bot_account: false,
+      bio: None,
+      banner: None,
+      updated: None,
+      inbox_url: inserted_timmy.inbox_url.to_owned(),
+      shared_inbox_url: None,
+      matrix_user_id: None,
+    });
+
+    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 = PostReportQueryBuilder::create(&conn, inserted_timmy.id)
+      .list()
+      .unwrap();
+    assert_eq!(reports_after_resolve[0], expected_sara_report_view);
+
+    // Make sure the counts are correct
+    let report_count_after_resolved =
+      PostReportView::get_report_count(&conn, inserted_timmy.id, None).unwrap();
+    assert_eq!(1, report_count_after_resolved);
+
+    Person::delete(&conn, inserted_timmy.id).unwrap();
+    Person::delete(&conn, inserted_sara.id).unwrap();
+    Person::delete(&conn, inserted_jessica.id).unwrap();
+    Community::delete(&conn, inserted_community.id).unwrap();
+  }
+}