crates/db_views/src/private_message_report_view.rs

   1 use crate::structs::PrivateMessageReportView;
   2 use diesel::{result::Error, ExpressionMethods, JoinOnDsl, NullableExpressionMethods, QueryDsl};
   3 use diesel_async::RunQueryDsl;
   4 use lemmy_db_schema::{
   5   newtypes::PrivateMessageReportId,
   6   schema::{person, private_message, private_message_report},
   7   source::{
   8     person::Person,
   9     private_message::PrivateMessage,
  10     private_message_report::PrivateMessageReport,
  11   },
  12   traits::JoinView,
  13   utils::{get_conn, limit_and_offset, DbPool},
  14 };
  15
  16 type PrivateMessageReportViewTuple = (
  17   PrivateMessageReport,
  18   PrivateMessage,
  19   Person,
  20   Person,
  21   Option<Person>,
  22 );
  23
  24 impl PrivateMessageReportView {
  25   /// returns the PrivateMessageReportView for the provided report_id
  26   ///
  27   /// * `report_id` - the report id to obtain
  28   pub async fn read(
  29     pool: &mut DbPool<'_>,
  30     report_id: PrivateMessageReportId,
  31   ) -> Result<Self, Error> {
  32     let conn = &mut get_conn(pool).await?;
  33     let (person_alias_1, person_alias_2) = diesel::alias!(person as person1, person as person2);
  34
  35     let (private_message_report, private_message, private_message_creator, creator, resolver) =
  36       private_message_report::table
  37         .find(report_id)
  38         .inner_join(private_message::table)
  39         .inner_join(person::table.on(private_message::creator_id.eq(person::id)))
  40         .inner_join(
  41           person_alias_1
  42             .on(private_message_report::creator_id.eq(person_alias_1.field(person::id))),
  43         )
  44         .left_join(
  45           person_alias_2.on(
  46             private_message_report::resolver_id.eq(person_alias_2.field(person::id).nullable()),
  47           ),
  48         )
  49         .select((
  50           private_message_report::all_columns,
  51           private_message::all_columns,
  52           person::all_columns,
  53           person_alias_1.fields(person::all_columns),
  54           person_alias_2.fields(person::all_columns).nullable(),
  55         ))
  56         .first::<PrivateMessageReportViewTuple>(conn)
  57         .await?;
  58
  59     Ok(Self {
  60       private_message_report,
  61       private_message,
  62       private_message_creator,
  63       creator,
  64       resolver,
  65     })
  66   }
  67
  68   /// Returns the current unresolved post report count for the communities you mod
  69   pub async fn get_report_count(pool: &mut DbPool<'_>) -> Result<i64, Error> {
  70     use diesel::dsl::count;
  71     let conn = &mut get_conn(pool).await?;
  72
  73     private_message_report::table
  74       .inner_join(private_message::table)
  75       .filter(private_message_report::resolved.eq(false))
  76       .into_boxed()
  77       .select(count(private_message_report::id))
  78       .first::<i64>(conn)
  79       .await
  80   }
  81 }
  82
  83 #[derive(Default)]
  84 pub struct PrivateMessageReportQuery {
  85   pub page: Option<i64>,
  86   pub limit: Option<i64>,
  87   pub unresolved_only: Option<bool>,
  88 }
  89
  90 impl PrivateMessageReportQuery {
  91   pub async fn list(self, pool: &mut DbPool<'_>) -> Result<Vec<PrivateMessageReportView>, Error> {
  92     let conn = &mut get_conn(pool).await?;
  93     let (person_alias_1, person_alias_2) = diesel::alias!(person as person1, person as person2);
  94
  95     let mut query = private_message_report::table
  96       .inner_join(private_message::table)
  97       .inner_join(person::table.on(private_message::creator_id.eq(person::id)))
  98       .inner_join(
  99         person_alias_1.on(private_message_report::creator_id.eq(person_alias_1.field(person::id))),
 100       )
 101       .left_join(
 102         person_alias_2
 103           .on(private_message_report::resolver_id.eq(person_alias_2.field(person::id).nullable())),
 104       )
 105       .select((
 106         private_message_report::all_columns,
 107         private_message::all_columns,
 108         person::all_columns,
 109         person_alias_1.fields(person::all_columns),
 110         person_alias_2.fields(person::all_columns).nullable(),
 111       ))
 112       .into_boxed();
 113
 114     if self.unresolved_only.unwrap_or(false) {
 115       query = query.filter(private_message_report::resolved.eq(false));
 116     }
 117
 118     let (limit, offset) = limit_and_offset(self.page, self.limit)?;
 119
 120     query = query
 121       .order_by(private_message::published.desc())
 122       .limit(limit)
 123       .offset(offset);
 124
 125     let res = query.load::<PrivateMessageReportViewTuple>(conn).await?;
 126
 127     Ok(
 128       res
 129         .into_iter()
 130         .map(PrivateMessageReportView::from_tuple)
 131         .collect(),
 132     )
 133   }
 134 }
 135
 136 impl JoinView for PrivateMessageReportView {
 137   type JoinTuple = PrivateMessageReportViewTuple;
 138   fn from_tuple(a: Self::JoinTuple) -> Self {
 139     Self {
 140       private_message_report: a.0,
 141       private_message: a.1,
 142       private_message_creator: a.2,
 143       creator: a.3,
 144       resolver: a.4,
 145     }
 146   }
 147 }
 148
 149 #[cfg(test)]
 150 mod tests {
 151   use crate::private_message_report_view::PrivateMessageReportQuery;
 152   use lemmy_db_schema::{
 153     source::{
 154       instance::Instance,
 155       person::{Person, PersonInsertForm},
 156       private_message::{PrivateMessage, PrivateMessageInsertForm},
 157       private_message_report::{PrivateMessageReport, PrivateMessageReportForm},
 158     },
 159     traits::{Crud, Reportable},
 160     utils::build_db_pool_for_tests,
 161   };
 162   use serial_test::serial;
 163
 164   #[tokio::test]
 165   #[serial]
 166   async fn test_crud() {
 167     let pool = &build_db_pool_for_tests().await;
 168     let pool = &mut pool.into();
 169
 170     let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string())
 171       .await
 172       .unwrap();
 173
 174     let new_person_1 = PersonInsertForm::builder()
 175       .name("timmy_mrv".into())
 176       .public_key("pubkey".to_string())
 177       .instance_id(inserted_instance.id)
 178       .build();
 179     let inserted_timmy = Person::create(pool, &new_person_1).await.unwrap();
 180
 181     let new_person_2 = PersonInsertForm::builder()
 182       .name("jessica_mrv".into())
 183       .public_key("pubkey".to_string())
 184       .instance_id(inserted_instance.id)
 185       .build();
 186     let inserted_jessica = Person::create(pool, &new_person_2).await.unwrap();
 187
 188     // timmy sends private message to jessica
 189     let pm_form = PrivateMessageInsertForm::builder()
 190       .creator_id(inserted_timmy.id)
 191       .recipient_id(inserted_jessica.id)
 192       .content("something offensive".to_string())
 193       .build();
 194     let pm = PrivateMessage::create(pool, &pm_form).await.unwrap();
 195
 196     // jessica reports private message
 197     let pm_report_form = PrivateMessageReportForm {
 198       creator_id: inserted_jessica.id,
 199       original_pm_text: pm.content.clone(),
 200       private_message_id: pm.id,
 201       reason: "its offensive".to_string(),
 202     };
 203     let pm_report = PrivateMessageReport::report(pool, &pm_report_form)
 204       .await
 205       .unwrap();
 206
 207     let reports = PrivateMessageReportQuery::default()
 208       .list(pool)
 209       .await
 210       .unwrap();
 211     assert_eq!(1, reports.len());
 212     assert!(!reports[0].private_message_report.resolved);
 213     assert_eq!(inserted_timmy.name, reports[0].private_message_creator.name);
 214     assert_eq!(inserted_jessica.name, reports[0].creator.name);
 215     assert_eq!(pm_report.reason, reports[0].private_message_report.reason);
 216     assert_eq!(pm.content, reports[0].private_message.content);
 217
 218     let new_person_3 = PersonInsertForm::builder()
 219       .name("admin_mrv".into())
 220       .public_key("pubkey".to_string())
 221       .instance_id(inserted_instance.id)
 222       .build();
 223     let inserted_admin = Person::create(pool, &new_person_3).await.unwrap();
 224
 225     // admin resolves the report (after taking appropriate action)
 226     PrivateMessageReport::resolve(pool, pm_report.id, inserted_admin.id)
 227       .await
 228       .unwrap();
 229
 230     let reports = PrivateMessageReportQuery {
 231       unresolved_only: (Some(false)),
 232       ..Default::default()
 233     }
 234     .list(pool)
 235     .await
 236     .unwrap();
 237     assert_eq!(1, reports.len());
 238     assert!(reports[0].private_message_report.resolved);
 239     assert!(reports[0].resolver.is_some());
 240     assert_eq!(
 241       inserted_admin.name,
 242       reports[0].resolver.as_ref().unwrap().name
 243     );
 244
 245     Instance::delete(pool, inserted_instance.id).await.unwrap();
 246   }
 247 }