From 8d428d90a8c50ac2b397cf3da026df30f8d02b60 Mon Sep 17 00:00:00 2001
From: slice <tinyslices@gmail.com>
Date: Wed, 26 Apr 2023 20:32:52 -0700
Subject: [PATCH] borgus

---
 .../migration.sql                               |  8 ++++++++
 prisma/schema.prisma                            |  2 +-
 src/app/me/AboutMe.tsx                          |  8 ++++++++
 src/app/oauth/discord/redirect/route.ts         | 17 ++++++++++++-----
 src/app/oauth/github/redirect/route.ts          | 13 ++++++++++++-
 src/components/AvatarChanger.module.css         |  4 ++--
 src/ldap.ts                                     | 16 ++++++++++++++++
 7 files changed, 59 insertions(+), 9 deletions(-)
 create mode 100644 prisma/migrations/20230427033108_make_username_unique/migration.sql

diff --git a/prisma/migrations/20230427033108_make_username_unique/migration.sql b/prisma/migrations/20230427033108_make_username_unique/migration.sql
new file mode 100644
index 0000000..54942a5
--- /dev/null
+++ b/prisma/migrations/20230427033108_make_username_unique/migration.sql
@@ -0,0 +1,8 @@
+/*
+  Warnings:
+
+  - A unique constraint covering the columns `[username]` on the table `User` will be added. If there are existing duplicate values, this will fail.
+
+*/
+-- CreateIndex
+CREATE UNIQUE INDEX "User_username_key" ON "User"("username");
diff --git a/prisma/schema.prisma b/prisma/schema.prisma
index 6699e0c..fcf95a9 100644
--- a/prisma/schema.prisma
+++ b/prisma/schema.prisma
@@ -18,7 +18,7 @@ model AuthTicket {
 
 model User {
   id         Int         @id @default(autoincrement())
-  username   String?
+  username   String?     @unique
   authTicket AuthTicket?
 
   discordAuth DiscordAuth?
diff --git a/src/app/me/AboutMe.tsx b/src/app/me/AboutMe.tsx
index 85df785..b921c5f 100644
--- a/src/app/me/AboutMe.tsx
+++ b/src/app/me/AboutMe.tsx
@@ -226,6 +226,14 @@ export default function AboutMe({ info }: { info: UserInfo }) {
           )}
         </Formik>
       </PrettyForm>
+
+      <hr className={styles.divider} />
+      <h2 className={styles.header}>Connections</h2>
+      <div>
+        <p>discord: {info.discordId}</p>
+        <p>github: {info.githubId}</p>
+      </div>
+
       <hr className={styles.divider} />
       <input
         type="button"
diff --git a/src/app/oauth/discord/redirect/route.ts b/src/app/oauth/discord/redirect/route.ts
index 46dd533..bdb9095 100644
--- a/src/app/oauth/discord/redirect/route.ts
+++ b/src/app/oauth/discord/redirect/route.ts
@@ -11,6 +11,7 @@ import prisma from "@/prisma";
 import { v4 } from "uuid";
 import * as ldap from "@/ldap";
 import { getLogger } from "@/logger";
+import { AuthState, getAuthState, getUser } from "@/auth";
 
 export async function GET(request: Request) {
   const logger = getLogger("/oauth/discord/redirect");
@@ -66,6 +67,13 @@ export async function GET(request: Request) {
     return new Response("not permitted to register account", { status: 403 });
   }
 
+  let userId = null;
+  const authState = await getAuthState();
+  if (authState === AuthState.LoggedIn) {
+    const currentUser = await getUser();
+    userId = currentUser?.id;
+  }
+
   // - create the discord auth data in prisma, which will make the user if it doesn't exist
   // - get the user from the discord auth data
   // - either create a new auth ticket or invalidate the old one
@@ -80,11 +88,10 @@ export async function GET(request: Request) {
       accessToken: tokenBody.access_token,
       refreshToken: tokenBody.refresh_token,
       expiresAt: new Date(Date.now() + tokenBody.expires_in * 1000),
-      user: {
-        create: {
-          username: null
-        }
-      }
+      user:
+        userId != null
+          ? { connect: { id: userId } }
+          : { create: { username: null } }
     },
     update: {
       accessToken: tokenBody.access_token,
diff --git a/src/app/oauth/github/redirect/route.ts b/src/app/oauth/github/redirect/route.ts
index e52d184..1a9b04a 100644
--- a/src/app/oauth/github/redirect/route.ts
+++ b/src/app/oauth/github/redirect/route.ts
@@ -9,6 +9,7 @@ import {
 import prisma from "@/prisma";
 import * as ldap from "@/ldap";
 import { v4 } from "uuid";
+import { AuthState, getAuthState, getUser } from "@/auth";
 
 const logger = getLogger("/oauth/github/redirect");
 
@@ -61,12 +62,22 @@ export async function GET(request: Request) {
     return new Response("not permitted to register account", { status: 403 });
   }
 
+  let userId = null;
+  const authState = await getAuthState();
+  if (authState === AuthState.LoggedIn) {
+    const currentUser = await getUser();
+    userId = currentUser?.id;
+  }
+
   const githubAuth = await prisma.gitHubAuth.upsert({
     where: { id: githubUser.id },
     create: {
       id: githubUser.id,
       accessToken,
-      user: { create: { username: null } }
+      user:
+        userId != null
+          ? { connect: { id: userId } }
+          : { create: { username: null } }
     },
     update: { accessToken }
   });
diff --git a/src/components/AvatarChanger.module.css b/src/components/AvatarChanger.module.css
index 9f95934..7be6bd7 100644
--- a/src/components/AvatarChanger.module.css
+++ b/src/components/AvatarChanger.module.css
@@ -13,8 +13,8 @@
 }
 
 .avatarChanger button svg {
-  width: 1.5em;
-  height: 1.5em;
+  width: 1.2em;
+  height: 1.2em;
   margin-right: 0.5em;
 }
 
diff --git a/src/ldap.ts b/src/ldap.ts
index d6f9a37..5b09f7f 100644
--- a/src/ldap.ts
+++ b/src/ldap.ts
@@ -4,6 +4,7 @@ import { gql } from "./__generated__";
 import { BerWriter } from "asn1";
 import { User } from "@prisma/client";
 import { ensureJpg } from "@/image";
+import prisma from "./prisma";
 
 export type LLDAPAuthResponse = {
   token: string;
@@ -19,6 +20,9 @@ export type UserInfo = {
   displayName: string;
   email: string;
   avatar?: string;
+
+  discordId?: string;
+  githubId?: string;
 };
 
 async function getLdapClient() {
@@ -202,10 +206,22 @@ export async function getUserInfo(user: User) {
     ? `data:image/jpeg;base64,${mutationAvatar}`
     : undefined;
 
+  const dbUser = await prisma.user.findFirst({
+    where: {
+      username: user.username
+    },
+    include: {
+      discordAuth: true,
+      githubAuth: true
+    }
+  });
+
   const userInfo: UserInfo = {
     username: mutation.data.user.id,
     displayName: mutation.data.user.displayName,
     email: mutation.data.user.email,
+    discordId: dbUser?.discordAuth?.id,
+    githubId: dbUser?.githubAuth?.id?.toString(),
     avatar
   };