3 files changed, 94 insertions, 20 deletions
diff --git a/src/handler/admin.ts b/src/handler/admin.ts
new file mode 100644
index 0000000..a626e50
--- /dev/null
+++ b/src/handler/admin.ts
@@ -0,0 +1,12 @@
+import type { HandlerFunc, Context } from "https://deno.land/x/abc@master/mod.ts";
+import * as log from "https://deno.land/std/log/mod.ts";
+import { isAdmin } from "../util/user.ts";
+import { isSetup } from "../util/server.ts";
+
+export const render: HandlerFunc = async (c: Context) => {
+    if (await isAdmin(c))
+        return await c.render("./src/views/admin.ejs", { initial: false });
+    else if (!(await isSetup()))
+        return await c.render("./src/views/admin.ejs", { initial: true });
+    return c.redirect("/");
+}
diff --git a/src/handler/fileView.ts b/src/handler/fileView.ts
index eb9c3a1..29bc7a9 100644
--- a/src/handler/fileView.ts
+++ b/src/handler/fileView.ts
@@ -1,6 +1,8 @@
 import type { HandlerFunc, Context } from "https://deno.land/x/abc@master/mod.ts";
 import { getFiles } from "../util/files.ts";
+import { getCurrentUser } from "../util/user.ts";
 
 export const handlePath: HandlerFunc = async (c: Context) => {
-    return await c.render("./src/views/index.html", { files: await getFiles(c.path) });
-};
+    if (!(await getCurrentUser(c))) return c.redirect("/user/login");  // TODO: Handle shared files
+    return await c.render("./src/views/index.ejs", { files: await getFiles(c) });
+}
diff --git a/src/handler/user.ts b/src/handler/user.ts
index e194008..2c3ac8d 100644
--- a/src/handler/user.ts
+++ b/src/handler/user.ts
@@ -1,25 +1,85 @@
 import type { HandlerFunc, Context } from "https://deno.land/x/abc@master/mod.ts";
-import db, {loginData} from "../db/user.ts";
+import db, { loginData } from "../db/user.ts";
+import * as log from "https://deno.land/std/log/mod.ts";
+import { getCurrentUser, isAdmin } from "../util/user.ts";
+import { isSetup } from "../util/server.ts";
+import { deleteCookie } from "https://deno.land/std/http/cookie.ts";
+
 
-export const index: HandlerFunc = async (c: Context) => c.params.name;
 export const register: HandlerFunc = async (c: Context) => {
-    const { username, email, password } = await c.body();
-    const success = await db.createUser(email, username, password);
-    // TODO: Send email
-    return {success};
-};
+    if (!(await isAdmin(c)) && await isSetup()) return { success: false }; // I'm tired: not sure if this works
+    // TODO: How to handle register
+    const { username, email, password, admin } = await c.body();
+    try {
+        const success = await db.createUser(email, username, password, admin !== undefined ? admin : false);
+        return { success };
+    } catch (e) {
+        return { success: false };
+    }
+}
+export const renderLogin: HandlerFunc = async (c: Context) => {
+    if (await getCurrentUser(c)) return c.redirect("/");
+    return await c.render("./src/views/login.ejs");
+}
 export const login: HandlerFunc = async (c: Context) => {
     const { username, password } = await c.body();
-    const data: loginData = await db.login(username, password);
-    if (data.success) {
-        c.setCookie({
-            name: "uid",
-            value: data.uid!.toString(),
-        });
-        c.setCookie({
-            name: "verification",
-            value: data.verification!,
-        })
+    try {
+        const data: loginData = await db.login(username, password);
+        if (data.success) {
+            c.setCookie({
+                name: "uid",
+                value: data.uid!.toString(),
+                path: "/",
+            });
+            c.setCookie({
+                name: "verification",
+                value: data.verification!,
+                path: "/",
+            });
+        }
+        return { success: data.success };
+    } catch (e) {
+        log.error(e);
+        return { success: false };
+    }
+}
+export const logout: HandlerFunc = async (c: Context) => {
+    deleteCookie(c.response, "uid");
+    deleteCookie(c.response, "verification");
+    c.redirect("/");
+}
+export const changeTheme: HandlerFunc = async (c: Context) => {
+    try {
+        const currentUser = await getCurrentUser(c);
+        if (!currentUser) return { success: false };
+        await db.changeTheme(currentUser.id);
+        return { success: true };
+    } catch (e) {
+        log.error(e);
+        return { success: false };
+    }
+}
+export const setAdmin: HandlerFunc = async (c: Context) => {
+    const { uid, state } = await c.body();
+    try {
+        const currentUser = await getCurrentUser(c);
+        if (!(currentUser && currentUser.isAdmin)) return { success: false };
+        await db.setAdminState(uid, state);
+        return { success: true };
+    } catch (e) {
+        log.error(e);
+        return { success: false };
+    }
+}
+export const updatePassword: HandlerFunc = async (c: Context) => {
+    const currentUser = await getCurrentUser(c);
+    if (!currentUser) return { success: false };
+    const { currentPassword, newPassword } = await c.body();
+    try {
+        await db.changePassword(currentUser.id, currentPassword, newPassword);
+        return { success: true };
+    } catch (e) {
+        log.error(e);
+        return { success: false };
     }
-    return {"success": data.success};
 }