🎉 first commit

app/routes/api.chat.$action/fork.server.ts

@@ -0,0 +1,167 @@
+import type { ActionFunctionArgs } from '@remix-run/node';
+import { getUserChatById } from '~/lib/.server/chat';
+import { prisma } from '~/lib/.server/prisma';
+import { errorResponse, successResponse } from '~/utils/api-response';
+import { createScopedLogger } from '~/utils/logger';
+
+const logger = createScopedLogger('api.chat.fork');
+
+export type HandleForkActionArgs = ActionFunctionArgs & {
+  userId: string;
+};
+
+/**
+ * 处理复制聊天操作
+ */
+export async function handleForkAction({ request, userId }: HandleForkActionArgs) {
+  try {
+    const { sourceChatId, messageId } = await request.json();
+
+    if (!sourceChatId) {
+      return errorResponse(400, '源聊天ID不能为空');
+    }
+
+    const sourceChat = await getUserChatById(sourceChatId, userId);
+    if (!sourceChat) {
+      return errorResponse(404, '找不到源聊天');
+    }
+
+    // 使用事务处理整个复制过程，确保数据一致性
+    return await prisma.$transaction(async (tx) => {
+      logger.debug(`开始复制聊天 ${sourceChatId} 的数据...`);
+
+      const metadata =
+        sourceChat.metadata &&
+        typeof sourceChat.metadata === 'object' &&
+        !Array.isArray(sourceChat.metadata) &&
+        sourceChat.metadata !== null
+          ? (sourceChat.metadata as Record<string, any>)
+          : undefined;
+
+      // 创建新聊天
+      const newChat = await tx.chat.create({
+        data: {
+          userId,
+          description: `${sourceChat.description || 'Chat'} (Copy)`,
+          urlId: sourceChat.urlId || undefined,
+          metadata,
+        },
+      });
+
+      logger.debug(`为用户 ${userId} 创建了聊天副本: ${newChat.id}`);
+
+      // 检查是否有消息需要复制
+      if (sourceChat.messages && sourceChat.messages.length > 0) {
+        // 根据messageId过滤消息
+        let messagesToCopy = sourceChat.messages;
+
+        // 如果指定了messageId，过滤消息
+        if (messageId) {
+          const targetIndex = messagesToCopy.findIndex((msg) => msg.id === messageId);
+
+          if (targetIndex === -1) {
+            await tx.chat.delete({ where: { id: newChat.id } });
+            logger.warn('在聊天中找不到指定的消息', { sourceChatId, messageId });
+            return errorResponse(404, '在聊天中找不到指定的消息');
+          }
+
+          // 只保留从 0 到 targetIndex 的消息
+          messagesToCopy = messagesToCopy.slice(0, targetIndex + 1);
+
+          logger.debug(`将复制聊天 ${sourceChatId} 的前 ${messagesToCopy.length} 条消息（到消息ID: ${messageId}）`);
+        } else {
+          logger.debug(`将复制聊天 ${sourceChatId} 的全部 ${messagesToCopy.length} 条消息`);
+        }
+
+        // 准备批量创建消息的数据
+        // 由于 prisma 中 output 与 input 类型不一致，需要手动复制 https://github.com/prisma/prisma/issues/9247
+        const messageCreateData = messagesToCopy.map((msg) => ({
+          chatId: newChat.id,
+          userId,
+          role: msg.role,
+          content: msg.content,
+          annotations: msg.annotations || undefined,
+          metadata: msg.metadata || undefined,
+          parts: msg.parts || undefined,
+          revisionId: msg.revisionId || undefined,
+          isDiscarded: msg.isDiscarded || false,
+        }));
+
+        logger.debug('批量创建消息数据', JSON.stringify(messageCreateData));
+
+        // 使用批量创建消息函数创建消息
+        await tx.message.createMany({
+          data: messageCreateData,
+        });
+
+        logger.debug(`为聊天 ${newChat.id} 批量创建了 ${messageCreateData.length} 条消息`);
+
+        const newMessages = await tx.message.findMany({
+          where: { chatId: newChat.id },
+          orderBy: { createdAt: 'asc' },
+        });
+
+        // 创建映射：原消息ID -> 新消息对象
+        const messageMapping = messagesToCopy.reduce(
+          (map, oldMsg, index) => {
+            map[oldMsg.id] = newMessages[index];
+            return map;
+          },
+          {} as Record<string, any>,
+        );
+
+        const pageToCreate = messagesToCopy
+          .filter((msg) => msg.page != null)
+          .map((msg) => {
+            const page = msg.page!;
+            return {
+              messageId: messageMapping[msg.id].id,
+              pages: JSON.parse(JSON.stringify(page.pages)),
+            };
+          });
+
+        // 批量创建 Page 项目数据
+        if (pageToCreate.length > 0) {
+          await tx.page.createMany({
+            data: pageToCreate,
+          });
+          logger.debug(`为聊天 ${newChat.id} 批量创建了 ${pageToCreate.length} 个Page项目`);
+        }
+
+        // 收集需要创建的区块数据
+        const sectionsToCreate = [];
+        for (const msg of messagesToCopy) {
+          if (msg.sections && msg.sections.length > 0) {
+            for (const section of msg.sections) {
+              sectionsToCreate.push({
+                messageId: messageMapping[msg.id].id,
+                type: section.type,
+                action: section.action,
+                actionId: section.actionId,
+                pageName: section.pageName,
+                content: section.content,
+                domId: section.domId,
+                sort: section.sort,
+                rootDomId: section.rootDomId,
+              });
+            }
+          }
+        }
+
+        // 批量创建区块数据
+        if (sectionsToCreate.length > 0) {
+          await tx.section.createMany({
+            data: sectionsToCreate,
+          });
+          logger.debug(`为聊天 ${newChat.id} 批量创建了 ${sectionsToCreate.length} 个区块`);
+        }
+      }
+
+      // 返回新聊天ID
+      return successResponse(newChat.id, '聊天复制成功');
+    });
+  } catch (error) {
+    logger.error('复制聊天失败:', error);
+    return errorResponse(500, '服务器处理请求失败');
+  }
+}