From 163d8b71c4e445061a8eb68c7fffb9e089bc8858 Mon Sep 17 00:00:00 2001
From: David Todd <c0de@c0defox.es>
Date: Sat, 18 Jan 2020 21:38:52 -0600
Subject: [PATCH] Add simple API

---
 api/__init__.py |  9 +++++
 api/api.py      | 95 +++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 104 insertions(+)
 create mode 100644 api/__init__.py
 create mode 100644 api/api.py

diff --git a/api/__init__.py b/api/__init__.py
new file mode 100644
index 0000000..c24270d
--- /dev/null
+++ b/api/__init__.py
@@ -0,0 +1,9 @@
+#!/usr/bin/env python3
+
+"""
+    Creates a python module for the API
+
+    Usage: from api import API
+"""
+
+from .api import API
diff --git a/api/api.py b/api/api.py
new file mode 100644
index 0000000..1f5cefd
--- /dev/null
+++ b/api/api.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+# Copyright 2020 - David Todd (c0de@c0defox.es)
+# Licensed under the MIT License (https://opensource.org/licenses/MIT)
+
+"""
+    This file contains the API for a simple bookmarking application
+"""
+
+import json
+import datetime
+from pprint import pprint
+from uuid import UUID
+from bottle import response
+from .db import Database
+
+class API:
+    """
+        This class contains various methods for a simple bookmarking application API
+
+        As of right now, these are mostly just to pass data through to the database.
+        Stuff should be sanitized here though
+    """
+
+    def __init__(self):
+        """
+            Setup some internal variables, such as various constants
+            and the database connection
+        """
+        self.database = Database('bookmarks.db')
+        self.dt_fmt = "%H:%M:%S on %B %d %Y"
+        self.response_type = 'application/json'
+        self.format_date = datetime.datetime.strftime
+
+    def save_bookmark(self, title, uri):
+        """
+            Save the bookmark with `title` and `uri`
+
+            Returns a JSON object containing
+            the UUID of the saved bookmark
+        """
+        response.content_type = self.response_type
+        bookmark = self.database.save_bookmark(uri, title)
+
+        return json.dumps({
+            'uuid': bookmark.hex
+        })
+
+    def get_bookmark(self, bookmark_id):
+        """
+            Get the bookmark with the provided `bookmark_id`
+
+            `bookmark_id` is of type string
+            Returns a JSON object containing
+            everything about the bookmark
+        """
+        response.content_type = self.response_type
+        bookmark = self.database.get_bookmark(UUID(bookmark_id))
+
+        # 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)
+
+    def get_all_bookmarks(self):
+        """
+            Gets all of the bookmarks that are currently saved
+
+            Returns a JSON object containing
+            everything about each bookmark
+        """
+        response.content_type = self.response_type
+        bookmarks = self.database.get_all_bookmarks()
+
+        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 ''
+            } for bookmark in bookmarks
+        ]
+
+        return json.dumps(bookmark_object)
+
+    def delete_bookmark(self, bookmark_id):
+        """
+        """
+        response.content_type = self.response_type