useScrollLocker.js 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. "use strict";
  2. var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard").default;
  3. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = useScrollLocker;
  8. var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
  9. var React = _interopRequireWildcard(require("react"));
  10. var _dynamicCSS = require("rc-util/lib/Dom/dynamicCSS");
  11. var _useLayoutEffect = _interopRequireDefault(require("rc-util/lib/hooks/useLayoutEffect"));
  12. var _getScrollBarSize = require("rc-util/lib/getScrollBarSize");
  13. var _util = require("./util");
  14. var UNIQUE_ID = "rc-util-locker-".concat(Date.now());
  15. var uuid = 0;
  16. function useScrollLocker(lock) {
  17. var mergedLock = !!lock;
  18. var _React$useState = React.useState(function () {
  19. uuid += 1;
  20. return "".concat(UNIQUE_ID, "_").concat(uuid);
  21. }),
  22. _React$useState2 = (0, _slicedToArray2.default)(_React$useState, 1),
  23. id = _React$useState2[0];
  24. (0, _useLayoutEffect.default)(function () {
  25. if (mergedLock) {
  26. var scrollbarSize = (0, _getScrollBarSize.getTargetScrollBarSize)(document.body).width;
  27. var isOverflow = (0, _util.isBodyOverflowing)();
  28. (0, _dynamicCSS.updateCSS)("\nhtml body {\n overflow-y: hidden;\n ".concat(isOverflow ? "width: calc(100% - ".concat(scrollbarSize, "px);") : '', "\n}"), id);
  29. } else {
  30. (0, _dynamicCSS.removeCSS)(id);
  31. }
  32. return function () {
  33. (0, _dynamicCSS.removeCSS)(id);
  34. };
  35. }, [mergedLock, id]);
  36. }