From ea9ae6120ff4ad967a612b24560c3f6b4ffbbaea Mon Sep 17 00:00:00 2001
From: Michael Fatemi <myfatemi04@gmail.com>
Date: Sat, 10 Apr 2021 16:47:19 -0400
Subject: [PATCH] add access token middleware

---
 src/api.ts   | 1 +
 src/auth.ts  | 7 ++++++-
 src/index.ts | 2 ++
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/api.ts b/src/api.ts
index fa6925a..fa7cdeb 100644
--- a/src/api.ts
+++ b/src/api.ts
@@ -18,6 +18,7 @@ router.get('/user', (req, res) => {
 	}
 
 	let userID = req.query.userID;
+	console.log(req.session, userID);
 	if (userID === '@me') {
 		userID = req.session.accountID;
 	}
diff --git a/src/auth.ts b/src/auth.ts
index c3a1c4b..82709d0 100644
--- a/src/auth.ts
+++ b/src/auth.ts
@@ -26,5 +26,10 @@ export function createSession(accountID: string): string {
 }
 
 export function getSessionByToken(token: string): SessionData {
-	return sessions[token];
+	console.log(sessions);
+	if (token in sessions) {
+		return sessions[token];
+	} else {
+		return { accountID: null };
+	}
 }
diff --git a/src/index.ts b/src/index.ts
index cd9382e..52a52fc 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -7,11 +7,13 @@ import * as bodyParser from 'body-parser';
 import cors from 'cors';
 import express from 'express';
 import * as api from './api';
+import { sessionMiddleware } from './sessionMiddleware';
 
 const app = express();
 
 app.use(cors());
 app.use(bodyParser.json());
+app.use(sessionMiddleware);
 
 app.use('/api', api.router);