UNPKG

1.11 kBJavaScriptView Raw
1import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
2import * as React from 'react';
3import { updateCSS, removeCSS } from "rc-util/es/Dom/dynamicCSS";
4import useLayoutEffect from "rc-util/es/hooks/useLayoutEffect";
5import getScrollBarSize from "rc-util/es/getScrollBarSize";
6import { isBodyOverflowing } from "./util";
7var UNIQUE_ID = "rc-util-locker-".concat(Date.now());
8var uuid = 0;
9export default function useScrollLocker(lock) {
10 var mergedLock = !!lock;
11 var _React$useState = React.useState(function () {
12 uuid += 1;
13 return "".concat(UNIQUE_ID, "_").concat(uuid);
14 }),
15 _React$useState2 = _slicedToArray(_React$useState, 1),
16 id = _React$useState2[0];
17 useLayoutEffect(function () {
18 if (mergedLock) {
19 var scrollbarSize = getScrollBarSize();
20 var isOverflow = isBodyOverflowing();
21 updateCSS("\nhtml body {\n overflow-y: hidden;\n ".concat(isOverflow ? "width: calc(100% - ".concat(scrollbarSize, "px);") : '', "\n}"), id);
22 } else {
23 removeCSS(id);
24 }
25 return function () {
26 removeCSS(id);
27 };
28 }, [mergedLock, id]);
29}
\No newline at end of file