fix chat

src/server/routers/chat.ts

@@ -2,45 +2,71 @@ import { auth } from '@clerk/nextjs/server'
 import { publicProcedure, router } from "../trpc";
 import { TRPCError } from "@trpc/server";
 import { db } from '~/server/db'
-import { chatSession, systemSettings } from "../dbschema/schema";
+import { chatMessage,
+chatSession, systemSettings } from "../dbschema/schema";
 import { isAdmin } from '~/app/actions';
 import { z } from 'zod';
+import { eq } from 'drizzle-orm';
 export const chatRouter = router({
   getSession: publicProcedure.query(async () => {
     const { userId } = await auth();
     if (userId == null) {
-      throw new TRPCError({message: "chat is only available to signed in users",code: 'UNAUTHORIZED'});
+      throw new TRPCError({ message: "chat is only available to signed in users", code: 'UNAUTHORIZED' });
     }
     let session = await db.query.chatSession.findFirst({
-      with: {
-        messages: true
-      },
       where(fields, operators) {
-          return operators.eq(fields.userId,userId)
+        return operators.eq(fields.userId, userId)
       },
     })
     if (session !== undefined) {
       return session;
     }
-    let newSession = await db.insert(chatSession).values({userId: userId}).returning().execute().then((r) => r.at(0));
+    let newSession = await db.insert(chatSession).values({ userId: userId }).returning().execute().then((r) => r.at(0));
     if (newSession == undefined) {
-      throw new TRPCError({message: "failed to create session", code:"INTERNAL_SERVER_ERROR"});
+      throw new TRPCError({ message: "failed to create session", code: "INTERNAL_SERVER_ERROR" });
     }
     session = await db.query.chatSession.findFirst({
       with: {
         messages: true
       },
       where(fields, operators) {
-          return operators.eq(fields.id,newSession.id)
+        return operators.eq(fields.id, newSession.id)
       },
     })
     if (session == undefined) {
-      throw new TRPCError({message: "session not found", code:"NOT_FOUND"});
+      throw new TRPCError({ message: "session not found", code: "NOT_FOUND" });
     }
     if (session !== undefined) {
       return session;
     }
   }),
+  getMessages: publicProcedure.input(z.string()).query(async ({input}) => {
+    let res = await db.query.chatMessage.findMany({
+      where(fields,operators) {
+        return operators.eq(fields.sessionId,input)
+      }
+    })
+    return res;
+  }),
+  clearChat: publicProcedure.mutation(async () => {
+    console.log("deleting session")
+    const { userId } = await auth();
+    if (userId == null) {
+      throw new TRPCError({ message: "chat is only available to signed in users", code: 'UNAUTHORIZED' });
+    }
+    let session = await db.query.chatSession.findFirst({
+      with: {
+        messages: true
+      },
+      where(fields, operators) {
+        return operators.eq(fields.userId, userId)
+      },
+    })
+    if (session != undefined) {
+      db.delete(chatMessage).where(eq(chatMessage.sessionId,session.id)).execute()
+    }
+
+  }),
   getSystemPrompt: publicProcedure.query(async () => {
     const row = await db.select().from(systemSettings).limit(1).then((r) => r[0])
     return row?.systemPropmt ?? ''