feat: widgets and ai chat

2026-04-27 11:40:38 +01:00
parent ae1801f85f
commit 5717dd8004
391 changed files with 28177 additions and 22467 deletions
@@ -0,0 +1,96 @@
+import { and, desc, eq, inArray } from 'drizzle-orm';
+import * as schema from '@db/schema';
+import { db } from '@db/drizzle';
+
+/** Cap on how many documents we keep per conversation. Older ones get pruned on insert. */
+const MAX_DOCUMENTS_PER_CONVERSATION = 10;
+
+export interface ChatDocumentRecord {
+  id: string;
+  conversationId: string;
+  courseId: string;
+  userId: string;
+  assetId: string | null;
+  fileName: string;
+  mimeType: string;
+  text: string;
+  wordCount: number;
+  pageCount: number | null;
+  createdAt: string;
+}
+
+export async function createChatDocument(record: {
+  id: string;
+  conversationId: string;
+  courseId: string;
+  userId: string;
+  assetId: string | null;
+  fileName: string;
+  mimeType: string;
+  text: string;
+  wordCount: number;
+  pageCount: number | null;
+}): Promise<void> {
+  try {
+    await db.transaction(async (tx) => {
+      await tx.insert(schema.aiChatDocument).values(record);
+
+      const all = await tx
+        .select({ id: schema.aiChatDocument.id })
+        .from(schema.aiChatDocument)
+        .where(eq(schema.aiChatDocument.conversationId, record.conversationId))
+        .orderBy(desc(schema.aiChatDocument.createdAt));
+
+      if (all.length > MAX_DOCUMENTS_PER_CONVERSATION) {
+        const toDelete = all.slice(MAX_DOCUMENTS_PER_CONVERSATION).map((row) => row.id);
+
+        await tx.delete(schema.aiChatDocument).where(inArray(schema.aiChatDocument.id, toDelete));
+      }
+    });
+  } catch (error) {
+    console.error('createChatDocument error:', error);
+    throw new Error('Failed to persist chat document');
+  }
+}
+
+export async function getChatDocument(documentId: string): Promise<ChatDocumentRecord | null> {
+  try {
+    const [row] = await db
+      .select()
+      .from(schema.aiChatDocument)
+      .where(eq(schema.aiChatDocument.id, documentId))
+      .limit(1);
+
+    return row ?? null;
+  } catch (error) {
+    console.error('getChatDocument error:', error);
+    throw new Error('Failed to fetch chat document');
+  }
+}
+
+export async function getChatDocumentsByIds(documentIds: string[]): Promise<ChatDocumentRecord[]> {
+  if (documentIds.length === 0) return [];
+
+  try {
+    return await db.select().from(schema.aiChatDocument).where(inArray(schema.aiChatDocument.id, documentIds));
+  } catch (error) {
+    console.error('getChatDocumentsByIds error:', error);
+    throw new Error('Failed to fetch chat documents');
+  }
+}
+
+export async function listChatDocumentsByConversation(
+  conversationId: string,
+  userId: string
+): Promise<ChatDocumentRecord[]> {
+  try {
+    return await db
+      .select()
+      .from(schema.aiChatDocument)
+      .where(and(eq(schema.aiChatDocument.conversationId, conversationId), eq(schema.aiChatDocument.userId, userId)))
+      .orderBy(desc(schema.aiChatDocument.createdAt));
+  } catch (error) {
+    console.error('listChatDocumentsByConversation error:', error);
+    throw new Error('Failed to list chat documents');
+  }
+}
@@ -0,0 +1,158 @@
+import { and, eq, desc } from 'drizzle-orm';
+import * as schema from '@db/schema';
+import { db } from '@db/drizzle';
+
+const MAX_CONVERSATIONS_PER_COURSE = 50;
+
+export interface ChatConversationSummary {
+  id: string;
+  title: string | null;
+  createdAt: string;
+  updatedAt: string;
+}
+
+// ─── List all conversations for a user in a course ───────────────────────────
+
+export async function listChatConversations(courseId: string, userId: string): Promise<ChatConversationSummary[]> {
+  try {
+    const rows = await db
+      .select({
+        id: schema.aiChatConversation.id,
+        title: schema.aiChatConversation.title,
+        createdAt: schema.aiChatConversation.createdAt,
+        updatedAt: schema.aiChatConversation.updatedAt
+      })
+      .from(schema.aiChatConversation)
+      .where(and(eq(schema.aiChatConversation.courseId, courseId), eq(schema.aiChatConversation.userId, userId)))
+      .orderBy(desc(schema.aiChatConversation.updatedAt))
+      .limit(MAX_CONVERSATIONS_PER_COURSE);
+
+    return rows;
+  } catch (error) {
+    console.error('listChatConversations error:', error);
+    throw new Error('Failed to list chat conversations');
+  }
+}
+
+// ─── Get a single conversation with messages ─────────────────────────────────
+
+export async function getChatConversation(conversationId: string, userId: string) {
+  try {
+    const [row] = await db
+      .select()
+      .from(schema.aiChatConversation)
+      .where(and(eq(schema.aiChatConversation.id, conversationId), eq(schema.aiChatConversation.userId, userId)))
+      .limit(1);
+
+    return row ?? null;
+  } catch (error) {
+    console.error('getChatConversation error:', error);
+    throw new Error('Failed to fetch chat conversation');
+  }
+}
+
+// ─── Create a new conversation ───────────────────────────────────────────────
+
+export async function createChatConversation(
+  courseId: string,
+  userId: string,
+  title?: string
+): Promise<{ id: string; title: string | null }> {
+  try {
+    const [row] = await db
+      .insert(schema.aiChatConversation)
+      .values({
+        courseId,
+        userId,
+        title: title || 'New conversation',
+        messages: []
+      })
+      .returning({ id: schema.aiChatConversation.id, title: schema.aiChatConversation.title });
+
+    return row;
+  } catch (error) {
+    console.error('createChatConversation error:', error);
+    throw new Error('Failed to create chat conversation');
+  }
+}
+
+// ─── Save messages to a conversation ─────────────────────────────────────────
+
+export async function saveChatMessages(conversationId: string, userId: string, messages: unknown[], title?: string) {
+  try {
+    const updateData: Record<string, unknown> = {
+      messages,
+      updatedAt: new Date().toISOString()
+    };
+
+    if (title) {
+      updateData.title = title;
+    }
+
+    await db
+      .update(schema.aiChatConversation)
+      .set(updateData)
+      .where(and(eq(schema.aiChatConversation.id, conversationId), eq(schema.aiChatConversation.userId, userId)));
+  } catch (error) {
+    console.error('saveChatMessages error:', error);
+    throw new Error('Failed to save chat messages');
+  }
+}
+
+// ─── Delete a conversation ───────────────────────────────────────────────────
+
+export async function deleteChatConversation(conversationId: string, userId: string) {
+  try {
+    await db
+      .delete(schema.aiChatConversation)
+      .where(and(eq(schema.aiChatConversation.id, conversationId), eq(schema.aiChatConversation.userId, userId)));
+  } catch (error) {
+    console.error('deleteChatConversation error:', error);
+    throw new Error('Failed to delete chat conversation');
+  }
+}
+
+// ─── Update conversation title ───────────────────────────────────────────────
+
+export async function updateConversationTitle(conversationId: string, userId: string, title: string): Promise<void> {
+  try {
+    await db
+      .update(schema.aiChatConversation)
+      .set({ title, updatedAt: new Date().toISOString() })
+      .where(and(eq(schema.aiChatConversation.id, conversationId), eq(schema.aiChatConversation.userId, userId)));
+  } catch (error) {
+    console.error('updateConversationTitle error:', error);
+    throw new Error('Failed to update conversation title');
+  }
+}
+
+// ─── Legacy compat (keep old exports working during migration) ───────────────
+
+export async function getChatHistory(courseId: string, userId: string) {
+  const conversations = await listChatConversations(courseId, userId);
+
+  if (conversations.length === 0) return [];
+
+  const latestConversation = await getChatConversation(conversations[0].id, userId);
+
+  return (latestConversation?.messages ?? []) as unknown[];
+}
+
+export async function saveChatHistory(courseId: string, userId: string, messages: unknown[]) {
+  const conversations = await listChatConversations(courseId, userId);
+
+  if (conversations.length > 0) {
+    await saveChatMessages(conversations[0].id, userId, messages);
+  } else {
+    const newConversation = await createChatConversation(courseId, userId);
+    await saveChatMessages(newConversation.id, userId, messages);
+  }
+}
+
+export async function deleteChatHistory(courseId: string, userId: string) {
+  const conversations = await listChatConversations(courseId, userId);
+
+  for (const conversation of conversations) {
+    await deleteChatConversation(conversation.id, userId);
+  }
+}
@@ -0,0 +1,2 @@
+export * from './chat-history';
+export * from './chat-document';