60 lines
1.4 KiB
TypeScript
60 lines
1.4 KiB
TypeScript
'use client';
|
|
|
|
import { create } from 'zustand';
|
|
import { registerTenantHeaderBridge } from '@/lib/api/client';
|
|
|
|
const KEY = 'smart_tenant_id';
|
|
|
|
type TenantState = {
|
|
/** 显式选择的租户；null 表示不额外传 X-Tenant-ID，由 Bearer 默认租户生效 */
|
|
tenantId: string | null;
|
|
setTenantId: (id: string | null) => void;
|
|
/** 从当前用户资料同步（登录后首次） */
|
|
hydrateFromUserTenant: (tenantId: string | undefined | null) => void;
|
|
};
|
|
|
|
function load(): string | null {
|
|
if (typeof window === 'undefined') {
|
|
return null;
|
|
}
|
|
try {
|
|
const raw = sessionStorage.getItem(KEY);
|
|
if (raw === '' || raw === 'null') {
|
|
return null;
|
|
}
|
|
return raw;
|
|
} catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
export const useTenantStore = create<TenantState>((set, get) => ({
|
|
tenantId: typeof window !== 'undefined' ? load() : null,
|
|
|
|
setTenantId: (id) => {
|
|
set({ tenantId: id });
|
|
if (typeof window !== 'undefined') {
|
|
if (id) {
|
|
sessionStorage.setItem(KEY, id);
|
|
} else {
|
|
sessionStorage.removeItem(KEY);
|
|
}
|
|
}
|
|
},
|
|
|
|
hydrateFromUserTenant: (tid) => {
|
|
if (!tid) {
|
|
return;
|
|
}
|
|
const cur = get().tenantId;
|
|
if (cur == null || cur === '') {
|
|
set({ tenantId: tid });
|
|
if (typeof window !== 'undefined') {
|
|
sessionStorage.setItem(KEY, tid);
|
|
}
|
|
}
|
|
},
|
|
}));
|
|
|
|
registerTenantHeaderBridge(() => useTenantStore.getState().tenantId);
|