reka-ui
Version:
Vue port for Radix UI Primitives.
105 lines (102 loc) • 3.96 kB
JavaScript
import { createSharedComposable, useEventListener } from '@vueuse/core';
import { tryOnBeforeUnmount, isClient, isIOS } from '@vueuse/shared';
import { computed, ref, watch, nextTick } from 'vue';
import { defu } from 'defu';
import { i as injectConfigProviderContext } from '../ConfigProvider/ConfigProvider.js';
const useBodyLockStackCount = createSharedComposable(() => {
const map = ref(/* @__PURE__ */ new Map());
const initialOverflow = ref();
const locked = computed(() => {
for (const value of map.value.values()) {
if (value)
return true;
}
return false;
});
const context = injectConfigProviderContext({
scrollBody: ref(true)
});
let stopTouchMoveListener = null;
const resetBodyStyle = () => {
document.body.style.paddingRight = "";
document.body.style.marginRight = "";
document.body.style.pointerEvents = "";
document.body.style.removeProperty("--scrollbar-width");
document.body.style.overflow = initialOverflow.value ?? "";
isIOS && stopTouchMoveListener?.();
initialOverflow.value = void 0;
};
watch(locked, (val, oldVal) => {
if (!isClient)
return;
if (!val) {
if (oldVal)
resetBodyStyle();
return;
}
if (initialOverflow.value === void 0)
initialOverflow.value = document.body.style.overflow;
const verticalScrollbarWidth = window.innerWidth - document.documentElement.clientWidth;
const defaultConfig = { padding: verticalScrollbarWidth, margin: 0 };
const config = context.scrollBody?.value ? typeof context.scrollBody.value === "object" ? defu({
padding: context.scrollBody.value.padding === true ? verticalScrollbarWidth : context.scrollBody.value.padding,
margin: context.scrollBody.value.margin === true ? verticalScrollbarWidth : context.scrollBody.value.margin
}, defaultConfig) : defaultConfig : { padding: 0, margin: 0 };
if (verticalScrollbarWidth > 0) {
document.body.style.paddingRight = typeof config.padding === "number" ? `${config.padding}px` : String(config.padding);
document.body.style.marginRight = typeof config.margin === "number" ? `${config.margin}px` : String(config.margin);
document.body.style.setProperty("--scrollbar-width", `${verticalScrollbarWidth}px`);
document.body.style.overflow = "hidden";
}
if (isIOS) {
stopTouchMoveListener = useEventListener(
document,
"touchmove",
(e) => preventDefault(e),
{ passive: false }
);
}
nextTick(() => {
document.body.style.pointerEvents = "none";
document.body.style.overflow = "hidden";
});
}, { immediate: true, flush: "sync" });
return map;
});
function useBodyScrollLock(initialState) {
const id = Math.random().toString(36).substring(2, 7);
const map = useBodyLockStackCount();
map.value.set(id, initialState ?? false);
const locked = computed({
get: () => map.value.get(id) ?? false,
set: (value) => map.value.set(id, value)
});
tryOnBeforeUnmount(() => {
map.value.delete(id);
});
return locked;
}
function checkOverflowScroll(ele) {
const style = window.getComputedStyle(ele);
if (style.overflowX === "scroll" || style.overflowY === "scroll" || style.overflowX === "auto" && ele.clientWidth < ele.scrollWidth || style.overflowY === "auto" && ele.clientHeight < ele.scrollHeight) {
return true;
} else {
const parent = ele.parentNode;
if (!(parent instanceof Element) || parent.tagName === "BODY")
return false;
return checkOverflowScroll(parent);
}
}
function preventDefault(rawEvent) {
const e = rawEvent || window.event;
const _target = e.target;
if (_target instanceof Element && checkOverflowScroll(_target))
return false;
if (e.touches.length > 1)
return true;
if (e.preventDefault && e.cancelable)
e.preventDefault();
return false;
}
export { useBodyScrollLock as u };
//# sourceMappingURL=useBodyScrollLock.js.map