95 lines
3.7 KiB
JavaScript
95 lines
3.7 KiB
JavaScript
import { injectConfigProviderContext } from "../ConfigProvider/ConfigProvider.js";
|
|
import { computed, nextTick, ref, watch } from "vue";
|
|
import { createSharedComposable, useEventListener } from "@vueuse/core";
|
|
import { isClient, isIOS, tryOnBeforeUnmount } from "@vueuse/shared";
|
|
import { defu } from "defu";
|
|
|
|
//#region src/shared/useBodyScrollLock.ts
|
|
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.documentElement.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.documentElement.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;
|
|
}
|
|
|
|
//#endregion
|
|
export { useBodyScrollLock };
|
|
//# sourceMappingURL=useBodyScrollLock.js.map
|