From 8ab4682519c80a2bd97eec4f99944afcbf0d7579 Mon Sep 17 00:00:00 2001
From: one <wangan.cs@gmail.com>
Date: Sun, 6 Jul 2025 19:51:59 +0800
Subject: [PATCH] fix: hide scrollbars on capturing (#7867)

---
 src/renderer/src/assets/styles/scrollbar.scss | 8 ++++++++
 src/renderer/src/utils/image.ts               | 6 ++++++
 2 files changed, 14 insertions(+)

diff --git a/src/renderer/src/assets/styles/scrollbar.scss b/src/renderer/src/assets/styles/scrollbar.scss
index c5df842f78..21039de9c2 100644
--- a/src/renderer/src/assets/styles/scrollbar.scss
+++ b/src/renderer/src/assets/styles/scrollbar.scss
@@ -49,3 +49,11 @@ pre:not(.shiki)::-webkit-scrollbar-thumb {
   --color-scrollbar-thumb: var(--color-scrollbar-thumb-light);
   --color-scrollbar-thumb-hover: var(--color-scrollbar-thumb-light-hover);
 }
+
+/* 用于截图时隐藏滚动条
+ * FIXME: 临时方案，因为 html-to-image 没有正确处理伪元素。
+ */
+.hide-scrollbar,
+.hide-scrollbar * {
+  scrollbar-width: none !important;
+}
diff --git a/src/renderer/src/utils/image.ts b/src/renderer/src/utils/image.ts
index ee52739b7c..25ba33aabf 100644
--- a/src/renderer/src/utils/image.ts
+++ b/src/renderer/src/utils/image.ts
@@ -68,6 +68,9 @@ export const captureScrollableDiv = async (divRef: React.RefObject<HTMLDivElemen
 
       const originalScrollTop = div.scrollTop
 
+      // Hide scrollbars during capture
+      div.classList.add('hide-scrollbar')
+
       // Modify styles to show full content
       div.style.height = 'auto'
       div.style.maxHeight = 'none'
@@ -134,6 +137,9 @@ export const captureScrollableDiv = async (divRef: React.RefObject<HTMLDivElemen
     } catch (error) {
       console.error('Error capturing scrollable div:', error)
       throw error
+    } finally {
+      // Remove scrollbar hiding class
+      divRef.current?.classList.remove('hide-scrollbar')
     }
   }