import { useEffect, useState } from 'react'

export const ScreenSize = Object.freeze({
  xs: 531,
  sm: 941,
  lg: 1481
})

export const useScreenSize = () => {
  const { width } = useWindowSize()
  if (!width || width <= ScreenSize.xs) {
    return ScreenSize.xs
  } else if (width <= ScreenSize.sm) {
    return ScreenSize.sm
  } else {
    return ScreenSize.lg
  }
}

// Hook
function useWindowSize() {
  // Initialize state with undefined width/height so server and client renders match
  // Learn more here: https://joshwcomeau.com/react/the-perils-of-rehydration/
  const [windowSize, setWindowSize] = useState({
    width: undefined,
    height: undefined
  })
  useEffect(() => {
    // Handler to call on window resize
    function handleResize() {
      // Set window width/height to state
      setWindowSize({
        width: window.innerWidth,
        height: window.innerHeight
      })
    }
    // Add event listener
    window.addEventListener('resize', handleResize)
    // Call handler right away so state gets updated with initial window size
    handleResize()
    // Remove event listener on cleanup
    return () => window.removeEventListener('resize', handleResize)
  }, []) // Empty array ensures that effect is only run on mount
  return windowSize
}