Implement logic

frontend/src/ui/state.ts

@@ -1,13 +1,121 @@
 import { create } from "zustand";
+import type { Message, User, UserProfile } from "../core/types";
 
 type Page = "login" | "register" | "chat"
 
+interface ChatState {
+    messages: Message[];
+    currentChat: string;
+    activeTab: "chats" | "channels" | "contacts" | "dms";
+    dmUsers: User[];
+    activeDm: { userId: number; username: string; publicKey: string | null } | null;
+}
+
+interface UserState {
+    currentUser: User | null;
+    authToken: string | null;
+}
+
 interface AppState {
     currentPage: Page;
     setCurrentPage: (page: Page) => void;
+    
+    // Chat state
+    chat: ChatState;
+    addMessage: (message: Message) => void;
+    updateMessage: (messageId: number, updatedMessage: Partial<Message>) => void;
+    removeMessage: (messageId: number) => void;
+    setCurrentChat: (chat: string) => void;
+    setActiveTab: (tab: ChatState["activeTab"]) => void;
+    setDmUsers: (users: User[]) => void;
+    setActiveDm: (dm: ChatState["activeDm"]) => void;
+    clearMessages: () => void;
+    
+    // User state
+    user: UserState;
+    setUser: (token: string, user: User) => void;
+    logout: () => void;
 }
 
-export const useAppState = create<AppState>((set) => ({
+export const useAppState = create<AppState>((set, get) => ({
     currentPage: "login", // default page
-    setCurrentPage: (page: Page) => set({ currentPage: page })
+    setCurrentPage: (page: Page) => set({ currentPage: page }),
+    
+    // Chat state
+    chat: {
+        messages: [],
+        currentChat: "Общий чат",
+        activeTab: "chats",
+        dmUsers: [],
+        activeDm: null
+    },
+    addMessage: (message: Message) => set((state) => ({
+        chat: {
+            ...state.chat,
+            messages: [...state.chat.messages, message]
+        }
+    })),
+    updateMessage: (messageId: number, updatedMessage: Partial<Message>) => set((state) => ({
+        chat: {
+            ...state.chat,
+            messages: state.chat.messages.map(msg => 
+                msg.id === messageId ? { ...msg, ...updatedMessage } : msg
+            )
+        }
+    })),
+    removeMessage: (messageId: number) => set((state) => ({
+        chat: {
+            ...state.chat,
+            messages: state.chat.messages.filter(msg => msg.id !== messageId)
+        }
+    })),
+    clearMessages: () => set((state) => ({
+        chat: {
+            ...state.chat,
+            messages: []
+        }
+    })),
+    setCurrentChat: (chat: string) => set((state) => ({
+        chat: {
+            ...state.chat,
+            currentChat: chat
+        }
+    })),
+    setActiveTab: (tab: ChatState["activeTab"]) => set((state) => ({
+        chat: {
+            ...state.chat,
+            activeTab: tab
+        }
+    })),
+    setDmUsers: (users: User[]) => set((state) => ({
+        chat: {
+            ...state.chat,
+            dmUsers: users
+        }
+    })),
+    setActiveDm: (dm: ChatState["activeDm"]) => set((state) => ({
+        chat: {
+            ...state.chat,
+            activeDm: dm
+        }
+    })),
+    
+    // User state
+    user: {
+        currentUser: null,
+        authToken: null
+    },
+    setUser: (token: string, user: User) => set((state) => ({
+        user: {
+            currentUser: user,
+            authToken: token
+        }
+    })),
+    logout: () => set((state) => ({
+        user: {
+            currentUser: null,
+            authToken: null
+        },
+        currentPage: "login"
+    }))
 }));