/
use-media-query.ts
58 lines (48 loc) · 1.63 KB
/
use-media-query.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import { useEffect, useRef, useState } from 'react';
export interface UseMediaQueryOptions {
getInitialValueInEffect: boolean;
}
type MediaQueryCallback = (event: { matches: boolean; media: string }) => void;
/**
* Older versions of Safari (shipped withCatalina and before) do not support addEventListener on matchMedia
* https://stackoverflow.com/questions/56466261/matchmedia-addlistener-marked-as-deprecated-addeventlistener-equivalent
* */
function attachMediaListener(query: MediaQueryList, callback: MediaQueryCallback) {
try {
query.addEventListener('change', callback);
return () => query.removeEventListener('change', callback);
} catch (e) {
query.addListener(callback);
return () => query.removeListener(callback);
}
}
function getInitialValue(query: string, initialValue?: boolean) {
if (typeof initialValue === 'boolean') {
return initialValue;
}
if (typeof window !== 'undefined' && 'matchMedia' in window) {
return window.matchMedia(query).matches;
}
return false;
}
export function useMediaQuery(
query: string,
initialValue?: boolean,
{ getInitialValueInEffect }: UseMediaQueryOptions = {
getInitialValueInEffect: true,
}
) {
const [matches, setMatches] = useState(
getInitialValueInEffect ? initialValue : getInitialValue(query)
);
const queryRef = useRef<MediaQueryList>();
useEffect(() => {
if ('matchMedia' in window) {
queryRef.current = window.matchMedia(query);
setMatches(queryRef.current.matches);
return attachMediaListener(queryRef.current, (event) => setMatches(event.matches));
}
return undefined;
}, [query]);
return matches;
}