-
+
+
+
| null = null
const SAVE_DEBOUNCE = 800
-// Server-side persistence via Vite dev middleware
+// Server-side persistence via Vite dev middleware (fallback for dev mode without IDB)
async function loadServerChats(): Promise<{ conversations: Map; activeId: string | null }> {
const empty = { conversations: new Map(), activeId: null }
if (!isDev) return empty
@@ -43,6 +51,19 @@ function saveServerChats(conversations: Map, activeId: str
}, SAVE_DEBOUNCE)
}
+// Debounced IDB save for a single conversation
+const idbSaveTimers = new Map>()
+
+function debouncedIDBSave(conv: Conversation) {
+ if (!useIDB) return
+ const existing = idbSaveTimers.get(conv.id)
+ if (existing) clearTimeout(existing)
+ idbSaveTimers.set(conv.id, setTimeout(() => {
+ idbSave(conv).catch(() => {})
+ idbSaveTimers.delete(conv.id)
+ }, SAVE_DEBOUNCE))
+}
+
export const useChatStore = defineStore('chat', () => {
const conversations = ref>(new Map())
const activeConversationId = ref(null)
@@ -55,25 +76,55 @@ export const useChatStore = defineStore('chat', () => {
const webSearchEnabled = ref(localStorage.getItem('aiui-web-search') !== 'false')
const chatCollapsed = ref(localStorage.getItem('aiui-chat-collapsed') !== 'false')
- // Load chats from server on startup — _loaded gate prevents the watcher
- // from overwriting the file with empty data before the load completes
- if (isDev) {
- loadServerChats().then((data) => {
+ // Load chats: try IndexedDB first, fall back to dev-chats middleware
+ async function loadChats() {
+ if (useIDB) {
+ try {
+ const idbConversations = await idbLoadAll()
+ if (idbConversations.size > 0) {
+ conversations.value = idbConversations
+ const savedActiveId = localStorage.getItem('aiui-active-conversation')
+ activeConversationId.value =
+ savedActiveId && idbConversations.has(savedActiveId)
+ ? savedActiveId
+ : [...idbConversations.keys()].pop() ?? null
+ loaded.value = true
+ _loaded = true
+ return
+ }
+ } catch {
+ // IDB failed, fall through to dev-chats
+ }
+ }
+
+ if (isDev) {
+ const data = await loadServerChats()
if (data.conversations.size > 0) {
conversations.value = data.conversations
activeConversationId.value =
data.activeId && data.conversations.has(data.activeId)
? data.activeId
: [...data.conversations.keys()][0] ?? null
+ // Migrate existing dev-chats to IndexedDB
+ if (useIDB) {
+ for (const conv of data.conversations.values()) {
+ idbSave(conv).catch(() => {})
+ }
+ }
}
- loaded.value = true
- _loaded = true
- })
- } else {
+ }
loaded.value = true
_loaded = true
}
+ loadChats()
+
+ // Persist active conversation ID to localStorage
+ watch(activeConversationId, (id) => {
+ if (id) localStorage.setItem('aiui-active-conversation', id)
+ else localStorage.removeItem('aiui-active-conversation')
+ })
+
watch(panelSide, (val) => {
localStorage.setItem('aiui-panel-side', val)
})
@@ -89,6 +140,7 @@ export const useChatStore = defineStore('chat', () => {
watch(
[conversations, activeConversationId],
([conv, active]) => {
+ // Dev-chats middleware fallback
saveServerChats(conv as Map, active as string | null)
},
{ deep: true }
@@ -116,6 +168,7 @@ export const useChatStore = defineStore('chat', () => {
}
conversations.value.set(id, conversation)
activeConversationId.value = id
+ debouncedIDBSave(conversation)
return id
}
@@ -135,6 +188,7 @@ export const useChatStore = defineStore('chat', () => {
conv.title = message.content.slice(0, 60) + (message.content.length > 60 ? '...' : '')
}
+ debouncedIDBSave(conv)
return msg
}
@@ -143,13 +197,17 @@ export const useChatStore = defineStore('chat', () => {
if (!conv || conv.messages.length === 0) return
const last = conv.messages[conv.messages.length - 1]
last.content += text
+ debouncedIDBSave(conv)
}
function setMessageWebResults(conversationId: string, messageId: string, results: WebSearchResult[]) {
const conv = conversations.value.get(conversationId)
if (!conv) return
const msg = conv.messages.find((m: { id: string }) => m.id === messageId)
- if (msg) msg.webResults = results
+ if (msg) {
+ msg.webResults = results
+ debouncedIDBSave(conv)
+ }
}
function switchSide() {
@@ -168,6 +226,7 @@ export const useChatStore = defineStore('chat', () => {
function deleteConversation(id: string) {
conversations.value.delete(id)
+ if (useIDB) idbDelete(id).catch(() => {})
if (activeConversationId.value === id) {
const remaining = conversationList.value
activeConversationId.value = remaining.length > 0 ? remaining[0].id : null