From 1e29c37036e34477efe89dfac42f913f6f1772c6 Mon Sep 17 00:00:00 2001
From: dap <15891557205@163.com>
Date: Wed, 7 Aug 2024 11:55:13 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=B6=85=E7=BA=A7=E7=AE=A1=E7=90=86?=
 =?UTF-8?q?=E5=91=98=E7=A7=9F=E6=88=B7=E5=88=87=E6=8D=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/web-antd/src/api/system/tenant/index.ts  | 28 ++++++
 .../web-antd/src/api/system/tenant/model.d.ts | 16 ++++
 .../src/components/TenantToggle/index.vue     | 88 +++++++++++++++++++
 .../src/components/TenantToggle/readme.md     |  1 +
 apps/web-antd/src/layouts/basic.vue           |  4 +
 apps/web-antd/src/store/tenant.ts             | 39 ++++++++
 6 files changed, 176 insertions(+)
 create mode 100644 apps/web-antd/src/api/system/tenant/index.ts
 create mode 100644 apps/web-antd/src/api/system/tenant/model.d.ts
 create mode 100644 apps/web-antd/src/components/TenantToggle/index.vue
 create mode 100644 apps/web-antd/src/components/TenantToggle/readme.md
 create mode 100644 apps/web-antd/src/store/tenant.ts

diff --git a/apps/web-antd/src/api/system/tenant/index.ts b/apps/web-antd/src/api/system/tenant/index.ts
new file mode 100644
index 00000000..856d7bc2
--- /dev/null
+++ b/apps/web-antd/src/api/system/tenant/index.ts
@@ -0,0 +1,28 @@
+import { requestClient } from '#/api/request';
+
+enum Api {
+  root = '/system/tenant',
+  tenantDynamic = '/system/tenant/dynamic',
+  tenantDynamicClear = '/system/tenant/dynamic/clear',
+  tenantExport = '/system/tenant/export',
+  tenantList = '/system/tenant/list',
+  tenantStatus = '/system/tenant/changeStatus',
+  tenantSyncPackage = '/system/tenant/syncTenantPackage',
+}
+
+/**
+ * 动态切换租户
+ * @param tenantId 租户ID
+ * @returns void
+ */
+export function tenantDynamicToggle(tenantId: string) {
+  return requestClient.get<void>(`${Api.tenantDynamic}/${tenantId}`);
+}
+
+/**
+ * 清除 动态切换租户
+ * @returns void
+ */
+export function tenantDynamicClear() {
+  return requestClient.get<void>(Api.tenantDynamicClear);
+}
diff --git a/apps/web-antd/src/api/system/tenant/model.d.ts b/apps/web-antd/src/api/system/tenant/model.d.ts
new file mode 100644
index 00000000..309ad587
--- /dev/null
+++ b/apps/web-antd/src/api/system/tenant/model.d.ts
@@ -0,0 +1,16 @@
+export interface Tenant {
+  accountCount: number;
+  address?: string;
+  companyName: string;
+  contactPhone: string;
+  contactUserName: string;
+  domain?: string;
+  expireTime?: string;
+  id: number;
+  intro: string;
+  licenseNumber?: any;
+  packageId?: string;
+  remark?: string;
+  status: string;
+  tenantId: string;
+}
diff --git a/apps/web-antd/src/components/TenantToggle/index.vue b/apps/web-antd/src/components/TenantToggle/index.vue
new file mode 100644
index 00000000..72f8a19f
--- /dev/null
+++ b/apps/web-antd/src/components/TenantToggle/index.vue
@@ -0,0 +1,88 @@
+<script setup lang="ts">
+import { computed, onMounted, ref, unref } from 'vue';
+import { useRouter } from 'vue-router';
+
+import { useAccess } from '@vben/access';
+import { DEFAULT_HOME_PATH } from '@vben/constants';
+import { useTabs } from '@vben/hooks';
+
+import { message, Select } from 'ant-design-vue';
+import { storeToRefs } from 'pinia';
+
+import { tenantDynamicClear, tenantDynamicToggle } from '#/api/system/tenant';
+import { useTenantStore } from '#/store/tenant';
+
+const { hasAccessByRoles } = useAccess();
+
+// 上一次选择的租户
+const lastSelected = ref<string>();
+// 当前选择租户的id
+const selected = ref<string>();
+
+const tenantStore = useTenantStore();
+const { initTenant, setChecked } = tenantStore;
+const { tenantEnable, tenantList } = storeToRefs(tenantStore);
+
+const showToggle = computed<boolean>(() => {
+  // 超级管理员 && 启用租户
+  return hasAccessByRoles(['superadmin']) && unref(tenantEnable);
+});
+
+onMounted(async () => {
+  if (!hasAccessByRoles(['superadmin'])) {
+    return;
+  }
+  await initTenant();
+});
+
+const { closeAllTabs } = useTabs();
+const router = useRouter();
+function close(checked: boolean) {
+  // store设置状态
+  setChecked(checked);
+  // 需要关闭全部标签页
+  closeAllTabs();
+  // 切换完加载首页
+  router.push(DEFAULT_HOME_PATH);
+}
+
+/**
+ * 为什么要用any ide报错😅 实际类型为string
+ */
+async function onSelected(tenantId: any, option: any) {
+  if (unref(lastSelected) === tenantId) {
+    // createMessage.info('选择一致');
+    return;
+  }
+  await tenantDynamicToggle(tenantId);
+  lastSelected.value = tenantId;
+  message.success(`切换当前租户为: ${option.title}`);
+  close(true);
+}
+
+async function onDeselect() {
+  await tenantDynamicClear();
+  message.success('还原为默认租户');
+  lastSelected.value = '';
+  close(false);
+}
+</script>
+
+<template>
+  <div v-if="showToggle" class="mr-[8px] hidden md:block">
+    <Select
+      v-model:value="selected"
+      :field-names="{ label: 'companyName', value: 'tenantId' }"
+      :options="tenantList"
+      allow-clear
+      class="w-60"
+      placeholder="选择租户"
+      @deselect="onDeselect"
+      @select="onSelected"
+    >
+      <template #suffixIcon>
+        <span class="icon-mdi--company"></span>
+      </template>
+    </Select>
+  </div>
+</template>
diff --git a/apps/web-antd/src/components/TenantToggle/readme.md b/apps/web-antd/src/components/TenantToggle/readme.md
new file mode 100644
index 00000000..fb186816
--- /dev/null
+++ b/apps/web-antd/src/components/TenantToggle/readme.md
@@ -0,0 +1 @@
+租户切换组件
diff --git a/apps/web-antd/src/layouts/basic.vue b/apps/web-antd/src/layouts/basic.vue
index a5c771b4..81c8aef1 100644
--- a/apps/web-antd/src/layouts/basic.vue
+++ b/apps/web-antd/src/layouts/basic.vue
@@ -16,6 +16,7 @@ import { openWindow } from '@vben/utils';
 
 import { message } from 'ant-design-vue';
 
+import TenantToggle from '#/components/TenantToggle/index.vue';
 import { $t } from '#/locales';
 import { resetRoutes } from '#/router';
 import { useAuthStore, useNotifyStore } from '#/store';
@@ -77,6 +78,9 @@ function handleViewAll() {
 
 <template>
   <BasicLayout @clear-preferences-and-logout="handleLogout">
+    <template #header-right-1>
+      <TenantToggle />
+    </template>
     <template #user-dropdown>
       <UserDropdown
         :avatar
diff --git a/apps/web-antd/src/store/tenant.ts b/apps/web-antd/src/store/tenant.ts
new file mode 100644
index 00000000..1238273a
--- /dev/null
+++ b/apps/web-antd/src/store/tenant.ts
@@ -0,0 +1,39 @@
+import { ref } from 'vue';
+
+import { defineStore } from 'pinia';
+
+import {
+  tenantList as tenantListApi,
+  type TenantOption,
+} from '#/api/core/auth';
+
+export const useTenantStore = defineStore('app-tenant', () => {
+  const checked = ref(false);
+  const tenantEnable = ref(true);
+  const tenantList = ref<TenantOption[]>([]);
+
+  async function initTenant() {
+    const { tenantEnabled, voList } = await tenantListApi();
+    tenantEnable.value = tenantEnabled;
+    tenantList.value = voList;
+  }
+
+  async function setChecked(_checked: boolean) {
+    checked.value = _checked;
+  }
+
+  function $reset() {
+    checked.value = false;
+    tenantEnable.value = true;
+    tenantList.value = [];
+  }
+
+  return {
+    $reset,
+    checked,
+    initTenant,
+    setChecked,
+    tenantEnable,
+    tenantList,
+  };
+});