first commit

Co-Authored-By: Codex <noreply@openai.com>
2026-08-05 08:54:23 +08:00 · 2026-05-19 07:53:53 +08:00
commit 472bf8b732
133 changed files with 16869 additions and 0 deletions
@@ -0,0 +1,78 @@
+"use client";
+
+import { create } from "zustand";
+import { persist } from "zustand/middleware";
+
+import { AUTH_TOKEN_KEY, fetchCurrentUser, login, register, type AuthPayload, type AuthUser } from "@/services/api/auth";
+
+type UserStore = {
+  token: string;
+  user: AuthUser | null;
+  isReady: boolean;
+  isLoading: boolean;
+  setSession: (token: string, user: AuthUser) => void;
+  clearSession: () => void;
+  hydrateUser: () => Promise<void>;
+  login: (payload: AuthPayload) => Promise<AuthUser>;
+  register: (payload: AuthPayload) => Promise<AuthUser>;
+};
+
+export const useUserStore = create<UserStore>()(
+  persist(
+    (set, get) => ({
+      token: "",
+      user: null,
+      isReady: false,
+      isLoading: false,
+      setSession: (token, user) => set({ token, user, isReady: true }),
+      clearSession: () => set({ token: "", user: null, isReady: true }),
+      hydrateUser: async () => {
+        const token = get().token;
+        if (!token) {
+          set({ user: null, isReady: true });
+          return;
+        }
+        set({ isLoading: true });
+        try {
+          const user = await fetchCurrentUser(token);
+          if (user.role === "guest") {
+            set({ token: "", user: null, isReady: true, isLoading: false });
+            return;
+          }
+          set({ user, isReady: true, isLoading: false });
+        } catch {
+          set({ token: "", user: null, isReady: true, isLoading: false });
+        }
+      },
+      login: async (payload) => {
+        set({ isLoading: true });
+        try {
+          const session = await login(payload);
+          set({ token: session.token, user: session.user, isReady: true, isLoading: false });
+          return session.user;
+        } catch (error) {
+          set({ isLoading: false });
+          throw error;
+        }
+      },
+      register: async (payload) => {
+        set({ isLoading: true });
+        try {
+          const session = await register(payload);
+          set({ token: session.token, user: session.user, isReady: true, isLoading: false });
+          return session.user;
+        } catch (error) {
+          set({ isLoading: false });
+          throw error;
+        }
+      },
+    }),
+    {
+      name: AUTH_TOKEN_KEY,
+      partialize: (state) => ({ token: state.token }),
+      onRehydrateStorage: () => (state) => {
+        if (state) state.isReady = false;
+      },
+    },
+  ),
+);