61 lines
1.8 KiB
TypeScript
61 lines
1.8 KiB
TypeScript
import { create } from 'zustand';
|
||
import { persist, createJSONStorage } from 'zustand/middleware';
|
||
import type { AdminInfo } from '@/lib/types';
|
||
|
||
const TOKEN_COOKIE = process.env.NEXT_PUBLIC_TOKEN_KEY ?? 'admin_token';
|
||
const TOKEN_LS_KEY = TOKEN_COOKIE; // 文档要求：JWT 仅前端 localStorage 存储
|
||
|
||
/** 把 token 同步到 Cookie，便于 Edge middleware 在 SSR 前拦截 */
|
||
function syncCookie(token: string | null): void {
|
||
if (typeof document === 'undefined') return;
|
||
if (token) {
|
||
const expires = new Date(Date.now() + 7 * 24 * 60 * 60 * 1000).toUTCString();
|
||
document.cookie = `${TOKEN_COOKIE}=${encodeURIComponent(token)}; expires=${expires}; path=/; SameSite=Lax`;
|
||
} else {
|
||
document.cookie = `${TOKEN_COOKIE}=; expires=Thu, 01 Jan 1970 00:00:00 GMT; path=/;`;
|
||
}
|
||
}
|
||
|
||
/** 同步到 localStorage，axios 拦截器从这里取 */
|
||
function syncLocalStorage(token: string | null): void {
|
||
if (typeof window === 'undefined') return;
|
||
if (token) {
|
||
window.localStorage.setItem(TOKEN_LS_KEY, token);
|
||
} else {
|
||
window.localStorage.removeItem(TOKEN_LS_KEY);
|
||
}
|
||
}
|
||
|
||
interface AdminState {
|
||
token: string | null;
|
||
admin: AdminInfo | null;
|
||
setLogin: (token: string, admin: AdminInfo) => void;
|
||
logout: () => void;
|
||
hasAuth: () => boolean;
|
||
}
|
||
|
||
export const useAdminStore = create<AdminState>()(
|
||
persist(
|
||
(set, get) => ({
|
||
token: null,
|
||
admin: null,
|
||
setLogin: (token, admin) => {
|
||
syncLocalStorage(token);
|
||
syncCookie(token);
|
||
set({ token, admin });
|
||
},
|
||
logout: () => {
|
||
syncLocalStorage(null);
|
||
syncCookie(null);
|
||
set({ token: null, admin: null });
|
||
},
|
||
hasAuth: () => Boolean(get().token),
|
||
}),
|
||
{
|
||
name: 'corp_admin_session',
|
||
storage: createJSONStorage(() => localStorage),
|
||
partialize: (s) => ({ token: s.token, admin: s.admin }),
|
||
},
|
||
),
|
||
);
|