From 23054d7750ce01a550b2d4aa6a8a49b9732809c3 Mon Sep 17 00:00:00 2001
From: xlsea <m@xlsea.cn>
Date: Thu, 24 Apr 2025 21:58:07 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=96=B0=E5=A2=9E=E7=A7=9F=E6=88=B7?=
 =?UTF-8?q?=E4=B8=8B=E6=8B=89=E7=BB=84=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/custom/tenant-select.vue       | 40 +++++++++++++++++++
 src/layouts/modules/global-header/index.vue   | 11 +++++
 src/service/api/system/tenant.ts              |  8 ++++
 src/typings/components.d.ts                   |  1 +
 src/typings/storage.d.ts                      |  2 +
 .../_builtin/login/modules/pwd-login.vue      |  2 +-
 src/views/_builtin/login/modules/register.vue |  2 +-
 7 files changed, 64 insertions(+), 2 deletions(-)
 create mode 100644 src/components/custom/tenant-select.vue

diff --git a/src/components/custom/tenant-select.vue b/src/components/custom/tenant-select.vue
new file mode 100644
index 00000000..84b846ae
--- /dev/null
+++ b/src/components/custom/tenant-select.vue
@@ -0,0 +1,40 @@
+<script setup lang="ts">
+import { ref } from 'vue';
+import type { SelectOption } from 'naive-ui';
+import { useLoading } from '@sa/hooks';
+import { fetchTenantList } from '@/service/api';
+
+defineOptions({ name: 'TenantSelect' });
+
+const value = defineModel<CommonType.IdType>('value', { required: false, default: '000000' });
+const enabled = defineModel<boolean>('enabled', { required: false, default: false });
+
+const tenantOption = ref<SelectOption[]>([]);
+const { loading, startLoading, endLoading } = useLoading();
+
+async function handleFetchTenantList() {
+  startLoading();
+  const { data, error } = await fetchTenantList();
+  if (error) return;
+  enabled.value = data.tenantEnabled;
+  tenantOption.value = data.voList.map(tenant => {
+    return {
+      label: tenant.companyName,
+      value: tenant.tenantId
+    };
+  });
+  endLoading();
+}
+
+handleFetchTenantList();
+</script>
+
+<template>
+  <NSelect
+    v-if="enabled"
+    v-model:value="value"
+    placeholder="请选择/输入公司名称"
+    :options="tenantOption"
+    :loading="loading"
+  />
+</template>
diff --git a/src/layouts/modules/global-header/index.vue b/src/layouts/modules/global-header/index.vue
index 592048db..093c2154 100644
--- a/src/layouts/modules/global-header/index.vue
+++ b/src/layouts/modules/global-header/index.vue
@@ -1,7 +1,10 @@
 <script setup lang="ts">
+import { ref, watch } from 'vue';
 import { useFullscreen } from '@vueuse/core';
 import { GLOBAL_HEADER_MENU_ID } from '@/constants/app';
+import { fetchChangeTenant } from '@/service/api/system/tenant';
 import { useAppStore } from '@/store/modules/app';
+import { useAuthStore } from '@/store/modules/auth';
 import { useThemeStore } from '@/store/modules/theme';
 import GlobalLogo from '../global-logo/index.vue';
 import GlobalBreadcrumb from '../global-breadcrumb/index.vue';
@@ -25,8 +28,15 @@ interface Props {
 defineProps<Props>();
 
 const appStore = useAppStore();
+const authStore = useAuthStore();
 const themeStore = useThemeStore();
 const { isFullscreen, toggle } = useFullscreen();
+
+const tenantId = ref<CommonType.IdType>(authStore.userInfo?.user?.tenantId || '000000');
+
+watch(tenantId, async () => {
+  await fetchChangeTenant(tenantId.value);
+});
 </script>
 
 <template>
@@ -38,6 +48,7 @@ const { isFullscreen, toggle } = useFullscreen();
       <GlobalBreadcrumb v-if="!appStore.isMobile" class="ml-12px" />
     </div>
     <div class="h-full flex-y-center justify-end">
+      <TenantSelect v-if="authStore.userInfo?.user?.userId === 1" class="mr-12px w-150px" />
       <GlobalSearch />
       <FullScreen v-if="!appStore.isMobile" :full="isFullscreen" @click="toggle" />
       <LangSwitch
diff --git a/src/service/api/system/tenant.ts b/src/service/api/system/tenant.ts
index 2ebdaf03..ad556f5d 100644
--- a/src/service/api/system/tenant.ts
+++ b/src/service/api/system/tenant.ts
@@ -46,3 +46,11 @@ export function fetchSyncTenantDict() {
     method: 'get'
   });
 }
+
+/** 动态切换租户 */
+export function fetchChangeTenant(tenantId: CommonType.IdType) {
+  return request<boolean>({
+    url: `/system/tenant/dynamic/${tenantId}`,
+    method: 'get'
+  });
+}
diff --git a/src/typings/components.d.ts b/src/typings/components.d.ts
index d8e8b9a2..ae66ac78 100644
--- a/src/typings/components.d.ts
+++ b/src/typings/components.d.ts
@@ -141,6 +141,7 @@ declare module 'vue' {
     TableHeaderOperation: typeof import('./../components/advanced/table-header-operation.vue')['default']
     TableRowCheckAlert: typeof import('./../components/advanced/table-row-check-alert.vue')['default']
     TableSiderLayout: typeof import('./../components/advanced/table-sider-layout.vue')['default']
+    TenantSelect: typeof import('./../components/custom/tenant-select.vue')['default']
     ThemeSchemaSwitch: typeof import('./../components/common/theme-schema-switch.vue')['default']
     WaveBg: typeof import('./../components/custom/wave-bg.vue')['default']
   }
diff --git a/src/typings/storage.d.ts b/src/typings/storage.d.ts
index f80663c4..688c9eda 100644
--- a/src/typings/storage.d.ts
+++ b/src/typings/storage.d.ts
@@ -44,5 +44,7 @@ declare namespace StorageType {
     };
     /** The login form rember */
     loginRember: Api.Auth.PwdLoginForm;
+    /** The tenant id */
+    tenantId: CommonType.IdType;
   }
 }
diff --git a/src/views/_builtin/login/modules/pwd-login.vue b/src/views/_builtin/login/modules/pwd-login.vue
index cd9667d5..c43855dd 100644
--- a/src/views/_builtin/login/modules/pwd-login.vue
+++ b/src/views/_builtin/login/modules/pwd-login.vue
@@ -122,7 +122,7 @@ async function handleSocialLogin(type: Api.System.SocialSource) {
 <template>
   <NForm ref="formRef" :model="model" :rules="rules" size="large" :show-label="false" @keyup.enter="handleSubmit">
     <NFormItem v-if="tenantEnabled" path="tenantId">
-      <NSelect v-model:value="model.tenantId" placeholder="请选择/输入公司名称" :options="tenantOption" />
+      <TenantSelect v-model:value="model.tenantId" :enabled="tenantEnabled" />
     </NFormItem>
     <NFormItem path="username">
       <NInput v-model:value="model.username" :placeholder="$t('page.login.common.userNamePlaceholder')" />
diff --git a/src/views/_builtin/login/modules/register.vue b/src/views/_builtin/login/modules/register.vue
index ae86a39b..87a5f7fa 100644
--- a/src/views/_builtin/login/modules/register.vue
+++ b/src/views/_builtin/login/modules/register.vue
@@ -95,7 +95,7 @@ handleFetchCaptchaCode();
 <template>
   <NForm ref="formRef" :model="model" :rules="rules" size="large" :show-label="false" @keyup.enter="handleSubmit">
     <NFormItem v-if="tenantEnabled" path="tenantId">
-      <NSelect v-model:value="model.tenantId" placeholder="请选择/输入公司名称" :options="tenantOption" />
+      <TenantSelect v-model:value="model.tenantId" :enabled="tenantEnabled" />
     </NFormItem>
     <NFormItem path="username">
       <NInput v-model:value="model.username" :placeholder="$t('page.login.common.userNamePlaceholder')" />