File size: 2,864 Bytes
f56a29b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | /**
* Chat Storage - Persist chat sessions to IndexedDB
*
* Independent from stage/scene storage cycle.
* Handles serialization, truncation, and batch writes.
*/
import type { ChatSession, ChatMessageMetadata, SessionStatus } from '@/lib/types/chat';
import type { UIMessage } from 'ai';
import { db, type ChatSessionRecord } from './database';
/** Maximum messages per session to avoid IndexedDB bloat */
const MAX_MESSAGES_PER_SESSION = 200;
/**
* Save chat sessions for a stage to IndexedDB.
* - Active sessions are saved as 'interrupted' (streaming context lost on refresh)
* - pendingToolCalls are cleared (runtime-only state)
* - Messages are truncated to MAX_MESSAGES_PER_SESSION
*/
export async function saveChatSessions(stageId: string, sessions: ChatSession[]): Promise<void> {
if (!sessions || sessions.length === 0) {
// Delete all sessions for this stage if empty
await db.chatSessions.where('stageId').equals(stageId).delete();
return;
}
const records: ChatSessionRecord[] = sessions.map((session) => ({
id: session.id,
stageId,
type: session.type,
title: session.title,
// Mark active sessions as interrupted (streaming context lost on refresh)
status: (session.status === 'active' ? 'interrupted' : session.status) as SessionStatus,
// Truncate messages and strip non-serializable data
messages: session.messages.slice(-MAX_MESSAGES_PER_SESSION),
config: session.config,
toolCalls: session.toolCalls,
pendingToolCalls: [], // Clear runtime state
createdAt: session.createdAt,
updatedAt: session.updatedAt,
sceneId: session.sceneId,
lastActionIndex: session.lastActionIndex,
}));
await db.transaction('rw', db.chatSessions, async () => {
// Delete old sessions for this stage, then bulk insert new ones
await db.chatSessions.where('stageId').equals(stageId).delete();
await db.chatSessions.bulkPut(records);
});
}
/**
* Load chat sessions for a stage from IndexedDB.
* Returns sessions sorted by createdAt.
*/
export async function loadChatSessions(stageId: string): Promise<ChatSession[]> {
const records = await db.chatSessions.where('stageId').equals(stageId).sortBy('createdAt');
return records.map((record) => ({
id: record.id,
type: record.type,
title: record.title,
status: record.status,
messages: record.messages as UIMessage<ChatMessageMetadata>[],
config: record.config,
toolCalls: record.toolCalls,
pendingToolCalls: record.pendingToolCalls,
createdAt: record.createdAt,
updatedAt: record.updatedAt,
sceneId: record.sceneId,
lastActionIndex: record.lastActionIndex,
}));
}
/**
* Delete all chat sessions for a stage.
*/
export async function deleteChatSessions(stageId: string): Promise<void> {
await db.chatSessions.where('stageId').equals(stageId).delete();
}
|