From a52eae2964e8e86bf3e587f17d607025f07efdbb Mon Sep 17 00:00:00 2001
From: Nathan Perry <avaglir@gmail.com>
Date: Thu, 3 May 2018 22:08:54 -0400
Subject: support message deletion

---
 src/db/mod.rs    | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 src/db/models.rs | 19 +++++++++++++++++++
 src/db/schema.rs | 12 ++++++++++++
 3 files changed, 81 insertions(+)

(limited to 'src/db')

diff --git a/src/db/mod.rs b/src/db/mod.rs
index 8657a4f..0bf2f1d 100644
--- a/src/db/mod.rs
+++ b/src/db/mod.rs
@@ -35,6 +35,56 @@ pub fn find_meme<T: AsRef<str>>(conn: &PgConnection, search: T) -> Result<Meme>
         .map_err(Error::from)
 }
 
+pub fn delete_meme<T: AsRef<str>>(conn: &PgConnection, search: T, deleted_by: u64) -> Result<()> {
+    conn.transaction::<(), Error, _>(|| {
+        let deleted = memes::table
+            .filter(memes::title.eq(search.as_ref()))
+            .first::<Meme>(conn)?;
+
+        ::diesel::delete(memes::table)
+            .filter(memes::id.eq(deleted.id))
+            .execute(conn)?;
+
+        if let Some(image_id) = deleted.image_id {
+            let count = memes::table
+                .filter(memes::image_id.eq(image_id))
+                .count()
+                .execute(conn)?;
+
+            if count == 0 {
+                ::diesel::delete(images::table)
+                    .filter(images::id.eq(image_id))
+                    .execute(conn)?;
+            }
+        }
+
+        if let Some(audio_id) = deleted.audio_id {
+            let count = memes::table
+                .select(::diesel::dsl::count_star())
+                .filter(memes::audio_id.eq(audio_id))
+                .execute(conn)?;
+
+            if count == 0 {
+                ::diesel::delete(audio::table)
+                    .filter(audio::id.eq(audio_id))
+                    .execute(conn)?;
+            }
+        }
+
+        let tombstone = NewTombstone {
+            deleted_by: deleted_by as i64,
+            metadata_id: deleted.metadata_id,
+            meme_id: deleted.id,
+        };
+
+        let _ = ::diesel::insert_into(tombstones::table)
+            .values(&tombstone)
+            .execute(conn)?;
+
+        Ok(())
+    })
+}
+
 pub fn rand_text(conn: &PgConnection) -> Result<Meme> {
     memes::table
         .filter(memes::content.is_not_null())
diff --git a/src/db/models.rs b/src/db/models.rs
index 6ac8b18..d25b593 100644
--- a/src/db/models.rs
+++ b/src/db/models.rs
@@ -203,3 +203,22 @@ pub struct NewAuditRecord {
     pub updated_by: i64,
     pub metadata_id: i32,
 }
+
+#[derive(Queryable, Identifiable, PartialEq, Debug)]
+#[table_name="tombstones"]
+pub struct Tombstone {
+    pub id: i32,
+    pub deleted: NaiveDateTime,
+    pub deleted_by: i64,
+    pub metadata_id: i32,
+    pub meme_id: i32,
+}
+
+
+#[derive(Insertable, PartialEq, Debug)]
+#[table_name="tombstones"]
+pub struct NewTombstone {
+    pub deleted_by: i64,
+    pub metadata_id: i32,
+    pub meme_id: i32,
+}
diff --git a/src/db/schema.rs b/src/db/schema.rs
index 1822204..8749731 100644
--- a/src/db/schema.rs
+++ b/src/db/schema.rs
@@ -45,12 +45,23 @@ table! {
     }
 }
 
+table! {
+    tombstones (id) {
+        id -> Int4,
+        meme_id -> Int4,
+        deleted_by -> Int8,
+        deleted_at -> Timestamp,
+        metadata_id -> Nullable<Int4>,
+    }
+}
+
 joinable!(audio -> metadata (metadata_id));
 joinable!(audit_records -> metadata (metadata_id));
 joinable!(images -> metadata (metadata_id));
 joinable!(memes -> audio (audio_id));
 joinable!(memes -> images (image_id));
 joinable!(memes -> metadata (metadata_id));
+joinable!(tombstones -> metadata (metadata_id));
 
 allow_tables_to_appear_in_same_query!(
     audio,
@@ -58,4 +69,5 @@ allow_tables_to_appear_in_same_query!(
     images,
     memes,
     metadata,
+    tombstones,
 );
-- 
cgit v1.3.1