50 lines
1.2 KiB
TypeScript
50 lines
1.2 KiB
TypeScript
"use client";
|
|
import { create } from "zustand";
|
|
import { auth } from "@/lib/firebase";
|
|
import { User, onAuthStateChanged, signOut, getIdToken } from "firebase/auth";
|
|
|
|
interface AuthState {
|
|
user: User | null;
|
|
token: string | null;
|
|
loading: boolean;
|
|
setUser: (user: User | null) => void;
|
|
setToken: (token: string | null) => void;
|
|
setLoading: (loading: boolean) => void;
|
|
fetchToken: (user: User) => Promise<void>;
|
|
logout: () => Promise<void>;
|
|
}
|
|
|
|
export const useAuthStore = create<AuthState>((set) => ({
|
|
user: null,
|
|
token: null,
|
|
loading: true,
|
|
|
|
setUser: (user) => set({ user }),
|
|
setToken: (token) => set({ token }),
|
|
setLoading: (loading) => set({ loading }),
|
|
|
|
fetchToken: async (user) => {
|
|
const idToken = await getIdToken(user, true);
|
|
set({ token: idToken });
|
|
},
|
|
|
|
logout: async () => {
|
|
await signOut(auth);
|
|
set({ user: null, token: null, loading: false });
|
|
},
|
|
}));
|
|
|
|
// ✅ Firebase observer
|
|
onAuthStateChanged(auth, async (user) => {
|
|
const store = useAuthStore.getState();
|
|
|
|
if (user) {
|
|
store.setUser(user);
|
|
await store.fetchToken(user);
|
|
store.setLoading(false);
|
|
} else {
|
|
store.setUser(null);
|
|
store.setToken(null);
|
|
store.setLoading(false);
|
|
}
|
|
});
|