@unif/react-native-chat-sdk 0.4.2 → 0.5.4

package/src/hooks/useXConversations.ts

@@ -6,6 +6,7 @@
 
 import { useState, useCallback, useEffect, useRef } from 'react';
 import { createHistoryStore } from '../stores/historyStore';
+import { getOrCreateChatStore } from '../stores/chatStore';
 import type { ChatStorage } from '../stores/storage';
 import type { ChatMessage } from '../types/message';
 
@@ -23,7 +24,7 @@ export function useXConversations(options: UseXConversationsOptions) {
   const history = historyRef.current;
 
   const [sessions, setSessions] = useState(history.sessions);
-  const [activeId, setActiveId] = useState('');
+  const [activeId, setActiveIdState] = useState('');
 
   // 初始化加载
   useEffect(() => {
@@ -36,16 +37,20 @@ export function useXConversations(options: UseXConversationsOptions) {
     // eslint-disable-next-line react-hooks/exhaustive-deps
   }, []);
 
+  const setActiveId = useCallback((id: string) => {
+    setActiveIdState(id);
+  }, []);
+
   const switchSession = useCallback(
     async (id: string): Promise<ChatMessage[]> => {
-      setActiveId(id);
+      setActiveIdState(id);
       return await history.loadSessionMessages(id);
     },
     [history]
   );
 
   const newSession = useCallback(() => {
-    setActiveId('');
+    setActiveIdState('');
   }, []);
 
   const deleteSession = useCallback(
@@ -53,7 +58,7 @@ export function useXConversations(options: UseXConversationsOptions) {
       await history.deleteSession(id);
       setSessions([...history.sessions]);
       if (activeId === id) {
-        setActiveId('');
+        setActiveIdState('');
       }
     },
     [history, activeId]
@@ -67,6 +72,27 @@ export function useXConversations(options: UseXConversationsOptions) {
     [history]
   );
 
+  const saveSession = useCallback(
+    async (id: string, title: string, messages: ChatMessage[]) => {
+      await history.saveSession(id, title, messages);
+      setSessions([...history.sessions]);
+    },
+    [history]
+  );
+
+  const loadSessionMessages = useCallback(
+    async (id: string): Promise<ChatMessage[]> => {
+      const msgs = await history.loadSessionMessages(id);
+      // 将消息加载到对应的内存 store
+      if (msgs.length > 0) {
+        const store = getOrCreateChatStore(id);
+        store.getState().loadSession(msgs);
+      }
+      return msgs;
+    },
+    [history]
+  );
+
   const updateSession = useCallback(
     (id: string, data: Partial<{ title: string }>) => {
       const session = history.sessions.find((s) => s.id === id);
@@ -81,10 +107,13 @@ export function useXConversations(options: UseXConversationsOptions) {
   return {
     sessions,
     activeId,
+    setActiveId,
     switchSession,
     newSession,
     deleteSession,
     archiveSession,
+    saveSession,
+    loadSessionMessages,
     updateSession,
   };
 }

package/src/index.ts

@@ -47,6 +47,11 @@ export type { DeepSeekChatProviderConfig } from './providers/DeepSeekChatProvide
 
 // Stores
 export type { ChatStorage } from './stores/storage';
+export {
+  getOrCreateChatStore,
+  deleteChatStore,
+  getChatStoreMessages,
+} from './stores/chatStore';
 
 // Hooks
 export { useXChat } from './hooks/useXChat';

package/src/stores/chatStore.ts

@@ -35,6 +35,26 @@ interface ChatState {
   loadSession: (messages: ChatMessage[]) => void;
 }
 
+// 全局 store 管理器 — 多会话内存常驻
+const chatStoreMap = new Map<string, ReturnType<typeof createChatStore>>();
+
+export function getOrCreateChatStore(key: string) {
+  let store = chatStoreMap.get(key);
+  if (!store) {
+    store = createChatStore();
+    chatStoreMap.set(key, store);
+  }
+  return store;
+}
+
+export function deleteChatStore(key: string) {
+  chatStoreMap.delete(key);
+}
+
+export function getChatStoreMessages(key: string) {
+  return chatStoreMap.get(key)?.getState()?.messages;
+}
+
 export const createChatStore = () =>
   create<ChatState>((set, get) => ({
     messages: [],

package/src/stores/historyStore.ts

@@ -24,6 +24,11 @@ export interface HistoryStore {
     title: string,
     messages: ChatMessage[]
   ) => Promise<void>;
+  saveSession: (
+    id: string,
+    title: string,
+    messages: ChatMessage[]
+  ) => Promise<void>;
   loadSessionMessages: (id: string) => Promise<ChatMessage[]>;
   deleteSession: (id: string) => Promise<void>;
   clearAll: () => Promise<void>;
@@ -88,6 +93,42 @@ export function createHistoryStore(storage: ChatStorage): HistoryStore {
       await saveIndex();
     },
 
+    saveSession: async (id, title, messages) => {
+      await loadIndex();
+      if (messages.length === 0) return;
+
+      await storage.setItem(
+        `${MESSAGE_KEY_PREFIX}${id}`,
+        JSON.stringify(messages)
+      );
+
+      const lastMsg = messages[0];
+      const summary: SessionSummary = {
+        id,
+        title,
+        lastMessage: lastMsg?.text?.slice(0, 50) ?? '',
+        timestamp: Date.now(),
+        messageCount: messages.length,
+      };
+
+      const idx = sessions.findIndex((s) => s.id === id);
+      if (idx >= 0) {
+        sessions[idx] = summary;
+      } else {
+        sessions.unshift(summary);
+      }
+
+      // FIFO 淘汰
+      if (sessions.length > MAX_SESSIONS) {
+        const removed = sessions.splice(MAX_SESSIONS);
+        await Promise.all(
+          removed.map((s) => storage.removeItem(`${MESSAGE_KEY_PREFIX}${s.id}`))
+        );
+      }
+
+      await saveIndex();
+    },
+
     loadSessionMessages: async (id) => {
       const raw = await storage.getItem(`${MESSAGE_KEY_PREFIX}${id}`);
       if (!raw) return [];