From fb0b3ff9bc39de6c6dfeefd09bdf5be311fb1118 Mon Sep 17 00:00:00 2001
From: ansengarvin <45224464+ansengarvin@users.noreply.github.com>
Date: Sun, 25 Feb 2024 19:36:04 -0800
Subject: [PATCH] Added ability to set and clear token headers from front end

---
 frontend/src/lib/backend.ts       | 13 +++++++++++++
 frontend/src/routes/Edit.svelte   |  4 +++-
 frontend/src/routes/Login.svelte  |  3 ++-
 frontend/src/routes/Upload.svelte |  3 ++-
 4 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/frontend/src/lib/backend.ts b/frontend/src/lib/backend.ts
index a9fc598d..ec2f25bd 100644
--- a/frontend/src/lib/backend.ts
+++ b/frontend/src/lib/backend.ts
@@ -1,6 +1,19 @@
 export * from "./openapi";
 export { DefaultService as Backend } from "./openapi";
 import { OpenAPI } from "./openapi";
+import Cookies from "js-cookie";
 
 export const BACKEND_URL = "http://localhost:8000";
 OpenAPI.BASE = BACKEND_URL;
+
+// Sets the token header to the stored authentication token from the cookie
+export function setToken() {
+	console.log("Setting Token")
+	OpenAPI.TOKEN = Cookies.get("auth")
+}
+
+// Sets the token header to nothing.
+export function clearToken() {
+	console.log("Clearing token.")
+	OpenAPI.TOKEN = undefined
+}
diff --git a/frontend/src/routes/Edit.svelte b/frontend/src/routes/Edit.svelte
index 0a23cfad..0c22fee9 100644
--- a/frontend/src/routes/Edit.svelte
+++ b/frontend/src/routes/Edit.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { Backend, UploadError, type ProteinEntry } from "../lib/backend";
+	import { Backend, UploadError, setToken, type ProteinEntry } from "../lib/backend";
 	import { Button, Input, Label, Helper, Select } from "flowbite-svelte";
 	import { navigate } from "svelte-routing";
 	import { onMount } from "svelte";
@@ -140,6 +140,7 @@
 					on:click={async () => {
 						if (entry) {
 							try {
+								setToken()
 								const err = await Backend.editProteinEntry({
 									newName: name,
 									oldName: entry.name,
@@ -179,6 +180,7 @@
 				<Button
 					color="red"
 					on:click={async () => {
+						setToken()
 						await Backend.deleteProteinEntry(urlId);
 						navigate("/");
 					}}>Delete Protein Entry</Button
diff --git a/frontend/src/routes/Login.svelte b/frontend/src/routes/Login.svelte
index 88d85fa2..3428c901 100644
--- a/frontend/src/routes/Login.svelte
+++ b/frontend/src/routes/Login.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { Backend, type LoginResponse } from "../lib/backend";
+	import { Backend, clearToken, type LoginResponse } from "../lib/backend";
 	import { Button, Label, Input } from "flowbite-svelte";
 	import Cookies from "js-cookie";
 	import { onMount } from "svelte";
@@ -11,6 +11,7 @@
 	 * We want to do this to avoid bugs, and to let the user log out.
 	 */
 	 onMount(async () => {
+		clearToken()
 		Cookies.remove("auth")
 		$user.loggedIn = false
 		$user.username = ""
diff --git a/frontend/src/routes/Upload.svelte b/frontend/src/routes/Upload.svelte
index 4693152c..0bf08cb5 100644
--- a/frontend/src/routes/Upload.svelte
+++ b/frontend/src/routes/Upload.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { Backend, UploadError } from "../lib/backend";
+	import { Backend, UploadError, setToken } from "../lib/backend";
 	import {
 		Fileupload,
 		Button,
@@ -108,6 +108,7 @@
 
 					const pdbFileStr = await fileToString(file);
 					try {
+						setToken()
 						const err = await Backend.uploadProteinEntry({
 							name,
 							description,