From 8e6c1abf19d1fd9db659f27512d1cc0eeeacdca8 Mon Sep 17 00:00:00 2001
From: Li Kui <90845831+likui628@users.noreply.github.com>
Date: Thu, 11 Jul 2024 11:05:01 +0800
Subject: [PATCH] feat: login dialog (#37)

* chore: login-dialog demo

* Merge branch 'main' into login-dialog

* chore: update dialog

* Merge branch 'main' into login-dialog

* chore: accept login params

* chore: redirect to login or show login dialog
---
 apps/web-antd/src/layouts/basic.vue           | 23 +++++--
 apps/web-antd/src/router/guard.ts             | 23 +++++--
 apps/web-antd/src/store/modules/access.ts     |  8 +++
 .../ui-kit/layout-ui/src/vben-layout.vue      |  1 +
 packages/business/layouts/package.json        |  1 +
 .../business/layouts/src/basic/layout.vue     |  4 ++
 .../business/layouts/src/widgets/index.ts     |  1 +
 .../layouts/src/widgets/login-dialog/index.ts |  1 +
 .../src/widgets/login-dialog/login-dialog.vue | 48 ++++++++++++++
 .../universal-ui/src/authentication/index.ts  |  6 +-
 .../universal-ui/src/authentication/login.vue | 65 +------------------
 .../src/authentication/typings.ts             | 63 ++++++++++++++++++
 pnpm-lock.yaml                                |  3 +
 13 files changed, 174 insertions(+), 73 deletions(-)
 create mode 100644 packages/business/layouts/src/widgets/login-dialog/index.ts
 create mode 100644 packages/business/layouts/src/widgets/login-dialog/login-dialog.vue

diff --git a/apps/web-antd/src/layouts/basic.vue b/apps/web-antd/src/layouts/basic.vue
index 0dcbb7c7..42cb3ea1 100644
--- a/apps/web-antd/src/layouts/basic.vue
+++ b/apps/web-antd/src/layouts/basic.vue
@@ -1,12 +1,16 @@
 <script lang="ts" setup>
-import type { NotificationItem } from '@vben/layouts';
-
-import { computed, ref } from 'vue';
+import { computed, ref, toRefs } from 'vue';
 import { useRouter } from 'vue-router';
 
 import { LOGIN_PATH } from '@vben/constants';
 import { IcRoundCreditScore, MdiDriveDocument, MdiGithub } from '@vben/icons';
-import { BasicLayout, Notification, UserDropdown } from '@vben/layouts';
+import {
+  BasicLayout,
+  LoginDialog,
+  Notification,
+  NotificationItem,
+  UserDropdown,
+} from '@vben/layouts';
 import { openWindow } from '@vben/utils';
 import { preferences } from '@vben-core/preferences';
 
@@ -80,7 +84,8 @@ const menus = computed(() => [
 ]);
 
 const appStore = useAppStore();
-const { userInfo } = useAccessStore();
+const accessStore = useAccessStore();
+const { showLoginDialog, userInfo } = toRefs(accessStore);
 const router = useRouter();
 
 async function handleLogout() {
@@ -118,5 +123,13 @@ function handleMakeAll() {
         @make-all="handleMakeAll"
       />
     </template>
+    <template #dialog>
+      <LoginDialog
+        :open="showLoginDialog"
+        password-placeholder="123456"
+        username-placeholder="vben"
+        @login="accessStore.authLogin"
+      />
+    </template>
   </BasicLayout>
 </template>
diff --git a/apps/web-antd/src/router/guard.ts b/apps/web-antd/src/router/guard.ts
index 9b494d2f..0c71a66b 100644
--- a/apps/web-antd/src/router/guard.ts
+++ b/apps/web-antd/src/router/guard.ts
@@ -92,10 +92,25 @@ function setupAccessGuard(router: Router) {
 
     // 生成路由表
     // 当前登录用户拥有的角色标识列表
-    const userInfo =
-      accessStore.userInfo || (await accessStore.fetchUserInfo());
-
-    const userRoles = userInfo.roles ?? [];
+    let userRoles: string[] = [];
+    try {
+      const userInfo =
+        accessStore.userInfo || (await accessStore.fetchUserInfo());
+      userRoles = userInfo.roles ?? [];
+    } catch (error: any) {
+      if (error.status === 409) {
+        accessStore.setShowLoginDialog(true);
+      } else if (error.status === 401) {
+        accessStore.reset();
+        return {
+          path: LOGIN_PATH,
+          // 如不需要，直接删除 query
+          query: { redirect: encodeURIComponent(to.fullPath) },
+          // 携带当前跳转的页面，登录后重新跳转该页面
+          replace: true,
+        };
+      }
+    }
 
     // 生成菜单和路由
     const { accessibleMenus, accessibleRoutes } = await generateAccess({
diff --git a/apps/web-antd/src/store/modules/access.ts b/apps/web-antd/src/store/modules/access.ts
index a17979c2..9bbe54fe 100644
--- a/apps/web-antd/src/store/modules/access.ts
+++ b/apps/web-antd/src/store/modules/access.ts
@@ -17,6 +17,11 @@ export const useAccessStore = defineStore('access', () => {
   const router = useRouter();
   const loading = ref(false);
 
+  const showLoginDialog = ref(false);
+  function setShowLoginDialog(value: boolean) {
+    showLoginDialog.value = value;
+  }
+
   const accessToken = computed(() => coreStoreAccess.accessToken);
   const userRoles = computed(() => coreStoreAccess.userRoles);
   const userInfo = computed(() => coreStoreAccess.userInfo);
@@ -65,6 +70,7 @@ export const useAccessStore = defineStore('access', () => {
         coreStoreAccess.setUserInfo(userInfo);
         coreStoreAccess.setAccessCodes(accessCodes);
 
+        showLoginDialog.value = false;
         onSuccess
           ? await onSuccess?.()
           : await router.push(userInfo.homePath || DEFAULT_HOME_PATH);
@@ -99,6 +105,8 @@ export const useAccessStore = defineStore('access', () => {
     reset,
     setAccessMenus,
     setAccessRoutes,
+    setShowLoginDialog,
+    showLoginDialog,
     userInfo,
     userRoles,
   };
diff --git a/packages/@core/ui-kit/layout-ui/src/vben-layout.vue b/packages/@core/ui-kit/layout-ui/src/vben-layout.vue
index 6177fe00..f1f06691 100644
--- a/packages/@core/ui-kit/layout-ui/src/vben-layout.vue
+++ b/packages/@core/ui-kit/layout-ui/src/vben-layout.vue
@@ -565,6 +565,7 @@ function handleOpenMenu() {
         <slot name="footer"></slot>
       </LayoutFooter>
     </div>
+    <slot name="extra"></slot>
     <div
       v-if="maskVisible"
       :style="maskStyle"
diff --git a/packages/business/layouts/package.json b/packages/business/layouts/package.json
index c1a595c8..3e29dfe3 100644
--- a/packages/business/layouts/package.json
+++ b/packages/business/layouts/package.json
@@ -48,6 +48,7 @@
     "@vben-core/stores": "workspace:*",
     "@vben-core/tabs-ui": "workspace:*",
     "@vben-core/toolkit": "workspace:*",
+    "@vben/universal-ui": "workspace:*",
     "@vueuse/core": "^10.11.0",
     "vue": "^3.4.31",
     "vue-router": "^4.4.0"
diff --git a/packages/business/layouts/src/basic/layout.vue b/packages/business/layouts/src/basic/layout.vue
index 3ab28288..112c2798 100644
--- a/packages/business/layouts/src/basic/layout.vue
+++ b/packages/business/layouts/src/basic/layout.vue
@@ -277,5 +277,9 @@ function clearPreferencesAndLogout() {
         />
       </LayoutFooter>
     </template>
+
+    <template #extra>
+      <slot name="dialog"></slot>
+    </template>
   </VbenAdminLayout>
 </template>
diff --git a/packages/business/layouts/src/widgets/index.ts b/packages/business/layouts/src/widgets/index.ts
index 718d1a73..03303c8a 100644
--- a/packages/business/layouts/src/widgets/index.ts
+++ b/packages/business/layouts/src/widgets/index.ts
@@ -4,6 +4,7 @@ export { default as CozeAssistant } from './coze-assistant.vue';
 export * from './global-search';
 export { default as LanguageToggle } from './language-toggle.vue';
 export { default as AuthenticationLayoutToggle } from './layout-toggle.vue';
+export * from './login-dialog';
 export * from './notification';
 export * from './preferences';
 export * from './theme-toggle';
diff --git a/packages/business/layouts/src/widgets/login-dialog/index.ts b/packages/business/layouts/src/widgets/login-dialog/index.ts
new file mode 100644
index 00000000..879f0864
--- /dev/null
+++ b/packages/business/layouts/src/widgets/login-dialog/index.ts
@@ -0,0 +1 @@
+export { default as LoginDialog } from './login-dialog.vue';
diff --git a/packages/business/layouts/src/widgets/login-dialog/login-dialog.vue b/packages/business/layouts/src/widgets/login-dialog/login-dialog.vue
new file mode 100644
index 00000000..3ad397cf
--- /dev/null
+++ b/packages/business/layouts/src/widgets/login-dialog/login-dialog.vue
@@ -0,0 +1,48 @@
+<script setup lang="ts">
+import { computed } from 'vue';
+
+import {
+  AuthenticationLogin,
+  AuthenticationProps,
+  LoginAndRegisterParams,
+} from '@vben/universal-ui';
+import { Dialog, DialogContent } from '@vben-core/shadcn-ui';
+
+interface Props extends AuthenticationProps {
+  open: boolean;
+}
+
+defineOptions({
+  name: 'LoginDialog',
+});
+
+const props = withDefaults(defineProps<Props>(), {
+  open: false,
+});
+
+const emit = defineEmits<{
+  login: [LoginAndRegisterParams];
+}>();
+
+const loginProps = computed(() => {
+  const { open: _, ...rest } = props;
+  return rest;
+});
+</script>
+
+<template>
+  <div>
+    <Dialog :open="open" class="flex items-center justify-center">
+      <DialogContent
+        class="top-[50%] w-full translate-y-[-50%] border-none p-0 shadow-xl sm:w-[600px] sm:rounded-2xl"
+      >
+        <div class="p-4">
+          <AuthenticationLogin
+            v-bind="loginProps"
+            @submit="(e) => emit('login', e)"
+          />
+        </div>
+      </DialogContent>
+    </Dialog>
+  </div>
+</template>
diff --git a/packages/business/universal-ui/src/authentication/index.ts b/packages/business/universal-ui/src/authentication/index.ts
index 906fc563..f82360a4 100644
--- a/packages/business/universal-ui/src/authentication/index.ts
+++ b/packages/business/universal-ui/src/authentication/index.ts
@@ -3,4 +3,8 @@ export { default as AuthenticationForgetPassword } from './forget-password.vue';
 export { default as AuthenticationLogin } from './login.vue';
 export { default as AuthenticationQrCodeLogin } from './qrcode-login.vue';
 export { default as AuthenticationRegister } from './register.vue';
-export type { LoginAndRegisterParams, LoginCodeParams } from './typings';
+export type {
+  AuthenticationProps,
+  LoginAndRegisterParams,
+  LoginCodeParams,
+} from './typings';
diff --git a/packages/business/universal-ui/src/authentication/login.vue b/packages/business/universal-ui/src/authentication/login.vue
index 61d04f35..12e55c0c 100644
--- a/packages/business/universal-ui/src/authentication/login.vue
+++ b/packages/business/universal-ui/src/authentication/login.vue
@@ -1,6 +1,4 @@
 <script setup lang="ts">
-import type { LoginEmits } from './typings';
-
 import { computed, reactive } from 'vue';
 import { useRouter } from 'vue-router';
 
@@ -14,68 +12,9 @@ import {
 
 import Title from './auth-title.vue';
 import ThirdPartyLogin from './third-party-login.vue';
+import { AuthenticationProps, LoginEmits } from './typings';
 
-interface Props {
-  /**
-   * @zh_CN 验证码登录路径
-   */
-  codeLoginPath?: string;
-
-  /**
-   * @zh_CN 忘记密码路径
-   */
-  forgetPasswordPath?: string;
-
-  /**
-   * @zh_CN 是否处于加载处理状态
-   */
-  loading?: boolean;
-
-  /**
-   * @zh_CN 密码占位符
-   */
-  passwordPlaceholder?: string;
-
-  /**
-   * @zh_CN 二维码登录路径
-   */
-  qrCodeLoginPath?: string;
-
-  /**
-   * @zh_CN 注册路径
-   */
-  registerPath?: string;
-
-  /**
-   * @zh_CN 是否显示验证码登录
-   */
-  showCodeLogin?: boolean;
-
-  /**
-   * @zh_CN 是否显示忘记密码
-   */
-  showForgetPassword?: boolean;
-
-  /**
-   * @zh_CN 是否显示二维码登录
-   */
-  showQrcodeLogin?: boolean;
-
-  /**
-   * @zh_CN 是否显示注册按钮
-   */
-  showRegister?: boolean;
-
-  /**
-   * @zh_CN 是否显示第三方登录
-   */
-  showThirdPartyLogin?: boolean;
-
-  /**
-   * @zh_CN 用户名占位符
-   */
-  usernamePlaceholder?: string;
-}
+interface Props extends AuthenticationProps {}
 
 defineOptions({
   name: 'AuthenticationLogin',
diff --git a/packages/business/universal-ui/src/authentication/typings.ts b/packages/business/universal-ui/src/authentication/typings.ts
index 06ba1da8..6bd6b4dd 100644
--- a/packages/business/universal-ui/src/authentication/typings.ts
+++ b/packages/business/universal-ui/src/authentication/typings.ts
@@ -1,3 +1,65 @@
+interface AuthenticationProps {
+  /**
+   * @zh_CN 验证码登录路径
+   */
+  codeLoginPath?: string;
+
+  /**
+   * @zh_CN 忘记密码路径
+   */
+  forgetPasswordPath?: string;
+
+  /**
+   * @zh_CN 是否处于加载处理状态
+   */
+  loading?: boolean;
+
+  /**
+   * @zh_CN 密码占位符
+   */
+  passwordPlaceholder?: string;
+
+  /**
+   * @zh_CN 二维码登录路径
+   */
+  qrCodeLoginPath?: string;
+
+  /**
+   * @zh_CN 注册路径
+   */
+  registerPath?: string;
+
+  /**
+   * @zh_CN 是否显示验证码登录
+   */
+  showCodeLogin?: boolean;
+
+  /**
+   * @zh_CN 是否显示忘记密码
+   */
+  showForgetPassword?: boolean;
+
+  /**
+   * @zh_CN 是否显示二维码登录
+   */
+  showQrcodeLogin?: boolean;
+
+  /**
+   * @zh_CN 是否显示注册按钮
+   */
+  showRegister?: boolean;
+
+  /**
+   * @zh_CN 是否显示第三方登录
+   */
+  showThirdPartyLogin?: boolean;
+
+  /**
+   * @zh_CN 用户名占位符
+   */
+  usernamePlaceholder?: string;
+}
+
 interface LoginAndRegisterParams {
   password: string;
   username: string;
@@ -21,6 +83,7 @@ interface RegisterEmits {
 }
 
 export type {
+  AuthenticationProps,
   LoginAndRegisterParams,
   LoginCodeEmits,
   LoginCodeParams,
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
index 4a6ed86a..afcc948e 100644
--- a/pnpm-lock.yaml
+++ b/pnpm-lock.yaml
@@ -878,6 +878,9 @@ importers:
       '@vben-core/toolkit':
         specifier: workspace:*
         version: link:../../@core/shared/toolkit
+      '@vben/universal-ui':
+        specifier: workspace:*
+        version: link:../universal-ui
       '@vueuse/core':
         specifier: ^10.11.0
         version: 10.11.0(vue@3.4.31(typescript@5.5.3))