mirror of
https://github.com/qier222/YesPlayMusic.git
synced 2025-02-28 16:38:40 +08:00
22 lines
461 B
TypeScript
22 lines
461 B
TypeScript
import { useEffect } from 'react'
|
|
|
|
const useLockMainScroll = (lock: boolean) => {
|
|
useEffect(() => {
|
|
const main = document.querySelector('#main') as HTMLElement | null
|
|
if (!main) {
|
|
throw new Error('Main element not found')
|
|
}
|
|
|
|
if (lock) {
|
|
main.style.overflow = 'hidden'
|
|
} else {
|
|
main.style.overflow = 'auto'
|
|
}
|
|
return () => {
|
|
main.style.overflow = 'auto'
|
|
}
|
|
}, [lock])
|
|
}
|
|
|
|
export default useLockMainScroll
|