element-plus
Version:
> TODO: description
50 lines (49 loc) • 1.95 kB
JavaScript
import { watch, isRef, onUnmounted } from 'vue';
import getScrollBarWidth from '../../utils/scrollbar-width';
import throwError from '../../utils/error';
import { addClass, removeClass, hasClass, getStyle, } from '../../utils/dom';
/**
* Hook that monitoring the ref value to lock or unlock the screen.
* When the trigger became true, it assumes modal is now opened and vice versa.
* @param trigger {Ref<boolean>}
*/
export default (trigger) => {
if (!isRef(trigger)) {
throwError('[useLockScreen]', 'You need to pass a ref param to this function');
}
let scrollBarWidth = 0;
let withoutHiddenClass = false;
let bodyPaddingRight = '0';
let computedBodyPaddingRight = 0;
onUnmounted(() => {
cleanup();
});
const cleanup = () => {
removeClass(document.body, 'el-popup-parent--hidden');
if (withoutHiddenClass) {
document.body.style.paddingRight = bodyPaddingRight;
}
};
watch(trigger, val => {
if (val) {
withoutHiddenClass = !hasClass(document.body, 'el-popup-parent--hidden');
if (withoutHiddenClass) {
bodyPaddingRight = document.body.style.paddingRight;
computedBodyPaddingRight = parseInt(getStyle(document.body, 'paddingRight'), 10);
}
scrollBarWidth = getScrollBarWidth();
const bodyHasOverflow = document.documentElement.clientHeight < document.body.scrollHeight;
const bodyOverflowY = getStyle(document.body, 'overflowY');
if (scrollBarWidth > 0 &&
(bodyHasOverflow || bodyOverflowY === 'scroll') &&
withoutHiddenClass) {
document.body.style.paddingRight =
computedBodyPaddingRight + scrollBarWidth + 'px';
}
addClass(document.body, 'el-popup-parent--hidden');
}
else {
cleanup();
}
});
};