-
Notifications
You must be signed in to change notification settings - Fork 0
/
useResizeObserver.js
55 lines (45 loc) · 1.28 KB
/
useResizeObserver.js
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
import { useCallback, useEffect, useRef, useState } from "react";
// import ResizeObserver from "resize-observer-polyfill";
function useResizeObserver(element) {
const [size, setSize] = useState();
const observer = useRef();
const callback = useRef();
const onResize = useCallback((fn) => {
callback.current = fn;
}, []);
useEffect(() => {
const current = element && element.current;
if (current && observer.current) {
observer.current.unobserve(current);
}
observer.current = new ResizeObserver((entries) => {
entries.forEach((entry) => {
if (
entry.contentRect &&
(entry.contentRect.width !== size?.width ||
entry.contentRect.height !== size?.height)
) {
setSize({
width: entry.contentRect.width,
height: entry.contentRect.height,
});
}
});
});
if (current && observer.current) {
observer.current.observe(current);
}
return () => {
if (observer && observer.current && current) {
observer.current.unobserve(current);
}
};
}, [element]);
useEffect(() => {
if (callback.current && size) {
callback.current(size);
}
}, [size]);
return [size, onResize];
}
export default useResizeObserver;