From 275ac5d9e81f3d0f7ffbd6fcccf346be478f563b Mon Sep 17 00:00:00 2001
From: Ben Kurtovic <ben.kurtovic@verizon.net>
Date: Fri, 31 Aug 2012 01:30:46 -0400
Subject: [PATCH] Implement do_delete().

---
 earwigbot/commands/notes.py | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/earwigbot/commands/notes.py b/earwigbot/commands/notes.py
index dfaf1f1..8674515 100644
--- a/earwigbot/commands/notes.py
+++ b/earwigbot/commands/notes.py
@@ -189,10 +189,10 @@ class Notes(Command):
         try:
             newtitle = data.args[2]
         except IndexError:
-            self.reply(data, "Please specify an entry to rename.")
+            self.reply(data, "Please specify a new name for the entry.")
             return
         if newtitle == data.args[1]:
-            self.reply(data, "The old and new titles are identical.")
+            self.reply(data, "The old and new names are identical.")
             return
 
         with sqlite.connect(self._dbfile) as conn, self._db_access_lock:
@@ -207,17 +207,40 @@ class Notes(Command):
                 self.reply(data, msg)
                 return
             conn.execute(query2, (newtitle, id_))
+
         msg = "Entry \x0302{0}\x0F renamed to \x0302{1}\x0F."
         self.reply(data, msg.format(data.args[1], newtitle))
 
     def do_delete(self, data):
         """Delete an entry from the notes database."""
+        query1 = """SELECT entry_id, user_host FROM entries
+                    INNER JOIN revisions ON entry_revision = rev_id
+                    INNER JOIN users ON rev_user = user_id
+                    WHERE entry_slug = ?"""
+        query2 = "DELETE FROM entries WHERE entry_id = ?"
+        query3 = "DELETE FROM revisions WHERE rev_entry = ?"
         try:
             slug = data.args[1].lower().replace("_", "").replace("-", "")
         except IndexError:
             self.reply(data, "Please specify an entry to delete.")
             return
 
+        with sqlite.connect(self._dbfile) as conn, self._db_access_lock:
+            try:
+                id_, author = conn.execute(query1, (slug,)).fetchone()
+            except (sqlite.OperationalError, TypeError):
+                msg = "Entry \x0302{0}\x0F not found.".format(data.args[1])
+                self.reply(data, msg)
+                return
+            if author != data.host and not permdb.is_admin(data):
+                msg = "You must be an author or a bot admin to delete this entry."
+                self.reply(data, msg)
+                return
+            conn.execute(query2, (id_))
+            conn.execute(query3, (id_))
+
+        self.reply(data, "Entry \x0302{0}\x0F deleted.".format(data.args[1]))
+
     def create_db(self, conn):
         """Initialize the notes database with its necessary tables."""
         script = """