From 876437db1f7347787badbdd48ac751dba111a752 Mon Sep 17 00:00:00 2001
From: Melissa Zhang <mzhang19096@gmail.com>
Date: Wed, 8 Jul 2020 08:57:21 -0700
Subject: store hypothesis API tokens to database

---
 src/client/util/CurrentUserUtils.ts         |  2 +-
 src/server/ApiManagers/HypothesisManager.ts | 14 ++++-------
 src/server/database.ts                      | 39 ++++++++++++++++++++++++++++-
 3 files changed, 44 insertions(+), 11 deletions(-)

(limited to 'src')

diff --git a/src/client/util/CurrentUserUtils.ts b/src/client/util/CurrentUserUtils.ts
index 8099228c6..cb2a025cb 100644
--- a/src/client/util/CurrentUserUtils.ts
+++ b/src/client/util/CurrentUserUtils.ts
@@ -424,7 +424,7 @@ export class CurrentUserUtils {
             { title: "Drag a document previewer", label: "Prev", icon: "expand", click: 'openOnRight(getCopy(this.dragFactory, true))', drag: 'getCopy(this.dragFactory,true)', dragFactory: doc.emptyDocHolder as Doc },
             { title: "Toggle a Calculator REPL", label: "repl", icon: "calculator", click: 'addOverlayWindow("ScriptingRepl", { x: 300, y: 100, width: 200, height: 200, title: "Scripting REPL" })' },
             { title: "Connect a Google Account", label: "Google Account", icon: "external-link-alt", click: 'GoogleAuthenticationManager.Instance.fetchOrGenerateAccessToken(true)' },
-            { title: "Connect a Hypothesis Account", label: "Hypothesis Account", icon: "heading", click: 'HypothesisAuthenticationManager.Instance.fetchOrGenerateAccessToken(true)' },
+            { title: "Connect a Hypothesis Account", label: "Hypothesis Account", icon: "heading", click: 'HypothesisAuthenticationManager.Instance.fetchAccessToken(true)' },
         ];
 
     }
diff --git a/src/server/ApiManagers/HypothesisManager.ts b/src/server/ApiManagers/HypothesisManager.ts
index 33badbc42..73c707a55 100644
--- a/src/server/ApiManagers/HypothesisManager.ts
+++ b/src/server/ApiManagers/HypothesisManager.ts
@@ -13,11 +13,8 @@ export default class HypothesisManager extends ApiManager {
             method: Method.GET,
             subscription: "/readHypothesisAccessToken",
             secureHandler: async ({ user, res }) => {
-                if (existsSync(serverPathToFile(Directory.hypothesis, user.id))) {
-                    const read = readFileSync(serverPathToFile(Directory.hypothesis, user.id), "base64") || "";
-                    console.log("READ = " + read);
-                    res.send(read);
-                } else res.send("");
+                const credentials = await Database.Auxiliary.HypothesisAccessToken.Fetch(user.id);
+                res.send(credentials?.hypothesisApiKey ?? "");
             }
         });
 
@@ -25,9 +22,8 @@ export default class HypothesisManager extends ApiManager {
             method: Method.POST,
             subscription: "/writeHypothesisAccessToken",
             secureHandler: async ({ user, req, res }) => {
-                const write = req.body.authenticationCode;
-                console.log("WRITE = " + write);
-                res.send(await writeFile(serverPathToFile(Directory.hypothesis, user.id), write, "base64", () => { }));
+                await Database.Auxiliary.HypothesisAccessToken.Write(user.id, req.body.authenticationCode);
+                res.send();
             }
         });
 
@@ -35,7 +31,7 @@ export default class HypothesisManager extends ApiManager {
             method: Method.GET,
             subscription: "/revokeHypothesisAccessToken",
             secureHandler: async ({ user, res }) => {
-                await Database.Auxiliary.GoogleAccessToken.Revoke("dash-hyp-" + user.id);
+                await Database.Auxiliary.HypothesisAccessToken.Revoke(user.id);
                 res.send();
             }
         });
diff --git a/src/server/database.ts b/src/server/database.ts
index 2372cbcf2..7fbab357b 100644
--- a/src/server/database.ts
+++ b/src/server/database.ts
@@ -304,7 +304,8 @@ export namespace Database {
          */
         export enum AuxiliaryCollections {
             GooglePhotosUploadHistory = "uploadedFromGooglePhotos",
-            GoogleAccess = "googleAuthentication"
+            GoogleAccess = "googleAuthentication",
+            HypothesisAccess = "hypothesisAuthentication"
         }
 
         /**
@@ -405,6 +406,42 @@ export namespace Database {
 
         }
 
+        export namespace HypothesisAccessToken {
+            /**
+             * Format stored in database.
+             */
+            interface StoredCredentials {
+                userId: string;
+                hypothesisApiKey: string;
+                _id?: string;
+            }
+
+            /**
+             * Writes the @param enrichedCredentials to the database, associated
+             * with @param userId for later retrieval and updating. 
+             */
+            export const Write = async (userId: string, hypothesisApiKey: string) => {
+                return Instance.insert({ userId, hypothesisApiKey }, AuxiliaryCollections.HypothesisAccess);
+            };
+
+            /**
+             * Retrieves the credentials associaed with @param userId
+             * and optionally removes their database id according to @param removeId. 
+             */
+            export const Fetch = async (userId: string, removeId = true): Promise<Opt<StoredCredentials>> => {
+                return SanitizedSingletonQuery<StoredCredentials>({ userId }, AuxiliaryCollections.HypothesisAccess, removeId);
+            };
+
+            /**
+            * Revokes the credentials associated with @param userId. 
+            */
+            export const Revoke = async (userId: string) => {
+                const entry = await Fetch(userId, false);
+                if (entry) {
+                    Instance.delete({ _id: entry._id }, AuxiliaryCollections.HypothesisAccess);
+                }
+            };
+        }
     }
 
 }
-- 
cgit v1.2.3-70-g09d2