From f96d49722e68e7d6f6972f95129f17646d97c462 Mon Sep 17 00:00:00 2001
From: colin <colin.in@foxmail.com>
Date: Fri, 28 Mar 2025 19:01:47 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=99=BB=E5=BD=95=E9=A1=B5=E5=A2=9E?=
 =?UTF-8?q?=E5=8A=A0=E7=A7=9F=E6=88=B7=E9=80=89=E6=8B=A9=E6=8E=A7=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../src/views/_core/authentication/login.vue  | 46 +++++++++++++++++--
 1 file changed, 41 insertions(+), 5 deletions(-)

diff --git a/apps/vben5/apps/app-antd/src/views/_core/authentication/login.vue b/apps/vben5/apps/app-antd/src/views/_core/authentication/login.vue
index 16e9a25ab..7af4fc667 100644
--- a/apps/vben5/apps/app-antd/src/views/_core/authentication/login.vue
+++ b/apps/vben5/apps/app-antd/src/views/_core/authentication/login.vue
@@ -1,26 +1,38 @@
 <script lang="ts" setup>
 import type { TwoFactorError } from '@abp/account';
 
-import type { VbenFormSchema } from '@vben/common-ui';
+import type { ExtendedFormApi, VbenFormSchema } from '@vben/common-ui';
 import type { Recordable } from '@vben/types';
 
-import { computed } from 'vue';
+import { computed, nextTick, onMounted, useTemplateRef } from 'vue';
 
 import { AuthenticationLogin, useVbenModal, z } from '@vben/common-ui';
 import { $t } from '@vben/locales';
 
+import { useAbpStore } from '@abp/core';
+
+import { useAbpConfigApi } from '#/api/core/useAbpConfigApi';
 import { useAuthStore } from '#/store';
 
 import TwoFactorLogin from './two-factor-login.vue';
 
+interface LoginInstance {
+  getFormApi(): ExtendedFormApi | undefined;
+}
+
 defineOptions({ name: 'Login' });
 
+const abpStore = useAbpStore();
 const authStore = useAuthStore();
 
+const { getConfigApi } = useAbpConfigApi();
+
+const login = useTemplateRef<LoginInstance>('login');
+
 const formSchema = computed((): VbenFormSchema[] => {
-  return [
+  let schemas: VbenFormSchema[] = [
     {
-      component: 'VbenInput',
+      component: 'Input',
       componentProps: {
         placeholder: $t('authentication.usernameTip'),
       },
@@ -29,7 +41,7 @@ const formSchema = computed((): VbenFormSchema[] => {
       rules: z.string().min(1, { message: $t('authentication.usernameTip') }),
     },
     {
-      component: 'VbenInputPassword',
+      component: 'InputPassword',
       componentProps: {
         placeholder: $t('authentication.password'),
       },
@@ -38,10 +50,31 @@ const formSchema = computed((): VbenFormSchema[] => {
       rules: z.string().min(1, { message: $t('authentication.passwordTip') }),
     },
   ];
+  if (abpStore.application?.multiTenancy.isEnabled) {
+    schemas = [
+      {
+        component: 'TenantSelect',
+        fieldName: 'tenant',
+        componentProps: {
+          onChange: onInit,
+        },
+      },
+      ...schemas,
+    ];
+  }
+  return schemas;
 });
 const [TwoFactorModal, twoFactorModalApi] = useVbenModal({
   connectedComponent: TwoFactorLogin,
 });
+async function onInit() {
+  const abpConfig = await getConfigApi();
+  abpStore.setApplication(abpConfig);
+  nextTick(() => {
+    const formApi = login.value?.getFormApi();
+    formApi?.setFieldValue('tenant', abpConfig.currentTenant.name);
+  });
+}
 async function onLogin(params: Recordable<any>) {
   try {
     await authStore.authLogin(params);
@@ -59,11 +92,14 @@ function onTwoFactorError(params: Recordable<any>, error: any) {
     twoFactorModalApi.open();
   }
 }
+
+onMounted(onInit);
 </script>
 
 <template>
   <div>
     <AuthenticationLogin
+      ref="login"
       :form-schema="formSchema"
       :loading="authStore.loginLoading"
       @submit="onLogin"