From 64ee81c327bbb31180d53a7496f71146b9987993 Mon Sep 17 00:00:00 2001
From: xingyu4j <xingyu4j@vip.qq.com>
Date: Tue, 6 May 2025 14:11:46 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E5=88=87=E6=8D=A2?=
 =?UTF-8?q?=E7=A7=9F=E6=88=B7=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/web-antd/src/api/request.ts              |  8 +++
 apps/web-antd/src/api/system/tenant/index.ts  |  7 +++
 apps/web-antd/src/layouts/basic.vue           |  4 ++
 .../layouts/components/tenant-dropdown.vue    | 63 +++++++++++++++++++
 packages/stores/src/modules/access.ts         | 21 +++++++
 5 files changed, 103 insertions(+)
 create mode 100644 apps/web-antd/src/layouts/components/tenant-dropdown.vue

diff --git a/apps/web-antd/src/api/request.ts b/apps/web-antd/src/api/request.ts
index 5b5c15fb..7f78987b 100644
--- a/apps/web-antd/src/api/request.ts
+++ b/apps/web-antd/src/api/request.ts
@@ -80,6 +80,10 @@ function createRequestClient(baseURL: string, options?: RequestClientOptions) {
       config.headers['tenant-id'] = tenantEnable
         ? accessStore.tenantId
         : undefined;
+      // 只有登录时，才设置 visit-tenant-id 访问租户
+      config.headers['visit-tenant-id'] = tenantEnable
+        ? accessStore.visitTenantId
+        : undefined;
       return config;
     },
   });
@@ -136,6 +140,10 @@ baseRequestClient.addRequestInterceptor({
     config.headers['tenant-id'] = tenantEnable
       ? accessStore.tenantId
       : undefined;
+    // 只有登录时，才设置 visit-tenant-id 访问租户
+    config.headers['visit-tenant-id'] = tenantEnable
+      ? accessStore.visitTenantId
+      : undefined;
     return config;
   },
 });
diff --git a/apps/web-antd/src/api/system/tenant/index.ts b/apps/web-antd/src/api/system/tenant/index.ts
index cefacc51..3bed9249 100644
--- a/apps/web-antd/src/api/system/tenant/index.ts
+++ b/apps/web-antd/src/api/system/tenant/index.ts
@@ -39,6 +39,13 @@ export function getTenant(id: number) {
   );
 }
 
+/** 获取租户精简信息列表 */
+export function getTenantList() {
+  return requestClient.get<SystemTenantApi.Tenant[]>(
+    '/system/tenant/simple-list',
+  );
+}
+
 /** 新增租户 */
 export function createTenant(data: SystemTenantApi.Tenant) {
   return requestClient.post('/system/tenant/create', data);
diff --git a/apps/web-antd/src/layouts/basic.vue b/apps/web-antd/src/layouts/basic.vue
index 86f44466..2777f4a3 100644
--- a/apps/web-antd/src/layouts/basic.vue
+++ b/apps/web-antd/src/layouts/basic.vue
@@ -34,6 +34,7 @@ import { useAuthStore } from '#/store';
 import LoginForm from '#/views/_core/authentication/login.vue';
 
 import Help from './components/help.vue';
+import TenantDropdown from './components/tenant-dropdown.vue';
 
 const userStore = useUserStore();
 const authStore = useAuthStore();
@@ -202,6 +203,9 @@ watch(
         @read="handleNotificationRead"
       />
     </template>
+    <template #header-right-1>
+      <TenantDropdown class="w-30 mr-2" />
+    </template>
     <template #extra>
       <AuthenticationLoginExpiredModal
         v-model:open="accessStore.loginExpired"
diff --git a/apps/web-antd/src/layouts/components/tenant-dropdown.vue b/apps/web-antd/src/layouts/components/tenant-dropdown.vue
new file mode 100644
index 00000000..013958d4
--- /dev/null
+++ b/apps/web-antd/src/layouts/components/tenant-dropdown.vue
@@ -0,0 +1,63 @@
+<script lang="ts" setup>
+import { computed, onMounted, ref } from 'vue';
+
+import { useAccess } from '@vben/access';
+import { isTenantEnable, useRefresh } from '@vben/hooks';
+import { useAccessStore } from '@vben/stores';
+
+import { Button, Dropdown, Menu } from 'ant-design-vue';
+
+import { getTenantList } from '#/api/system/tenant';
+
+const { refresh } = useRefresh();
+
+const { hasAccessByCodes } = useAccess();
+const accessStore = useAccessStore();
+
+const tenantEnable = isTenantEnable();
+
+const visitTenantList = computed(() => {
+  if (tenantEnable) {
+    const list = accessStore.visitTenantId.map((item) => ({
+      label: item.name,
+      id: item.id,
+    }));
+    return list;
+  }
+  return [];
+});
+
+const tenant = ref<string>(
+  visitTenantList.value.find((item) => item.id === accessStore.tenantId)
+    ?.label || '切换租户',
+);
+
+function handleClick(id: number | undefined) {
+  if (id) {
+    accessStore.setTenantId(id);
+    refresh();
+    window.location.reload();
+  }
+}
+
+onMounted(async () => {
+  if (tenantEnable) {
+    const resp = await getTenantList();
+    accessStore.setVisitTenantId(resp);
+  }
+});
+</script>
+<template>
+  <Dropdown v-if="tenantEnable && hasAccessByCodes(['system:tenant:visit'])">
+    <template #overlay>
+      <Menu>
+        <template v-for="item in visitTenantList" :key="item.key">
+          <Menu.Item @click="handleClick(item.id)">
+            {{ item.label }}
+          </Menu.Item>
+        </template>
+      </Menu>
+    </template>
+    <Button> {{ tenant }} </Button>
+  </Dropdown>
+</template>
diff --git a/packages/stores/src/modules/access.ts b/packages/stores/src/modules/access.ts
index 54c2b95a..b8183f01 100644
--- a/packages/stores/src/modules/access.ts
+++ b/packages/stores/src/modules/access.ts
@@ -6,6 +6,18 @@ import { acceptHMRUpdate, defineStore } from 'pinia';
 
 type AccessToken = null | string;
 
+type VisitTenantId = {
+  accountCount: number;
+  contactMobile: string;
+  contactName: string;
+  expireTime: Date;
+  id?: number;
+  name: string;
+  packageId: number;
+  status: number;
+  website: string;
+};
+
 interface AccessState {
   /**
    * 权限码
@@ -47,6 +59,10 @@ interface AccessState {
    * 登录租户编号
    */
   tenantId: null | number;
+  /**
+   * 访问租户编号
+   */
+  visitTenantId: VisitTenantId[];
 }
 
 /**
@@ -101,6 +117,9 @@ export const useAccessStore = defineStore('core-access', {
     setTenantId(tenantId: null | number) {
       this.tenantId = tenantId;
     },
+    setVisitTenantId(visitTenantId: VisitTenantId[]) {
+      this.visitTenantId = visitTenantId;
+    },
     unlockScreen() {
       this.isLockScreen = false;
       this.lockScreenPassword = undefined;
@@ -113,6 +132,7 @@ export const useAccessStore = defineStore('core-access', {
       'refreshToken',
       'accessCodes',
       'tenantId',
+      'visitTenantId',
       'isLockScreen',
       'lockScreenPassword',
     ],
@@ -128,6 +148,7 @@ export const useAccessStore = defineStore('core-access', {
     loginExpired: false,
     refreshToken: null,
     tenantId: null,
+    visitTenantId: [],
   }),
 });