39 lines
943 B
TypeScript
39 lines
943 B
TypeScript
import { useEffect, useMemo, useState } from 'react';
|
|
import ResizeObserver from 'resize-observer-polyfill';
|
|
|
|
type UseDimensionsRect = { width: number, height: number };
|
|
type UseDimensionsResult = [Element | null, any, any]
|
|
|
|
const defaultState: UseDimensionsRect = {
|
|
width: 0,
|
|
height: 0,
|
|
};
|
|
|
|
const useDimensions = (): UseDimensionsResult => {
|
|
const [element, setRef] = useState<Element | null>(null);
|
|
const [rect, setRect] = useState<UseDimensionsRect>(defaultState);
|
|
|
|
const observer = useMemo(
|
|
() =>
|
|
new ResizeObserver((entries: any) => {
|
|
if (entries[0]) {
|
|
const { width, height } = entries[0].contentRect;
|
|
setRect({ width, height });
|
|
}
|
|
}),
|
|
[],
|
|
);
|
|
|
|
useEffect((): any => {
|
|
if (!element) return null;
|
|
observer.observe(element);
|
|
|
|
return () => {
|
|
observer.disconnect();
|
|
};
|
|
}, [element]);
|
|
|
|
return [element, setRef, rect];
|
|
};
|
|
|
|
export { useDimensions };
|