From 0b664d69797b3ca4409424247783f6e9e9a0b178 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Mon, 16 Mar 2026 11:15:39 +0800
Subject: [PATCH] fix: There is a record of tool calls, and exporting a PDF
 displays a blank one

---
 ui/src/components/pdf-export/index.vue | 183 +++++++++----------------
 1 file changed, 68 insertions(+), 115 deletions(-)

diff --git a/ui/src/components/pdf-export/index.vue b/ui/src/components/pdf-export/index.vue
index b51d1ba0cc8..3cdb67388b9 100644
--- a/ui/src/components/pdf-export/index.vue
+++ b/ui/src/components/pdf-export/index.vue
@@ -17,7 +17,6 @@
         justify-content: center;
       "
     >
-      <div ref="cloneContainerRef" style="width: 100%"></div>
       <div ref="svgContainerRef"></div>
     </div>
     <template #footer>
@@ -39,17 +38,18 @@
     </template>
   </el-dialog>
 </template>
+
 <script setup lang="ts">
 import * as htmlToImage from 'html-to-image'
 import { ref, nextTick } from 'vue'
-import html2Canvas from 'html2canvas'
 import { jsPDF } from 'jspdf'
 
 const loading = ref<boolean>(false)
 const svgContainerRef = ref()
-const cloneContainerRef = ref()
 const dialogVisible = ref<boolean>(false)
-const isSafari = /^((?!chrome|android).)*safari/i.test(navigator.userAgent)
+
+// 保存原始元素引用，用于导出
+const originalElement = ref<HTMLElement | null>(null)
 
 const open = (element: HTMLElement | null) => {
   dialogVisible.value = true
@@ -58,118 +58,73 @@ const open = (element: HTMLElement | null) => {
     loading.value = false
     return
   }
-  const cElement = element.cloneNode(true) as HTMLElement
-  setTimeout(() => {
-    nextTick(() => {
-      cloneContainerRef.value.appendChild(cElement)
-      htmlToImage
-        .toSvg(cElement, {
-          pixelRatio: 1,
-          quality: 1,
-          onImageErrorHandler: (
-            event: Event | string,
-            source?: string,
-            lineno?: number,
-            colno?: number,
-            error?: Error,
-          ) => {
-            console.log(event, source, lineno, colno, error)
-          },
-        })
-        .then((dataUrl) => {
-          if (isSafari) {
-            // Safari: 跳过 SVG data URI，直接用 toCanvas
-            return htmlToImage
-              .toCanvas(cElement, {
-                pixelRatio: 1,
-                quality: 1,
-              })
-              .then((canvas) => {
-                cloneContainerRef.value.style.display = 'none'
-                canvas.style.width = '100%'
-                canvas.style.height = 'auto'
-                svgContainerRef.value.appendChild(canvas)
-                svgContainerRef.value.style.height = canvas.height + 'px'
-              })
-          } else {
-            // Chrome 等：保持原逻辑
-            return fetch(dataUrl)
-              .then((response) => {
-                return response.text()
-              })
-              .then((text) => {
-                const parser = new DOMParser()
-                const svgDoc = parser.parseFromString(text, 'image/svg+xml')
-                cloneContainerRef.value.style.display = 'none'
-                const svgElement = svgDoc.documentElement
-                svgContainerRef.value.appendChild(svgElement)
-                svgContainerRef.value.style.height = svgElement.scrollHeight + 'px'
-              })
-          }
-        })
-        .finally(() => {
-          loading.value = false
-        })
-        .catch((e) => {
-          console.error(e)
-          loading.value = false
-        })
-    })
-  }, 1)
+
+  // 保存原始元素引用
+  originalElement.value = element
+
+  nextTick(() => {
+    htmlToImage
+      .toCanvas(element, {
+        pixelRatio: window.devicePixelRatio || 1,
+        quality: 1,
+        skipFonts: false,
+        backgroundColor: '#ffffff',
+      })
+      .then((canvas) => {
+        // 清空之前的内容
+        svgContainerRef.value.innerHTML = ''
+        canvas.style.width = '100%'
+        canvas.style.height = 'auto'
+        svgContainerRef.value.appendChild(canvas)
+      })
+      .finally(() => {
+        loading.value = false
+      })
+      .catch((e) => {
+        console.error(e)
+        loading.value = false
+      })
+  })
 }
 
 const exportPDF = () => {
   loading.value = true
   setTimeout(() => {
-    nextTick(() => {
-      if (isSafari) {
-        // Safari: 直接取已有的 canvas
-        const canvas = svgContainerRef.value.querySelector('canvas')
-        if (canvas) {
-          generatePDF(canvas)
-        }
-        loading.value = false
-      } else {
-        html2Canvas(svgContainerRef.value, {
-          logging: false,
-          allowTaint: true,
-          useCORS: true,
+    nextTick(async () => {
+      try {
+        const targetEl = originalElement.value
+        if (!targetEl) return
+        const canvas = await htmlToImage.toCanvas(targetEl, {
+          pixelRatio: 2,
+          quality: 1,
+          skipFonts: false,
+          backgroundColor: '#ffffff',
         })
-          .then((canvas) => {
-            generatePDF(canvas)
-          })
-          .finally(() => {
-            loading.value = false
-          })
+        generatePDF(canvas)
+      } catch (e) {
+        console.error('PDF export error:', e)
+      } finally {
+        loading.value = false
       }
     })
   })
 }
 
 const generatePDF = (canvas: HTMLCanvasElement) => {
-  const newCanvas = document.createElement('canvas')
-  newCanvas.width = canvas.width
-  newCanvas.height = canvas.height
-  const ctx = newCanvas.getContext('2d')!
-  ctx.fillStyle = '#ffffff'
-  ctx.fillRect(0, 0, newCanvas.width, newCanvas.height)
-  ctx.drawImage(canvas, 0, 0)
-
   const doc = new jsPDF('p', 'mm', 'a4')
-  const imgData = newCanvas.toDataURL('image/jpeg', 1)
+  const imgData = canvas.toDataURL('image/jpeg', 1)
   const pageWidth = doc.internal.pageSize.getWidth()
   const pageHeight = doc.internal.pageSize.getHeight()
   const imgWidth = pageWidth
-  const imgHeight = (newCanvas.height * imgWidth) / newCanvas.width
+  const imgHeight = (canvas.height * imgWidth) / canvas.width
 
-  doc.addImage(imgData, 'jpeg', 0, 0, imgWidth, imgHeight)
+  doc.addImage(imgData, 'JPEG', 0, 0, imgWidth, imgHeight)
 
   let heightLeft = imgHeight - pageHeight
-
   while (heightLeft > 0) {
     const position = -(imgHeight - heightLeft)
     doc.addPage()
-    doc.addImage(imgData, 'jpeg', 0, position, imgWidth, imgHeight)
+    doc.addImage(imgData, 'JPEG', 0, position, imgWidth, imgHeight)
     heightLeft -= pageHeight
   }
 
@@ -179,33 +134,27 @@ const generatePDF = (canvas: HTMLCanvasElement) => {
 const exportJepg = () => {
   loading.value = true
   setTimeout(() => {
-    nextTick(() => {
-      if (isSafari) {
-        // Safari: 直接取已有的 canvas
-        const canvas = svgContainerRef.value.querySelector('canvas')
-        if (canvas) {
-          downloadJpeg(canvas)
-        }
-        loading.value = false
-      } else {
-        html2Canvas(svgContainerRef.value, {
-          logging: false,
-          allowTaint: true,
-          useCORS: true,
+    nextTick(async () => {
+      try {
+        const targetEl = originalElement.value
+        if (!targetEl) return
+        const canvas = await htmlToImage.toCanvas(targetEl, {
+          pixelRatio: window.devicePixelRatio || 1,
+          quality: 1,
+          skipFonts: false,
+          backgroundColor: '#ffffff',
         })
-          .then((canvas) => {
-            downloadJpeg(canvas)
-          })
-          .finally(() => {
-            loading.value = false
-          })
+        downloadJpeg(canvas)
+      } catch (e) {
+        console.error('JPEG export error:', e)
+      } finally {
+        loading.value = false
       }
     })
   }, 1)
 }
 
 const downloadJpeg = (canvas: HTMLCanvasElement) => {
-  // 创建新 canvas，先填充白色背景
   const newCanvas = document.createElement('canvas')
   newCanvas.width = canvas.width
   newCanvas.height = canvas.height
@@ -225,8 +174,12 @@ const downloadJpeg = (canvas: HTMLCanvasElement) => {
 
 const close = () => {
   dialogVisible.value = false
+  originalElement.value = null
+  // 清空预览内容
+  if (svgContainerRef.value) {
+    svgContainerRef.value.innerHTML = ''
+  }
 }
 
 defineExpose({ open, close })
 </script>
-<style lang="scss" scoped></style>