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/commands/meme.rs | 11 ++++++++---
 src/db/mod.rs        | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 src/db/models.rs     | 19 +++++++++++++++++++
 src/db/schema.rs     | 12 ++++++++++++
 src/main.rs          |  8 ++++++++
 5 files changed, 97 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/commands/meme.rs b/src/commands/meme.rs
index 6c78fc5..d1c90d1 100644
--- a/src/commands/meme.rs
+++ b/src/commands/meme.rs
@@ -61,11 +61,16 @@ pub fn addmeme(_: &mut Context, msg: &Message, mut args: Args) -> Result<()> {
         metadata_id: 0,
     }.save(&conn, msg.author.id.0).map(|_| {})?;
 
-    return msg.react("👌");
+    msg.react("👌")
 }
 
-pub fn delmeme(_: &mut Context, msg: &Message, _: Args) -> Result<()> {
-    send(msg.channel_id, "hwaet", msg.tts)
+pub fn delmeme(_: &mut Context, msg: &Message, mut args: Args) -> Result<()> {
+    let title = args.single_quoted::<String>()?;
+
+    let conn = connection()?;
+    delete_meme(&conn, &title, msg.author.id.0)?;
+
+    msg.react("💀")
 }
 
 pub fn renamememe(_: &mut Context, msg: &Message, _: Args) -> Result<()> {
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,
 );
diff --git a/src/main.rs b/src/main.rs
index a2a4e6d..ff5ca24 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -59,6 +59,14 @@ impl EventHandler for Handler {
         if guild.is_none() {
             info!("bot isn't in configured guild. join here: {:?}", OAUTH_URL.as_str());
         }
+
+        #[cfg(debug_assertions)] {
+            let _ = guild.map(|g| g.id().edit_nickname(Some("thulani (dev)")));
+        }
+
+        #[cfg(not(debug_assertions))] {
+            let _ = guild.map(|g| g.id().edit_nickname(Some("thulani")));
+        }
     }
 }
 
-- 
cgit v1.3.1