feat(web-ai): persist chat history locally

2026-04-06 15:58:00 +08:00
parent 4c6aeb3e6c
commit 1ea483016f
3 changed files with 181 additions and 6 deletions
@@ -0,0 +1,91 @@
+import { localDb, type LocalAiChatSessionRecord } from "@/services/local-db";
+import type { WebAiChannel } from "@/services/ai-api";
+
+export type LocalAiChatMessageRecord = {
+  id: string;
+  role: "user" | "assistant" | "system";
+  content: string;
+  meta?: string;
+};
+
+export type SaveLocalAiChatSessionInput = {
+  userId: string;
+  channel: WebAiChannel;
+  sessionId: string | null;
+  messages: LocalAiChatMessageRecord[];
+};
+
+export type LocalAiChatSessionSnapshot = {
+  channel: WebAiChannel;
+  sessionId: string | null;
+  messages: LocalAiChatMessageRecord[];
+};
+
+function createSessionKey(userId: string, channel: WebAiChannel): string {
+  return `${userId}:${channel}`;
+}
+
+function parseMessages(messagesJson: string): LocalAiChatMessageRecord[] {
+  try {
+    const parsed = JSON.parse(messagesJson) as unknown;
+    if (!Array.isArray(parsed)) {
+      return [];
+    }
+
+    return parsed.filter((item): item is LocalAiChatMessageRecord => {
+      if (!item || typeof item !== "object") {
+        return false;
+      }
+
+      const record = item as Record<string, unknown>;
+      return (
+        typeof record["id"] === "string" &&
+        (record["role"] === "user" ||
+          record["role"] === "assistant" ||
+          record["role"] === "system") &&
+        typeof record["content"] === "string" &&
+        (record["meta"] === undefined || typeof record["meta"] === "string")
+      );
+    });
+  } catch {
+    return [];
+  }
+}
+
+function toSnapshot(record: LocalAiChatSessionRecord): LocalAiChatSessionSnapshot {
+  return {
+    channel: record.channel,
+    sessionId: record.sessionId,
+    messages: parseMessages(record.messagesJson)
+  };
+}
+
+export async function listLocalAiChatSessions(
+  userId: string
+): Promise<LocalAiChatSessionSnapshot[]> {
+  const records = await localDb.aiChatSessions.where("userId").equals(userId).toArray();
+  return records.map(toSnapshot);
+}
+
+export async function saveLocalAiChatSession(
+  input: SaveLocalAiChatSessionInput
+): Promise<LocalAiChatSessionRecord> {
+  const record: LocalAiChatSessionRecord = {
+    key: createSessionKey(input.userId, input.channel),
+    userId: input.userId,
+    channel: input.channel,
+    sessionId: input.sessionId,
+    messagesJson: JSON.stringify(input.messages),
+    updatedAt: Date.now()
+  };
+
+  await localDb.aiChatSessions.put(record);
+  return record;
+}
+
+export async function deleteLocalAiChatSession(
+  userId: string,
+  channel: WebAiChannel
+): Promise<void> {
+  await localDb.aiChatSessions.delete(createSessionKey(userId, channel));
+}