From c8072d10025cc81ed0e0f3d7616bfcd0a01c61af Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Wed, 17 Dec 2025 10:00:50 +0800
Subject: [PATCH] fix: refactor wecomQrCode component to use dynamic iframe
 creation and improve cleanup logic

---
 ui/src/locales/lang/zh-CN/workflow.ts         |   2 +-
 .../user-login/scanCompinents/wecomQrCode.vue | 105 +++++++++++++-----
 .../login/scanCompinents/wecomQrCode.vue      |   1 -
 3 files changed, 79 insertions(+), 29 deletions(-)

diff --git a/ui/src/locales/lang/zh-CN/workflow.ts b/ui/src/locales/lang/zh-CN/workflow.ts
index 46f3d0417..63ca37a3f 100644
--- a/ui/src/locales/lang/zh-CN/workflow.ts
+++ b/ui/src/locales/lang/zh-CN/workflow.ts
@@ -138,7 +138,7 @@ export default {
         fileUploadType: {
           label: '上传的文件类型',
           documentText: '需要使用“文档内容提取”节点解析文档内容',
-          imageText: '需要使用“视觉模型”节点解析图片内容',
+          imageText: '需要使用“图片理解”节点解析图片内容',
           audioText: '需要使用“语音转文本”节点解析音频内容',
           videoText: '需要使用“视频理解”节点解析视频内容',
           otherText: '需要自行解析该类型文件',
diff --git a/ui/src/views/chat/user-login/scanCompinents/wecomQrCode.vue b/ui/src/views/chat/user-login/scanCompinents/wecomQrCode.vue
index 086b9188d..98cb976d1 100644
--- a/ui/src/views/chat/user-login/scanCompinents/wecomQrCode.vue
+++ b/ui/src/views/chat/user-login/scanCompinents/wecomQrCode.vue
@@ -1,14 +1,15 @@
 <template>
-  <iframe :src="iframeUrl" width="100%" height="380px" frameborder="0"
-          style="margin-top: -30px"></iframe>
+  <div id="wecom-qr" class="wecom-qr flex"></div>
 </template>
 
 <script lang="ts" setup>
-import {ref, nextTick, defineProps} from 'vue'
-import {getBrowserLang} from '@/locales'
-import {useRoute} from "vue-router";
+import { nextTick, defineProps, onBeforeUnmount } from 'vue'
+import { useRoute, useRouter } from 'vue-router'
+import { getBrowserLang } from '@/locales'
+import useStore from '@/stores'
 
-const route = useRoute()
+const WE_COM_ORIGIN = 'https://login.work.weixin.qq.com'
+const LOGIN_STATE = 'fit2cloud-wecom-qr'
 const props = defineProps<{
   config: {
     app_secret: string
@@ -19,38 +20,88 @@ const props = defineProps<{
   }
 }>()
 
+const router = useRouter()
+const route = useRoute()
+const { chatUser } = useStore()
+
 const {
-  params: {accessToken},
+  params: { accessToken },
 } = route as any
 
-const iframeUrl = ref('')
-const init = async () => {
-  await nextTick() // 确保DOM已更新
-  const data = {
-    corpId: props.config.corp_id,
-    agentId: props.config.agent_id,
-    redirectUri: props.config.callback_url,
-  }
-  let lang = localStorage.getItem('MaxKB-locale') || getBrowserLang() || 'en-US'
-  if (lang === 'en-US') {
-    lang = 'en'
-  } else {
-    lang = 'zh'
-  }
-  const redirectUri = encodeURIComponent(data.redirectUri)
-  console.log('redirectUri', data.redirectUri)
-  // 手动构建生成二维码的url
-  iframeUrl.value = `https://login.work.weixin.qq.com/wwlogin/sso/login?login_type=CorpApp&appid=${data.corpId}&agentid=${data.agentId}&redirect_uri=${redirectUri}&state=${accessToken}&lang=${lang}&lang=${lang}&panel_size=small`
+let iframe: HTMLIFrameElement | null = null
+
+function createTransparentIFrame(el: string) {
+  const container = document.querySelector(el)
+  if (!container) return null
+
+  const iframeEl = document.createElement('iframe')
+  iframeEl.style.cssText = `
+    display: block;
+    border: none;
+    background: transparent;
+  `
+  iframeEl.referrerPolicy = 'origin'
+  iframeEl.setAttribute('frameborder', '0')
+  iframeEl.setAttribute('allowtransparency', 'true')
+  iframeEl.setAttribute('allow', 'local-network-access')
+  container.appendChild(iframeEl)
+  return iframeEl
 }
 
+function getLang() {
+  const lang = localStorage.getItem('MaxKB-locale') || getBrowserLang()
+  return lang === 'en-US' ? 'en' : 'zh'
+}
+
+function cleanup() {
+  iframe?.remove()
+  iframe = null
+}
+
+const init = async () => {
+  await nextTick()
+
+  iframe = createTransparentIFrame('#wecom-qr')
+  if (!iframe) return
+
+  const redirectUri = encodeURIComponent(props.config.callback_url)
+
+  iframe.src =
+    `${WE_COM_ORIGIN}/wwlogin/sso/login` +
+    `?login_type=CorpApp` +
+    `&appid=${props.config.corp_id}` +
+    `&agentid=${props.config.agent_id}` +
+    `&redirect_uri=${redirectUri}` +
+    `&state=${accessToken}` +
+    `&lang=${getLang()}` +
+    `&panel_size=small` +
+    `&redirect_type=self`
+  iframe.addEventListener('load', (e) => {
+    if (iframe?.contentWindow) {
+      iframe.contentWindow.postMessage('getToken', '*')
+    }
+  })
+  window.addEventListener('message', (event) => {
+    if (event.data.type === 'token') {
+      chatUser.setToken(event.data.value)
+      router.push({
+        name: 'chat',
+        params: { accessToken },
+        query: route.query,
+      })
+    }
+  })
+}
+
+onBeforeUnmount(cleanup)
+
 init()
 </script>
 
 <style scoped lang="scss">
 #wecom-qr {
   margin-top: -20px;
-  height: 331px;
+  height: 360px;
   justify-content: center;
-
 }
 </style>
diff --git a/ui/src/views/login/scanCompinents/wecomQrCode.vue b/ui/src/views/login/scanCompinents/wecomQrCode.vue
index 022d19b6f..bcf35c36b 100644
--- a/ui/src/views/login/scanCompinents/wecomQrCode.vue
+++ b/ui/src/views/login/scanCompinents/wecomQrCode.vue
@@ -32,7 +32,6 @@ const init = async () => {
     lang = 'zh'
   }
   const redirectUri = encodeURIComponent(data.redirectUri)
-  console.log('redirectUri', data.redirectUri)
   // 手动构建生成二维码的url
   iframeUrl.value = `https://login.work.weixin.qq.com/wwlogin/sso/login?login_type=CorpApp&appid=${data.corpId}&agentid=${data.agentId}&redirect_uri=${redirectUri}&state=fit2cloud-wecom-qr&lang=${lang}&panel_size=small`
 }