3 files changed, 57 insertions, 15 deletions

diff --git a/src/util/files.ts b/src/util/files.ts
index 49c5ebb..a1ce5d9 100644
--- a/src/util/files.ts
+++ b/src/util/files.ts
@@ -1,27 +1,25 @@
 import { walk, ensureDirSync, existsSync } from "https://deno.land/std/fs/mod.ts";
+import type { Context } from "https://deno.land/x/abc@master/mod.ts";
+import { getUserCookies } from "./user.ts";
 
-const TEMP_USER_ID = 42; // TODO: FIX
-
-export const cleanPath = (path: string): string => {
-    createUserDirectory(TEMP_USER_ID);
-
+export const cleanPath = (path: string, uid: number): string => {
     return (
         "data/" +
-        TEMP_USER_ID +
+        uid +
         "/" +
         path
-            .replace("/files/", "")
+            .replace(/\/files\/?/, "")
             .replace("../", "") // TODO: Fix relative ../
             .replace("./", "")
             .replace(/([^:]\/)\/+/g, "$1")
     );
-};
-
-export const getFiles = async (path: string) => {
-    const newPath = path ? path : "";
+}
 
-    createUserDirectory(TEMP_USER_ID);
-    const dataPath: string = cleanPath(newPath);
+export const getFiles = async (c: Context) => {
+    const path = c.path ? c.path : "";
+    const uid = getUserCookies(c).uid;
+    createUserDirectory(uid);  // TODO: Consider doing this in db/user/createUser => performance?
+    const dataPath: string = cleanPath(path, uid);
 
     if (!existsSync(dataPath)) return [];
 
@@ -30,9 +28,9 @@ export const getFiles = async (path: string) => {
         files.push(entry.path);
     }
     return files;
-};
+}
 
 export const createUserDirectory = (uid: number) => {
     ensureDirSync("data/" + uid);
     // TODO: Give user access to dir
-};
+}
diff --git a/src/util/server.ts b/src/util/server.ts
new file mode 100644
index 0000000..665c14f
--- /dev/null
+++ b/src/util/server.ts
@@ -0,0 +1,14 @@
+import DBController from "../db/DBController.ts";
+import * as log from "https://deno.land/std/log/mod.ts";
+
+const controller = new DBController();
+
+export const isSetup = async (): Promise<boolean> => {
+    try {
+        const users = await controller.query("SELECT id FROM users");
+        return users.length > 0
+    } catch (e) {
+        log.error(e);
+        return false;
+    }
+}
diff --git a/src/util/user.ts b/src/util/user.ts
new file mode 100644
index 0000000..f632a11
--- /dev/null
+++ b/src/util/user.ts
@@ -0,0 +1,30 @@
+import type { userData } from "../db/user.ts";
+import db from "../db/user.ts";
+import * as log from "https://deno.land/std/log/mod.ts";
+import type { Context } from "https://deno.land/x/abc@master/mod.ts";
+
+export const getCurrentUser = async (c: Context): Promise<userData | undefined> => {
+    const cookies = getUserCookies(c);
+    try {
+        return await db.getUserByVerificationId(cookies.uid, cookies.verification) as userData;
+    } catch (e) {
+        log.error(e);
+        return undefined;
+    }
+}
+
+export const getUserCookies = (c: Context): userCookies => {
+    const uid = parseInt(c.cookies["uid"]);
+    const verification = c.cookies["verification"];
+    return { uid, verification };
+}
+
+export const isAdmin = async (c: Context): Promise<boolean> => {
+    const user = await getCurrentUser(c);
+    return (user && user.isAdmin) as boolean;
+}
+
+export interface userCookies {
+    uid: number;
+    verification: string;
+}