From 60a839a328b3d20aaf29920649cc6219c0a1626f Mon Sep 17 00:00:00 2001
From: David Todd <c0de@c0defox.es>
Date: Sat, 18 Jan 2020 22:07:26 -0600
Subject: [PATCH] Add ability to update bookmarks

---
 api/api.py | 58 ++++++++++++++++++++++++++++++++++++++++++++----------
 server.py  |  8 ++++++++
 2 files changed, 56 insertions(+), 10 deletions(-)

diff --git a/api/api.py b/api/api.py
index 9d2a708..e3969c3 100644
--- a/api/api.py
+++ b/api/api.py
@@ -31,6 +31,19 @@ class API:
         self.response_type = 'application/json'
         self.format_date = datetime.datetime.strftime
 
+    def _get_bookmark_object(self, bookmark):
+        """
+            Shortcut to return the `bookmark_object` that several methods return
+        """
+        # Maybe I need to use `sqlite3.Cursor.description` or `sqlite3.Row.keys()`
+        return {
+            'uuid': bookmark[0].hex,
+            'uri': bookmark[1],
+            'title': bookmark[2],
+            'date_created': self.format_date(bookmark[3], self.dt_fmt),
+            'date_updated': self.format_date(bookmark[4], self.dt_fmt) if bookmark[4] != None else '',
+        }
+
     def save_bookmark(self, title, uri):
         """
             Save the bookmark with `title` and `uri`
@@ -59,16 +72,7 @@ class API:
         if bookmark == None:
             return abort(404, "Provided bookmark doesn't exist or has been deleted")
 
-        # Maybe I need to use `sqlite3.Cursor.description` or `sqlite3.Row.keys()`
-        bookmark_object = {
-            'uuid': bookmark[0].hex,
-            'uri': bookmark[1],
-            'title': bookmark[2],
-            'date_created': self.format_date(bookmark[3], self.dt_fmt),
-            'date_updated': self.format_date(bookmark[4], self.dt_fmt) if bookmark[4] != None else '',
-        }
-
-        return json.dumps(bookmark_object)
+        return json.dumps(self._get_bookmark_object(bookmark))
 
     def get_all_bookmarks(self):
         """
@@ -107,3 +111,37 @@ class API:
             'uuid': bookmark_id,
             'bookmark_deleted': self.database.delete_bookmark(UUID(bookmark_id)),
         })
+
+    def update_bookmark_title(self, bookmark_id, title):
+        """
+            Updates the bookmark that is associated
+            with `bookmark_id` with the provided `title`
+
+            Returns a JSON object containing
+            everything about the bookmark
+        """
+        response.content_type = self.response_type
+        bookmark = self.database.update_bookmark_title(bookmark_id, title)
+
+        if bookmark == None:
+            return abort(404, "Provided bookmark doesn't exist or has been deleted")
+
+        return json.dumps(self._get_bookmark_object(bookmark))
+
+    def update_bookmark_uri(self, bookmark_id, uri):
+        """
+            Updates the bookmark that is associated
+            with `bookmark_id` with the provided `uri`
+
+            This action will change the `bookmark_id`
+
+            Returns a JSON object containing
+            everything about the bookmark
+        """
+        response.content_type = self.response_type
+        bookmark = self.database.update_bookmark_uri(bookmark_id, uri)
+
+        if bookmark == None:
+            return abort(404, "Provided bookmark doesn't exist or has been deleted")
+
+        return json.dumps(self._get_bookmark_object(bookmark))
diff --git a/server.py b/server.py
index e49c12f..711c286 100644
--- a/server.py
+++ b/server.py
@@ -33,5 +33,13 @@ def get_bookmark(bookmark_id):
 def delete_bookmark(bookmark_id):
     return _api.delete_bookmark(bookmark_id)
 
+@route('/update/title/<bookmark_id>/<title>')
+def update_bookmark_title(bookmark_id, title):
+    return _api.update_bookmark_title(bookmark_id, title)
+
+@route('/update/uri/<bookmark_id>/<uri:path>')
+def update_bookmark_uri(bookmark_id, uri):
+    return _api.update_bookmark_uri(bookmark_id, uri)
+
 if __name__ == '__main__':
     run(host='localhost', port=8080)