byhl-zt/pro-components/utils/hooks/useMediaQuery/index.ts

export const MediaQueryEnum = {
  xs: {
    maxWidth: 575,
    matchMedia: '(max-width: 575px)'
  },
  sm: {
    minWidth: 576,
    maxWidth: 767,
    matchMedia: '(min-width: 576px) and (max-width: 767px)'
  },
  md: {
    minWidth: 768,
    maxWidth: 991,
    matchMedia: '(min-width: 768px) and (max-width: 991px)'
  },
  lg: {
    minWidth: 992,
    maxWidth: 1199,
    matchMedia: '(min-width: 992px) and (max-width: 1199px)'
  },
  xl: {
    minWidth: 1200,
    maxWidth: 1599,
    matchMedia: '(min-width: 1200px) and (max-width: 1599px)'
  },
  xxl: {
    minWidth: 1600,
    matchMedia: '(min-width: 1600px)'
  }
}

export type MediaQueryKey = keyof typeof MediaQueryEnum

/**
 * loop query screen className
 * Array.find will throw a error
 * `Rendered more _hooks than during the previous render.`
 * So should use Array.forEach
 */
export const getScreenClassName = () => {
  let className: MediaQueryKey = 'md'
  // support ssr
  if (typeof window === 'undefined') {
    return className
  }
  const mediaQueryKey = (Object.keys(MediaQueryEnum) as MediaQueryKey[]).find(key => {
    const { matchMedia } = MediaQueryEnum[key]
    if (window.matchMedia(matchMedia).matches) {
      return true
    }
    return false
  })
  className = mediaQueryKey as unknown as MediaQueryKey
  return className
}

const useMedia = () => {
  const colSpan = ref<string>(getScreenClassName())

  Object.keys(MediaQueryEnum).forEach(key => {
    const { matchMedia } = MediaQueryEnum[key as MediaQueryKey]
    const query = window.matchMedia(matchMedia)
    if (query.matches) {
      colSpan.value = key
    }
    query.onchange = e => {
      if (e.matches) {
        colSpan.value = key
      }
    }
  })

  return colSpan
}

export default useMedia