import { useMemo, useSyncExternalStore } from 'react';

import { useE2EERoom } from './useE2EERoom';
import type { E2ERoomState } from '../../../lib/e2ee/E2ERoomState';

export const useE2EERoomState = (rid: string) => {
	const e2eRoom = useE2EERoom(rid);

	const subscribeE2EERoomState = useMemo(
		() =>
			[
				(callback: () => void): (() => void) => (e2eRoom ? e2eRoom.onStateChange(callback) : () => undefined),
				(): E2ERoomState | undefined => (e2eRoom ? e2eRoom.getState() : undefined),
			] as const,
		[e2eRoom],
	);

	return useSyncExternalStore(...subscribeE2EERoomState);
};