making routes and saving chats to db

Author: Aayush-Agnihotri

.env.template

@@ -22,4 +22,4 @@ USE_SECURE_COOKIES=
 SPECIAL_TOKEN=
 
 OLLAMA_ENDPOINT=
-DATABASE_URL=
+MONGO_DATABASE_URL=

prisma/schema.prisma

@@ -1,41 +1,33 @@
 datasource db {
   provider = "mongodb"
-  url      = env("DATABASE_URL")
+  url      = env("MONGO_DATABASE_URL")
 }
 
 generator client {
   provider = "prisma-client-js"
 }
 
-model Email {
-  id    String @id @default(auto()) @map("_id") @db.ObjectId
-  email String @unique
-  user  User?
-}
-
 model User {
-  id      String  @id @default(auto()) @map("_id") @db.ObjectId
-  uid     String  @unique
-  isLead  Boolean
-  email   Email   @relation(fields: [emailId], references: [id])
-  emailId String  @unique @db.ObjectId
-  chats   Chat[]
+  email    String  @id @map("_id")
+  isAppDev Boolean
+  chats    Chat[]
 }
 
 model Chat {
-  uuid     String    @id @map("_id")
-  summary  String
-  user     User      @relation(fields: [userId], references: [uid])
-  userId   String
-  messages Message[]
+  id        String    @id @map("_id")
+  summary   String
+  updatedAt DateTime
+  user      User      @relation(fields: [userEmail], references: [email], onDelete: Cascade)
+  userEmail String
+  messages  Message[]
 }
 
 model Message {
   id        String   @id @default(auto()) @map("_id") @db.ObjectId
   content   String
   images    String[]
-  timestamp DateTime @default(now())
+  timestamp DateTime
   sender    String
-  chat      Chat     @relation(fields: [chatId], references: [uuid])
+  chat      Chat     @relation(fields: [chatId], references: [id], onDelete: Cascade)
   chatId    String
 }

scripts/prepopulateEmails.ts

@@ -1,16 +1,13 @@
 import { PrismaClient } from '@prisma/client';
+import { upsertAppDevUser } from '@/app/utils/databaseUtils';
 
 const prisma = new PrismaClient();
 
 const appDevEmails = ['a@cornell.edu', 'b@cornell.edu', 'c@cornell.edu'];
 
 async function main() {
   for (const email of appDevEmails) {
-    await prisma.email.upsert({
-      where: { email: email },
-      create: { email: email },
-      update: {},
-    });
+    await upsertAppDevUser(email);
   }
 }
 

src/app/api/authenticate/route.ts

@@ -1,22 +1,16 @@
+import { getUserByEmail, upsertUser } from '@/app/utils/databaseUtils';
 import { NextRequest, NextResponse } from 'next/server';
-// import { getEmail, upsertUserFromEmail } from '@/app/utils/databaseUtils';
 
 export async function GET(request: NextRequest) {
-  // const uid = request.headers.get('uid');
-  // if (!uid) {
-  //   return NextResponse.json({ message: 'No uid found' }, { status: 400 });
-  // }
+  const email = request.headers.get('email');
+  if (!email) {
+    return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
+  }
 
-  // const userEmail = request.headers.get('email')!;
-  // const email = await getEmail(userEmail);
-  // if (!email) {
-  //   return NextResponse.json(
-  //     { message: 'Only Cornell AppDev members can use AI Dev' },
-  //     { status: 401 },
-  //   );
-  // }
-
-  // const newUser = await upsertUserFromEmail(email, uid);
-  // return NextResponse.json({ success: true, user: newUser });
-  return NextResponse.json({ success: true });
+  const user = await getUserByEmail(email);
+  if (!user) {
+    await upsertUser(email);
+  }
+  // TODO: Return back user to frontend to store, should contain name, pfp, email, isAppDev
+  return NextResponse.json({});
 }

src/app/api/chats/[...chatId]/route.ts

@@ -1,10 +1,71 @@
 import { NextRequest, NextResponse } from 'next/server';
-import { getMessagesByChatId } from '@/app/utils/databaseUtils';
+import {
+  createChat,
+  createMessage,
+  deleteChatById,
+  getChatById,
+  getChatsByEmail,
+  getMessagesByChatId,
+  updateChatTimestamp,
+} from '@/app/utils/databaseUtils';
 
 export async function GET(request: NextRequest) {
-  // TODO: Do Authorization check
-  const uid = request.headers.get('uid');
+  const email = request.headers.get('email');
+  if (!email) {
+    return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
+  }
+
   const chatId = request.nextUrl.pathname.split('/')[3];
+  const chat = await getChatById(chatId);
+  if (!chat) {
+    return NextResponse.json({ message: 'Chat not found' }, { status: 404 });
+  }
+
+  if (chat.userEmail !== email) {
+    return NextResponse.json({ message: 'Unauthorized' }, { status: 401 });
+  }
+
   const messages = await getMessagesByChatId(chatId);
   return NextResponse.json({ messages: messages });
 }
+
+export async function POST(request: NextRequest) {
+  const email = request.headers.get('email');
+  if (!email) {
+    return NextResponse.json({ message: 'No email' }, { status: 400 });
+  }
+
+  const chatId = request.nextUrl.pathname.split('/')[3];
+  const { content, images, timestamp, sender } = await request.json();
+  if (!content) {
+    return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
+  }
+
+  const chat = await getChatById(chatId);
+  if (!chat) {
+    await createChat(email, chatId, content, timestamp);
+  }
+  await createMessage(chatId, content, images, timestamp, sender);
+  await updateChatTimestamp(chatId, timestamp);
+  return NextResponse.json({});
+}
+
+export async function DELETE(request: NextRequest) {
+  const email = request.headers.get('email');
+  if (!email) {
+    return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
+  }
+
+  const chatId = request.nextUrl.pathname.split('/')[3];
+  const chat = await getChatById(chatId);
+  if (!chat) {
+    return NextResponse.json({ message: 'Chat not found' }, { status: 404 });
+  }
+
+  if (chat.userEmail !== email) {
+    return NextResponse.json({ message: 'Unauthorized' }, { status: 401 });
+  }
+
+  await deleteChatById(chatId);
+  return NextResponse.json({ chats: await getChatsByEmail(email) });
+}

src/app/api/chats/route.ts

@@ -1,44 +1,12 @@
 import { NextRequest, NextResponse } from 'next/server';
-import {
-  deleteChatById,
-  getChatsByUserId,
-  getChatById,
-  createChatWithMessage,
-} from '@/app/utils/databaseUtils';
+import { getChatsByEmail } from '@/app/utils/databaseUtils';
 
 export async function GET(request: NextRequest) {
-  const uid = request.headers.get('uid');
-  const chats = await getChatsByUserId(uid!);
-  return NextResponse.json({ chats: chats });
-}
-
-export async function POST(request: NextRequest) {
-  const uid = request.headers.get('uid');
-  const { chatId, message } = await request.json();
-  if (!chatId || !message) {
+  const email = request.headers.get('email');
+  if (!email) {
     return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
   }
 
-  await createChatWithMessage(uid!, chatId, message);
-  return NextResponse.json({});
-}
-
-export async function DELETE(request: NextRequest) {
-  const uid = request.headers.get('uid');
-  const { chatId } = await request.json();
-  if (!chatId) {
-    return NextResponse.json({ message: 'Invalid request' }, { status: 400 });
-  }
-
-  const chat = await getChatById(chatId);
-  if (!chat) {
-    return NextResponse.json({ message: 'Chat not found' }, { status: 404 });
-  }
-
-  if (chat.userId !== uid) {
-    return NextResponse.json({ message: 'Unauthorized' }, { status: 401 });
-  }
-
-  await deleteChatById(chatId);
-  return NextResponse.json({ chats: await getChatsByUserId(uid!) });
+  const chats = await getChatsByEmail(email);
+  return NextResponse.json({ chats: chats });
 }