From 2844a2bd71dcf1eb17a53992c10129b7496332e0 Mon Sep 17 00:00:00 2001
From: Armand Philippot <git@armandphilippot.com>
Date: Tue, 31 Oct 2023 17:41:43 +0100
Subject: feat(components): add an Overlay component

* add useScrollbarWidth hook
* add useScrollLock hook
* add a new component to lock scroll with an overlay (it can be useful
especially on small screens to prevent background contents to be
scrolled)
---
 src/utils/hooks/use-scroll-lock/use-scroll-lock.ts | 58 ++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 src/utils/hooks/use-scroll-lock/use-scroll-lock.ts

(limited to 'src/utils/hooks/use-scroll-lock/use-scroll-lock.ts')

diff --git a/src/utils/hooks/use-scroll-lock/use-scroll-lock.ts b/src/utils/hooks/use-scroll-lock/use-scroll-lock.ts
new file mode 100644
index 0000000..64fc5e8
--- /dev/null
+++ b/src/utils/hooks/use-scroll-lock/use-scroll-lock.ts
@@ -0,0 +1,58 @@
+import { useCallback, useEffect, useRef, useMemo } from 'react';
+import { useScrollBarWidth } from '../use-scrollbar-width';
+
+type Styles = {
+  overflow: string;
+  paddingRight: string;
+};
+
+/**
+ * React hook to lock/unlock the scroll on the body.
+ *
+ * @param {boolean} [isScrollLocked] - Should the scroll be locked?
+ */
+export const useScrollLock = (isScrollLocked = false) => {
+  const scrollbarWidth = useScrollBarWidth();
+  const initialStyles = useRef<Styles | null>(null);
+  const lockedStyles: Styles = useMemo(() => {
+    return {
+      overflow: 'hidden',
+      paddingRight: `calc(${
+        initialStyles.current?.paddingRight ?? 0
+      } + ${scrollbarWidth}px)`,
+    };
+  }, [scrollbarWidth]);
+
+  useEffect(() => {
+    const computedStyle =
+      typeof window === 'undefined'
+        ? undefined
+        : window.getComputedStyle(document.body);
+
+    initialStyles.current = {
+      overflow: computedStyle?.overflow ?? '',
+      paddingRight: computedStyle?.paddingRight ?? '',
+    };
+  }, []);
+
+  const lockScroll = useCallback(() => {
+    document.body.style.overflow = lockedStyles.overflow;
+    document.body.style.paddingRight = lockedStyles.paddingRight;
+  }, [lockedStyles]);
+
+  const unlockScroll = useCallback(() => {
+    document.body.style.overflow = initialStyles.current?.overflow ?? '';
+    document.body.style.paddingRight =
+      initialStyles.current?.paddingRight ?? '';
+  }, []);
+
+  useEffect(() => {
+    if (typeof window === 'undefined') return undefined;
+
+    if (isScrollLocked) lockScroll();
+
+    return () => {
+      unlockScroll();
+    };
+  }, [isScrollLocked, lockScroll, unlockScroll]);
+};
-- 
cgit v1.2.3