From e85c94d21d95a169bd8fdf4e88310a7c76834dc9 Mon Sep 17 00:00:00 2001
From: Milo Schwartz <mschwartz10612@gmail.com>
Date: Wed, 2 Oct 2024 20:49:05 -0400
Subject: [PATCH] verifyTotp endpoint

---
 server/routers/auth/verifyTotp.ts | 35 ++++++++++++++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/server/routers/auth/verifyTotp.ts b/server/routers/auth/verifyTotp.ts
index e338f929..75ad54f3 100644
--- a/server/routers/auth/verifyTotp.ts
+++ b/server/routers/auth/verifyTotp.ts
@@ -5,7 +5,11 @@ import { fromError } from "zod-validation-error";
 import { decodeHex } from "oslo/encoding";
 import { TOTPController } from "oslo/otp";
 import HttpCode from "@server/types/HttpCode";
-import { verifySession, lucia, unauthorized } from "@server/auth";
+import { verifySession, unauthorized } from "@server/auth";
+import { response } from "@server/utils";
+import { db } from "@server/db";
+import { users } from "@server/db/schema";
+import { eq } from "drizzle-orm";
 
 export const verifyTotpBody = z.object({
     code: z.string(),
@@ -39,4 +43,33 @@ export async function verifyTotp(
     if (!session) {
         return unauthorized();
     }
+
+    if (!user.twoFactorSecret) {
+        return createHttpError(
+            HttpCode.BAD_REQUEST,
+            "User has not requested two-factor authentication",
+        );
+    }
+
+    const totpController = new TOTPController();
+    const valid = await totpController.verify(
+        user.twoFactorSecret,
+        decodeHex(code),
+    );
+
+    if (valid) {
+        // if valid, enable two-factor authentication; the totp secret is no longer temporary
+        await db
+            .update(users)
+            .set({ twoFactorEnabled: true })
+            .where(eq(users.id, user.id));
+    }
+
+    return response<{ valid: boolean }>(res, {
+        data: { valid },
+        success: true,
+        error: false,
+        message: valid ? "Code is valid" : "Code is invalid",
+        status: HttpCode.OK,
+    });
 }