chore: 根据不同的容器dom进行获取scroll 的宽度

Tencent · Oct 14, 2024 · 44f9c8a · 44f9c8a
1 parent 993a49c
commit 44f9c8a
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/js/utils/getScrollbarWidth.ts b/js/utils/getScrollbarWidth.ts
@@ -35,7 +35,14 @@ export function getScrollbarWidthWithCSS() {
   return scrollbarWidth;
 }
 
-// 获取 body 下滚动条宽度
-export function getScrollbarWidth() {
-  return window.innerWidth - document.documentElement.clientWidth;
+/**
+ * @description: Calculate scroll bar width
+ * @param container  Container used to calculate scrollbar width
+ * @default container: document.body
+ */
+export function getScrollbarWidth(container: HTMLElement = document.body) {
+  if (container === document.body) {
+    return window.innerWidth - document.documentElement.clientWidth;
+  }
+  return container.offsetWidth - container.clientWidth;
 }