From dfaab3a2906954e01d9d6dd1bd0eb59d26fc911d Mon Sep 17 00:00:00 2001
From: NotNite <hi@notnite.com>
Date: Wed, 26 Apr 2023 14:19:58 -0400
Subject: [PATCH] redirect if in weird states

---
 src/app/login/LoginForm.tsx | 49 ++++++++++++++++++++++++++++++++++
 src/app/login/page.tsx      | 52 +++++++------------------------------
 src/app/page.tsx            |  4 +--
 src/auth.ts                 | 16 ++++++++++++
 4 files changed, 75 insertions(+), 46 deletions(-)
 create mode 100644 src/app/login/LoginForm.tsx

diff --git a/src/app/login/LoginForm.tsx b/src/app/login/LoginForm.tsx
new file mode 100644
index 0000000..6fa3ee6
--- /dev/null
+++ b/src/app/login/LoginForm.tsx
@@ -0,0 +1,49 @@
+"use client";
+
+import React from "react";
+
+// TODO: use input from register & un programmer art this
+export default function LoginForm() {
+  const usernameRef = React.useRef<HTMLInputElement>(null);
+  const passwordRef = React.useRef<HTMLInputElement>(null);
+
+  return (
+    <form
+      style={{ display: "flex", flexDirection: "column" }}
+      onSubmit={async (e) => {
+        e.preventDefault();
+
+        const username = usernameRef.current?.value ?? "";
+        const password = passwordRef.current?.value ?? "";
+
+        const req = await fetch("/api/login", {
+          method: "POST",
+          headers: {
+            "Content-Type": "application/json"
+          },
+          body: JSON.stringify({
+            username,
+            password
+          })
+        });
+
+        if (req.status === 200) {
+          const res: { ticket: string } = await req.json();
+          document.cookie = `ticket=${res.ticket}; path=/;`;
+          window.location.href = "/me";
+        } else {
+          // todo error handling lol
+        }
+      }}
+    >
+      <input type="text" placeholder="Username" ref={usernameRef} required />
+      <input
+        type="password"
+        placeholder="Password"
+        ref={passwordRef}
+        required
+      />
+      <input type="submit" value="Login" />
+    </form>
+  );
+}
diff --git a/src/app/login/page.tsx b/src/app/login/page.tsx
index 538bd1c..9c7605c 100644
--- a/src/app/login/page.tsx
+++ b/src/app/login/page.tsx
@@ -1,52 +1,18 @@
-"use client";
-
 import styles from "@/app/page.module.css";
 import React from "react";
+import LoginForm from "./LoginForm";
+import { AuthState, getAuthState } from "@/auth";
+import { redirect } from "next/navigation";
 
-// TODO: use input from register & un programmer art this
-export default function Page() {
-  const usernameRef = React.useRef<HTMLInputElement>(null);
-  const passwordRef = React.useRef<HTMLInputElement>(null);
+export default async function Page() {
+  const state = await getAuthState();
+
+  if (state === AuthState.Registering) redirect("/register");
+  if (state === AuthState.LoggedIn) redirect("/me");
 
   return (
     <main className={styles.main}>
-      <form
-        style={{ display: "flex", flexDirection: "column" }}
-        onSubmit={async (e) => {
-          e.preventDefault();
-
-          const username = usernameRef.current?.value ?? "";
-          const password = passwordRef.current?.value ?? "";
-
-          const req = await fetch("/api/login", {
-            method: "POST",
-            headers: {
-              "Content-Type": "application/json"
-            },
-            body: JSON.stringify({
-              username,
-              password
-            })
-          });
-
-          if (req.status === 200) {
-            const res: { ticket: string } = await req.json();
-            document.cookie = `ticket=${res.ticket}; path=/;`;
-            window.location.href = "/me";
-          } else {
-            // todo error handling lol
-          }
-        }}
-      >
-        <input type="text" placeholder="Username" ref={usernameRef} required />
-        <input
-          type="password"
-          placeholder="Password"
-          ref={passwordRef}
-          required
-        />
-        <input type="submit" value="Login" />
-      </form>
+      <LoginForm />
     </main>
   );
 }
diff --git a/src/app/page.tsx b/src/app/page.tsx
index cb019ad..8e00a1e 100644
--- a/src/app/page.tsx
+++ b/src/app/page.tsx
@@ -13,9 +13,7 @@ export default function Home() {
         }}
       >
         <a href="/login">login</a>
-        <a href="/me">me</a>
-        <a href="/oauth/discord/login">discord</a>
-        <a href="/register">register</a>
+        <a href="/oauth/discord/login">register (discord)</a>
       </p>
     </main>
   );
diff --git a/src/auth.ts b/src/auth.ts
index 5941098..acaefd7 100644
--- a/src/auth.ts
+++ b/src/auth.ts
@@ -94,3 +94,19 @@ export async function createAuthTicket(username: string) {
 
   return authTicket.ticket;
 }
+
+export enum AuthState {
+  LoggedOut,
+  Registering,
+  LoggedIn
+}
+
+export async function getAuthState() {
+  const user = await getUser();
+  if (user === null) return AuthState.LoggedOut;
+
+  const info = ldap.getUserInfo(user);
+  if (info === null) return AuthState.Registering;
+
+  return AuthState.LoggedIn;
+}