136 lines
3.3 KiB
TypeScript
136 lines
3.3 KiB
TypeScript
import { store } from "@/store";
|
|
import TenantAPI, { type TenantInfo } from "@/api/system/tenant-api";
|
|
|
|
const TENANT_ID_KEY = "current_tenant_id";
|
|
const TENANT_INFO_KEY = "current_tenant_info";
|
|
|
|
/**
|
|
* 租户 Store
|
|
*/
|
|
export const useTenantStore = defineStore("tenant", () => {
|
|
// 当前租户ID
|
|
const currentTenantId = ref<number | null>(null);
|
|
// 当前租户信息
|
|
const currentTenant = ref<TenantInfo | null>(null);
|
|
// 用户可访问的租户列表
|
|
const tenantList = ref<TenantInfo[]>([]);
|
|
|
|
/**
|
|
* 初始化租户信息
|
|
* 从 localStorage 恢复上次使用的租户
|
|
*/
|
|
function initTenant() {
|
|
const savedTenantId = localStorage.getItem(TENANT_ID_KEY);
|
|
const savedTenantInfo = localStorage.getItem(TENANT_INFO_KEY);
|
|
|
|
if (savedTenantId) {
|
|
currentTenantId.value = Number(savedTenantId);
|
|
}
|
|
|
|
if (savedTenantInfo) {
|
|
try {
|
|
currentTenant.value = JSON.parse(savedTenantInfo);
|
|
} catch (e) {
|
|
console.error("解析租户信息失败", e);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取用户租户列表
|
|
*/
|
|
function fetchTenantList() {
|
|
return new Promise<TenantInfo[]>((resolve, reject) => {
|
|
TenantAPI.getTenantList()
|
|
.then((data) => {
|
|
tenantList.value = data || [];
|
|
resolve(data || []);
|
|
})
|
|
.catch((error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 设置当前租户
|
|
*
|
|
* @param tenant 租户信息
|
|
*/
|
|
function setCurrentTenant(tenant: TenantInfo) {
|
|
currentTenantId.value = tenant.id;
|
|
currentTenant.value = tenant;
|
|
|
|
// 保存到 localStorage
|
|
localStorage.setItem(TENANT_ID_KEY, String(tenant.id));
|
|
localStorage.setItem(TENANT_INFO_KEY, JSON.stringify(tenant));
|
|
}
|
|
|
|
/**
|
|
* 切换租户
|
|
*
|
|
* @param tenantId 目标租户ID
|
|
*/
|
|
function switchTenant(tenantId: number) {
|
|
return new Promise<void>((resolve, reject) => {
|
|
TenantAPI.switchTenant(tenantId)
|
|
.then((tenantInfo) => {
|
|
// 后端返回切换后的租户信息
|
|
if (tenantInfo) {
|
|
setCurrentTenant(tenantInfo);
|
|
} else {
|
|
// 如果后端未返回，从租户列表中找到对应的租户信息
|
|
const tenant = tenantList.value.find((t) => t.id === tenantId);
|
|
if (tenant) {
|
|
setCurrentTenant(tenant);
|
|
} else {
|
|
// 如果列表中没有，重新获取租户信息
|
|
TenantAPI.getCurrentTenant()
|
|
.then((info) => {
|
|
if (info) {
|
|
setCurrentTenant(info);
|
|
}
|
|
})
|
|
.catch(console.error);
|
|
}
|
|
}
|
|
resolve();
|
|
})
|
|
.catch((error) => {
|
|
reject(error);
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* 清除租户信息
|
|
*/
|
|
function clearTenant() {
|
|
currentTenantId.value = null;
|
|
currentTenant.value = null;
|
|
tenantList.value = [];
|
|
localStorage.removeItem(TENANT_ID_KEY);
|
|
localStorage.removeItem(TENANT_INFO_KEY);
|
|
}
|
|
|
|
// 初始化
|
|
initTenant();
|
|
|
|
return {
|
|
currentTenantId,
|
|
currentTenant,
|
|
tenantList,
|
|
fetchTenantList,
|
|
setCurrentTenant,
|
|
switchTenant,
|
|
clearTenant,
|
|
};
|
|
});
|
|
|
|
/**
|
|
* 在组件外部使用 TenantStore 的钩子函数
|
|
*/
|
|
export function useTenantStoreHook() {
|
|
return useTenantStore(store);
|
|
}
|