import type {RefObject} from 'react';
import {useState} from 'react';
import {useHookableRef} from '../useHookableRef/index.js';
import {useRafCallback} from '../useRafCallback/index.js';
import type {UseResizeObserverCallback} from '../useResizeObserver/index.js';
import {useResizeObserver} from '../useResizeObserver/index.js';

export type Measures = {
	width: number;
	height: number;
};

/**
 * Uses ResizeObserver to track element dimensions and re-render component when they change.
 *
 * @param enabled Whether resize observer is enabled or not.
 */
export function useMeasure<T extends Element>(enabled = true): [Measures | undefined, RefObject<T | null>] {
	const [element, setElement] = useState<T | null>(null);
	const elementRef = useHookableRef<T | null>(null, (v) => {
		setElement(v);

		return v;
	});

	const [measures, setMeasures] = useState<Measures>();
	const [observerHandler] = useRafCallback<UseResizeObserverCallback>((entry) => {
		setMeasures({width: entry.contentRect.width, height: entry.contentRect.height});
	});

	useResizeObserver(element, observerHandler, enabled);

	return [measures, elementRef];
}