Reports (#165)

* Reports WIP * Finish reports * Clippy fixes Co-authored-by: Geometrically <geometrically@pop-os.localdomain>
2021-02-13 12:11:13 -07:00
parent 3df740702c
commit 109d7d87bd
21 changed files with 1158 additions and 238 deletions
--- a/src/database/models/report_item.rs
+++ b/src/database/models/report_item.rs
@@ -0,0 +1,153 @@
+use super::ids::*;
+
+pub struct Report {
+    pub id: ReportId,
+    pub report_type_id: ReportTypeId,
+    pub mod_id: Option<ModId>,
+    pub version_id: Option<VersionId>,
+    pub user_id: Option<UserId>,
+    pub body: String,
+    pub reporter: UserId,
+    pub created: chrono::DateTime<chrono::Utc>,
+}
+
+pub struct QueryReport {
+    pub id: ReportId,
+    pub report_type: String,
+    pub mod_id: Option<ModId>,
+    pub version_id: Option<VersionId>,
+    pub user_id: Option<UserId>,
+    pub body: String,
+    pub reporter: UserId,
+    pub created: chrono::DateTime<chrono::Utc>,
+}
+
+impl Report {
+    pub async fn insert(
+        &self,
+        transaction: &mut sqlx::Transaction<'_, sqlx::Postgres>,
+    ) -> Result<(), sqlx::error::Error> {
+        sqlx::query!(
+            "
+            INSERT INTO reports (
+                id, report_type_id, mod_id, version_id, user_id,
+                body, reporter
+            )
+            VALUES (
+                $1, $2, $3, $4, $5,
+                $6, $7
+            )
+            ",
+            self.id as ReportId,
+            self.report_type_id as ReportTypeId,
+            self.mod_id.map(|x| x.0 as i64),
+            self.version_id.map(|x| x.0 as i64),
+            self.user_id.map(|x| x.0 as i64),
+            self.body,
+            self.reporter as UserId
+        )
+        .execute(&mut *transaction)
+        .await?;
+
+        Ok(())
+    }
+
+    pub async fn get<'a, E>(id: ReportId, exec: E) -> Result<Option<QueryReport>, sqlx::Error>
+    where
+        E: sqlx::Executor<'a, Database = sqlx::Postgres> + Copy,
+    {
+        let result = sqlx::query!(
+            "
+            SELECT rt.name, r.mod_id, r.version_id, r.user_id, r.body, r.reporter, r.created
+            FROM reports r
+            INNER JOIN report_types rt ON rt.id = r.report_type_id
+            WHERE r.id = $1
+            ",
+            id as ReportId,
+        )
+        .fetch_optional(exec)
+        .await?;
+
+        if let Some(row) = result {
+            Ok(Some(QueryReport {
+                id,
+                report_type: row.name,
+                mod_id: row.mod_id.map(ModId),
+                version_id: row.version_id.map(VersionId),
+                user_id: row.user_id.map(UserId),
+                body: row.body,
+                reporter: UserId(row.reporter),
+                created: row.created,
+            }))
+        } else {
+            Ok(None)
+        }
+    }
+
+    pub async fn get_many<'a, E>(
+        version_ids: Vec<ReportId>,
+        exec: E,
+    ) -> Result<Vec<QueryReport>, sqlx::Error>
+    where
+        E: sqlx::Executor<'a, Database = sqlx::Postgres> + Copy,
+    {
+        use futures::stream::TryStreamExt;
+
+        let version_ids_parsed: Vec<i64> = version_ids.into_iter().map(|x| x.0).collect();
+        let versions = sqlx::query!(
+            "
+            SELECT r.id, rt.name, r.mod_id, r.version_id, r.user_id, r.body, r.reporter, r.created
+            FROM reports r
+            INNER JOIN report_types rt ON rt.id = r.report_type_id
+            WHERE r.id IN (SELECT * FROM UNNEST($1::bigint[]))
+            ",
+            &version_ids_parsed
+        )
+        .fetch_many(exec)
+        .try_filter_map(|e| async {
+            Ok(e.right().map(|row| QueryReport {
+                id: ReportId(row.id),
+                report_type: row.name,
+                mod_id: row.mod_id.map(ModId),
+                version_id: row.version_id.map(VersionId),
+                user_id: row.user_id.map(UserId),
+                body: row.body,
+                reporter: UserId(row.reporter),
+                created: row.created,
+            }))
+        })
+        .try_collect::<Vec<QueryReport>>()
+        .await?;
+
+        Ok(versions)
+    }
+
+    pub async fn remove_full<'a, E>(id: ReportId, exec: E) -> Result<Option<()>, sqlx::Error>
+    where
+        E: sqlx::Executor<'a, Database = sqlx::Postgres> + Copy,
+    {
+        let result = sqlx::query!(
+            "
+            SELECT EXISTS(SELECT 1 FROM reports WHERE id = $1)
+            ",
+            id as ReportId
+        )
+        .fetch_one(exec)
+        .await?;
+
+        if !result.exists.unwrap_or(false) {
+            return Ok(None);
+        }
+
+        sqlx::query!(
+            "
+            DELETE FROM reports WHERE id = $1
+            ",
+            id as ReportId,
+        )
+        .execute(exec)
+        .await?;
+
+        Ok(Some(()))
+    }
+}