crates/db_schema/src/impls/private_message_report.rs

   1 use crate::{
   2   newtypes::{PersonId, PrivateMessageReportId},
   3   schema::private_message_report::dsl::{private_message_report, resolved, resolver_id, updated},
   4   source::private_message_report::{PrivateMessageReport, PrivateMessageReportForm},
   5   traits::Reportable,
   6   utils::{get_conn, naive_now, DbPool},
   7 };
   8 use diesel::{
   9   dsl::{insert_into, update},
  10   result::Error,
  11   ExpressionMethods,
  12   QueryDsl,
  13 };
  14 use diesel_async::RunQueryDsl;
  15
  16 #[async_trait]
  17 impl Reportable for PrivateMessageReport {
  18   type Form = PrivateMessageReportForm;
  19   type IdType = PrivateMessageReportId;
  20
  21   async fn report(pool: &DbPool, pm_report_form: &PrivateMessageReportForm) -> Result<Self, Error> {
  22     let conn = &mut get_conn(pool).await?;
  23     insert_into(private_message_report)
  24       .values(pm_report_form)
  25       .get_result::<Self>(conn)
  26       .await
  27   }
  28
  29   async fn resolve(
  30     pool: &DbPool,
  31     report_id: Self::IdType,
  32     by_resolver_id: PersonId,
  33   ) -> Result<usize, Error> {
  34     let conn = &mut get_conn(pool).await?;
  35     update(private_message_report.find(report_id))
  36       .set((
  37         resolved.eq(true),
  38         resolver_id.eq(by_resolver_id),
  39         updated.eq(naive_now()),
  40       ))
  41       .execute(conn)
  42       .await
  43   }
  44
  45   async fn unresolve(
  46     pool: &DbPool,
  47     report_id: Self::IdType,
  48     by_resolver_id: PersonId,
  49   ) -> Result<usize, Error> {
  50     let conn = &mut get_conn(pool).await?;
  51     update(private_message_report.find(report_id))
  52       .set((
  53         resolved.eq(false),
  54         resolver_id.eq(by_resolver_id),
  55         updated.eq(naive_now()),
  56       ))
  57       .execute(conn)
  58       .await
  59   }
  60 }