From c2df6e04e25a9978ff17dbff7df38be795204d12 Mon Sep 17 00:00:00 2001
From: guoyuqi <xiaomeinvG@126.com>
Date: Tue, 4 Jun 2024 20:12:40 +0800
Subject: [PATCH] =?UTF-8?q?feat(=E7=B3=BB=E7=BB=9F=E7=AE=A1=E7=90=86):=20?=
 =?UTF-8?q?=E6=96=B0=E5=A2=9E=E9=92=89=E9=92=89=E7=99=BB=E9=99=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../sdk/util/FilterChainUtils.java            |  2 +
 frontend/src/api/modules/user/index.ts        | 14 ++++-
 frontend/src/api/requrls/user.ts              |  2 +
 frontend/src/models/user.ts                   |  7 +++
 .../views/login/components/dingTalkQrCode.vue | 61 ++++++++++++++++---
 .../src/views/login/components/tabQrCode.vue  | 11 ++--
 6 files changed, 81 insertions(+), 16 deletions(-)

diff --git a/backend/framework/sdk/src/main/java/io/metersphere/sdk/util/FilterChainUtils.java b/backend/framework/sdk/src/main/java/io/metersphere/sdk/util/FilterChainUtils.java
index a5a7fb1005..de0febe456 100644
--- a/backend/framework/sdk/src/main/java/io/metersphere/sdk/util/FilterChainUtils.java
+++ b/backend/framework/sdk/src/main/java/io/metersphere/sdk/util/FilterChainUtils.java
@@ -12,7 +12,9 @@ public class FilterChainUtils {
         filterChainDefinitionMap.put("/ldap/login", "anon");
         filterChainDefinitionMap.put("/authentication/get-list", "anon");
         filterChainDefinitionMap.put("/we_com/info", "anon");
+        filterChainDefinitionMap.put("/ding_talk/info", "anon");
         filterChainDefinitionMap.put("/sso/callback/we_com", "anon");
+        filterChainDefinitionMap.put("/auth", "anon");
         filterChainDefinitionMap.put("/setting/get/platform/param", "anon");
         filterChainDefinitionMap.put("/signout", "anon");
         filterChainDefinitionMap.put("/is-login", "anon");
diff --git a/frontend/src/api/modules/user/index.ts b/frontend/src/api/modules/user/index.ts
index a9c61283e8..debf21aa26 100644
--- a/frontend/src/api/modules/user/index.ts
+++ b/frontend/src/api/modules/user/index.ts
@@ -7,8 +7,10 @@ import {
   DisableLocalConfigUrl,
   EnableAPIKEYUrl,
   EnableLocalConfigUrl,
+  GeDingInfoUrl,
   GetAPIKEYListUrl,
   getAuthenticationUrl,
+  GetDingCallbackUrl,
   GetInfoUrl,
   GetLocalConfigUrl,
   GetMenuListUrl,
@@ -46,7 +48,7 @@ import type {
   UpdateLocalConfigParams,
   UpdatePswParams,
 } from '@/models/user';
-import { WecomInfo } from '@/models/user';
+import { DingInfo, WecomInfo } from '@/models/user';
 
 import type { RouteRecordNormalized } from 'vue-router';
 
@@ -83,6 +85,16 @@ export function getWeComCallback(code: string) {
   );
 }
 
+export function getDingInfo() {
+  return MSR.get<DingInfo>({ url: GeDingInfoUrl }, { ignoreCancelToken: true, errorMessageMode: 'none' });
+}
+
+export function getDingCallback(code: string) {
+  return MSR.get<LoginRes>(
+    { url: GetDingCallbackUrl, params: { code } },
+    { ignoreCancelToken: true, errorMessageMode: 'none' }
+  );
+}
 export function logout() {
   return MSR.get<LoginRes>({ url: LogoutUrl });
 }
diff --git a/frontend/src/api/requrls/user.ts b/frontend/src/api/requrls/user.ts
index d78a4b0c2f..87f42c3a5c 100644
--- a/frontend/src/api/requrls/user.ts
+++ b/frontend/src/api/requrls/user.ts
@@ -29,3 +29,5 @@ export const GetPlatformOrgOptionUrl = '/user/platform/switch-option'; // 个人
 export const GetWeComInfoUrl = '/we_com/info'; // 获取企业微信登陆的配置信息
 export const GetWeComCallbackUrl = '/sso/callback/we_com'; // 获取企业微信登陆的回调信息
 export const GetPlatformParamUrl = '/setting/get/platform/param';
+export const GeDingInfoUrl = '/ding_talk/info'; // 获取企业微信登陆的配置信息
+export const GetDingCallbackUrl = '/auth'; // 获取企业微信登陆的回调信息
diff --git a/frontend/src/models/user.ts b/frontend/src/models/user.ts
index 37ea8ab5ea..bb05ac3b98 100644
--- a/frontend/src/models/user.ts
+++ b/frontend/src/models/user.ts
@@ -21,6 +21,13 @@ export interface WecomInfo {
   callBack?: string;
 }
 
+// 企业微信对接信息
+export interface DingInfo {
+  agentId?: string;
+  state?: string;
+  callBack?: string;
+}
+
 // 更新本地执行配置
 export interface UpdateLocalConfigParams {
   id: string;
diff --git a/frontend/src/views/login/components/dingTalkQrCode.vue b/frontend/src/views/login/components/dingTalkQrCode.vue
index e59b06ff95..13a86603a9 100644
--- a/frontend/src/views/login/components/dingTalkQrCode.vue
+++ b/frontend/src/views/login/components/dingTalkQrCode.vue
@@ -3,16 +3,32 @@
 </template>
 
 <script lang="ts" setup>
+  import { useRouter } from 'vue-router';
   import { useScriptTag } from '@vueuse/core';
+  import { Message } from '@arco-design/web-vue';
+
+  import { getProjectInfo } from '@/api/modules/project-management/basicInfo';
+  import { getDingCallback, getDingInfo } from '@/api/modules/user';
+  import { useI18n } from '@/hooks/useI18n';
+  import { NO_PROJECT_ROUTE_NAME, NO_RESOURCE_ROUTE_NAME } from '@/router/constants';
+  import { useAppStore, useUserStore } from '@/store';
+  import useLicenseStore from '@/store/modules/setting/license';
+  import { setLoginExpires } from '@/utils/auth';
+  import { getFirstRouteNameByPermission, routerNameHasPermission } from '@/utils/permission';
+
+  const { t } = useI18n();
+
+  const userStore = useUserStore();
+  const appStore = useAppStore();
+  const licenseStore = useLicenseStore();
+  const router = useRouter();
 
   const { load } = useScriptTag('https://g.alicdn.com/dingding/h5-dingtalk-login/0.21.0/ddlogin.js');
 
-  const a = encodeURIComponent('https://s0my5tnf41e2.ngrok.xiaomiqiu123.top/');
-
-  const url = `https://login.dingtalk.com/oauth2/auth?redirect_uri=${a}&response_type=code&client_id=dinglsfxhodjquu4gq2x&scope=openid&state=dddd&prompt=consent`;
-
   const initActive = async () => {
+    const data = await getDingInfo();
     await load(true);
+    const url = encodeURIComponent(data.callBack ? data.callBack : '');
     window.DTFrameLogin(
       {
         id: 'ding-talk-qr',
@@ -20,17 +36,42 @@
         height: 300,
       },
       {
-        redirect_uri: a,
-        client_id: 'dinglsfxhodjquu4gq2x',
+        redirect_uri: url,
+        client_id: data.agentId ? data.agentId : '',
         scope: 'openid',
         response_type: 'code',
-        state: 'xxxxxxxxx',
+        state: 'fit2cloud-ding-qr',
         prompt: 'consent',
       },
-      (loginResult) => {
+      async (loginResult) => {
         const { redirectUrl, authCode, state } = loginResult;
-        // 这里可以直接进行重定向
-        window.location.href = redirectUrl;
+        const dingCallback = getDingCallback(authCode);
+        userStore.qrCodeLogin(await dingCallback);
+        Message.success(t('login.form.login.success'));
+        const { redirect, ...othersQuery } = router.currentRoute.value.query;
+        const redirectHasPermission =
+          redirect &&
+          ![NO_RESOURCE_ROUTE_NAME, NO_PROJECT_ROUTE_NAME].includes(redirect as string) &&
+          routerNameHasPermission(redirect as string, router.getRoutes());
+        const currentRouteName = getFirstRouteNameByPermission(router.getRoutes());
+        const [res] = await Promise.all([getProjectInfo(appStore.currentProjectId), licenseStore.getValidateLicense()]); // 登录前校验 license 避免进入页面后无license状态
+        if (!res || res.deleted) {
+          router.push({
+            name: NO_PROJECT_ROUTE_NAME,
+          });
+        }
+        if (res) {
+          appStore.setCurrentMenuConfig(res?.moduleIds || []);
+        }
+        setLoginExpires();
+        router.push({
+          name: redirectHasPermission ? (redirect as string) : currentRouteName,
+          query: {
+            ...othersQuery,
+            orgId: appStore.currentOrgId,
+            pId: appStore.currentProjectId,
+          },
+        });
         // 也可以在不跳转页面的情况下，使用code进行授权
         console.log(authCode);
       },
diff --git a/frontend/src/views/login/components/tabQrCode.vue b/frontend/src/views/login/components/tabQrCode.vue
index 7297814f8c..e09f786577 100644
--- a/frontend/src/views/login/components/tabQrCode.vue
+++ b/frontend/src/views/login/components/tabQrCode.vue
@@ -1,10 +1,10 @@
 <template>
-  <a-tabs v-model:active-key="activeName" class="tabPlatform" @change="handleClick">
-    <a-tab-pane key="wecom" :title="t('project.messageManagement.WE_COM')" class="font-[16px]"></a-tab-pane>
-    <a-tab-pane key="dingtalk" :title="t('project.messageManagement.DING_TALK')" class="font-[16px]"></a-tab-pane>
+  <a-radio-group v-model:active-key="activeName" type="button" class="tabPlatform" @change="handleClick">
+    <a-radio value="wecom">{{ t('project.messageManagement.WE_COM') }}</a-radio>
+    <a-radio value="dingtalk">{{ t('project.messageManagement.DING_TALK') }}</a-radio>
     <!--    <a-tab-pane key="lark" :title="t('project.messageManagement.LARK')"></a-tab-pane>
     <a-tab-pane key="larksuite" :title="t('project.messageManagement.LARK_SUITE')"></a-tab-pane>-->
-  </a-tabs>
+  </a-radio-group>
   <div v-if="activeName === 'wecom'" class="login-qrcode">
     <div class="qrcode">
       <wecom-qr v-if="activeName === 'wecom'" />
@@ -56,7 +56,7 @@
       }
     }
   };
-  function handleClick(val: string | number) {
+  function handleClick(val: string | number | boolean) {
     if (typeof val === 'string') {
       activeName.value = val;
     }
@@ -84,6 +84,7 @@
       overflow: hidden;
       border-radius: 8px;
       background: #ffffff;
+      flex-direction: column;
     }
     .title {
       display: flex;