新的提交

src/stores/chatStore.ts

@@ -0,0 +1,116 @@
+import { create } from 'zustand'
+import type { ChatSession, Message, Contact } from '../types/models'
+
+export interface ChatState {
+  // 连接状态
+  isConnected: boolean
+  isConnecting: boolean
+  connectionError: string | null
+  
+  // 会话列表
+  sessions: ChatSession[]
+  filteredSessions: ChatSession[]
+  currentSessionId: string | null
+  isLoadingSessions: boolean
+  
+  // 消息
+  messages: Message[]
+  isLoadingMessages: boolean
+  isLoadingMore: boolean
+  hasMoreMessages: boolean
+  
+  // 联系人缓存
+  contacts: Map<string, Contact>
+  
+  // 搜索
+  searchKeyword: string
+  
+  // 操作
+  setConnected: (connected: boolean) => void
+  setConnecting: (connecting: boolean) => void
+  setConnectionError: (error: string | null) => void
+  setSessions: (sessions: ChatSession[]) => void
+  setFilteredSessions: (sessions: ChatSession[]) => void
+  setCurrentSession: (sessionId: string | null) => void
+  setLoadingSessions: (loading: boolean) => void
+  setMessages: (messages: Message[]) => void
+  appendMessages: (messages: Message[], prepend?: boolean) => void
+  setLoadingMessages: (loading: boolean) => void
+  setLoadingMore: (loading: boolean) => void
+  setHasMoreMessages: (hasMore: boolean) => void
+  setContacts: (contacts: Contact[]) => void
+  addContact: (contact: Contact) => void
+  setSearchKeyword: (keyword: string) => void
+  reset: () => void
+}
+
+export const useChatStore = create<ChatState>((set, get) => ({
+  isConnected: false,
+  isConnecting: false,
+  connectionError: null,
+  sessions: [],
+  filteredSessions: [],
+  currentSessionId: null,
+  isLoadingSessions: false,
+  messages: [],
+  isLoadingMessages: false,
+  isLoadingMore: false,
+  hasMoreMessages: true,
+  contacts: new Map(),
+  searchKeyword: '',
+
+  setConnected: (connected) => set({ isConnected: connected }),
+  setConnecting: (connecting) => set({ isConnecting: connecting }),
+  setConnectionError: (error) => set({ connectionError: error }),
+  
+  setSessions: (sessions) => set({ sessions, filteredSessions: sessions }),
+  setFilteredSessions: (sessions) => set({ filteredSessions: sessions }),
+  
+  setCurrentSession: (sessionId) => set({ 
+    currentSessionId: sessionId,
+    messages: [],
+    hasMoreMessages: true
+  }),
+  
+  setLoadingSessions: (loading) => set({ isLoadingSessions: loading }),
+  
+  setMessages: (messages) => set({ messages }),
+  
+  appendMessages: (newMessages, prepend = false) => set((state) => ({
+    messages: prepend 
+      ? [...newMessages, ...state.messages]
+      : [...state.messages, ...newMessages]
+  })),
+  
+  setLoadingMessages: (loading) => set({ isLoadingMessages: loading }),
+  setLoadingMore: (loading) => set({ isLoadingMore: loading }),
+  setHasMoreMessages: (hasMore) => set({ hasMoreMessages: hasMore }),
+  
+  setContacts: (contacts) => set({ 
+    contacts: new Map(contacts.map(c => [c.username, c])) 
+  }),
+  
+  addContact: (contact) => set((state) => {
+    const newContacts = new Map(state.contacts)
+    newContacts.set(contact.username, contact)
+    return { contacts: newContacts }
+  }),
+  
+  setSearchKeyword: (keyword) => set({ searchKeyword: keyword }),
+  
+  reset: () => set({
+    isConnected: false,
+    isConnecting: false,
+    connectionError: null,
+    sessions: [],
+    filteredSessions: [],
+    currentSessionId: null,
+    isLoadingSessions: false,
+    messages: [],
+    isLoadingMessages: false,
+    isLoadingMore: false,
+    hasMoreMessages: true,
+    contacts: new Map(),
+    searchKeyword: ''
+  })
+}))