feat: Models Ai chat integration

Author: yashmakhija

bun.lock

@@ -0,0 +1,54 @@
+"use server";
+
+import { auth } from "@/server/auth";
+import { db } from "@/server/db";
+
+type MessageRole = "user" | "system" | "assistant" | "data";
+
+export async function getChat(chatId: string) {
+  try {
+    const authResult = await auth();
+    if (!authResult?.user?.id) {
+      return { success: false, error: "Unauthorized" };
+    }
+
+    const userId = authResult.user.id;
+
+
+    const chat = await db.chat.findFirst({
+      where: {
+        id: chatId,
+        userId,
+      },
+      include: {
+        messages: {
+          orderBy: {
+            createdAt: "asc",
+          },
+        },
+      },
+    });
+
+    if (!chat) {
+      return { success: false, error: "Chat not found" };
+    }
+
+    
+    const formattedMessages = chat.messages.map((message) => ({
+      id: message.id,
+      role: message.role as MessageRole,
+      content: message.content,
+    }));
+
+    return {
+      success: true,
+      chat: {
+        id: chat.id,
+        messages: formattedMessages,
+      },
+    };
+  } catch (error) {
+    console.error("Failed to get chat:", error);
+    return { success: false, error: "Failed to get chat" };
+  }
+}

src/actions/getChat.ts

@@ -0,0 +1,46 @@
+"use server";
+
+import { auth } from "@/server/auth";
+import { db } from "@/server/db";
+
+export async function getChats() {
+  try {
+    const authResult = await auth();
+    if (!authResult?.user?.id) {
+      return { success: false, error: "Unauthorized", chats: [] };
+    }
+
+    const userId = authResult.user.id;
+
+
+    const chats = await db.chat.findMany({
+      where: {
+        userId,
+      },
+      include: {
+        messages: {
+          take: 1,
+          orderBy: {
+            createdAt: "desc",
+          },
+        },
+      },
+      orderBy: {
+        updatedAt: "desc",
+      },
+    });
+
+    return {
+      success: true,
+      chats: chats.map((chat) => ({
+        id: chat.id,
+        createdAt: chat.createdAt,
+        updatedAt: chat.updatedAt,
+        lastMessage: chat.messages[0]?.content ?? "",
+      })),
+    };
+  } catch (error) {
+    console.error("Failed to get chats:", error);
+    return { success: false, error: "Failed to get chats", chats: [] };
+  }
+}

src/actions/getChats.ts

@@ -0,0 +1,71 @@
+"use server";
+
+import { auth } from "@/server/auth";
+import { db } from "@/server/db";
+
+type MessageRole = "user" | "system" | "assistant" | "data";
+
+interface Message {
+  id: string;
+  role: MessageRole;
+  content: string;
+}
+
+export async function saveChat(messages: Message[]) {
+  try {
+    const authResult = await auth();
+    if (!authResult?.user?.id) {
+      return { success: false, error: "Unauthorized" };
+    }
+
+    const userId = authResult.user.id;
+
+
+    let chat = await db.chat.findFirst({
+      where: {
+        userId,
+        messages: {
+          some: {
+            id: messages[0]?.id,
+          },
+        },
+      },
+    });
+
+    if (!chat) {
+
+      chat ??= await db.chat.create({
+        data: {
+          userId,
+        },
+      });
+    }
+
+
+    for (const message of messages) {
+
+      const existingMessage = await db.message.findFirst({
+        where: {
+          id: message.id,
+          chatId: chat.id,
+        },
+      });
+
+      if (!existingMessage) {
+        await db.message.create({
+          data: {
+            id: message.id,
+            chatId: chat.id,
+            content: message.content,
+            role: message.role,
+          },
+        });
+      }
+    }
+
+    return { success: true, chatId: chat.id };
+  } catch (error) {
+    console.error("Failed to save chat:", error);
+    return { success: false, error: "Failed to save chat" };
+  }
+}