2021-03-16 21:40:45 +08:00
|
|
|
import { watch, isRef, onUnmounted } from 'vue'
|
2020-09-09 21:18:08 +08:00
|
|
|
|
|
|
|
import getScrollBarWidth from '@element-plus/utils/scrollbar-width'
|
2021-09-10 10:01:17 +08:00
|
|
|
import { throwError } from '@element-plus/utils/error'
|
2020-09-09 21:18:08 +08:00
|
|
|
import {
|
|
|
|
addClass,
|
|
|
|
removeClass,
|
|
|
|
hasClass,
|
|
|
|
getStyle,
|
|
|
|
} from '@element-plus/utils/dom'
|
|
|
|
|
|
|
|
import type { Ref } from 'vue'
|
|
|
|
|
|
|
|
/**
|
|
|
|
* 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: Ref<boolean>) => {
|
|
|
|
if (!isRef(trigger)) {
|
|
|
|
throwError(
|
|
|
|
'[useLockScreen]',
|
2021-09-04 19:29:28 +08:00
|
|
|
'You need to pass a ref param to this function'
|
2020-09-09 21:18:08 +08:00
|
|
|
)
|
|
|
|
}
|
|
|
|
let scrollBarWidth = 0
|
|
|
|
let withoutHiddenClass = false
|
|
|
|
let bodyPaddingRight = '0'
|
|
|
|
let computedBodyPaddingRight = 0
|
2021-03-16 21:40:45 +08:00
|
|
|
|
|
|
|
onUnmounted(() => {
|
|
|
|
cleanup()
|
|
|
|
})
|
|
|
|
|
|
|
|
const cleanup = () => {
|
|
|
|
removeClass(document.body, 'el-popup-parent--hidden')
|
|
|
|
if (withoutHiddenClass) {
|
|
|
|
document.body.style.paddingRight = bodyPaddingRight
|
|
|
|
}
|
|
|
|
}
|
2021-09-04 19:29:28 +08:00
|
|
|
watch(trigger, (val) => {
|
2020-09-09 21:18:08 +08:00
|
|
|
if (val) {
|
|
|
|
withoutHiddenClass = !hasClass(document.body, 'el-popup-parent--hidden')
|
|
|
|
if (withoutHiddenClass) {
|
|
|
|
bodyPaddingRight = document.body.style.paddingRight
|
|
|
|
computedBodyPaddingRight = parseInt(
|
|
|
|
getStyle(document.body, 'paddingRight'),
|
2021-09-04 19:29:28 +08:00
|
|
|
10
|
2020-09-09 21:18:08 +08:00
|
|
|
)
|
|
|
|
}
|
|
|
|
scrollBarWidth = getScrollBarWidth()
|
|
|
|
const bodyHasOverflow =
|
|
|
|
document.documentElement.clientHeight < document.body.scrollHeight
|
|
|
|
const bodyOverflowY = getStyle(document.body, 'overflowY')
|
|
|
|
if (
|
|
|
|
scrollBarWidth > 0 &&
|
|
|
|
(bodyHasOverflow || bodyOverflowY === 'scroll') &&
|
|
|
|
withoutHiddenClass
|
|
|
|
) {
|
2021-09-17 09:18:24 +08:00
|
|
|
document.body.style.paddingRight = `${
|
|
|
|
computedBodyPaddingRight + scrollBarWidth
|
|
|
|
}px`
|
2020-09-09 21:18:08 +08:00
|
|
|
}
|
|
|
|
addClass(document.body, 'el-popup-parent--hidden')
|
|
|
|
} else {
|
2021-03-16 21:40:45 +08:00
|
|
|
cleanup()
|
2020-09-09 21:18:08 +08:00
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|